Update agent.py

agent.py

@@ -46,7 +46,6 @@ class BasicAgent():
         print(f"Agent received question: {question}.")
         messages=[HumanMessage(content=question)]
         response = self.agent.invoke({"messages":messages})
-        print(f"Got response {response}")
         answer = response['messages'][-1].content
         print(f"Agent returning answer: {answer}")
         return answer
@@ -58,9 +57,10 @@ class BasicAgent():
         You are a general AI assistant. I will ask you a question. Report your thoughts, and finish your answer with the following template: FINAL ANSWER: [YOUR FINAL ANSWER]. 
         YOUR FINAL ANSWER should be a number OR as few words as possible OR a comma separated list of numbers and/or strings. If you are asked for a number, don't use comma to write your number neither use units such as $ or percent sign unless specified otherwise. If you are asked for a string, don't use articles, neither abbreviations (e.g. for cities), and write the digits in plain text unless specified otherwise. If you are asked for a comma separated list, apply the above rules depending of whether the element to be put in the list is a number or a string.
         """)
-        print("Calling assistant with state: ", state["messages"])
+        updated_messages = [self.chat_with_tools.invoke([sys_msg] + state["messages"])]
+        print("Calling assistant with state: ",  state["messages"])
         return {
-            "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],            
+            "messages":  state["messages"]],            
         }