MariaFilipkowska commited on
Commit
03a9ea6
·
verified ·
1 Parent(s): dc53596

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -217,13 +217,12 @@ class BasicAgent:
217
  if answer:
218
  alfred_super = build_agent(chat1)
219
  answer_fin = alfred_super.invoke(
220
- input={"messages": [f"Verify the answer to the following question. QUESTION: {message}, ANSWER: {answer}"]},config={"recursion_limit": 6})['messages'][-1].content
221
  answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer_fin, flags=re.M))
222
  answer_fin = answer_fin.replace('FINAL ANSWER:', '')
223
  answer_fin = answer_fin.replace('FINAL ANSWER', '')
224
  answer_fin = answer_fin.replace('YOUR ', '')
225
- answer_fin = answer_fin.replace('*', '')
226
- print(f"Agent returning fixed answer: {answer_fin}")
227
 
228
  return answer_fin
229
 
 
217
  if answer:
218
  alfred_super = build_agent(chat1)
219
  answer_fin = alfred_super.invoke(
220
+ 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
221
  answer_fin = "".join(re.findall(r'(FINAL ANSWER.*)', answer_fin, flags=re.M))
222
  answer_fin = answer_fin.replace('FINAL ANSWER:', '')
223
  answer_fin = answer_fin.replace('FINAL ANSWER', '')
224
  answer_fin = answer_fin.replace('YOUR ', '')
225
+ print(f"Agent returning fixed answer: {answer_fin}")
 
226
 
227
  return answer_fin
228