Update agent.py

agent.py

@@ -195,12 +195,12 @@ class GaiaAgentState(TypedDict):
     tool_results: List[Any]
     final_answer: str
     
-def input_parser(state: GaiaAgentState) -> dict:
-    return {
-      "question": state["question"],
-      "files":    state.get("files", []),
-      "level":    state.get("level", 1)
-    }
+# def input_parser(state: GaiaAgentState) -> dict:
+#     return {
+#       "question": state["question"],
+#       "files":    state.get("files", []),
+#       "level":    state.get("level", 1)
+#     }
 
 # === 2. TOOL STUBS ===
 
@@ -440,13 +440,14 @@ class EsiAgent:
         # 1. Build & compile your graph
         builder = StateGraph(GaiaAgentState)
 
-        builder.add_node("parse",       input_parser)
+        # 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,   "parse")
-        builder.add_edge("parse",   "planner")
+        # builder.add_edge(START,   "parse")
+        # builder.add_edge("parse",   "planner")
+        builder.add_edge(START,   "planner")
         builder.add_edge("planner",    "execute")
         builder.add_edge("execute", "synthesize")
         builder.add_edge("synthesize", END)