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,
|