Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,9 @@ class BasicAgent:
|
|
| 27 |
response = state_dict["messages"][-1]
|
| 28 |
try:
|
| 29 |
response = response.content
|
| 30 |
-
|
|
|
|
|
|
|
| 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."
|