ORromu commited on
Commit
943311e
·
verified ·
1 Parent(s): 0f1bf73

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -25,8 +25,7 @@ class BasicAgent:
25
  messages = self.graph.invoke({"messages": messages})
26
  print(messages)
27
  answer = messages['messages'][-1].content
28
- #answer = answer[answer.index("FINAL ANSWER: ") + 14:]
29
- return answer[14:]
30
 
31
  def run_and_submit_all( profile: gr.OAuthProfile | None):
32
  """
 
25
  messages = self.graph.invoke({"messages": messages})
26
  print(messages)
27
  answer = messages['messages'][-1].content
28
+ return answer.split("FINAL ANSWER:")[-1].strip()
 
29
 
30
  def run_and_submit_all( profile: gr.OAuthProfile | None):
31
  """