Update agent.py

agent.py

@@ -197,9 +197,9 @@ class GaiaAgentState(TypedDict):
     
 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)
+      "question": state["question"],
+      "files":    state.get("files", []),
+      "level":    state.get("level", 1)
     }
 
 # === 2. TOOL STUBS ===