Toumaima commited on
Commit
7d13d18
·
verified ·
1 Parent(s): c87b88e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -32,7 +32,7 @@ class BasicAgent:
32
 
33
  def format_final_answer(self, answer: str) -> str:
34
  cleaned = " ".join(answer.split())
35
- return f"FINAL ANSWER: "+{cleaned}
36
 
37
  def check_commutativity(self):
38
  S = ['a', 'b', 'c', 'd', 'e']
@@ -86,8 +86,8 @@ class BasicAgent:
86
  messages=[{"role": "user", "content": full_prompt}]
87
  )
88
  answer = response.choices[0].message.content
89
- if "FINAL ANSWER:" in answer:
90
- return answer.split("FINAL ANSWER:")[-1].strip()
91
  else:
92
  return self.format_final_answer(answer)
93
  except Exception as e:
 
32
 
33
  def format_final_answer(self, answer: str) -> str:
34
  cleaned = " ".join(answer.split())
35
+ return f"FINAL ANSWER: {cleaned}
36
 
37
  def check_commutativity(self):
38
  S = ['a', 'b', 'c', 'd', 'e']
 
86
  messages=[{"role": "user", "content": full_prompt}]
87
  )
88
  answer = response.choices[0].message.content
89
+ if "FINAL ANSWER: " in answer:
90
+ return answer.split("FINAL ANSWER: ")[-1].strip()
91
  else:
92
  return self.format_final_answer(answer)
93
  except Exception as e: