diff --git a/src/shared/agent-tool-restrictions.test.ts b/src/shared/agent-tool-restrictions.test.ts new file mode 100644 index 00000000..96daad77 --- /dev/null +++ b/src/shared/agent-tool-restrictions.test.ts @@ -0,0 +1,33 @@ +import { describe, expect, test } from "bun:test" +import { + getAgentToolRestrictions, + hasAgentToolRestrictions, +} from "./agent-tool-restrictions" + +describe("agent-tool-restrictions", () => { + test("athena restrictions include call_omo_agent", () => { + //#given + //#when + const restrictions = getAgentToolRestrictions("athena") + //#then + expect(restrictions.write).toBe(false) + expect(restrictions.edit).toBe(false) + expect(restrictions.call_omo_agent).toBe(false) + }) + + test("council-member restrictions include call_omo_agent", () => { + //#given + //#when + const restrictions = getAgentToolRestrictions("council-member") + //#then + expect(restrictions.call_omo_agent).toBe(false) + }) + + test("hasAgentToolRestrictions returns true for athena", () => { + //#given + //#when + const result = hasAgentToolRestrictions("athena") + //#then + expect(result).toBe(true) + }) +}) diff --git a/src/shared/agent-tool-restrictions.ts b/src/shared/agent-tool-restrictions.ts index 6c7e2257..9f453921 100644 --- a/src/shared/agent-tool-restrictions.ts +++ b/src/shared/agent-tool-restrictions.ts @@ -14,7 +14,7 @@ const EXPLORATION_AGENT_DENYLIST: Record = { } const ATHENA_RESTRICTIONS = permissionToToolBooleans( - createAgentToolRestrictions(["write", "edit"]).permission + createAgentToolRestrictions(["write", "edit", "call_omo_agent"]).permission ) const AGENT_RESTRICTIONS: Record> = {