Update agent.py

agent.py

@@ -471,7 +471,7 @@ class EsiAgent:
         final_state = self.graph.invoke(state)
         print(final_state)
         # 4) Pull out the final answer
-        return final_state.final_answer
+        return final_state['final_answer']
 # Optional: allow running agent.py directly for quick tests
 if __name__ == "__main__":
     agent = EsiAgent()