"""
Test the intelligent routing system to show how the orchestrator makes decisions.
"""

import os
import sys

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

from agent import answer_gaia_question

def test_intelligent_routing():
    """Test cases that demonstrate the orchestrator's decision-making capabilities."""
    
    test_cases = [
        {
            "question": "What is the capital of France?",
            "expected_behavior": "Direct answer (simple factual question)"
        },
        {
            "question": "Calculate the sum of values in column A of the Excel file data.xlsx",
            "expected_behavior": "Route to retriever agent (file processing)"
        },
        {
            "question": "What is the current CEO of OpenAI as of 2024?",
            "expected_behavior": "Route to research agent (current information)"
        },
        {
            "question": "If a paper has a p-value of 0.04 and there were 1000 papers, how many would be false positives?",
            "expected_behavior": "Route to math agent (calculations)"
        },
        {
            "question": "List the prime numbers between 1 and 20, comma-separated",
            "expected_behavior": "Route to math agent OR direct answer"
        }
    ]
    
    print("🧠 Testing Intelligent Routing System")
    print("=" * 60)
    
    for i, test_case in enumerate(test_cases, 1):
        print(f"\n📝 Test {i}: {test_case['question']}")
        print(f"🎯 Expected: {test_case['expected_behavior']}")
        print("-" * 40)
        
        try:
            # This will show the orchestrator's decision-making process
            print("🔄 Processing...")
            answer = answer_gaia_question(test_case['question'], debug=True)
            print(f"✅ Final Result: {answer}")
            
        except Exception as e:
            print(f"❌ Error: {e}")
            import traceback
            traceback.print_exc()
        
        print("-" * 60)

if __name__ == "__main__":
    test_intelligent_routing()