Omnibus commited on
Commit
dc21b34
1 Parent(s): 4a7dea6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -370,9 +370,7 @@ NAME_TO_FUNC = {
370
  "MAIN": call_main,
371
  "UPDATE-TASK": call_set_task,
372
  "SEARCH_ENGINE": find_all,
373
- "WEBSITE_SCRAPE": find_all,
374
-
375
-
376
  }
377
 
378
 
@@ -391,7 +389,7 @@ def run_action(purpose, task, history, action_name, action_input):
391
 
392
  assert action_name in NAME_TO_FUNC
393
 
394
- print("RUN: ", action_name, action_input)
395
  return NAME_TO_FUNC[action_name](purpose, task, history, action_input)
396
  else:
397
  history += "observation: The TOOL I tried to use returned an error, I need to select a tool from: (UPDATE-TASK, SEARCH_ENGINE, WEBSITE_SCRAPE, COMPLETE)\n"
 
370
  "MAIN": call_main,
371
  "UPDATE-TASK": call_set_task,
372
  "SEARCH_ENGINE": find_all,
373
+ "SCRAPE_WEBSITE": find_all,
 
 
374
  }
375
 
376
 
 
389
 
390
  assert action_name in NAME_TO_FUNC
391
 
392
+ print("RUN: " + action_name + "ACTION_INPUT: " + action_input)
393
  return NAME_TO_FUNC[action_name](purpose, task, history, action_input)
394
  else:
395
  history += "observation: The TOOL I tried to use returned an error, I need to select a tool from: (UPDATE-TASK, SEARCH_ENGINE, WEBSITE_SCRAPE, COMPLETE)\n"