refactor(athena): consolidate tool restriction deny lists to direct boolean records

This commit is contained in:
ismeth 2026-02-20 14:29:34 +01:00 committed by YeonGyu-Kim
parent f9fdd08481
commit 21202ee877

View File

@ -4,8 +4,6 @@
* true = tool allowed, false = tool denied.
*/
import { createAgentToolRestrictions } from "./permission-compat"
const EXPLORATION_AGENT_DENYLIST: Record<string, boolean> = {
write: false,
edit: false,
@ -13,10 +11,6 @@ const EXPLORATION_AGENT_DENYLIST: Record<string, boolean> = {
call_omo_agent: false,
}
const ATHENA_RESTRICTIONS = permissionToToolBooleans(
createAgentToolRestrictions(["write", "edit", "call_omo_agent"]).permission
)
const AGENT_RESTRICTIONS: Record<string, Record<string, boolean>> = {
explore: EXPLORATION_AGENT_DENYLIST,
@ -49,7 +43,11 @@ const AGENT_RESTRICTIONS: Record<string, Record<string, boolean>> = {
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<string, Record<string, boolean>> = {
},
}
function permissionToToolBooleans(
permission: Record<string, "ask" | "allow" | "deny">
): Record<string, boolean> {
return Object.fromEntries(
Object.entries(permission).map(([tool, value]) => [tool, value === "allow"])
)
}
export function getAgentToolRestrictions(agentName: string): Record<string, boolean> {
return AGENT_RESTRICTIONS[agentName]
?? Object.entries(AGENT_RESTRICTIONS).find(([key]) => key.toLowerCase() === agentName.toLowerCase())?.[1]