huytofu92 commited on
Commit
226e692
·
1 Parent(s): bd5c59c

agent specs

Browse files
Files changed (1) hide show
  1. mini_agents.py +3 -3
mini_agents.py CHANGED
@@ -10,7 +10,8 @@ MODEL_CHOICES = {
10
  "vlm": ["Qwen/Qwen2.5-VL-7B-Instruct"],
11
  "code": ["Qwen/Qwen2.5-Coder-32B-Instruct"],
12
  "arithmetic": ["Qwen/Qwen2.5-Coder-7B-Instruct"],
13
- "pandas": ["Qwen/Qwen2.5-Coder-7B-Instruct"]
 
14
  }
15
 
16
  audio_model = InferenceClientModel(
@@ -90,10 +91,9 @@ operation_manager = CodeAgent(
90
  )
91
 
92
  master_agent = CodeAgent(
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",
 
10
  "vlm": ["Qwen/Qwen2.5-VL-7B-Instruct"],
11
  "code": ["Qwen/Qwen2.5-Coder-32B-Instruct"],
12
  "arithmetic": ["Qwen/Qwen2.5-Coder-7B-Instruct"],
13
+ "pandas": ["Qwen/Qwen2.5-Coder-7B-Instruct"],
14
+ "master": ["Qwen/Qwen2.5-Coder-32B-Instruct", "deepseek-ai/DeepSeek-V3-0324"]
15
  }
16
 
17
  audio_model = InferenceClientModel(
 
91
  )
92
 
93
  master_agent = CodeAgent(
94
+ model=MODEL_CHOICES["master"][0],
95
  managed_agents=[multimodal_manager, operation_manager],
96
  tools=[sort_list],
 
97
  max_steps=16,
98
  planning_interval=4,
99
  name="master_agent",