Update agent.py

agent.py

@@ -391,7 +391,7 @@ class EsiAgent:
         self.tool_numbers = level * 3
         def assistant(state):
             if self.tool_numbers == 0:
-              return state["messages"]
+              return {"messages": state["messages"]}
             self.tool_numbers -= 1
             # Prepend sys_msg only if it's not already there
             return {"messages": [llm_with_tools.invoke(state["messages"])]}
@@ -414,10 +414,10 @@ class EsiAgent:
         """
         # Build a minimal GAIA payload (if your MessagesState expects more, adjust here)
         print(33333333333333333333333333333333333333333333333333333333333)
-        print(self.sys_msg)
-        print(question)
-        print(type(self.sys_msg))
-        print(type(question))
+        # print(self.sys_msg)
+        # print(question)
+        # print(type(self.sys_msg))
+        # print(type(question))
         
         initial_state = {
             # "messages": [self.sys_msg]+[HumanMessage(content=question)],