Spaces:
Sleeping
Sleeping
agent specs
Browse files- 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["
|
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",
|