diff --git a/src/agents/atlas/agent.ts b/src/agents/atlas/agent.ts index 6f968b783..c583ada5a 100644 --- a/src/agents/atlas/agent.ts +++ b/src/agents/atlas/agent.ts @@ -17,7 +17,6 @@ import type { AvailableAgent, AvailableSkill, AvailableCategory } from "../dynam import { buildCategorySkillsDelegationGuide } from "../dynamic-agent-prompt-builder" import type { CategoryConfig } from "../../config/schema" import { mergeCategories } from "../../shared/merge-categories" -import { createAgentToolRestrictions } from "../../shared/permission-compat" import { getDefaultAtlasPrompt } from "./default" import { getGptAtlasPrompt } from "./gpt" @@ -100,11 +99,6 @@ function buildDynamicOrchestratorPrompt(ctx?: OrchestratorContext): string { } export function createAtlasAgent(ctx: OrchestratorContext): AgentConfig { - const restrictions = createAgentToolRestrictions([ - "task", - "call_omo_agent", - ]) - const baseConfig = { description: "Orchestrates work via task() to complete ALL tasks in a todo list until fully done. (Atlas - OhMyOpenCode)", @@ -113,7 +107,6 @@ export function createAtlasAgent(ctx: OrchestratorContext): AgentConfig { temperature: 0.1, prompt: buildDynamicOrchestratorPrompt(ctx), color: "#10B981", - ...restrictions, } return baseConfig as AgentConfig diff --git a/src/agents/tool-restrictions.test.ts b/src/agents/tool-restrictions.test.ts index 685acbc1f..85facdc54 100644 --- a/src/agents/tool-restrictions.test.ts +++ b/src/agents/tool-restrictions.test.ts @@ -4,6 +4,7 @@ import { createLibrarianAgent } from "./librarian" import { createExploreAgent } from "./explore" import { createMomusAgent } from "./momus" import { createMetisAgent } from "./metis" +import { createAtlasAgent } from "./atlas" const TEST_MODEL = "anthropic/claude-sonnet-4-5" @@ -96,4 +97,18 @@ describe("read-only agent tool restrictions", () => { } }) }) + + describe("Atlas", () => { + test("allows delegation tools for orchestration", () => { + // given + const agent = createAtlasAgent({ model: TEST_MODEL }) + + // when + const permission = (agent.permission ?? {}) as Record + + // then + expect(permission["task"]).toBeUndefined() + expect(permission["call_omo_agent"]).toBeUndefined() + }) + }) })