Spaces:
Sleeping
Sleeping
fix agent configs
Browse files- mini_agents.py +3 -10
mini_agents.py
CHANGED
@@ -22,7 +22,6 @@ audio_model = InferenceClientModel(
|
|
22 |
audio_agent = CodeAgent(
|
23 |
model=audio_model,
|
24 |
tools=[audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization],
|
25 |
-
verbose=True,
|
26 |
max_steps=4,
|
27 |
name="Audio Agent",
|
28 |
description="This agent is responsible for rocessing audio, transcribing audio and extracting text from it."
|
@@ -37,7 +36,6 @@ vlm_model = InferenceClientModel(
|
|
37 |
vlm_agent = CodeAgent(
|
38 |
model=vlm_model,
|
39 |
tools=[download_image, image_processing, object_detection, ocr_scan],
|
40 |
-
verbose=True,
|
41 |
max_steps=4,
|
42 |
name="VLM Agent",
|
43 |
description="This agent is responsible for downloading images, processing images, detecting objects in them and extracting text from them."
|
@@ -52,7 +50,6 @@ arithmetic_model = InferenceClientModel(
|
|
52 |
arithmetic_agent = CodeAgent(
|
53 |
model=arithmetic_model,
|
54 |
tools=[operate_two_numbers, convert_number],
|
55 |
-
verbose=True,
|
56 |
max_steps=4,
|
57 |
name="Arithmetic Agent",
|
58 |
description="This agent is responsible for performing arithmetic operations on two numbers."
|
@@ -67,7 +64,6 @@ pandas_model = InferenceClientModel(
|
|
67 |
pandas_agent = CodeAgent(
|
68 |
model=pandas_model,
|
69 |
tools=[to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby],
|
70 |
-
verbose=True,
|
71 |
max_steps=4,
|
72 |
name="Pandas Agent",
|
73 |
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."
|
@@ -77,9 +73,8 @@ multimodal_manager = CodeAgent(
|
|
77 |
model=MODEL_CHOICES["code"][0],
|
78 |
managed_agents=[audio_agent, vlm_agent],
|
79 |
tools=[sort_list],
|
80 |
-
verbose=True,
|
81 |
max_steps=8,
|
82 |
-
|
83 |
name="Multimodal Manager",
|
84 |
description="This agent is responsible for managing the audio and vlm agents."
|
85 |
)
|
@@ -88,9 +83,8 @@ operation_manager = CodeAgent(
|
|
88 |
model=MODEL_CHOICES["code"][0],
|
89 |
managed_agents=[arithmetic_agent, pandas_agent],
|
90 |
tools=[sort_list],
|
91 |
-
verbose=True,
|
92 |
max_steps=8,
|
93 |
-
|
94 |
name="Operation Manager",
|
95 |
description="This agent is responsible for managing the arithmetic and pandas agents."
|
96 |
)
|
@@ -99,10 +93,9 @@ master_agent = CodeAgent(
|
|
99 |
model=MODEL_CHOICES["code"][0],
|
100 |
managed_agents=[multimodal_manager, operation_manager],
|
101 |
tools=[sort_list],
|
102 |
-
verbose=True,
|
103 |
verbose_level=2,
|
104 |
max_steps=16,
|
105 |
-
|
106 |
name="Master Agent",
|
107 |
description="This agent is responsible for managing the multimodal and operation managers."
|
108 |
)
|
|
|
22 |
audio_agent = CodeAgent(
|
23 |
model=audio_model,
|
24 |
tools=[audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization],
|
|
|
25 |
max_steps=4,
|
26 |
name="Audio Agent",
|
27 |
description="This agent is responsible for rocessing audio, transcribing audio and extracting text from it."
|
|
|
36 |
vlm_agent = CodeAgent(
|
37 |
model=vlm_model,
|
38 |
tools=[download_image, image_processing, object_detection, ocr_scan],
|
|
|
39 |
max_steps=4,
|
40 |
name="VLM Agent",
|
41 |
description="This agent is responsible for downloading images, processing images, detecting objects in them and extracting text from them."
|
|
|
50 |
arithmetic_agent = CodeAgent(
|
51 |
model=arithmetic_model,
|
52 |
tools=[operate_two_numbers, convert_number],
|
|
|
53 |
max_steps=4,
|
54 |
name="Arithmetic Agent",
|
55 |
description="This agent is responsible for performing arithmetic operations on two numbers."
|
|
|
64 |
pandas_agent = CodeAgent(
|
65 |
model=pandas_model,
|
66 |
tools=[to_dataframe, to_json, get_dataframe_data, get_dataframe_column, get_dataframe_row, get_dataframe_groupby],
|
|
|
67 |
max_steps=4,
|
68 |
name="Pandas Agent",
|
69 |
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."
|
|
|
73 |
model=MODEL_CHOICES["code"][0],
|
74 |
managed_agents=[audio_agent, vlm_agent],
|
75 |
tools=[sort_list],
|
|
|
76 |
max_steps=8,
|
77 |
+
planning_interval=4,
|
78 |
name="Multimodal Manager",
|
79 |
description="This agent is responsible for managing the audio and vlm agents."
|
80 |
)
|
|
|
83 |
model=MODEL_CHOICES["code"][0],
|
84 |
managed_agents=[arithmetic_agent, pandas_agent],
|
85 |
tools=[sort_list],
|
|
|
86 |
max_steps=8,
|
87 |
+
planning_interval=4,
|
88 |
name="Operation Manager",
|
89 |
description="This agent is responsible for managing the arithmetic and pandas agents."
|
90 |
)
|
|
|
93 |
model=MODEL_CHOICES["code"][0],
|
94 |
managed_agents=[multimodal_manager, operation_manager],
|
95 |
tools=[sort_list],
|
|
|
96 |
verbose_level=2,
|
97 |
max_steps=16,
|
98 |
+
planning_interval=4,
|
99 |
name="Master Agent",
|
100 |
description="This agent is responsible for managing the multimodal and operation managers."
|
101 |
)
|