test/test_stage1.py

"""
Stage 1 Quick Verification Test
Author: @mangubee

Test that agent initialization and basic execution works.
"""

from src.agent import GAIAAgent
from src.config import Settings

print("\n" + "="*70)
print("Stage 1: Foundation Setup - Quick Verification")
print("="*70 + "\n")

# Test 1: Settings validation
print("Test 1: Checking configuration...")
settings = Settings()
api_keys = settings.validate_api_keys()
print(f"  API Keys configured:")
for service, is_set in api_keys.items():
    status = "✓" if is_set else "✗"
    print(f"    {status} {service}: {'SET' if is_set else 'NOT SET'}")
print(f"  Default LLM: {settings.default_llm_model}")

# Test 2: Agent initialization
print("\nTest 2: Initializing GAIAAgent...")
try:
    agent = GAIAAgent()
    print("  ✓ Agent initialized successfully")
except Exception as e:
    print(f"  ✗ Agent initialization failed: {e}")
    exit(1)

# Test 3: Basic question processing
print("\nTest 3: Processing test question...")
test_question = "What is the capital of France?"
try:
    answer = agent(test_question)
    print(f"  Question: {test_question}")
    print(f"  Answer: {answer}")
    print("  ✓ Question processed successfully")
except Exception as e:
    print(f"  ✗ Question processing failed: {e}")
    exit(1)

print("\n" + "="*70)
print("✓ Stage 1 verification complete - All systems ready!")
print("="*70 + "\n")
print("Next steps:")
print("1. [Optional] Test Gradio UI locally: PYTHONPATH=. uv run python app.py")
print("2. Push to HF Space to test deployment")
print("3. Proceed to Stage 2: Tool Development")