Omnibus commited on
Commit
ecf81e8
·
verified ·
1 Parent(s): 96343a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -182,7 +182,7 @@ def compress_data(c,purpose, task, history, result):
182
  resp = run_gpt(
183
  COMPRESS_DATA_PROMPT,
184
  stop_tokens=["observation:", "task:", "action:", "thought:"],
185
- max_tokens=2048,
186
  seed=seed,
187
  purpose=purpose,
188
  task=task,
@@ -414,7 +414,8 @@ def run(purpose,history):
414
  )
415
  yield result
416
  if action_name == "COMPLETE":
417
- return result
 
418
 
419
 
420
 
 
182
  resp = run_gpt(
183
  COMPRESS_DATA_PROMPT,
184
  stop_tokens=["observation:", "task:", "action:", "thought:"],
185
+ max_tokens=1024,
186
  seed=seed,
187
  purpose=purpose,
188
  task=task,
 
414
  )
415
  yield result
416
  if action_name == "COMPLETE":
417
+ break
418
+ return result
419
 
420
 
421