sampsong commited on
Commit
13bb12b
·
1 Parent(s): 14470b6

modify prompt to return without final answer

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -81,7 +81,9 @@ class BasicAgent:
81
  #print(messages[-1].__dict__)
82
  #print(f"full answer : {answer}")
83
  #print("Assistant Output:", messages[-1])
84
- return answer
 
 
85
 
86
  def run_and_submit_all( profile: gr.OAuthProfile | None):
87
  """
 
81
  #print(messages[-1].__dict__)
82
  #print(f"full answer : {answer}")
83
  #print("Assistant Output:", messages[-1])
84
+ final_answer = answer.split("FINAL ANSWER:")[-1]
85
+ print(final_answer)
86
+ return final_answer
87
 
88
  def run_and_submit_all( profile: gr.OAuthProfile | None):
89
  """