MariaFilipkowska commited on
Commit
7e6c848
·
verified ·
1 Parent(s): c257d04

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -157,7 +157,8 @@ class BasicAgent:
157
  print(message)
158
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
159
  answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
160
- answer = answer.replace('FINAL ANSWER: ', '')
 
161
  answer = answer.replace('[', '')
162
  nswer = answer.replace('*', '')
163
  fixed_answer = answer.replace(']', '')
 
157
  print(message)
158
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
159
  answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
160
+ answer = answer.replace('FINAL ANSWER:', '')
161
+ answer = answer.replace('FINAL ANSWER', '')
162
  answer = answer.replace('[', '')
163
  nswer = answer.replace('*', '')
164
  fixed_answer = answer.replace(']', '')