aynot commited on
Commit
1a6dacb
·
1 Parent(s): 8398fab

fix int answer

Browse files
Files changed (1) hide show
  1. agent.py +3 -1
agent.py CHANGED
@@ -25,7 +25,9 @@ class MyAgent:
25
 
26
  def get_answer(self, question):
27
  answer = self.agent.run(self.gaia_prompt + '\n' + question)
28
-
 
 
29
  fa_tmlp = "FINAL ANSWER:"
30
  if answer.startswith(fa_tmlp):
31
  answer = answer[len(fa_tmlp):]
 
25
 
26
  def get_answer(self, question):
27
  answer = self.agent.run(self.gaia_prompt + '\n' + question)
28
+
29
+ answer = str(answer)
30
+
31
  fa_tmlp = "FINAL ANSWER:"
32
  if answer.startswith(fa_tmlp):
33
  answer = answer[len(fa_tmlp):]