antonioschiro commited on
Commit
f43a0c6
·
verified ·
1 Parent(s): f87397d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -27,7 +27,9 @@ class BasicAgent:
27
  response = state_dict["messages"][-1]
28
  try:
29
  response = response.content
30
- response = response.split("Final answer:")[1].strip()
 
 
31
  return response
32
  except:
33
  return "Sorry, I can't answer to this question."
 
27
  response = state_dict["messages"][-1]
28
  try:
29
  response = response.content
30
+ if "Final answer:" in response:
31
+ response = response.split("Final answer:")[1]
32
+ response = response.strip()
33
  return response
34
  except:
35
  return "Sorry, I can't answer to this question."