Implementing Memory in Our Agent › Testing Your Memory System

def test_memory_system():
    """Test that memory works correctly."""
    assistant = PersonalAssistant(api_key=os.getenv("ANTHROPIC_API_KEY"))
    
    # Test 1: Store and retrieve
    assistant.chat("Remember that my birthday is July 20")
    response = assistant.chat("When is my birthday?")
    assert "july 20" in response.lower(), "Failed to retrieve birthday"
    
    # Test 2: Semantic search
    assistant.chat("Remember that I'm allergic to peanuts")
    response = assistant.chat("What should I avoid eating?")
    assert "peanut" in response.lower(), "Failed semantic search"
    
    # Test 3: Conversation context
    assistant.chat("I'm planning a trip to Japan")
    response = assistant.chat("What's the best time to visit?")
    assert "japan" in response.lower(), "Lost conversation context"
    
    # Test 4: Persistence
    assistant2 = PersonalAssistant(api_key=os.getenv("ANTHROPIC_API_KEY"))
    response = assistant2.chat("When is my birthday?")
    assert "july 20" in response.lower(), "Facts not persisted"
    
    print("All tests passed!")

test_memory_system()