huytofu92 commited on
Commit
69fc68f
·
1 Parent(s): 26a59c8
Files changed (1) hide show
  1. 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, read_python_file
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, read_python_file, *community_tools, *browser_tools, tavily_search_tool],
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,