huytofu92 commited on
Commit
8d5a7ee
·
1 Parent(s): 525242c

Change name

Browse files
Files changed (1) hide show
  1. 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="Audio Agent",
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="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,7 +51,7 @@ 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."
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="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,7 +75,7 @@ multimodal_manager = CodeAgent(
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,7 +85,7 @@ operation_manager = CodeAgent(
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,7 +96,7 @@ master_agent = CodeAgent(
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
 
 
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