Add logic to remove "Stdout: Output: " from LLM answer
Browse files
agent.py
CHANGED
|
@@ -231,4 +231,8 @@ class BasicAgent:
|
|
| 231 |
):
|
| 232 |
answer = answer[1:-1].strip()
|
| 233 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 234 |
return answer
|
|
|
|
| 231 |
):
|
| 232 |
answer = answer[1:-1].strip()
|
| 233 |
|
| 234 |
+
stuff_to_remove = ["Stdout: Output: "]
|
| 235 |
+
if answer.startswith(stuff_to_remove):
|
| 236 |
+
answer = answer[len(stuff_to_remove) :].strip()
|
| 237 |
+
|
| 238 |
return answer
|