test_brain_fix.py

#!/usr/bin/env python3
"""Test the brain fixes for encoding and method name issues"""
import asyncio
import sys
from pathlib import Path

# Add parent directory to path
parent_dir = Path(__file__).parent
sys.path.insert(0, str(parent_dir))

async def test_brain():
    try:
        # Test imports
        from core.brain import AutumnBrain
        from core.memory import AutumnMemory
        from config.personality import AutumnPersonality
        
        print("✓ Imports successful")
        
        # Initialize components
        memory = AutumnMemory()
        await memory.initialize()
        print("✓ Memory initialized")
        
        personality = AutumnPersonality()
        print("✓ Personality initialized")
        
        # Initialize brain
        brain = AutumnBrain(memory, personality)
        await brain.initialize()
        print("✓ Brain initialized")
        
        # Test simple query
        response = await brain.process("Hello")
        print(f"✓ Query processed: {response[:50]}...")
        
        # Test math query
        response = await brain.process("What is 2 + 2?")
        print(f"✓ Math query processed: {response[:50]}...")
        
        # Cleanup
        await brain.shutdown()
        print("✓ Brain shutdown")
        
        print("\n🎉 All tests passed! Brain fixes working correctly!")
        
    except Exception as e:
        print(f"❌ Error: {e}")
        import traceback
        traceback.print_exc()

if __name__ == "__main__":
    asyncio.run(test_brain())