From 31438fdf2af8bdc72e041f375afc0fd896be17a4 Mon Sep 17 00:00:00 2001 From: Mario Zechner Date: Sun, 4 Jan 2026 18:47:57 +0100 Subject: [PATCH] fix(tools): wrap ALL registry tools with hooks, not just active ones wrappedToolRegistry was only containing activeToolsArray (4 tools). Now wraps all tools from the registry so hooks can enable any tool. --- packages/coding-agent/src/core/sdk.ts | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/packages/coding-agent/src/core/sdk.ts b/packages/coding-agent/src/core/sdk.ts index 610f2717..a812320f 100644 --- a/packages/coding-agent/src/core/sdk.ts +++ b/packages/coding-agent/src/core/sdk.ts @@ -656,9 +656,11 @@ export async function createAgentSession(options: CreateAgentSessionOptions = {} let wrappedToolRegistry: Map | undefined; if (hookRunner) { activeToolsArray = wrapToolsWithHooks(activeToolsArray as AgentTool[], hookRunner); - // Also create a wrapped version of the registry for setTools + // Wrap ALL registry tools (not just active) so hooks can enable any + const allRegistryTools = Array.from(toolRegistry.values()); + const wrappedAllTools = wrapToolsWithHooks(allRegistryTools, hookRunner); wrappedToolRegistry = new Map(); - for (const tool of activeToolsArray as AgentTool[]) { + for (const tool of wrappedAllTools) { wrappedToolRegistry.set(tool.name, tool); } }