Update app.py

app.py

@@ -208,16 +208,28 @@ class BasicAgent:
         wrong_answers = ["Requests rate limit exceeded", "", " ", "  ", "insufficient information"]
         while not answer or answer in wrong_answers or answer.lower().startswith("error"):
             try:
-                alfred = build_agent(chat3)
+                alfred = build_agent(chat1)
                 answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
             except Exception as e:
                 print(e)
-                alfred = build_agent(chat1)
+                alfred = build_agent(chat3)
                 answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content            
         if answer:
             alfred_super = build_agent(chat1)
             answer_fin = alfred_super.invoke(
-                input={"messages": [HumanMessage(f"Verify the answer to the following question. QUESTION: {system_prompt.content}:  {message.content}, ANSWER: {answer}")]}, config={"recursion_limit": 6})['messages'][-1].content
+                input={"messages": [
+                    SystemMessage(
+                        content="Verify the answer to the following question. If the answer is correct, return it.\
+                        If it's incorrect, return just the corrected answer. \
+                        Do not add any comments or characters, your should return just the answer: \
+                        either a string, or a number, or a comma separated list."
+                        ),
+                    HumanMessage(
+                        content=f"{system_prompt}. 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))    
             answer_fin = answer_fin.replace('FINAL ANSWER:', '')
             answer_fin = answer_fin.replace('FINAL ANSWER', '')