New heterogeneous managed agent

agent_server/chat_completions.py

@@ -22,7 +22,9 @@ from agents.json_tool_calling_agents import (
 )
 
 from agents.agent_with_custom_beam_design_tools import generate_beam_agent
-
+from agents.manager_with_heterogeneous_agents import (
+    generate_manager_with_heterogeneous_agents,
+)
 
 # Model name from env
 MODEL_NAME = os.getenv("MODEL_NAME", "Qwen/Qwen3-1.7B")
@@ -85,6 +87,8 @@ def agent_for_model(model_name: str):
         return generate_generator_with_managed_critic()
     if model_name == "custom-agent-with-beam-design-tools":
         return generate_beam_agent()
+    if model_name == "manager-with-heterogeneous-agents":
+        return generate_manager_with_heterogeneous_agents()
     raise ValueError(f"Unknown model id: {model_name}")
 
 

agent_server/models.py

@@ -13,46 +13,52 @@ def models_payload() -> dict:
         "object": "list",
         "data": [
             {
-                "id": "custom-agent-with-beam-design-tools",
+                "id": MODEL_NAME,
                 "object": "model",
                 "created": now,
-                "owned_by": "you",
+                "owned_by": "upstream",
             },
             {
-                "id": "generator-with-managed-critic",
+                "id": f"{MODEL_NAME}-nothink",
+                "object": "model",
+                "created": now,
+                "owned_by": "upstream",
+            },
+            {
+                "id": "code-writing-agent-without-tools",
                 "object": "model",
                 "created": now,
                 "owned_by": "you",
             },
             {
-                "id": "tool-calling-agent-with-search-and-code",
+                "id": "code-writing-agent-with-search",
                 "object": "model",
                 "created": now,
                 "owned_by": "you",
             },
             {
-                "id": "code-writing-agent-without-tools",
+                "id": "tool-calling-agent-with-search-and-code",
                 "object": "model",
                 "created": now,
                 "owned_by": "you",
             },
             {
-                "id": "code-writing-agent-with-search",
+                "id": "custom-agent-with-beam-design-tools",
                 "object": "model",
                 "created": now,
                 "owned_by": "you",
             },
             {
-                "id": MODEL_NAME,
+                "id": "generator-with-managed-critic",
                 "object": "model",
                 "created": now,
-                "owned_by": "upstream",
+                "owned_by": "you",
             },
             {
-                "id": f"{MODEL_NAME}-nothink",
+                "id": "manager-with-heterogeneous-agents",
                 "object": "model",
                 "created": now,
-                "owned_by": "upstream",
+                "owned_by": "you",
             },
         ],
     }

agents/manager_with_heterogeneous_agents.py

@@ -0,0 +1,42 @@
+from __future__ import annotations
+
+import os
+
+import smolagents.models
+from smolagents import ToolCallingAgent
+import agents.code_writing_agents
+import agents.agent_with_custom_beam_design_tools
+
+INSTRUCTIONS = (
+    "You are a manager agent with access to two specialized subagents."
+    "Your primary task is to delegate tasks to the appropriate subagent based on their expertise."
+)
+AGENTS = [
+    agents.code_writing_agents.generate_code_writing_agent_with_search(),
+    agents.agent_with_custom_beam_design_tools.generate_beam_agent(),
+]
+
+
+# ---------------- Factory ----------------
+def generate_manager_with_heterogeneous_agents() -> ToolCallingAgent:
+    """
+    Returns a CodeAgent (generator) that manages a critic sub-agent.
+    The critic is exposed to the generator as a managed agent (callable like a tool).
+    """
+
+    model = smolagents.models.OpenAIServerModel(
+        model_id=os.getenv("AGENT_MODEL", ""),
+        api_base=os.getenv("UPSTREAM_OPENAI_BASE", "").rstrip("/"),
+        api_key=os.getenv("OPENAI_API_KEY"),
+    )
+
+    generator = ToolCallingAgent(
+        tools=[],  # keep toolbox minimal
+        model=model,
+        name="manager_with_heterogeneous_agents",
+        managed_agents=AGENTS,
+        instructions=INSTRUCTIONS,
+        add_base_tools=False,
+        max_steps=2,
+    )
+    return generator