MariaFilipkowska commited on
Commit
2df6b7d
·
verified ·
1 Parent(s): 1722934

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -208,22 +208,22 @@ class BasicAgent:
208
  wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
209
  while not answer or answer in wrong_answers or answer.lower().startswith("error"):
210
  try:
211
- alfred = build_agent(chat2)
212
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
213
  except Exception as e:
214
  alfred = build_agent(chat3)
215
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
216
- finally:
217
- alfred = build_agent(chat1)
218
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
219
- if answer:
220
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
221
  answer = answer.replace('FINAL ANSWER:', '')
222
  answer = answer.replace('FINAL ANSWER', '')
223
  answer = answer.replace('YOUR ', '')
224
  answer = answer.replace('*', '')
225
  print(f"Agent returning fixed answer: {answer}")
226
-
227
  return answer
228
 
229
 
 
208
  wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
209
  while not answer or answer in wrong_answers or answer.lower().startswith("error"):
210
  try:
211
+ alfred = build_agent(chat1)
212
+ answer1 = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
213
  except Exception as e:
214
  alfred = build_agent(chat3)
215
+ answer1 = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
216
+ if answer1:
217
+ alfred_super = build_agent(chat1)
218
+ answer = alfred_super.invoke(
219
+ input={"messages": [f"Verify the answer to the following question. QUESTION: {message}, ANSWER: {answer1}"]},config={"recursion_limit": 6})['messages'][-1].content
220
  answer = "".join(re.findall(r'(FINAL ANSWER.*)', answer, flags=re.M))
221
  answer = answer.replace('FINAL ANSWER:', '')
222
  answer = answer.replace('FINAL ANSWER', '')
223
  answer = answer.replace('YOUR ', '')
224
  answer = answer.replace('*', '')
225
  print(f"Agent returning fixed answer: {answer}")
226
+
227
  return answer
228
 
229