Update agent.py

agent.py

@@ -43,9 +43,10 @@ class BasicAgent():
 
     
     def __call__(self, question: str) -> str:
-        print(f"Agent received question (first 50 chars): {question[:50]}...")
+        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
@@ -57,7 +58,7 @@ 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...", state["messages"])
+        print("Calling assistant with state: ", state["messages"])
         return {
             "messages": [self.chat_with_tools.invoke([sys_msg] + state["messages"])],            
         }