Spaces:
Sleeping
Sleeping
Remove base tools
Browse files- 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=
|
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,
|