from gemini_mikrotik_integration import MikroTikGeminiAssistant
import os

def test():
    assistant = MikroTikGeminiAssistant()
    
    print("--- Test 1: CPU Load ---")
    ans1 = assistant.ask("Berapa CPU load router saat ini?")
    print(f"A: {ans1}")
    
    print("\n--- Test 2: History (Uptime) ---")
    ans2 = assistant.ask("Berapa uptimenya?")
    print(f"A: {ans2}")
    
    print("\n--- Test 3: Context Connection ---")
    ans3 = assistant.ask("Apakah angka-angka tersebut masih dalam batas normal?")
    print(f"A: {ans3}")

if __name__ == "__main__":
    test()