Spaces:
Sleeping
Sleeping
Fix bug
Browse files- mini_agents.py +2 -2
mini_agents.py
CHANGED
@@ -1,6 +1,6 @@
|
|
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,
|
4 |
from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
|
5 |
from vlm_tools import image_processing, object_detection_tool, ocr_scan_tool, extract_images_from_video, get_image_from_file_path, get_video_from_file_path
|
6 |
from audio_tools import transcribe_audio_tool, get_audio_from_file_path, noise_reduction, audio_segmentation, speaker_diarization
|
@@ -144,7 +144,7 @@ master_model = InferenceClientModel(
|
|
144 |
master_agent = CodeAgent(
|
145 |
model=master_model,
|
146 |
managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
|
147 |
-
tools=[sort_list, get_youtube_transcript_from_url,
|
148 |
add_base_tools=True,
|
149 |
max_steps=20,
|
150 |
additional_authorized_imports=AUTHORIZED_IMPORTS,
|
|
|
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
|
5 |
from vlm_tools import image_processing, object_detection_tool, ocr_scan_tool, extract_images_from_video, get_image_from_file_path, get_video_from_file_path
|
6 |
from audio_tools import transcribe_audio_tool, get_audio_from_file_path, noise_reduction, audio_segmentation, speaker_diarization
|
|
|
144 |
master_agent = CodeAgent(
|
145 |
model=master_model,
|
146 |
managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
|
147 |
+
tools=[sort_list, get_youtube_transcript_from_url, read_python_file_from_path, *community_tools, *browser_tools, tavily_search_tool],
|
148 |
add_base_tools=True,
|
149 |
max_steps=20,
|
150 |
additional_authorized_imports=AUTHORIZED_IMPORTS,
|