refined code of output
Browse files
agent.py
CHANGED
|
@@ -231,8 +231,11 @@ class BasicAgent:
|
|
| 231 |
):
|
| 232 |
answer = answer[1:-1].strip()
|
| 233 |
|
| 234 |
-
stuff_to_remove =
|
| 235 |
if answer.startswith(stuff_to_remove):
|
| 236 |
answer = answer[len(stuff_to_remove) :].strip()
|
| 237 |
|
|
|
|
|
|
|
|
|
|
| 238 |
return answer
|
|
|
|
| 231 |
):
|
| 232 |
answer = answer[1:-1].strip()
|
| 233 |
|
| 234 |
+
stuff_to_remove = "Stdout:"
|
| 235 |
if answer.startswith(stuff_to_remove):
|
| 236 |
answer = answer[len(stuff_to_remove) :].strip()
|
| 237 |
|
| 238 |
+
stuff_to_remove_bis = "Output: "
|
| 239 |
+
if answer.startswith(stuff_to_remove_bis):
|
| 240 |
+
answer = answer[len(stuff_to_remove_bis) :].strip()
|
| 241 |
return answer
|