Update agent.py

agent.py

@@ -446,7 +446,16 @@ class EsiAgent:
         just the answer text (stripped of any prefix).
         """
         # Wrap in the same state shape your graph expects
-        initial_state = {"messages": [HumanMessage(content=question)]}
+        payload = {
+            "task_id": None,
+            "question": question,
+            "files": [],      # no attachments
+            "level": 1        # default to Level 1
+        }
+        # 2) Wrap in your state object
+        initial_state = GaiaAgentState(raw_payload=payload)
+        
+        # initial_state = {"messages": [HumanMessage(content=question)]}
         # Invoke the graph
         final_state = self.graph.invoke(initial_state)
         # Extract the last message