huytofu92 commited on
Commit
c6ae84c
·
1 Parent(s): 6056455
Files changed (2) hide show
  1. community_tools.py +0 -2
  2. mini_agents.py +2 -2
community_tools.py CHANGED
@@ -28,5 +28,3 @@ def get_youtube_transcript_from_url(video_url: str)->str:
28
  docs = youtube_loader.load()
29
  transcript = docs[0].page_content
30
  return transcript
31
-
32
- get_youtube_transcript_from_url_tool = get_youtube_transcript_from_url()
 
28
  docs = youtube_loader.load()
29
  transcript = docs[0].page_content
30
  return transcript
 
 
mini_agents.py CHANGED
@@ -4,7 +4,7 @@ from tools import tavily_search_tool, visit_webpage_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 download_image, image_processing, object_detection_tool, ocr_scan_tool, extract_frames_from_video
6
  from audio_tools import transcribe_audio_tool, audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization
7
- from community_tools import community_tools, get_youtube_transcript_from_url_tool
8
  import os
9
  import logging
10
  import yaml
@@ -124,7 +124,7 @@ master_model = InferenceClientModel(
124
  master_agent = CodeAgent(
125
  model=master_model,
126
  managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
127
- tools=[sort_list, *community_tools, tavily_search_tool, visit_webpage_tool, get_youtube_transcript_from_url_tool],
128
  add_base_tools=True,
129
  max_steps=20,
130
  additional_authorized_imports=["math", "pandas", "json", "numpy", "io", "os", "logging", "yaml", "pyplot", "matplotlib", 'hmmlearn', 'pickle', 'sklearn', 'scipy', 'datetime', 'typing'],
 
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 download_image, image_processing, object_detection_tool, ocr_scan_tool, extract_frames_from_video
6
  from audio_tools import transcribe_audio_tool, audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization
7
+ from community_tools import community_tools, get_youtube_transcript_from_url
8
  import os
9
  import logging
10
  import yaml
 
124
  master_agent = CodeAgent(
125
  model=master_model,
126
  managed_agents=[audio_agent, vlm_agent, arithmetic_agent, pandas_agent],
127
+ tools=[sort_list, get_youtube_transcript_from_url, *community_tools, tavily_search_tool, visit_webpage_tool],
128
  add_base_tools=True,
129
  max_steps=20,
130
  additional_authorized_imports=["math", "pandas", "json", "numpy", "io", "os", "logging", "yaml", "pyplot", "matplotlib", 'hmmlearn', 'pickle', 'sklearn', 'scipy', 'datetime', 'typing'],