debug_test.py

import asyncio
from langraph_agent import build_graph
from langchain_core.messages import HumanMessage

async def test_agent():
    """Test the agent with a simple question"""
    try:
        print("Building graph...")
        graph = build_graph(provider="groq")
        
        print("Testing with a simple question...")
        question = "What is 2 + 3?"
        messages = [HumanMessage(content=question)]
        
        print("Invoking agent...")
        response = await graph.ainvoke({"messages": messages})
        
        print(f"Response type: {type(response)}")
        print(f"Response keys: {response.keys() if isinstance(response, dict) else 'Not a dict'}")
        
        if 'messages' in response and response['messages']:
            print(f"Number of messages: {len(response['messages'])}")
            print(f"Last message content: {response['messages'][-1].content}")
            return "SUCCESS"
        else:
            print("No messages in response or empty messages")
            return "FAILED - No messages"
    except Exception as e:
        print(f"Error in test: {e}")
        import traceback
        traceback.print_exc()
        return f"FAILED - {e}"

if __name__ == "__main__":
    result = asyncio.run(test_agent())
    print(f"Test result: {result}")