Update agent.py

agent.py

@@ -194,6 +194,13 @@ class GaiaAgentState(TypedDict):
     plan: List[Dict[str, Any]]
     tool_results: List[Any]
     final_answer: str
+    
+def input_parser(state: GaiaAgentState) -> dict:
+    return {
+      "question": state.raw_payload["question"],
+      "files":    state.raw_payload.get("files", []),
+      "level":    state.raw_payload.get("level", 1)
+    }
 
 # === 2. TOOL STUBS ===
 
@@ -432,12 +439,14 @@ class EsiAgent:
     def __init__(self):
         # 1. Build & compile your graph
         builder = StateGraph(GaiaAgentState)
-        
+
+        builder.add_node("parse",       input_parser)
         builder.add_node("planner",       planner_node)
         builder.add_node("execute",    tool_executor)
         builder.add_node("synthesize", synthesizer)
-        
-        builder.add_edge(START,   "planner")
+
+        builder.add_edge(START,   "parse")
+        builder.add_edge("parse",   "planner")
         builder.add_edge("planner",    "execute")
         builder.add_edge("execute", "synthesize")
         builder.add_edge("synthesize", END)
@@ -460,11 +469,23 @@ class EsiAgent:
             "files":    files or [],
             "level":    level
         }
-        print(2222222222, payload)
+        # initial = {
+        #       "question": question,
+        #       "files":    [],       # or your file list
+        #       "level":    1,
+        #       "task_id": 1,
+        #       "plan":     [],       # planner_node will overwrite this 
+        #       "tool_results": []    # executor will overwrite this
+        #       "final_answer": ""
+        #     }
+
+        # print(2222222222, payload)
 
         # 2) Wrap in your custom state class
         state = GaiaAgentState(raw_payload=payload)
-        print(33333333333, state)
+        # state = GaiaAgentState(dict)
+        
+        print(6666666666666, state)
         # 3) Invoke the graph end-to-end
         final_state = self.graph.invoke(state)