Spaces:
Sleeping
Sleeping
Fix import and agent types
Browse files- community_tools.py +2 -2
- mini_agents.py +8 -4
community_tools.py
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
from
|
2 |
-
from
|
3 |
from smolagents.tools import Tool
|
4 |
|
5 |
import os
|
|
|
1 |
+
from langchain_google_community import GooglePlacesTool
|
2 |
+
from langchain_community.agent_toolkits.load_tools import load_tools
|
3 |
from smolagents.tools import Tool
|
4 |
|
5 |
import os
|
mini_agents.py
CHANGED
@@ -1,10 +1,13 @@
|
|
1 |
-
from smolagents import CodeAgent, InferenceClientModel
|
2 |
from tools import sort_list, operate_two_numbers, convert_number
|
3 |
from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
|
4 |
from vlm_tools import download_image, image_processing, object_detection, ocr_scan
|
5 |
from audio_tools import audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization
|
6 |
from community_tools import community_tools
|
7 |
import os
|
|
|
|
|
|
|
8 |
|
9 |
MODEL_CHOICES = {
|
10 |
"audio": ["whisper-large-v3"],
|
@@ -71,7 +74,7 @@ pandas_agent = CodeAgent(
|
|
71 |
description="This agent is responsible for converting data to a dataframe, performing pandas operations on such dataframe and converting the dataframe back to a json or a csv file."
|
72 |
)
|
73 |
|
74 |
-
multimodal_manager =
|
75 |
model=MODEL_CHOICES["code"][0],
|
76 |
managed_agents=[audio_agent, vlm_agent],
|
77 |
tools=[sort_list],
|
@@ -81,7 +84,7 @@ multimodal_manager = MultiStepAgent(
|
|
81 |
description="This agent is responsible for managing the audio and vlm agents."
|
82 |
)
|
83 |
|
84 |
-
operation_manager =
|
85 |
model=MODEL_CHOICES["code"][0],
|
86 |
managed_agents=[arithmetic_agent, pandas_agent],
|
87 |
tools=[sort_list],
|
@@ -91,11 +94,12 @@ operation_manager = MultiStepAgent(
|
|
91 |
description="This agent is responsible for managing the arithmetic and pandas agents."
|
92 |
)
|
93 |
|
94 |
-
master_agent =
|
95 |
model=MODEL_CHOICES["master"][0],
|
96 |
managed_agents=[multimodal_manager, operation_manager],
|
97 |
tools=[sort_list, *community_tools],
|
98 |
max_steps=16,
|
|
|
99 |
planning_interval=4,
|
100 |
name="master_agent",
|
101 |
description="This agent is responsible for managing the multimodal and operation managers."
|
|
|
1 |
+
from smolagents import CodeAgent, InferenceClientModel
|
2 |
from tools import sort_list, operate_two_numbers, convert_number
|
3 |
from tools import to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby
|
4 |
from vlm_tools import download_image, image_processing, object_detection, ocr_scan
|
5 |
from audio_tools import audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization
|
6 |
from community_tools import community_tools
|
7 |
import os
|
8 |
+
import logging
|
9 |
+
|
10 |
+
logging.basicConfig(level=logging.DEBUG)
|
11 |
|
12 |
MODEL_CHOICES = {
|
13 |
"audio": ["whisper-large-v3"],
|
|
|
74 |
description="This agent is responsible for converting data to a dataframe, performing pandas operations on such dataframe and converting the dataframe back to a json or a csv file."
|
75 |
)
|
76 |
|
77 |
+
multimodal_manager = CodeAgent(
|
78 |
model=MODEL_CHOICES["code"][0],
|
79 |
managed_agents=[audio_agent, vlm_agent],
|
80 |
tools=[sort_list],
|
|
|
84 |
description="This agent is responsible for managing the audio and vlm agents."
|
85 |
)
|
86 |
|
87 |
+
operation_manager = CodeAgent(
|
88 |
model=MODEL_CHOICES["code"][0],
|
89 |
managed_agents=[arithmetic_agent, pandas_agent],
|
90 |
tools=[sort_list],
|
|
|
94 |
description="This agent is responsible for managing the arithmetic and pandas agents."
|
95 |
)
|
96 |
|
97 |
+
master_agent = CodeAgent(
|
98 |
model=MODEL_CHOICES["master"][0],
|
99 |
managed_agents=[multimodal_manager, operation_manager],
|
100 |
tools=[sort_list, *community_tools],
|
101 |
max_steps=16,
|
102 |
+
verbosity_level=logging.DEBUG,
|
103 |
planning_interval=4,
|
104 |
name="master_agent",
|
105 |
description="This agent is responsible for managing the multimodal and operation managers."
|