JPM34 commited on
Commit
685d9f9
·
1 Parent(s): 09ab61d

Add logic to remove "Stdout: Output: " from LLM answer

Browse files
Files changed (1) hide show
  1. agent.py +4 -0
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