diff --git a/src/shared/agent-tool-restrictions.ts b/src/shared/agent-tool-restrictions.ts index 9f453921..a4fc820a 100644 --- a/src/shared/agent-tool-restrictions.ts +++ b/src/shared/agent-tool-restrictions.ts @@ -4,8 +4,6 @@ * true = tool allowed, false = tool denied. */ -import { createAgentToolRestrictions } from "./permission-compat" - const EXPLORATION_AGENT_DENYLIST: Record = { write: false, edit: false, @@ -13,10 +11,6 @@ const EXPLORATION_AGENT_DENYLIST: Record = { call_omo_agent: false, } -const ATHENA_RESTRICTIONS = permissionToToolBooleans( - createAgentToolRestrictions(["write", "edit", "call_omo_agent"]).permission -) - const AGENT_RESTRICTIONS: Record> = { explore: EXPLORATION_AGENT_DENYLIST, @@ -49,7 +43,11 @@ const AGENT_RESTRICTIONS: Record> = { task: false, }, - athena: ATHENA_RESTRICTIONS, + athena: { + write: false, + edit: false, + call_omo_agent: false, + }, "council-member": { write: false, @@ -59,14 +57,6 @@ const AGENT_RESTRICTIONS: Record> = { }, } -function permissionToToolBooleans( - permission: Record -): Record { - return Object.fromEntries( - Object.entries(permission).map(([tool, value]) => [tool, value === "allow"]) - ) -} - export function getAgentToolRestrictions(agentName: string): Record { return AGENT_RESTRICTIONS[agentName] ?? Object.entries(AGENT_RESTRICTIONS).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1]