diff options
Diffstat (limited to 'packages/api/src')
| -rw-r--r-- | packages/api/src/agent-manager.ts | 21 | ||||
| -rw-r--r-- | packages/api/src/routes/tabs.ts | 31 |
2 files changed, 43 insertions, 9 deletions
diff --git a/packages/api/src/agent-manager.ts b/packages/api/src/agent-manager.ts index 07af249..9d222fa 100644 --- a/packages/api/src/agent-manager.ts +++ b/packages/api/src/agent-manager.ts @@ -24,6 +24,7 @@ import { refreshAccountCredentials, refreshAccountCredentialsAsync, resolveApiKey, + getSetting, } from "@dispatch/core"; import type { PermissionManager } from "./permission-manager.js"; import { setConfigGetter } from "./routes/config.js"; @@ -47,6 +48,7 @@ interface TabAgent { keyId: string | null; modelId: string | null; taskList: TaskList; + _lastPermKey?: string; } export class AgentManager { @@ -193,10 +195,16 @@ export class AgentManager { const effectiveKeyId = keyId ?? tabAgent.keyId; const effectiveModelId = modelId ?? tabAgent.modelId; - // If the override differs from what the current agent was built with, invalidate the cache + // Read tool permission settings from DB (default: read=allow, edit=ask, bash=ask) + const permRead = getSetting("perm_read") !== "ask"; + const permEdit = getSetting("perm_edit") === "allow"; + const permBash = getSetting("perm_bash") === "allow"; + const permKey = `${permRead}:${permEdit}:${permBash}`; + + // If the override differs or permissions changed, invalidate the cached agent if ( tabAgent.agent && - (effectiveKeyId !== tabAgent.keyId || effectiveModelId !== tabAgent.modelId) + (effectiveKeyId !== tabAgent.keyId || effectiveModelId !== tabAgent.modelId || permKey !== tabAgent._lastPermKey) ) { tabAgent.agent = null; } @@ -204,13 +212,14 @@ export class AgentManager { if (!tabAgent.agent) { const workingDirectory = process.env.DISPATCH_WORKING_DIR ?? process.cwd(); + // Build tools list based on permission settings const tools = [ - createReadFileTool(workingDirectory), - createWriteFileTool(workingDirectory), - createListFilesTool(workingDirectory), - createRunShellTool(workingDirectory), + ...(permRead ? [createReadFileTool(workingDirectory), createListFilesTool(workingDirectory)] : []), + ...(permEdit ? [createWriteFileTool(workingDirectory)] : []), + ...(permBash ? [createRunShellTool(workingDirectory)] : []), createTaskListTool(tabAgent.taskList), ]; + tabAgent._lastPermKey = permKey; const ruleset = configToRuleset(this.config); diff --git a/packages/api/src/routes/tabs.ts b/packages/api/src/routes/tabs.ts index 7fe05d9..288cd51 100644 --- a/packages/api/src/routes/tabs.ts +++ b/packages/api/src/routes/tabs.ts @@ -10,6 +10,7 @@ import { getMessagesForTab, getSetting, setSetting, + deleteSetting, } from "@dispatch/core"; export const tabsRoutes = new Hono(); @@ -41,9 +42,15 @@ tabsRoutes.get("/settings/title-model", (c) => { }); tabsRoutes.put("/settings/title-model", async (c) => { - const body = await c.req.json<{ keyId?: string; modelId?: string }>(); - if (body.keyId) setSetting("title_model_key_id", body.keyId); - if (body.modelId) setSetting("title_model_id", body.modelId); + const body = await c.req.json<{ keyId?: string | null; modelId?: string | null }>(); + if (body.keyId !== undefined) { + if (body.keyId) setSetting("title_model_key_id", body.keyId); + else deleteSetting("title_model_key_id"); + } + if (body.modelId !== undefined) { + if (body.modelId) setSetting("title_model_id", body.modelId); + else deleteSetting("title_model_id"); + } return c.json({ success: true }); }); @@ -72,6 +79,24 @@ tabsRoutes.patch("/:id", async (c) => { return c.json(tab); }); +// ─── Settings ───────────────────────────────────────────────── + +tabsRoutes.get("/settings/:key", (c) => { + const key = c.req.param("key"); + const value = getSetting(key); + return c.json({ value }); +}); + +tabsRoutes.put("/settings/:key", async (c) => { + const key = c.req.param("key"); + const body = await c.req.json<{ value?: string }>(); + if (typeof body.value !== "string") { + return c.json({ error: "value is required" }, 400); + } + setSetting(key, body.value); + return c.json({ success: true }); +}); + tabsRoutes.delete("/:id", (c) => { const id = c.req.param("id"); const mgr = getAgentManager(); |
