junaid/test_basic.py at master · penguin-io/junaid · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
"""
Quick test to verify imports and basic functionality.
"""

import sys
import os

# Add parent directory to path
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

print("Testing imports...")

try:
    from agent.junaid_agent import JunaidAgent

    print("✓ JunaidAgent imported successfully")
except Exception as e:
    print(f"✗ Error importing JunaidAgent: {e}")
    sys.exit(1)

print("\nTesting agent initialization...")

try:
    agent = JunaidAgent(tts_enabled=False)
    print("✓ Agent initialized successfully")
except Exception as e:
    print(f"✗ Error initializing agent: {e}")
    sys.exit(1)

print("\nTesting graph structure...")

try:
    graph = agent.graph
    print(f"✓ Graph compiled successfully")
    print(f"  Nodes: main_coordinator, computer_vision, robotic_arm")
except Exception as e:
    print(f"✗ Error with graph: {e}")
    sys.exit(1)

print("\n" + "=" * 50)
print("All basic tests passed!")
print("=" * 50)
print("\nAgent is ready to use.")
print("To test with actual inputs, run the backend server:")
print("  cd backend && python server.py")