Update agent.py

agent.py

@@ -421,22 +421,6 @@ def synthesizer(state: GaiaAgentState) -> dict:
     return {"final_answer": final}
 
 
-# builder = StateGraph(dict)
-from dataclasses import dataclass, field
-
-
-@dataclass
-class GaiaAgentState:
-    raw_payload: Dict[str, Any]
-    # these will get populated by the input_parser node:
-    question: str = ""
-    files:    List[str] = field(default_factory=list)
-    level:    int = 1
-    # downstream nodes can write into these:
-    plan:         List[Dict[str,Any]] = field(default_factory=list)
-    tool_results: List[Any]         = field(default_factory=list)
-    final_answer: str = ""
-
 
     
 
@@ -485,7 +469,7 @@ class EsiAgent:
         }
         state = GaiaAgentState(raw_payload=payload)
         final_state = self.graph.invoke(state)
-
+        print(final_state)
         # 4) Pull out the final answer
         return final_state.final_answer
 # Optional: allow running agent.py directly for quick tests