"""
Tests for AgGPT16
"""

from AgGPT16 import ask
import time

def test_ai():
    """Test the AI with various prompts"""
    print("Testing AgGPT16 AI")
    print("=" * 50)
    
    test_prompts = [
        "Hello, how are you?",
        "What is Python?",
        "Tell me about machine learning",
        "I'm feeling sad today",
        "What's your favorite color?",
        "Can you help me with coding?",
    ]
    
    for i, prompt in enumerate(test_prompts, 1):
        print(f"\n{i}. USER: {prompt}")
        print("-" * 40)
        
        start_time = time.time()
        try:
            response = ask(prompt)
            end_time = time.time()
            
            print(f"AI: {response}")
            print(f"⏱️ Response time: {end_time - start_time:.2f}s")
            
        except Exception as e:
            print(f"❌ Error: {e}")
        
        print()

if __name__ == "__main__":
    test_ai()