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 |
|