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

refined code of output

Browse files
Files changed (1) hide show
  1. agent.py +4 -1
agent.py CHANGED
@@ -231,8 +231,11 @@ class BasicAgent:
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
 
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