diff --git a/src/daemon/dispatcher.py b/src/daemon/dispatcher.py index 5f84fb5..ec2e493 100644 --- a/src/daemon/dispatcher.py +++ b/src/daemon/dispatcher.py @@ -45,14 +45,17 @@ class Dispatcher: registered_agents: Optional[List[str]] = None, spawner: Optional[Any] = None, counter: Optional[Any] = None, + capability_map: Optional[Dict[str, List[str]]] = None, ): """ Args: registered_agents: 项目注册的 Agent ID 列表 spawner: AgentSpawner 实例(执行实际 spawn) counter: ActiveAgentCounter 实例(控制并发) + capability_map: task_type → [agent_id] 映射(无 assignee 时的自动分配) """ self.registered_agents = set(registered_agents or []) + self.capability_map = capability_map or {} self.spawner = spawner self.counter = counter