MariaFilipkowska commited on
Commit
d5f3340
·
verified ·
1 Parent(s): 67f4669

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -9
app.py CHANGED
@@ -206,14 +206,9 @@ class BasicAgent:
206
  print(message)
207
  answer = None
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
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
213
- except Exception as e:
214
- print(e)
215
- alfred = build_agent(chat3)
216
- answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
217
  if answer:
218
  alfred_super = build_agent(chat3)
219
  answer_fin = alfred_super.invoke(
@@ -229,7 +224,7 @@ class BasicAgent:
229
  ]},
230
  config={"recursion_limit": 6}
231
  )['messages'][-1].content
232
- answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer_fin, flags=re.M))
233
  answer_fin = answer_fin.replace('FINAL ANSWER:', '')
234
  answer_fin = answer_fin.replace('FINAL ANSWER', '')
235
  answer_fin = answer_fin.replace('YOUR ', '')
 
206
  print(message)
207
  answer = None
208
  wrong_answers = ["Requests rate limit exceeded", "", " ", " ", "insufficient information"]
209
+ while not answer or answer in wrong_answers or answer.lower().startswith("error"):
210
+ alfred = build_agent(chat1)
211
+ answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 6})['messages'][-1].content
 
 
 
 
 
212
  if answer:
213
  alfred_super = build_agent(chat3)
214
  answer_fin = alfred_super.invoke(
 
224
  ]},
225
  config={"recursion_limit": 6}
226
  )['messages'][-1].content
227
+ #answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer_fin, flags=re.M))
228
  answer_fin = answer_fin.replace('FINAL ANSWER:', '')
229
  answer_fin = answer_fin.replace('FINAL ANSWER', '')
230
  answer_fin = answer_fin.replace('YOUR ', '')