Update agent.py

agent.py

@@ -58,9 +58,20 @@ class BasicAgent():
         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(state["messages"])]
+        response = self.chat_with_tools.invoke(state["messages"])
+
 #        updated_messages = [self.chat_with_tools.invoke([sys_msg] + state["messages"])]
+ 
+        # Ensure the response is not None
+        if response is None:
+            raise RuntimeError("chat_with_tools.invoke returned None")
+    
+        # Ensure response is a list if expected
+        if isinstance(response, list):
+            updated_messages = response
+        else:
+            updated_messages = [response]
+            
         return {
             "messages":  updated_messages,            
         }