Leonardo Kamigauti commited on
Commit
8885992
·
1 Parent(s): 2be2acd

Add UserInputTool

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -98,12 +98,13 @@ agent = CodeAgent(
98
  model=model,
99
  tools=[final_answer,
100
  search_kaggle_datasets,
 
101
  download_kaggle_dataset,
102
  image_generation_tool],
103
  max_steps=6,
104
  verbosity_level=1,
105
  grammar=None,
106
- planning_interval=None,
107
  name=None,
108
  description=None,
109
  prompt_templates=prompt_templates,
 
98
  model=model,
99
  tools=[final_answer,
100
  search_kaggle_datasets,
101
+ UserInputTool(),
102
  download_kaggle_dataset,
103
  image_generation_tool],
104
  max_steps=6,
105
  verbosity_level=1,
106
  grammar=None,
107
+ planning_interval=2,
108
  name=None,
109
  description=None,
110
  prompt_templates=prompt_templates,