huytofu92 commited on
Commit
20c68a2
·
1 Parent(s): 028764a

fix agent configs

Browse files
Files changed (1) hide show
  1. 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
- planning_steps=4,
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
- planning_steps=4,
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
- planning_steps=4,
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
  )