Spaces:
Sleeping
Sleeping
fix tools
Browse files- community_tools.py +0 -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,
|
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
|
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'],
|