Update agent.py

agent.py

@@ -453,22 +453,36 @@ class EsiAgent:
         level:    GAIA difficulty (1,2,3)
         """
         # 1) Build the raw_payload
+        print(11111111111, question)
         payload = {
             "task_id": None,
             "question": question,
             "files":    files or [],
             "level":    level
         }
+        print(2222222222, payload)
 
         # 2) Wrap in your custom state class
         state = GaiaAgentState(raw_payload=payload)
-
+        print(33333333333, state)
         # 3) Invoke the graph end-to-end
         final_state = self.graph.invoke(state)
 
         # 4) Pull out the final answer
         return final_state.final_answer
 
+
+        # final_state = graph.invoke({
+        #     "task_id": None,
+        #     "question": question,
+        #     "files": [],
+        #     "level": 1,
+        #     "plan": [],
+        #     "tool_results": [],
+        #     # plus any other initial keys your nodes expect
+        # })
+
+        
     # def run(self, question: str) -> str:
     #     """
     #     Runs the agent end‑to‑end on a single question string and returns