Update app.py

app.py

@@ -205,35 +205,22 @@ class BasicAgent:
         message = HumanMessage(content=question)
         print(message)
         answer = None
-        answer_fin = None
         wrong_answers = ["Requests rate limit exceeded", "", " ", "  ", "insufficient information"]
-        while not answer or answer in wrong_answers or answer.lower().startswith("error"):        
-            alfred = build_agent(chat1)
-            answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content            
+        while not answer or answer in wrong_answers or answer.lower().startswith("error"):   
+            try:
+                alfred = build_agent(chat3)
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content   
+            except:
+                alfred = build_agent(chat1)
+                answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content 
+     
         if answer:
-            alfred_super = build_agent(chat3)
-            while not answer_fin or answer_fin in wrong_answers or answer_fin.lower().startswith("error"):
-                answer_fin = alfred_super.invoke(
-                    input={"messages": [
-                        SystemMessage(
-                            content="Verify the answer to the following question. If the answer is correct, return the same answer unchanged.\
-                            If it's incorrect, return just the corrected answer. \
-                            Do not add any comments or characters, you should return just the final answer."
-                            ),
-                        HumanMessage(
-                            content=f"QUESTION: {message.content},  ANSWER: {answer}"
-                                )
-                    ]}, 
-                config={"recursion_limit": 6}
-            )['messages'][-1].content
-            #answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer_fin, flags=re.M))
-            if answer_fin:
-                answer_fin = answer_fin.replace('FINAL ANSWER:', '')
-                answer_fin = answer_fin.replace('FINAL ANSWER', '')
-                answer_fin = answer_fin.replace('YOUR ', '')
-                answer_fin = answer_fin.replace('*', '')
-                print(f"Agent returning fixed answer: {answer_fin}")       
-    
+            answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
+            answer_fin = answer_fin.replace('FINAL ANSWER:', '')
+            answer_fin = answer_fin.replace('FINAL ANSWER', '')
+            answer_fin = answer_fin.replace('YOUR ', '')
+            answer_fin = answer_fin.replace('*', '')
+            print(f"Agent returning fixed answer: {answer_fin}")
             return answer_fin