test_agent.py

from claims_agent import ClaimsIntakeAgent
import os

def test_agent():
    print("Testing ClaimsIntakeAgent...")
    agent = ClaimsIntakeAgent()
    
    # Check if API key is set
    if not os.getenv("GEMINI_API_KEY") or os.getenv("GEMINI_API_KEY") == "your_api_key_here":
        print("Skipping live API test: GEMINI_API_KEY not set.")
        return

    # Test with text only
    response = agent.process_claim("I scratched my bumper backing into a pole.")
    print("\nResponse to text input:")
    print(response)

if __name__ == "__main__":
    test_agent()