Update app.py

app.py

@@ -28,12 +28,19 @@ class BasicAgent:
         if self.graph is None:
             return "Error: Agent not properly initialized"
         
-        # Wrap the question in a HumanMessage from langchain_core
-        messages = [HumanMessage(content=question)]
+        # Create complete state structure that matches EnhancedAgentState
+        state = {
+            "messages": [HumanMessage(content=question)],
+            "query": question,  # This was the critical missing field
+            "agent_type": "",
+            "final_answer": "",
+            "perf": {},
+            "agno_resp": ""
+        }
         config = {"configurable": {"thread_id": f"eval_{hash(question)}"}}
         
         try:
-            result = self.graph.invoke({"messages": messages}, config)
+            result = self.graph.invoke(state, config)
             
             # Handle different response formats
             if isinstance(result, dict):