Spaces:
Sleeping
Sleeping
Change name
Browse files- mini_agents.py +7 -7
mini_agents.py
CHANGED
@@ -23,7 +23,7 @@ audio_agent = CodeAgent(
|
|
23 |
model=audio_model,
|
24 |
tools=[audio_to_base64, noise_reduction, audio_segmentation, speaker_diarization],
|
25 |
max_steps=4,
|
26 |
-
name="
|
27 |
description="This agent is responsible for rocessing audio, transcribing audio and extracting text from it."
|
28 |
)
|
29 |
|
@@ -37,7 +37,7 @@ vlm_agent = CodeAgent(
|
|
37 |
model=vlm_model,
|
38 |
tools=[download_image, image_processing, object_detection, ocr_scan],
|
39 |
max_steps=4,
|
40 |
-
name="
|
41 |
description="This agent is responsible for downloading images, processing images, detecting objects in them and extracting text from them."
|
42 |
)
|
43 |
|
@@ -51,7 +51,7 @@ arithmetic_agent = CodeAgent(
|
|
51 |
model=arithmetic_model,
|
52 |
tools=[operate_two_numbers, convert_number],
|
53 |
max_steps=4,
|
54 |
-
name="
|
55 |
description="This agent is responsible for performing arithmetic operations on two numbers."
|
56 |
)
|
57 |
|
@@ -65,7 +65,7 @@ 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="
|
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."
|
70 |
)
|
71 |
|
@@ -75,7 +75,7 @@ multimodal_manager = CodeAgent(
|
|
75 |
tools=[sort_list],
|
76 |
max_steps=8,
|
77 |
planning_interval=4,
|
78 |
-
name="
|
79 |
description="This agent is responsible for managing the audio and vlm agents."
|
80 |
)
|
81 |
|
@@ -85,7 +85,7 @@ operation_manager = CodeAgent(
|
|
85 |
tools=[sort_list],
|
86 |
max_steps=8,
|
87 |
planning_interval=4,
|
88 |
-
name="
|
89 |
description="This agent is responsible for managing the arithmetic and pandas agents."
|
90 |
)
|
91 |
|
@@ -96,7 +96,7 @@ master_agent = CodeAgent(
|
|
96 |
verbose_level=2,
|
97 |
max_steps=16,
|
98 |
planning_interval=4,
|
99 |
-
name="
|
100 |
description="This agent is responsible for managing the multimodal and operation managers."
|
101 |
)
|
102 |
|
|
|
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."
|
28 |
)
|
29 |
|
|
|
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."
|
42 |
)
|
43 |
|
|
|
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."
|
56 |
)
|
57 |
|
|
|
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."
|
70 |
)
|
71 |
|
|
|
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 |
)
|
81 |
|
|
|
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 |
)
|
91 |
|
|
|
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 |
)
|
102 |
|