MariaFilipkowska commited on
Commit
ddbb668
·
verified ·
1 Parent(s): 04a9c98

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -161,9 +161,10 @@ class BasicAgent:
161
  print(message)
162
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
163
  print(answer)
164
- answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
165
  answer = answer.replace('FINAL ANSWER: ', '')
166
  answer = answer.replace('[', '')
 
167
  fixed_answer = answer.replace(']', '')
168
  print(f"Agent returning fixed answer: {fixed_answer}")
169
  return fixed_answer
 
161
  print(message)
162
  answer = alfred.invoke(input={"messages": [system_prompt, message]},config={"recursion_limit": 3})['messages'][-1].content
163
  print(answer)
164
+ #answer = "".join(re.findall(r'(FINAL ANSWER:.*)', answer, flags=re.M))
165
  answer = answer.replace('FINAL ANSWER: ', '')
166
  answer = answer.replace('[', '')
167
+ nswer = answer.replace('*', '')
168
  fixed_answer = answer.replace(']', '')
169
  print(f"Agent returning fixed answer: {fixed_answer}")
170
  return fixed_answer