Spaces:
Running
Running
File size: 2,432 Bytes
ba68fc1 |
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 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 |
#!/usr/bin/env python3
"""
Refactored GAIA Solver using new modular architecture
"""
import os
import sys
from pathlib import Path
# Add the current directory to Python path for imports
current_dir = Path(__file__).parent
if str(current_dir) not in sys.path:
sys.path.insert(0, str(current_dir))
from gaia import GAIASolver, Config
def main():
"""Main function to test the refactored GAIA solver"""
print("🚀 GAIA Solver - Refactored Architecture")
print("=" * 50)
try:
# Initialize configuration
config = Config()
print(f"📊 Available models: {[m.value for m in config.get_available_models()]}")
print(f"🔧 Fallback chain: {[m.value for m in config.get_fallback_chain()]}")
# Initialize solver
solver = GAIASolver(config)
# Get system status
status = solver.get_system_status()
print(f"\n🖥️ System Status:")
print(f" Models: {len(status['models'])} providers")
print(f" Available: {status['available_providers']}")
print(f" Current: {status['current_provider']}")
# Test with a sample question
print("\n🧪 Testing with sample question...")
sample_question = {
"task_id": "test_001",
"question": "What is 2 + 2?",
"level": 1
}
result = solver.solve_question(sample_question)
print(f"\n📋 Results:")
print(f" Answer: {result.answer}")
print(f" Confidence: {result.confidence:.2f}")
print(f" Method: {result.method_used}")
print(f" Time: {result.execution_time:.2f}s")
# Test random question if available
print("\n🎲 Testing with random question...")
random_result = solver.solve_random_question()
if random_result:
print(f" Answer: {random_result.answer[:100]}...")
print(f" Confidence: {random_result.confidence:.2f}")
print(f" Time: {random_result.execution_time:.2f}s")
else:
print(" No random questions available")
except Exception as e:
print(f"❌ Error: {e}")
print("\n💡 Make sure you have API keys configured:")
print("1. GEMINI_API_KEY")
print("2. HUGGINGFACE_TOKEN")
print("3. KLUSTER_API_KEY (optional)")
if __name__ == "__main__":
main() |