Update app.py

app.py

@@ -189,7 +189,7 @@ 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, use only digits in your final answer. Don't use comma nor brackets 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. Wrap the list in [square brackets].\
+            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.\
             If there is a file attached, open the file and read it. \
             If you don't have enough references to answer, use your tools, search the web, run your code or convert data to a data frame, whatever helps. \
             If the question refers to an external content and there is no reference file attached, perform a web search and retrieve relevant information from the internet. \
@@ -198,18 +198,21 @@ class BasicAgent:
         )
         message = HumanMessage(content=question)
         print(message)
-        try:
-            alfred = build_agent(chat1)
-            answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
-        except Exception as e:
-            alfred = build_agent(chat2)
-            answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
-        answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))    
-        answer = answer.replace('FINAL ANSWER:', '')
-        answer = answer.replace('FINAL ANSWER', '')
-        answer = answer.replace('YOUR ', '')
-        answer = answer.replace('*', '')
-        print(f"Agent returning fixed answer: {answer}")        
+        answer = None
+        while not answer:
+            try:
+                alfred = build_agent(chat1)
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
+            except Exception as e:
+                alfred = build_agent(chat2)
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
+        if answer:
+            answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))    
+            answer = answer.replace('FINAL ANSWER:', '')
+            answer = answer.replace('FINAL ANSWER', '')
+            answer = answer.replace('YOUR ', '')
+            answer = answer.replace('*', '')
+            print(f"Agent returning fixed answer: {answer}")        
             
         return answer