huytofu92 commited on
Commit
3db6b5f
·
1 Parent(s): 5ec6e01

Remove base tools

Browse files
Files changed (1) hide show
  1. mini_agents.py +3 -1
mini_agents.py CHANGED
@@ -1,4 +1,5 @@
1
  from smolagents import CodeAgent, InferenceClientModel
 
2
  from tools import sort_list, operate_two_numbers, convert_number, load_dataframe_from_csv, load_dataframe_from_excel
3
  from tools import tavily_search_tool, read_python_file_from_path
4
  from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
@@ -156,6 +157,7 @@ class MasterAgentWrapper:
156
  sort_list,
157
  get_youtube_transcript_from_url,
158
  read_python_file_from_path,
 
159
  *community_tools,
160
  tavily_search_tool
161
  ]
@@ -164,7 +166,7 @@ class MasterAgentWrapper:
164
  model=master_model,
165
  managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
166
  tools=self.base_tools, # Initialize without browser tools
167
- add_base_tools=True,
168
  max_steps=20,
169
  additional_authorized_imports=AUTHORIZED_IMPORTS,
170
  verbosity_level=logging.INFO,
 
1
  from smolagents import CodeAgent, InferenceClientModel
2
+ from smolagents.default_tools import PythonInterpreterTool
3
  from tools import sort_list, operate_two_numbers, convert_number, load_dataframe_from_csv, load_dataframe_from_excel
4
  from tools import tavily_search_tool, read_python_file_from_path
5
  from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
 
157
  sort_list,
158
  get_youtube_transcript_from_url,
159
  read_python_file_from_path,
160
+ PythonInterpreterTool(),
161
  *community_tools,
162
  tavily_search_tool
163
  ]
 
166
  model=master_model,
167
  managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
168
  tools=self.base_tools, # Initialize without browser tools
169
+ add_base_tools=False,
170
  max_steps=20,
171
  additional_authorized_imports=AUTHORIZED_IMPORTS,
172
  verbosity_level=logging.INFO,