@code-yeongyu/senpi 2026.8.11 → 2026.8.12

package/dist/core/extensions/builtin/tool-search/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/tool-search/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,mBAAmB,EAAE,MAAM,2CAA2C,CAAC;AAEhF,OAAO,EAAE,gCAAgC,EAAE,4BAA4B,EAAE,MAAM,oBAAoB,CAAC;AACpG,OAAO,EAAE,oBAAoB,EAAE,8BAA8B,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACvG,OAAO,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAExE,MAAM,UAAU,yBAAyB,CAAC,OAA0B;IACnE,OAAO,CAAC,EAAgB,EAAQ,EAAE;QACjC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;QAExB,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,CAAC,MAAM,EAAE,GAAG,EAAE,EAAE;YACtC,OAAO,CAAC,YAAY,EAAE,CAAC;YACvB,OAAO,CAAC,yBAAyB,CAAC,GAAG,CAAC,cAAc,CAAC,UAAU,EAAE,CAAC,CAAC;QACpE,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,CAAC,KAAK,EAAE,EAAE;YAC1B,OAAO,CAAC,yBAAyB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,yBAAyB,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,OAAO,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC3E,IAAI,cAAc,GAAG,KAAK,CAAC;QAC3B,OAAO,CAAC,iBAAiB,CAAC,GAAG,EAAE;YAC9B,IAAI,cAAc;gBAAE,OAAO;YAC3B,cAAc,GAAG,IAAI,CAAC;YACtB,EAAE,CAAC,YAAY,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC,CAAC;QAChD,CAAC,CAAC,CAAC;QAEH,MAAM,aAAa,GAAG,IAAI,gCAAgC,CAAC;YAC1D,OAAO,EAAE,GAAG,EAAE;gBACb,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,CAAC;gBAC5C,OAAO,OAAO;qBACZ,UAAU,EAAE;qBACZ,IAAI,CACJ,CAAC,GAAG,EAAE,EAAE,CACP,CAAC,GAAG,CAAC,MAAM,KAAK,WAAW,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;oBACrD,CAAC,GAAG,CAAC,MAAM,KAAK,KAAK,IAAI,4BAA4B,EAAE,CAAC,CACzD,CAAC;YACJ,CAAC;YACD,UAAU,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,UAAU,EAAE;YACtC,iBAAiB,EAAE,CAAC,IAAI,EAAE,EAAE;gBAC3B,MAAM,IAAI,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;gBAC3E,OAAO,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,IAAI,CAAC,WAAW,EAAE,UAAU,EAAE,IAAI,CAAC,UAAU,EAAE,CAAC;YACxG,CAAC;YACD,YAAY,EAAE,CAAC,IAAI,EAAE,EAAE;gBACtB,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;gBAC9E,IAAI,GAAG,EAAE,MAAM,KAAK,KAAK;oBAAE,OAAO,4BAA4B,EAAE,CAAC;gBACjE,OAAO,GAAG,EAAE,MAAM,KAAK,WAAW,IAAI,CAAC,EAAE,CAAC,cAAc,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAC3E,CAAC;YACD,cAAc,EAAE,qBAAqB;SACrC,CAAC,CAAC;QACH,EAAE,CAAC,EAAE,CAAC,yBAAyB,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,aAAa,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,EAAE,GAAG,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC;QAClH,EAAE,CAAC,EAAE,CAAC,yBAAyB,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,aAAa,CAAC,kBAAkB,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7F,CAAC,CAAC;AACH,CAAC;AAED,SAAS,gBAAgB;IACxB,IAAI,CAAC;QACJ,mBAAmB,CAAC,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;QACrC,OAAO,IAAI,CAAC;IACb,CAAC;IAAC,MAAM,CAAC;QACR,OAAO,KAAK,CAAC;IACd,CAAC;AACF,CAAC;AAED,MAAM,CAAC,OAAO,UAAU,mBAAmB,CAAC,EAAgB;IAC3D,MAAM,OAAO,GAAG;QACf,WAAW,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE;QACnC,cAAc,EAAE,GAAG,EAAE,CAAC,EAAE,CAAC,cAAc,EAAE;QACzC,cAAc,EAAE,CAAC,KAAwB,EAAE,EAAE,CAAC,EAAE,CAAC,cAAc,CAAC,CAAC,GAAG,KAAK,CAAC,CAAC;KAC3E,CAAC;IACF,MAAM,YAAY,GAAG,gBAAgB,EAAE,CAAC;IACxC,MAAM,OAAO,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,oBAAoB,CAAC,OAAO,CAAC,CAAC;IAC9F,IAAI,YAAY;QAAE,8BAA8B,CAAC,OAAO,CAAC,CAAC;IAC1D,OAAO,yBAAyB,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC,CAAC;AAC/C,CAAC;AAED,OAAO,EAAE,oBAAoB,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AACvE,OAAO,EAAE,oBAAoB,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC","sourcesContent":["import { bindToProviderScope } from \"@earendil-works/pi-ai/node/provider-scope\";\nimport type { ExtensionAPI, ExtensionFactory } from \"../../types.ts\";\nimport { AnthropicNativeToolSearchAdapter, isMcpNativeToolSearchEnabled } from \"./native-search.ts\";\nimport { getToolSearchService, installScopedToolSearchService, ToolSearchService } from \"./service.ts\";\nimport { createToolSearchTool, TOOL_SEARCH_TOOL_NAME } from \"./tool.ts\";\n\nexport function createToolSearchExtension(service: ToolSearchService): ExtensionFactory {\n\treturn (pi: ExtensionAPI): void => {\n\t\tservice.bindRuntime(pi);\n\n\t\tpi.on(\"session_start\", (_event, ctx) => {\n\t\t\tservice.beginSession();\n\t\t\tservice.maybeRehydrateFromHistory(ctx.sessionManager.getEntries());\n\t\t});\n\t\tpi.on(\"context\", (event) => {\n\t\t\tservice.maybeRehydrateFromHistory(event.messages);\n\t\t});\n\t\tpi.registerLazyToolActivator((toolName) => service.activateTool(toolName));\n\t\tlet toolRegistered = false;\n\t\tservice.bindToolRegistrar(() => {\n\t\t\tif (toolRegistered) return;\n\t\t\ttoolRegistered = true;\n\t\t\tpi.registerTool(createToolSearchTool(service));\n\t\t});\n\n\t\tconst nativeAdapter = new AnthropicNativeToolSearchAdapter({\n\t\t\tenabled: () => {\n\t\t\t\tconst active = new Set(pi.getActiveTools());\n\t\t\t\treturn service\n\t\t\t\t\t.getCatalog()\n\t\t\t\t\t.some(\n\t\t\t\t\t\t(doc) =>\n\t\t\t\t\t\t\t(doc.source === \"extension\" && !active.has(doc.name)) ||\n\t\t\t\t\t\t\t(doc.source === \"mcp\" && isMcpNativeToolSearchEnabled()),\n\t\t\t\t\t);\n\t\t\t},\n\t\t\tgetCatalog: () => service.getCatalog(),\n\t\t\tgetToolDefinition: (name) => {\n\t\t\t\tconst tool = pi.getAllTools().find((candidate) => candidate.name === name);\n\t\t\t\treturn tool === undefined ? undefined : { description: tool.description, parameters: tool.parameters };\n\t\t\t},\n\t\t\tisDeferrable: (name) => {\n\t\t\t\tconst doc = service.getCatalog().find((candidate) => candidate.name === name);\n\t\t\t\tif (doc?.source === \"mcp\") return isMcpNativeToolSearchEnabled();\n\t\t\t\treturn doc?.source === \"extension\" && !pi.getActiveTools().includes(name);\n\t\t\t},\n\t\t\tsearchToolName: TOOL_SEARCH_TOOL_NAME,\n\t\t});\n\t\tpi.on(\"before_provider_request\", (event, ctx) => nativeAdapter.applyBeforeRequest(ctx.model?.api, event.payload));\n\t\tpi.on(\"after_provider_response\", (event) => nativeAdapter.noteResponseStatus(event.status));\n\t};\n}\n\nfunction hasProviderScope(): boolean {\n\ttry {\n\t\tbindToProviderScope(() => undefined);\n\t\treturn true;\n\t} catch {\n\t\treturn false;\n\t}\n}\n\nexport default function toolSearchExtension(pi: ExtensionAPI): void | Promise<void> {\n\tconst runtime = {\n\t\tgetAllTools: () => pi.getAllTools(),\n\t\tgetActiveTools: () => pi.getActiveTools(),\n\t\tsetActiveTools: (names: readonly string[]) => pi.setActiveTools([...names]),\n\t};\n\tconst sessionOwned = hasProviderScope();\n\tconst service = sessionOwned ? new ToolSearchService(runtime) : getToolSearchService(runtime);\n\tif (sessionOwned) installScopedToolSearchService(service);\n\treturn createToolSearchExtension(service)(pi);\n}\n\nexport { getToolSearchService, ToolSearchService } from \"./service.ts\";\nexport { createToolSearchTool, TOOL_SEARCH_TOOL_NAME } from \"./tool.ts\";\n"]}

package/dist/core/extensions/builtin/tool-search/native-search.d.ts

@@ -0,0 +1,52 @@
+import type { ToolSearchDocument } from "./engine/document.ts";
+export declare const ANTHROPIC_TOOL_SEARCH_TYPE = "tool_search_tool_bm25_20251119";
+export declare const ANTHROPIC_TOOL_SEARCH_NAME = "tool_search";
+/** Anthropic caps a request at 10k tools; beyond that native search is invalid. */
+export declare const ANTHROPIC_MAX_TOOLS = 10000;
+type McpNativeToolSearchGate = () => boolean;
+/** Bind the MCP-owned user setting without moving MCP configuration into the shared builtin. */
+export declare function installMcpNativeToolSearchGate(gate: McpNativeToolSearchGate): void;
+export declare function isMcpNativeToolSearchEnabled(): boolean;
+export interface NativeToolDefinition {
+    readonly description?: string;
+    readonly parameters?: unknown;
+}
+export interface AnthropicNativeInjectionConfig {
+    /** Never defer this tool (our custom tool_search), and it is non-deferrable. */
+    readonly searchToolName?: string;
+    /** True only for catalog tools eligible for native deferral in this request. */
+    readonly isDeferrable: (toolName: string) => boolean;
+    /** Current searchable catalog. Omit when only transforming resident tools. */
+    readonly getCatalog?: () => readonly ToolSearchDocument[];
+    /** Resolve the winning registered definition used to inject inactive schemas. */
+    readonly getToolDefinition?: (toolName: string) => NativeToolDefinition | undefined;
+}
+/**
+ * Pure payload transform. Injects eligible inactive catalog schemas, adds exactly
+ * one native search tool, and enforces Anthropic's HARD RULES: never defer the
+ * local search tool, never combine defer_loading with cache_control, keep at
+ * least one non-deferred tool, and skip entirely above the 10k tool cap.
+ */
+export declare function addAnthropicNativeToolSearch(api: string | undefined, payload: unknown, config: AnthropicNativeInjectionConfig): unknown;
+/** `tool_reference` blocks emitted inside a tool_result so the API expands the referenced tools. */
+export declare function buildToolReferenceBlocks(names: readonly string[]): {
+    type: "tool_reference";
+    name: string;
+}[];
+export interface AnthropicNativeAdapterDeps extends AnthropicNativeInjectionConfig {
+    /** Resolved provider/config gate. */
+    enabled(): boolean;
+    /** Invoked once when a 400 forces the local-search fallback. */
+    onFallback?(reason: string): void;
+}
+/** Session-scoped request injector with permanent 400 fallback for the session. */
+export declare class AnthropicNativeToolSearchAdapter {
+    #private;
+    constructor(deps: AnthropicNativeAdapterDeps);
+    applyBeforeRequest(api: string | undefined, payload: unknown): unknown;
+    noteResponseStatus(status: number): void;
+    get disabled(): boolean;
+    get fallbackReason(): string | null;
+}
+export {};
+//# sourceMappingURL=native-search.d.ts.map

package/dist/core/extensions/builtin/tool-search/native-search.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"native-search.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/tool-search/native-search.ts"],"names":[],"mappings":"AACA,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAM/D,eAAO,MAAM,0BAA0B,mCAAmC,CAAC;AAC3E,eAAO,MAAM,0BAA0B,gBAAgB,CAAC;AACxD,mFAAmF;AACnF,eAAO,MAAM,mBAAmB,QAAQ,CAAC;AAEzC,KAAK,uBAAuB,GAAG,MAAM,OAAO,CAAC;AAK7C,gGAAgG;AAChG,wBAAgB,8BAA8B,CAAC,IAAI,EAAE,uBAAuB,GAAG,IAAI,CAGlF;AAED,wBAAgB,4BAA4B,IAAI,OAAO,CAEtD;AAED,MAAM,WAAW,oBAAoB;IACpC,QAAQ,CAAC,WAAW,CAAC,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,UAAU,CAAC,EAAE,OAAO,CAAC;CAC9B;AAED,MAAM,WAAW,8BAA8B;IAC9C,gFAAgF;IAChF,QAAQ,CAAC,cAAc,CAAC,EAAE,MAAM,CAAC;IACjC,gFAAgF;IAChF,QAAQ,CAAC,YAAY,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,OAAO,CAAC;IACrD,8EAA8E;IAC9E,QAAQ,CAAC,UAAU,CAAC,EAAE,MAAM,SAAS,kBAAkB,EAAE,CAAC;IAC1D,iFAAiF;IACjF,QAAQ,CAAC,iBAAiB,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,KAAK,oBAAoB,GAAG,SAAS,CAAC;CACpF;AAMD;;;;;GAKG;AACH,wBAAgB,4BAA4B,CAC3C,GAAG,EAAE,MAAM,GAAG,SAAS,EACvB,OAAO,EAAE,OAAO,EAChB,MAAM,EAAE,8BAA8B,GACpC,OAAO,CAYT;AAkCD,oGAAoG;AACpG,wBAAgB,wBAAwB,CAAC,KAAK,EAAE,SAAS,MAAM,EAAE,GAAG;IAAE,IAAI,EAAE,gBAAgB,CAAC;IAAC,IAAI,EAAE,MAAM,CAAA;CAAE,EAAE,CAE7G;AAED,MAAM,WAAW,0BAA2B,SAAQ,8BAA8B;IACjF,qCAAqC;IACrC,OAAO,IAAI,OAAO,CAAC;IACnB,gEAAgE;IAChE,UAAU,CAAC,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAAC;CAClC;AAED,mFAAmF;AACnF,qBAAa,gCAAgC;;IAM5C,YAAY,IAAI,EAAE,0BAA0B,EAE3C;IAED,kBAAkB,CAAC,GAAG,EAAE,MAAM,GAAG,SAAS,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAMrE;IAED,kBAAkB,CAAC,MAAM,EAAE,MAAM,GAAG,IAAI,CAMvC;IAED,IAAI,QAAQ,IAAI,OAAO,CAEtB;IAED,IAAI,cAAc,IAAI,MAAM,GAAG,IAAI,CAElC;CACD"}

package/dist/core/extensions/builtin/tool-search/native-search.js

@@ -0,0 +1,116 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+// Anthropic Messages supports native BM25 tool search through request payload
+// injection alone. OpenAI native client-mode search remains intentionally out of
+// scope because it requires a provider-layer seam.
+export const ANTHROPIC_TOOL_SEARCH_TYPE = "tool_search_tool_bm25_20251119";
+export const ANTHROPIC_TOOL_SEARCH_NAME = "tool_search";
+/** Anthropic caps a request at 10k tools; beyond that native search is invalid. */
+export const ANTHROPIC_MAX_TOOLS = 10000;
+const scopedMcpNativeToolSearchGate = new AsyncLocalStorage();
+let mcpNativeToolSearchGate = () => false;
+/** Bind the MCP-owned user setting without moving MCP configuration into the shared builtin. */
+export function installMcpNativeToolSearchGate(gate) {
+    mcpNativeToolSearchGate = gate;
+    scopedMcpNativeToolSearchGate.enterWith(gate);
+}
+export function isMcpNativeToolSearchEnabled() {
+    return (scopedMcpNativeToolSearchGate.getStore() ?? mcpNativeToolSearchGate)();
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null;
+}
+/**
+ * Pure payload transform. Injects eligible inactive catalog schemas, adds exactly
+ * one native search tool, and enforces Anthropic's HARD RULES: never defer the
+ * local search tool, never combine defer_loading with cache_control, keep at
+ * least one non-deferred tool, and skip entirely above the 10k tool cap.
+ */
+export function addAnthropicNativeToolSearch(api, payload, config) {
+    if (api !== "anthropic-messages" || !isRecord(payload))
+        return payload;
+    const tools = Array.isArray(payload.tools) ? payload.tools : [];
+    const injected = injectInactiveCatalogTools(tools, config);
+    if (injected.length > ANTHROPIC_MAX_TOOLS)
+        return payload;
+    const deferred = injected.map((tool) => maybeDefer(tool, config));
+    const hasSearchTool = deferred.some((tool) => isRecord(tool) && tool.type === ANTHROPIC_TOOL_SEARCH_TYPE);
+    const nextTools = hasSearchTool
+        ? deferred
+        : [...deferred, { type: ANTHROPIC_TOOL_SEARCH_TYPE, name: ANTHROPIC_TOOL_SEARCH_NAME }];
+    if (nextTools.length > ANTHROPIC_MAX_TOOLS)
+        return payload;
+    return { ...payload, tools: nextTools };
+}
+function injectInactiveCatalogTools(tools, config) {
+    const catalog = config.getCatalog?.() ?? [];
+    const residentNames = new Set(tools.flatMap((tool) => (isRecord(tool) && typeof tool.name === "string" ? [tool.name] : [])));
+    const injected = [...tools];
+    for (const doc of catalog) {
+        if (doc.name === config.searchToolName || residentNames.has(doc.name) || !config.isDeferrable(doc.name))
+            continue;
+        const definition = config.getToolDefinition?.(doc.name);
+        if (definition?.parameters === undefined)
+            continue;
+        injected.push({
+            name: doc.name,
+            description: definition.description ?? doc.description ?? doc.label,
+            input_schema: definition.parameters,
+            defer_loading: true,
+        });
+        residentNames.add(doc.name);
+    }
+    return injected;
+}
+function maybeDefer(tool, config) {
+    if (!isRecord(tool))
+        return tool;
+    const name = typeof tool.name === "string" ? tool.name : undefined;
+    if (name === undefined)
+        return tool;
+    if (name === config.searchToolName)
+        return tool;
+    if (!config.isDeferrable(name))
+        return tool;
+    if ("cache_control" in tool)
+        return tool;
+    if (tool.defer_loading === true)
+        return tool;
+    return { ...tool, defer_loading: true };
+}
+/** `tool_reference` blocks emitted inside a tool_result so the API expands the referenced tools. */
+export function buildToolReferenceBlocks(names) {
+    return names.map((name) => ({ type: "tool_reference", name }));
+}
+/** Session-scoped request injector with permanent 400 fallback for the session. */
+export class AnthropicNativeToolSearchAdapter {
+    #disabled = false;
+    #injectedLastRequest = false;
+    #fallbackReason = null;
+    #deps;
+    constructor(deps) {
+        this.#deps = deps;
+    }
+    applyBeforeRequest(api, payload) {
+        this.#injectedLastRequest = false;
+        if (this.#disabled || !this.#deps.enabled())
+            return payload;
+        const next = addAnthropicNativeToolSearch(api, payload, this.#deps);
+        this.#injectedLastRequest = next !== payload;
+        return next;
+    }
+    noteResponseStatus(status) {
+        if (status !== 400 || !this.#injectedLastRequest || this.#disabled)
+            return;
+        this.#disabled = true;
+        this.#fallbackReason =
+            "Anthropic returned 400 for native tool-search; disabled it and fell back to local tool_search for this session.";
+        this.#deps.onFallback?.(this.#fallbackReason);
+    }
+    get disabled() {
+        return this.#disabled;
+    }
+    get fallbackReason() {
+        return this.#fallbackReason;
+    }
+}
+//# sourceMappingURL=native-search.js.map

package/dist/core/extensions/builtin/tool-search/native-search.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"native-search.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/tool-search/native-search.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AAGrD,8EAA8E;AAC9E,iFAAiF;AACjF,mDAAmD;AAEnD,MAAM,CAAC,MAAM,0BAA0B,GAAG,gCAAgC,CAAC;AAC3E,MAAM,CAAC,MAAM,0BAA0B,GAAG,aAAa,CAAC;AACxD,mFAAmF;AACnF,MAAM,CAAC,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAIzC,MAAM,6BAA6B,GAAG,IAAI,iBAAiB,EAA2B,CAAC;AACvF,IAAI,uBAAuB,GAA4B,GAAG,EAAE,CAAC,KAAK,CAAC;AAEnE,gGAAgG;AAChG,MAAM,UAAU,8BAA8B,CAAC,IAA6B;IAC3E,uBAAuB,GAAG,IAAI,CAAC;IAC/B,6BAA6B,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;AAC/C,CAAC;AAED,MAAM,UAAU,4BAA4B;IAC3C,OAAO,CAAC,6BAA6B,CAAC,QAAQ,EAAE,IAAI,uBAAuB,CAAC,EAAE,CAAC;AAChF,CAAC;AAkBD,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACpD,CAAC;AAED;;;;;GAKG;AACH,MAAM,UAAU,4BAA4B,CAC3C,GAAuB,EACvB,OAAgB,EAChB,MAAsC;IAEtC,IAAI,GAAG,KAAK,oBAAoB,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC;QAAE,OAAO,OAAO,CAAC;IACvE,MAAM,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IAChE,MAAM,QAAQ,GAAG,0BAA0B,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;IAC3D,IAAI,QAAQ,CAAC,MAAM,GAAG,mBAAmB;QAAE,OAAO,OAAO,CAAC;IAC1D,MAAM,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC;IAClE,MAAM,aAAa,GAAG,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,IAAI,KAAK,0BAA0B,CAAC,CAAC;IAC1G,MAAM,SAAS,GAAG,aAAa;QAC9B,CAAC,CAAC,QAAQ;QACV,CAAC,CAAC,CAAC,GAAG,QAAQ,EAAE,EAAE,IAAI,EAAE,0BAA0B,EAAE,IAAI,EAAE,0BAA0B,EAAE,CAAC,CAAC;IACzF,IAAI,SAAS,CAAC,MAAM,GAAG,mBAAmB;QAAE,OAAO,OAAO,CAAC;IAC3D,OAAO,EAAE,GAAG,OAAO,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;AACzC,CAAC;AAED,SAAS,0BAA0B,CAAC,KAAyB,EAAE,MAAsC;IACpG,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,CAAC;IAC5C,MAAM,aAAa,GAAG,IAAI,GAAG,CAC5B,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAC7F,CAAC;IACF,MAAM,QAAQ,GAAc,CAAC,GAAG,KAAK,CAAC,CAAC;IACvC,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;QAC3B,IAAI,GAAG,CAAC,IAAI,KAAK,MAAM,CAAC,cAAc,IAAI,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC;YAAE,SAAS;QAClH,MAAM,UAAU,GAAG,MAAM,CAAC,iBAAiB,EAAE,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;QACxD,IAAI,UAAU,EAAE,UAAU,KAAK,SAAS;YAAE,SAAS;QACnD,QAAQ,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,GAAG,CAAC,IAAI;YACd,WAAW,EAAE,UAAU,CAAC,WAAW,IAAI,GAAG,CAAC,WAAW,IAAI,GAAG,CAAC,KAAK;YACnE,YAAY,EAAE,UAAU,CAAC,UAAU;YACnC,aAAa,EAAE,IAAI;SACnB,CAAC,CAAC;QACH,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;IACD,OAAO,QAAQ,CAAC;AACjB,CAAC;AAED,SAAS,UAAU,CAAC,IAAa,EAAE,MAAsC;IACxE,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IACjC,MAAM,IAAI,GAAG,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACnE,IAAI,IAAI,KAAK,SAAS;QAAE,OAAO,IAAI,CAAC;IACpC,IAAI,IAAI,KAAK,MAAM,CAAC,cAAc;QAAE,OAAO,IAAI,CAAC;IAChD,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC;IAC5C,IAAI,eAAe,IAAI,IAAI;QAAE,OAAO,IAAI,CAAC;IACzC,IAAI,IAAI,CAAC,aAAa,KAAK,IAAI;QAAE,OAAO,IAAI,CAAC;IAC7C,OAAO,EAAE,GAAG,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC;AACzC,CAAC;AAED,oGAAoG;AACpG,MAAM,UAAU,wBAAwB,CAAC,KAAwB;IAChE,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;AAChE,CAAC;AASD,mFAAmF;AACnF,MAAM,OAAO,gCAAgC;IAC5C,SAAS,GAAG,KAAK,CAAC;IAClB,oBAAoB,GAAG,KAAK,CAAC;IAC7B,eAAe,GAAkB,IAAI,CAAC;IAC7B,KAAK,CAA6B;IAE3C,YAAY,IAAgC;QAC3C,IAAI,CAAC,KAAK,GAAG,IAAI,CAAC;IACnB,CAAC;IAED,kBAAkB,CAAC,GAAuB,EAAE,OAAgB;QAC3D,IAAI,CAAC,oBAAoB,GAAG,KAAK,CAAC;QAClC,IAAI,IAAI,CAAC,SAAS,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE;YAAE,OAAO,OAAO,CAAC;QAC5D,MAAM,IAAI,GAAG,4BAA4B,CAAC,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,KAAK,CAAC,CAAC;QACpE,IAAI,CAAC,oBAAoB,GAAG,IAAI,KAAK,OAAO,CAAC;QAC7C,OAAO,IAAI,CAAC;IACb,CAAC;IAED,kBAAkB,CAAC,MAAc;QAChC,IAAI,MAAM,KAAK,GAAG,IAAI,CAAC,IAAI,CAAC,oBAAoB,IAAI,IAAI,CAAC,SAAS;YAAE,OAAO;QAC3E,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;QACtB,IAAI,CAAC,eAAe;YACnB,iHAAiH,CAAC;QACnH,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;IAC/C,CAAC;IAED,IAAI,QAAQ;QACX,OAAO,IAAI,CAAC,SAAS,CAAC;IACvB,CAAC;IAED,IAAI,cAAc;QACjB,OAAO,IAAI,CAAC,eAAe,CAAC;IAC7B,CAAC;CACD","sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport type { ToolSearchDocument } from \"./engine/document.ts\";\n\n// Anthropic Messages supports native BM25 tool search through request payload\n// injection alone. OpenAI native client-mode search remains intentionally out of\n// scope because it requires a provider-layer seam.\n\nexport const ANTHROPIC_TOOL_SEARCH_TYPE = \"tool_search_tool_bm25_20251119\";\nexport const ANTHROPIC_TOOL_SEARCH_NAME = \"tool_search\";\n/** Anthropic caps a request at 10k tools; beyond that native search is invalid. */\nexport const ANTHROPIC_MAX_TOOLS = 10000;\n\ntype McpNativeToolSearchGate = () => boolean;\n\nconst scopedMcpNativeToolSearchGate = new AsyncLocalStorage<McpNativeToolSearchGate>();\nlet mcpNativeToolSearchGate: McpNativeToolSearchGate = () => false;\n\n/** Bind the MCP-owned user setting without moving MCP configuration into the shared builtin. */\nexport function installMcpNativeToolSearchGate(gate: McpNativeToolSearchGate): void {\n\tmcpNativeToolSearchGate = gate;\n\tscopedMcpNativeToolSearchGate.enterWith(gate);\n}\n\nexport function isMcpNativeToolSearchEnabled(): boolean {\n\treturn (scopedMcpNativeToolSearchGate.getStore() ?? mcpNativeToolSearchGate)();\n}\n\nexport interface NativeToolDefinition {\n\treadonly description?: string;\n\treadonly parameters?: unknown;\n}\n\nexport interface AnthropicNativeInjectionConfig {\n\t/** Never defer this tool (our custom tool_search), and it is non-deferrable. */\n\treadonly searchToolName?: string;\n\t/** True only for catalog tools eligible for native deferral in this request. */\n\treadonly isDeferrable: (toolName: string) => boolean;\n\t/** Current searchable catalog. Omit when only transforming resident tools. */\n\treadonly getCatalog?: () => readonly ToolSearchDocument[];\n\t/** Resolve the winning registered definition used to inject inactive schemas. */\n\treadonly getToolDefinition?: (toolName: string) => NativeToolDefinition | undefined;\n}\n\nfunction isRecord(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null;\n}\n\n/**\n * Pure payload transform. Injects eligible inactive catalog schemas, adds exactly\n * one native search tool, and enforces Anthropic's HARD RULES: never defer the\n * local search tool, never combine defer_loading with cache_control, keep at\n * least one non-deferred tool, and skip entirely above the 10k tool cap.\n */\nexport function addAnthropicNativeToolSearch(\n\tapi: string | undefined,\n\tpayload: unknown,\n\tconfig: AnthropicNativeInjectionConfig,\n): unknown {\n\tif (api !== \"anthropic-messages\" || !isRecord(payload)) return payload;\n\tconst tools = Array.isArray(payload.tools) ? payload.tools : [];\n\tconst injected = injectInactiveCatalogTools(tools, config);\n\tif (injected.length > ANTHROPIC_MAX_TOOLS) return payload;\n\tconst deferred = injected.map((tool) => maybeDefer(tool, config));\n\tconst hasSearchTool = deferred.some((tool) => isRecord(tool) && tool.type === ANTHROPIC_TOOL_SEARCH_TYPE);\n\tconst nextTools = hasSearchTool\n\t\t? deferred\n\t\t: [...deferred, { type: ANTHROPIC_TOOL_SEARCH_TYPE, name: ANTHROPIC_TOOL_SEARCH_NAME }];\n\tif (nextTools.length > ANTHROPIC_MAX_TOOLS) return payload;\n\treturn { ...payload, tools: nextTools };\n}\n\nfunction injectInactiveCatalogTools(tools: readonly unknown[], config: AnthropicNativeInjectionConfig): unknown[] {\n\tconst catalog = config.getCatalog?.() ?? [];\n\tconst residentNames = new Set(\n\t\ttools.flatMap((tool) => (isRecord(tool) && typeof tool.name === \"string\" ? [tool.name] : [])),\n\t);\n\tconst injected: unknown[] = [...tools];\n\tfor (const doc of catalog) {\n\t\tif (doc.name === config.searchToolName || residentNames.has(doc.name) || !config.isDeferrable(doc.name)) continue;\n\t\tconst definition = config.getToolDefinition?.(doc.name);\n\t\tif (definition?.parameters === undefined) continue;\n\t\tinjected.push({\n\t\t\tname: doc.name,\n\t\t\tdescription: definition.description ?? doc.description ?? doc.label,\n\t\t\tinput_schema: definition.parameters,\n\t\t\tdefer_loading: true,\n\t\t});\n\t\tresidentNames.add(doc.name);\n\t}\n\treturn injected;\n}\n\nfunction maybeDefer(tool: unknown, config: AnthropicNativeInjectionConfig): unknown {\n\tif (!isRecord(tool)) return tool;\n\tconst name = typeof tool.name === \"string\" ? tool.name : undefined;\n\tif (name === undefined) return tool;\n\tif (name === config.searchToolName) return tool;\n\tif (!config.isDeferrable(name)) return tool;\n\tif (\"cache_control\" in tool) return tool;\n\tif (tool.defer_loading === true) return tool;\n\treturn { ...tool, defer_loading: true };\n}\n\n/** `tool_reference` blocks emitted inside a tool_result so the API expands the referenced tools. */\nexport function buildToolReferenceBlocks(names: readonly string[]): { type: \"tool_reference\"; name: string }[] {\n\treturn names.map((name) => ({ type: \"tool_reference\", name }));\n}\n\nexport interface AnthropicNativeAdapterDeps extends AnthropicNativeInjectionConfig {\n\t/** Resolved provider/config gate. */\n\tenabled(): boolean;\n\t/** Invoked once when a 400 forces the local-search fallback. */\n\tonFallback?(reason: string): void;\n}\n\n/** Session-scoped request injector with permanent 400 fallback for the session. */\nexport class AnthropicNativeToolSearchAdapter {\n\t#disabled = false;\n\t#injectedLastRequest = false;\n\t#fallbackReason: string | null = null;\n\treadonly #deps: AnthropicNativeAdapterDeps;\n\n\tconstructor(deps: AnthropicNativeAdapterDeps) {\n\t\tthis.#deps = deps;\n\t}\n\n\tapplyBeforeRequest(api: string | undefined, payload: unknown): unknown {\n\t\tthis.#injectedLastRequest = false;\n\t\tif (this.#disabled || !this.#deps.enabled()) return payload;\n\t\tconst next = addAnthropicNativeToolSearch(api, payload, this.#deps);\n\t\tthis.#injectedLastRequest = next !== payload;\n\t\treturn next;\n\t}\n\n\tnoteResponseStatus(status: number): void {\n\t\tif (status !== 400 || !this.#injectedLastRequest || this.#disabled) return;\n\t\tthis.#disabled = true;\n\t\tthis.#fallbackReason =\n\t\t\t\"Anthropic returned 400 for native tool-search; disabled it and fell back to local tool_search for this session.\";\n\t\tthis.#deps.onFallback?.(this.#fallbackReason);\n\t}\n\n\tget disabled(): boolean {\n\t\treturn this.#disabled;\n\t}\n\n\tget fallbackReason(): string | null {\n\t\treturn this.#fallbackReason;\n\t}\n}\n"]}

package/dist/core/extensions/builtin/tool-search/service.d.ts

@@ -0,0 +1,38 @@
+import type { ExtensionAPI, ToolInfo } from "../../types.ts";
+import { type Bm25Result, type Bm25SearchOptions } from "./engine/bm25.ts";
+import type { ToolSearchDocument } from "./engine/document.ts";
+export interface ToolSearchFeederHooks {
+    /** Activate every named match, including names already active as stubs. */
+    activate(names: readonly string[]): void;
+}
+export interface ToolSearchRuntime {
+    getAllTools(): ToolInfo[];
+    getActiveTools(): string[];
+    setActiveTools(names: readonly string[]): void;
+}
+type RuntimeApi = Pick<ExtensionAPI, "getActiveTools" | "getAllTools" | "setActiveTools">;
+/** Session-scoped owner of the live generalized tool-search catalog. */
+export declare class ToolSearchService {
+    #private;
+    constructor(runtime: ToolSearchRuntime);
+    bindRuntime(runtime: RuntimeApi): void;
+    bindActivationRuntime(runtime: Pick<RuntimeApi, "getActiveTools" | "setActiveTools">): void;
+    /** Register the resident search tool only after a searchable catalog exists. */
+    bindToolRegistrar(register: () => void): void;
+    beginSession(): void;
+    /** Replace one source's catalog generation and activation hook. */
+    feed(source: "mcp", docs: readonly ToolSearchDocument[], hooks: ToolSearchFeederHooks): void;
+    getCatalog(): ToolSearchDocument[];
+    search(query: string, limit?: number, options?: Bm25SearchOptions): Bm25Result[];
+    /** Route all matches through their owning feeder, even when already active. */
+    activate(matches: readonly Bm25Result[]): string[];
+    activateTool(name: string): boolean;
+    /** Replay ownership-aware and legacy MCP history once per catalog generation. */
+    maybeRehydrateFromHistory(messages: readonly unknown[]): string[];
+}
+/** Make a session-owned service visible to later builtins loaded in the same provider scope. */
+export declare function installScopedToolSearchService(value: ToolSearchService): void;
+export declare function getToolSearchService(runtime?: ToolSearchRuntime): ToolSearchService;
+export declare function resetToolSearchServiceForTests(): void;
+export {};
+//# sourceMappingURL=service.d.ts.map

package/dist/core/extensions/builtin/tool-search/service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/tool-search/service.ts"],"names":[],"mappings":"AAEA,OAAO,KAAK,EAAE,YAAY,EAAE,QAAQ,EAAE,MAAM,gBAAgB,CAAC;AAC7D,OAAO,EAAE,KAAK,UAAU,EAAE,KAAK,iBAAiB,EAAkB,MAAM,kBAAkB,CAAC;AAC3F,OAAO,KAAK,EAAE,kBAAkB,EAAoB,MAAM,sBAAsB,CAAC;AAIjF,MAAM,WAAW,qBAAqB;IACrC,2EAA2E;IAC3E,QAAQ,CAAC,KAAK,EAAE,SAAS,MAAM,EAAE,GAAG,IAAI,CAAC;CACzC;AAED,MAAM,WAAW,iBAAiB;IACjC,WAAW,IAAI,QAAQ,EAAE,CAAC;IAC1B,cAAc,IAAI,MAAM,EAAE,CAAC;IAC3B,cAAc,CAAC,KAAK,EAAE,SAAS,MAAM,EAAE,GAAG,IAAI,CAAC;CAC/C;AAED,KAAK,UAAU,GAAG,IAAI,CAAC,YAAY,EAAE,gBAAgB,GAAG,aAAa,GAAG,gBAAgB,CAAC,CAAC;AAO1F,wEAAwE;AACxE,qBAAa,iBAAiB;;IAS7B,YAAY,OAAO,EAAE,iBAAiB,EAMrC;IAED,WAAW,CAAC,OAAO,EAAE,UAAU,GAAG,IAAI,CAErC;IAED,qBAAqB,CAAC,OAAO,EAAE,IAAI,CAAC,UAAU,EAAE,gBAAgB,GAAG,gBAAgB,CAAC,GAAG,IAAI,CAE1F;IAED,gFAAgF;IAChF,iBAAiB,CAAC,QAAQ,EAAE,MAAM,IAAI,GAAG,IAAI,CAE5C;IAED,YAAY,IAAI,IAAI,CAMnB;IAED,mEAAmE;IACnE,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,EAAE,SAAS,kBAAkB,EAAE,EAAE,KAAK,EAAE,qBAAqB,GAAG,IAAI,CAK3F;IAED,UAAU,IAAI,kBAAkB,EAAE,CAGjC;IAED,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,SAAK,EAAE,OAAO,GAAE,iBAAsB,GAAG,UAAU,EAAE,CAE/E;IAED,+EAA+E;IAC/E,QAAQ,CAAC,OAAO,EAAE,SAAS,UAAU,EAAE,GAAG,MAAM,EAAE,CAYjD;IAED,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAKlC;IAED,iFAAiF;IACjF,yBAAyB,CAAC,QAAQ,EAAE,SAAS,OAAO,EAAE,GAAG,MAAM,EAAE,CAkBhE;CAoDD;AAkCD,gGAAgG;AAChG,wBAAgB,8BAA8B,CAAC,KAAK,EAAE,iBAAiB,GAAG,IAAI,CAE7E;AAED,wBAAgB,oBAAoB,CAAC,OAAO,CAAC,EAAE,iBAAiB,GAAG,iBAAiB,CAanF;AAED,wBAAgB,8BAA8B,IAAI,IAAI,CAErD"}

package/dist/core/extensions/builtin/tool-search/service.js

@@ -0,0 +1,200 @@
+import { AsyncLocalStorage } from "node:async_hooks";
+import { basename, extname } from "node:path";
+import { buildBm25Index } from "./engine/bm25.js";
+import { deriveExtensionRegistrationId, rehydrate } from "./engine/marker.js";
+import { TOOL_SEARCH_TOOL_NAME } from "./tool.js";
+/** Session-scoped owner of the live generalized tool-search catalog. */
+export class ToolSearchService {
+    #runtime;
+    #feeds = new Map();
+    #extensionDocs = [];
+    #extensionFingerprint = "";
+    #registryGeneration = 0;
+    #historyScannedGeneration = -1;
+    #registerToolSearch;
+    constructor(runtime) {
+        this.#runtime = runtime;
+        this.#feeds.set("extension", {
+            docs: [],
+            hooks: { activate: (names) => this.#promoteExtensionTools(names) },
+        });
+    }
+    bindRuntime(runtime) {
+        this.#runtime = runtime;
+    }
+    bindActivationRuntime(runtime) {
+        this.#runtime = { ...this.#runtime, ...runtime };
+    }
+    /** Register the resident search tool only after a searchable catalog exists. */
+    bindToolRegistrar(register) {
+        this.#registerToolSearch = register;
+    }
+    beginSession() {
+        this.#feeds.delete("mcp");
+        this.#registryGeneration += 1;
+        this.#historyScannedGeneration = -1;
+        this.#refreshExtensionDocs();
+        this.#syncToolSearchLifecycle();
+    }
+    /** Replace one source's catalog generation and activation hook. */
+    feed(source, docs, hooks) {
+        const validDocs = docs.filter((doc) => isValidDocument(doc, source));
+        this.#feeds.set(source, { docs: validDocs, hooks });
+        this.#registryGeneration += 1;
+        this.#syncToolSearchLifecycle();
+    }
+    getCatalog() {
+        this.#refreshExtensionDocs();
+        return [...(this.#feeds.get("mcp")?.docs ?? []), ...this.#extensionDocs];
+    }
+    search(query, limit = 10, options = {}) {
+        return buildBm25Index(this.getCatalog()).search(query, limit, options);
+    }
+    /** Route all matches through their owning feeder, even when already active. */
+    activate(matches) {
+        const namesBySource = new Map();
+        for (const match of matches) {
+            const names = namesBySource.get(match.doc.source) ?? [];
+            names.push(match.name);
+            namesBySource.set(match.doc.source, names);
+        }
+        for (const [source, names] of namesBySource) {
+            this.#feeds.get(source)?.hooks.activate(names);
+        }
+        const active = new Set(this.#runtime.getActiveTools());
+        return matches.map(({ name }) => name).filter((name) => active.has(name));
+    }
+    activateTool(name) {
+        const doc = this.getCatalog().find((candidate) => candidate.name === name);
+        if (doc === undefined)
+            return false;
+        this.#feeds.get(doc.source)?.hooks.activate([name]);
+        return this.#runtime.getActiveTools().includes(name);
+    }
+    /** Replay ownership-aware and legacy MCP history once per catalog generation. */
+    maybeRehydrateFromHistory(messages) {
+        const catalog = this.getCatalog();
+        if (this.#historyScannedGeneration === this.#registryGeneration)
+            return [];
+        this.#historyScannedGeneration = this.#registryGeneration;
+        const docsByName = new Map(catalog.map((doc) => [
+            doc.name,
+            {
+                name: doc.name,
+                registrationId: doc.registrationId,
+                source: doc.source,
+                allowLazyActivation: true,
+            },
+        ]));
+        const restored = rehydrate(messages, docsByName);
+        if (restored.length > 0)
+            this.#activateNames(restored, catalog);
+        return restored;
+    }
+    #refreshExtensionDocs() {
+        const docs = this.#runtime
+            .getAllTools()
+            .filter((tool) => tool.exposure === "search" && tool.allowLazyActivation)
+            .map(extensionDocument)
+            .filter((doc) => doc !== undefined)
+            .sort((left, right) => left.name.localeCompare(right.name));
+        const fingerprint = JSON.stringify(docs);
+        if (fingerprint === this.#extensionFingerprint)
+            return;
+        this.#extensionDocs = docs;
+        this.#extensionFingerprint = fingerprint;
+        this.#feeds.set("extension", {
+            docs,
+            hooks: { activate: (names) => this.#promoteExtensionTools(names) },
+        });
+        this.#registryGeneration += 1;
+        this.#syncToolSearchLifecycle();
+    }
+    #activateNames(names, catalog = this.getCatalog()) {
+        const sourceByName = new Map(catalog.map((doc) => [doc.name, doc.source]));
+        const namesBySource = new Map();
+        for (const name of names) {
+            const source = sourceByName.get(name);
+            if (source === undefined)
+                continue;
+            const sourceNames = namesBySource.get(source) ?? [];
+            sourceNames.push(name);
+            namesBySource.set(source, sourceNames);
+        }
+        for (const [source, sourceNames] of namesBySource)
+            this.#feeds.get(source)?.hooks.activate(sourceNames);
+    }
+    #syncToolSearchLifecycle() {
+        const hasDocuments = this.#extensionDocs.length > 0 || (this.#feeds.get("mcp")?.docs.length ?? 0) > 0;
+        if (hasDocuments)
+            this.#registerToolSearch?.();
+        const current = this.#runtime.getActiveTools();
+        const isActive = current.includes(TOOL_SEARCH_TOOL_NAME);
+        if (hasDocuments === isActive)
+            return;
+        this.#runtime.setActiveTools(hasDocuments ? [...current, TOOL_SEARCH_TOOL_NAME] : current.filter((name) => name !== TOOL_SEARCH_TOOL_NAME));
+    }
+    #promoteExtensionTools(names) {
+        if (names.length === 0)
+            return;
+        const current = this.#runtime.getActiveTools();
+        const active = new Set(current);
+        const added = [...new Set(names.filter((name) => !active.has(name)))].sort();
+        this.#runtime.setActiveTools([...current, ...added]);
+    }
+}
+function extensionDocument(tool) {
+    if (tool.name.length === 0)
+        return undefined;
+    const ownerLabel = extensionOwnerLabel(tool);
+    const registrationId = deriveExtensionRegistrationId(tool.sourceInfo, tool.name);
+    if (registrationId.length === 0)
+        return undefined;
+    return {
+        name: tool.name,
+        label: tool.label,
+        aliases: [],
+        description: tool.description,
+        searchText: tool.searchText,
+        keywords: tool.searchKeywords,
+        source: "extension",
+        group: tool.searchGroup ?? ownerLabel,
+        ownerLabel,
+        registrationId,
+    };
+}
+function extensionOwnerLabel(tool) {
+    const fileName = basename(tool.sourceInfo.path);
+    const extension = extname(fileName);
+    return extension.length === 0 ? fileName : fileName.slice(0, -extension.length);
+}
+function isValidDocument(doc, source) {
+    return doc.source === source && doc.name.length > 0 && doc.registrationId.length > 0;
+}
+const scopedService = new AsyncLocalStorage();
+let service = null;
+/** Make a session-owned service visible to later builtins loaded in the same provider scope. */
+export function installScopedToolSearchService(value) {
+    scopedService.enterWith(value);
+}
+export function getToolSearchService(runtime) {
+    const scoped = scopedService.getStore();
+    if (scoped !== undefined) {
+        if (runtime !== undefined)
+            scoped.bindRuntime(runtime);
+        return scoped;
+    }
+    if (service === null) {
+        if (runtime === undefined)
+            throw new Error("ToolSearchService runtime is not bound");
+        service = new ToolSearchService(runtime);
+    }
+    else if (runtime !== undefined) {
+        service.bindRuntime(runtime);
+    }
+    return service;
+}
+export function resetToolSearchServiceForTests() {
+    service = null;
+}
+//# sourceMappingURL=service.js.map

package/dist/core/extensions/builtin/tool-search/service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"service.js","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/tool-search/service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAE9C,OAAO,EAA2C,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAE3F,OAAO,EAAE,6BAA6B,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC9E,OAAO,EAAE,qBAAqB,EAAE,MAAM,WAAW,CAAC;AAoBlD,wEAAwE;AACxE,MAAM,OAAO,iBAAiB;IAC7B,QAAQ,CAAoB;IACnB,MAAM,GAAG,IAAI,GAAG,EAA+B,CAAC;IACzD,cAAc,GAAyB,EAAE,CAAC;IAC1C,qBAAqB,GAAG,EAAE,CAAC;IAC3B,mBAAmB,GAAG,CAAC,CAAC;IACxB,yBAAyB,GAAG,CAAC,CAAC,CAAC;IAC/B,mBAAmB,CAA2B;IAE9C,YAAY,OAA0B;QACrC,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;QACxB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE;YAC5B,IAAI,EAAE,EAAE;YACR,KAAK,EAAE,EAAE,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE;SAClE,CAAC,CAAC;IACJ,CAAC;IAED,WAAW,CAAC,OAAmB;QAC9B,IAAI,CAAC,QAAQ,GAAG,OAAO,CAAC;IACzB,CAAC;IAED,qBAAqB,CAAC,OAA8D;QACnF,IAAI,CAAC,QAAQ,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,EAAE,GAAG,OAAO,EAAE,CAAC;IAClD,CAAC;IAED,gFAAgF;IAChF,iBAAiB,CAAC,QAAoB;QACrC,IAAI,CAAC,mBAAmB,GAAG,QAAQ,CAAC;IACrC,CAAC;IAED,YAAY;QACX,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;QAC1B,IAAI,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,yBAAyB,GAAG,CAAC,CAAC,CAAC;QACpC,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,IAAI,CAAC,wBAAwB,EAAE,CAAC;IACjC,CAAC;IAED,mEAAmE;IACnE,IAAI,CAAC,MAAa,EAAE,IAAmC,EAAE,KAA4B;QACpF,MAAM,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,eAAe,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,CAAC;QACrE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QACpD,IAAI,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,wBAAwB,EAAE,CAAC;IACjC,CAAC;IAED,UAAU;QACT,IAAI,CAAC,qBAAqB,EAAE,CAAC;QAC7B,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,IAAI,EAAE,CAAC,EAAE,GAAG,IAAI,CAAC,cAAc,CAAC,CAAC;IAC1E,CAAC;IAED,MAAM,CAAC,KAAa,EAAE,KAAK,GAAG,EAAE,EAAE,OAAO,GAAsB,EAAE;QAChE,OAAO,cAAc,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC;IACxE,CAAC;IAED,+EAA+E;IAC/E,QAAQ,CAAC,OAA8B;QACtC,MAAM,aAAa,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC5D,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;YAC7B,MAAM,KAAK,GAAG,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACxD,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;YACvB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,KAAK,MAAM,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,aAAa,EAAE,CAAC;YAC7C,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAChD,CAAC;QACD,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC,CAAC;QACvD,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC;IAC3E,CAAC;IAED,YAAY,CAAC,IAAY;QACxB,MAAM,GAAG,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,KAAK,IAAI,CAAC,CAAC;QAC3E,IAAI,GAAG,KAAK,SAAS;YAAE,OAAO,KAAK,CAAC;QACpC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC;QACpD,OAAO,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;IAED,iFAAiF;IACjF,yBAAyB,CAAC,QAA4B;QACrD,MAAM,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE,CAAC;QAClC,IAAI,IAAI,CAAC,yBAAyB,KAAK,IAAI,CAAC,mBAAmB;YAAE,OAAO,EAAE,CAAC;QAC3E,IAAI,CAAC,yBAAyB,GAAG,IAAI,CAAC,mBAAmB,CAAC;QAC1D,MAAM,UAAU,GAAG,IAAI,GAAG,CACzB,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC;YACpB,GAAG,CAAC,IAAI;YACR;gBACC,IAAI,EAAE,GAAG,CAAC,IAAI;gBACd,cAAc,EAAE,GAAG,CAAC,cAAc;gBAClC,MAAM,EAAE,GAAG,CAAC,MAAM;gBAClB,mBAAmB,EAAE,IAAI;aACzB;SACD,CAAC,CACF,CAAC;QACF,MAAM,QAAQ,GAAG,SAAS,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACjD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC;YAAE,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;QAChE,OAAO,QAAQ,CAAC;IACjB,CAAC;IAED,qBAAqB;QACpB,MAAM,IAAI,GAAG,IAAI,CAAC,QAAQ;aACxB,WAAW,EAAE;aACb,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,KAAK,QAAQ,IAAI,IAAI,CAAC,mBAAmB,CAAC;aACxE,GAAG,CAAC,iBAAiB,CAAC;aACtB,MAAM,CAAC,CAAC,GAAG,EAA6B,EAAE,CAAC,GAAG,KAAK,SAAS,CAAC;aAC7D,IAAI,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC;QACzC,IAAI,WAAW,KAAK,IAAI,CAAC,qBAAqB;YAAE,OAAO;QACvD,IAAI,CAAC,cAAc,GAAG,IAAI,CAAC;QAC3B,IAAI,CAAC,qBAAqB,GAAG,WAAW,CAAC;QACzC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,WAAW,EAAE;YAC5B,IAAI;YACJ,KAAK,EAAE,EAAE,QAAQ,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE;SAClE,CAAC,CAAC;QACH,IAAI,CAAC,mBAAmB,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,wBAAwB,EAAE,CAAC;IACjC,CAAC;IAED,cAAc,CAAC,KAAwB,EAAE,OAAO,GAAG,IAAI,CAAC,UAAU,EAAE;QACnE,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,CAAC,MAAM,CAAU,CAAC,CAAC,CAAC;QACpF,MAAM,aAAa,GAAG,IAAI,GAAG,EAA8B,CAAC;QAC5D,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YAC1B,MAAM,MAAM,GAAG,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;YACtC,IAAI,MAAM,KAAK,SAAS;gBAAE,SAAS;YACnC,MAAM,WAAW,GAAG,aAAa,CAAC,GAAG,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;YACpD,WAAW,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACvB,aAAa,CAAC,GAAG,CAAC,MAAM,EAAE,WAAW,CAAC,CAAC;QACxC,CAAC;QACD,KAAK,MAAM,CAAC,MAAM,EAAE,WAAW,CAAC,IAAI,aAAa;YAAE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;IACzG,CAAC;IAED,wBAAwB;QACvB,MAAM,YAAY,GAAG,IAAI,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,IAAI,CAAC,MAAM,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC;QACtG,IAAI,YAAY;YAAE,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC;QAC/C,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,CAAC,qBAAqB,CAAC,CAAC;QACzD,IAAI,YAAY,KAAK,QAAQ;YAAE,OAAO;QACtC,IAAI,CAAC,QAAQ,CAAC,cAAc,CAC3B,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,EAAE,qBAAqB,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,KAAK,qBAAqB,CAAC,CAC7G,CAAC;IACH,CAAC;IAED,sBAAsB,CAAC,KAAwB;QAC9C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO;QAC/B,MAAM,OAAO,GAAG,IAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC;QAChC,MAAM,KAAK,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC7E,IAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,GAAG,OAAO,EAAE,GAAG,KAAK,CAAC,CAAC,CAAC;IACtD,CAAC;CACD;AAED,SAAS,iBAAiB,CAAC,IAAc;IACxC,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAC7C,MAAM,UAAU,GAAG,mBAAmB,CAAC,IAAI,CAAC,CAAC;IAC7C,MAAM,cAAc,GAAG,6BAA6B,CAAC,IAAI,CAAC,UAAU,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACjF,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,SAAS,CAAC;IAClD,OAAO;QACN,IAAI,EAAE,IAAI,CAAC,IAAI;QACf,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,OAAO,EAAE,EAAE;QACX,WAAW,EAAE,IAAI,CAAC,WAAW;QAC7B,UAAU,EAAE,IAAI,CAAC,UAAU;QAC3B,QAAQ,EAAE,IAAI,CAAC,cAAc;QAC7B,MAAM,EAAE,WAAW;QACnB,KAAK,EAAE,IAAI,CAAC,WAAW,IAAI,UAAU;QACrC,UAAU;QACV,cAAc;KACd,CAAC;AACH,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAc;IAC1C,MAAM,QAAQ,GAAG,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;IAChD,MAAM,SAAS,GAAG,OAAO,CAAC,QAAQ,CAAC,CAAC;IACpC,OAAO,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;AACjF,CAAC;AAED,SAAS,eAAe,CAAC,GAAuB,EAAE,MAAwB;IACzE,OAAO,GAAG,CAAC,MAAM,KAAK,MAAM,IAAI,GAAG,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,GAAG,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,CAAC;AACtF,CAAC;AAED,MAAM,aAAa,GAAG,IAAI,iBAAiB,EAAqB,CAAC;AACjE,IAAI,OAAO,GAA6B,IAAI,CAAC;AAE7C,gGAAgG;AAChG,MAAM,UAAU,8BAA8B,CAAC,KAAwB;IACtE,aAAa,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;AAChC,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,OAA2B;IAC/D,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,EAAE,CAAC;IACxC,IAAI,MAAM,KAAK,SAAS,EAAE,CAAC;QAC1B,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QACvD,OAAO,MAAM,CAAC;IACf,CAAC;IACD,IAAI,OAAO,KAAK,IAAI,EAAE,CAAC;QACtB,IAAI,OAAO,KAAK,SAAS;YAAE,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QACrF,OAAO,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAC1C,CAAC;SAAM,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAClC,OAAO,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC;IACD,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,MAAM,UAAU,8BAA8B;IAC7C,OAAO,GAAG,IAAI,CAAC;AAChB,CAAC","sourcesContent":["import { AsyncLocalStorage } from \"node:async_hooks\";\nimport { basename, extname } from \"node:path\";\nimport type { ExtensionAPI, ToolInfo } from \"../../types.ts\";\nimport { type Bm25Result, type Bm25SearchOptions, buildBm25Index } from \"./engine/bm25.ts\";\nimport type { ToolSearchDocument, ToolSearchSource } from \"./engine/document.ts\";\nimport { deriveExtensionRegistrationId, rehydrate } from \"./engine/marker.ts\";\nimport { TOOL_SEARCH_TOOL_NAME } from \"./tool.ts\";\n\nexport interface ToolSearchFeederHooks {\n\t/** Activate every named match, including names already active as stubs. */\n\tactivate(names: readonly string[]): void;\n}\n\nexport interface ToolSearchRuntime {\n\tgetAllTools(): ToolInfo[];\n\tgetActiveTools(): string[];\n\tsetActiveTools(names: readonly string[]): void;\n}\n\ntype RuntimeApi = Pick<ExtensionAPI, \"getActiveTools\" | \"getAllTools\" | \"setActiveTools\">;\n\ninterface FeedState {\n\tdocs: ToolSearchDocument[];\n\thooks: ToolSearchFeederHooks;\n}\n\n/** Session-scoped owner of the live generalized tool-search catalog. */\nexport class ToolSearchService {\n\t#runtime: ToolSearchRuntime;\n\treadonly #feeds = new Map<ToolSearchSource, FeedState>();\n\t#extensionDocs: ToolSearchDocument[] = [];\n\t#extensionFingerprint = \"\";\n\t#registryGeneration = 0;\n\t#historyScannedGeneration = -1;\n\t#registerToolSearch: (() => void) | undefined;\n\n\tconstructor(runtime: ToolSearchRuntime) {\n\t\tthis.#runtime = runtime;\n\t\tthis.#feeds.set(\"extension\", {\n\t\t\tdocs: [],\n\t\t\thooks: { activate: (names) => this.#promoteExtensionTools(names) },\n\t\t});\n\t}\n\n\tbindRuntime(runtime: RuntimeApi): void {\n\t\tthis.#runtime = runtime;\n\t}\n\n\tbindActivationRuntime(runtime: Pick<RuntimeApi, \"getActiveTools\" | \"setActiveTools\">): void {\n\t\tthis.#runtime = { ...this.#runtime, ...runtime };\n\t}\n\n\t/** Register the resident search tool only after a searchable catalog exists. */\n\tbindToolRegistrar(register: () => void): void {\n\t\tthis.#registerToolSearch = register;\n\t}\n\n\tbeginSession(): void {\n\t\tthis.#feeds.delete(\"mcp\");\n\t\tthis.#registryGeneration += 1;\n\t\tthis.#historyScannedGeneration = -1;\n\t\tthis.#refreshExtensionDocs();\n\t\tthis.#syncToolSearchLifecycle();\n\t}\n\n\t/** Replace one source's catalog generation and activation hook. */\n\tfeed(source: \"mcp\", docs: readonly ToolSearchDocument[], hooks: ToolSearchFeederHooks): void {\n\t\tconst validDocs = docs.filter((doc) => isValidDocument(doc, source));\n\t\tthis.#feeds.set(source, { docs: validDocs, hooks });\n\t\tthis.#registryGeneration += 1;\n\t\tthis.#syncToolSearchLifecycle();\n\t}\n\n\tgetCatalog(): ToolSearchDocument[] {\n\t\tthis.#refreshExtensionDocs();\n\t\treturn [...(this.#feeds.get(\"mcp\")?.docs ?? []), ...this.#extensionDocs];\n\t}\n\n\tsearch(query: string, limit = 10, options: Bm25SearchOptions = {}): Bm25Result[] {\n\t\treturn buildBm25Index(this.getCatalog()).search(query, limit, options);\n\t}\n\n\t/** Route all matches through their owning feeder, even when already active. */\n\tactivate(matches: readonly Bm25Result[]): string[] {\n\t\tconst namesBySource = new Map<ToolSearchSource, string[]>();\n\t\tfor (const match of matches) {\n\t\t\tconst names = namesBySource.get(match.doc.source) ?? [];\n\t\t\tnames.push(match.name);\n\t\t\tnamesBySource.set(match.doc.source, names);\n\t\t}\n\t\tfor (const [source, names] of namesBySource) {\n\t\t\tthis.#feeds.get(source)?.hooks.activate(names);\n\t\t}\n\t\tconst active = new Set(this.#runtime.getActiveTools());\n\t\treturn matches.map(({ name }) => name).filter((name) => active.has(name));\n\t}\n\n\tactivateTool(name: string): boolean {\n\t\tconst doc = this.getCatalog().find((candidate) => candidate.name === name);\n\t\tif (doc === undefined) return false;\n\t\tthis.#feeds.get(doc.source)?.hooks.activate([name]);\n\t\treturn this.#runtime.getActiveTools().includes(name);\n\t}\n\n\t/** Replay ownership-aware and legacy MCP history once per catalog generation. */\n\tmaybeRehydrateFromHistory(messages: readonly unknown[]): string[] {\n\t\tconst catalog = this.getCatalog();\n\t\tif (this.#historyScannedGeneration === this.#registryGeneration) return [];\n\t\tthis.#historyScannedGeneration = this.#registryGeneration;\n\t\tconst docsByName = new Map(\n\t\t\tcatalog.map((doc) => [\n\t\t\t\tdoc.name,\n\t\t\t\t{\n\t\t\t\t\tname: doc.name,\n\t\t\t\t\tregistrationId: doc.registrationId,\n\t\t\t\t\tsource: doc.source,\n\t\t\t\t\tallowLazyActivation: true,\n\t\t\t\t},\n\t\t\t]),\n\t\t);\n\t\tconst restored = rehydrate(messages, docsByName);\n\t\tif (restored.length > 0) this.#activateNames(restored, catalog);\n\t\treturn restored;\n\t}\n\n\t#refreshExtensionDocs(): void {\n\t\tconst docs = this.#runtime\n\t\t\t.getAllTools()\n\t\t\t.filter((tool) => tool.exposure === \"search\" && tool.allowLazyActivation)\n\t\t\t.map(extensionDocument)\n\t\t\t.filter((doc): doc is ToolSearchDocument => doc !== undefined)\n\t\t\t.sort((left, right) => left.name.localeCompare(right.name));\n\t\tconst fingerprint = JSON.stringify(docs);\n\t\tif (fingerprint === this.#extensionFingerprint) return;\n\t\tthis.#extensionDocs = docs;\n\t\tthis.#extensionFingerprint = fingerprint;\n\t\tthis.#feeds.set(\"extension\", {\n\t\t\tdocs,\n\t\t\thooks: { activate: (names) => this.#promoteExtensionTools(names) },\n\t\t});\n\t\tthis.#registryGeneration += 1;\n\t\tthis.#syncToolSearchLifecycle();\n\t}\n\n\t#activateNames(names: readonly string[], catalog = this.getCatalog()): void {\n\t\tconst sourceByName = new Map(catalog.map((doc) => [doc.name, doc.source] as const));\n\t\tconst namesBySource = new Map<ToolSearchSource, string[]>();\n\t\tfor (const name of names) {\n\t\t\tconst source = sourceByName.get(name);\n\t\t\tif (source === undefined) continue;\n\t\t\tconst sourceNames = namesBySource.get(source) ?? [];\n\t\t\tsourceNames.push(name);\n\t\t\tnamesBySource.set(source, sourceNames);\n\t\t}\n\t\tfor (const [source, sourceNames] of namesBySource) this.#feeds.get(source)?.hooks.activate(sourceNames);\n\t}\n\n\t#syncToolSearchLifecycle(): void {\n\t\tconst hasDocuments = this.#extensionDocs.length > 0 || (this.#feeds.get(\"mcp\")?.docs.length ?? 0) > 0;\n\t\tif (hasDocuments) this.#registerToolSearch?.();\n\t\tconst current = this.#runtime.getActiveTools();\n\t\tconst isActive = current.includes(TOOL_SEARCH_TOOL_NAME);\n\t\tif (hasDocuments === isActive) return;\n\t\tthis.#runtime.setActiveTools(\n\t\t\thasDocuments ? [...current, TOOL_SEARCH_TOOL_NAME] : current.filter((name) => name !== TOOL_SEARCH_TOOL_NAME),\n\t\t);\n\t}\n\n\t#promoteExtensionTools(names: readonly string[]): void {\n\t\tif (names.length === 0) return;\n\t\tconst current = this.#runtime.getActiveTools();\n\t\tconst active = new Set(current);\n\t\tconst added = [...new Set(names.filter((name) => !active.has(name)))].sort();\n\t\tthis.#runtime.setActiveTools([...current, ...added]);\n\t}\n}\n\nfunction extensionDocument(tool: ToolInfo): ToolSearchDocument | undefined {\n\tif (tool.name.length === 0) return undefined;\n\tconst ownerLabel = extensionOwnerLabel(tool);\n\tconst registrationId = deriveExtensionRegistrationId(tool.sourceInfo, tool.name);\n\tif (registrationId.length === 0) return undefined;\n\treturn {\n\t\tname: tool.name,\n\t\tlabel: tool.label,\n\t\taliases: [],\n\t\tdescription: tool.description,\n\t\tsearchText: tool.searchText,\n\t\tkeywords: tool.searchKeywords,\n\t\tsource: \"extension\",\n\t\tgroup: tool.searchGroup ?? ownerLabel,\n\t\townerLabel,\n\t\tregistrationId,\n\t};\n}\n\nfunction extensionOwnerLabel(tool: ToolInfo): string {\n\tconst fileName = basename(tool.sourceInfo.path);\n\tconst extension = extname(fileName);\n\treturn extension.length === 0 ? fileName : fileName.slice(0, -extension.length);\n}\n\nfunction isValidDocument(doc: ToolSearchDocument, source: ToolSearchSource): boolean {\n\treturn doc.source === source && doc.name.length > 0 && doc.registrationId.length > 0;\n}\n\nconst scopedService = new AsyncLocalStorage<ToolSearchService>();\nlet service: ToolSearchService | null = null;\n\n/** Make a session-owned service visible to later builtins loaded in the same provider scope. */\nexport function installScopedToolSearchService(value: ToolSearchService): void {\n\tscopedService.enterWith(value);\n}\n\nexport function getToolSearchService(runtime?: ToolSearchRuntime): ToolSearchService {\n\tconst scoped = scopedService.getStore();\n\tif (scoped !== undefined) {\n\t\tif (runtime !== undefined) scoped.bindRuntime(runtime);\n\t\treturn scoped;\n\t}\n\tif (service === null) {\n\t\tif (runtime === undefined) throw new Error(\"ToolSearchService runtime is not bound\");\n\t\tservice = new ToolSearchService(runtime);\n\t} else if (runtime !== undefined) {\n\t\tservice.bindRuntime(runtime);\n\t}\n\treturn service;\n}\n\nexport function resetToolSearchServiceForTests(): void {\n\tservice = null;\n}\n"]}

package/dist/core/extensions/builtin/tool-search/tool.d.ts

@@ -0,0 +1,21 @@
+import { Type } from "typebox";
+import type { ToolDefinition } from "../../types.ts";
+import type { Bm25Result } from "./engine/bm25.ts";
+import type { ToolSearchSource } from "./engine/document.ts";
+import type { ToolSearchService } from "./service.ts";
+export declare const TOOL_SEARCH_TOOL_NAME = "tool_search";
+declare const ParamsSchema: Type.TObject<{
+    query: Type.TString;
+    source: Type.TOptional<Type.TUnion<[Type.TLiteral<"mcp">, Type.TLiteral<"extension">]>>;
+    group: Type.TOptional<Type.TString>;
+}>;
+export interface ToolSearchDetails {
+    query: string;
+    activated: string[];
+}
+type ToolSearchTool = ToolDefinition<typeof ParamsSchema, ToolSearchDetails, unknown>;
+/** Author the single shared tool-search definition registered by this builtin. */
+export declare function createToolSearchTool(service: ToolSearchService): ToolSearchTool;
+export declare function buildToolSearchResultText(query: string, matches: readonly Bm25Result[], source: ToolSearchSource | undefined, group: string | undefined): string;
+export {};
+//# sourceMappingURL=tool.d.ts.map

package/dist/core/extensions/builtin/tool-search/tool.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../../../../src/core/extensions/builtin/tool-search/tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAe,IAAI,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,gBAAgB,CAAC;AACrD,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,kBAAkB,CAAC;AACnD,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,sBAAsB,CAAC;AAE7D,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,cAAc,CAAC;AAEtD,eAAO,MAAM,qBAAqB,gBAAgB,CAAC;AAGnD,QAAA,MAAM,YAAY;;;;EAQhB,CAAC;AAGH,MAAM,WAAW,iBAAiB;IACjC,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,EAAE,CAAC;CACpB;AAED,KAAK,cAAc,GAAG,cAAc,CAAC,OAAO,YAAY,EAAE,iBAAiB,EAAE,OAAO,CAAC,CAAC;AAEtF,kFAAkF;AAClF,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,iBAAiB,GAAG,cAAc,CAyC/E;AAYD,wBAAgB,yBAAyB,CACxC,KAAK,EAAE,MAAM,EACb,OAAO,EAAE,SAAS,UAAU,EAAE,EAC9B,MAAM,EAAE,gBAAgB,GAAG,SAAS,EACpC,KAAK,EAAE,MAAM,GAAG,SAAS,GACvB,MAAM,CAwBR"}

package/dist/core/extensions/builtin/tool-search/tool.js

@@ -0,0 +1,95 @@
+import { Text } from "@earendil-works/pi-tui";
+import { Type } from "typebox";
+import { emitActivationMarker } from "./engine/marker.js";
+export const TOOL_SEARCH_TOOL_NAME = "tool_search";
+const MAX_RESULTS = 10;
+const ParamsSchema = Type.Object({
+    query: Type.String({ description: "Natural-language description of the capability you need." }),
+    source: Type.Optional(Type.Union([Type.Literal("mcp"), Type.Literal("extension")], {
+        description: "Optional: restrict the search to MCP or extension tools.",
+    })),
+    group: Type.Optional(Type.String({ description: "Optional: restrict the search to one catalog group." })),
+});
+/** Author the single shared tool-search definition registered by this builtin. */
+export function createToolSearchTool(service) {
+    return {
+        name: TOOL_SEARCH_TOOL_NAME,
+        label: "Tool search",
+        description: "Search the catalog of available tools by capability; matched tools are activated and become callable on your NEXT turn.",
+        promptSnippet: "Search available tool catalogs by capability; matched tools activate next turn.",
+        parameters: ParamsSchema,
+        prepareArguments: prepareToolSearchArguments,
+        executionMode: "parallel",
+        async execute(_toolCallId, params) {
+            const options = {
+                ...(params.source === undefined ? {} : { source: params.source }),
+                ...(params.group === undefined ? {} : { group: params.group }),
+            };
+            const matches = service.search(params.query, MAX_RESULTS, options);
+            const activated = service.activate(matches);
+            return {
+                content: [
+                    { type: "text", text: buildToolSearchResultText(params.query, matches, params.source, params.group) },
+                ],
+                details: { activated, query: params.query },
+            };
+        },
+        renderCall(args, theme) {
+            const source = args.source === undefined ? "" : ` source:${args.source}`;
+            const group = args.group === undefined ? "" : ` @${args.group}`;
+            return new Text(theme.fg("toolTitle", theme.bold(`${TOOL_SEARCH_TOOL_NAME} "${args.query}"${source}${group}`)), 0, 0);
+        },
+        renderResult(result, options, theme) {
+            const count = result.details?.activated.length ?? 0;
+            const title = options.isPartial
+                ? `${TOOL_SEARCH_TOOL_NAME}: searching`
+                : `${TOOL_SEARCH_TOOL_NAME}: ${count} tool(s) activated`;
+            return new Text(theme.fg("toolOutput", title), 0, 0);
+        },
+    };
+}
+function prepareToolSearchArguments(args) {
+    if (!isRecord(args))
+        return args;
+    const { server, ...rest } = args;
+    return {
+        ...rest,
+        ...(rest.source === undefined && server === undefined ? {} : { source: rest.source ?? "mcp" }),
+        ...(rest.group === undefined && server === undefined ? {} : { group: rest.group ?? server }),
+    };
+}
+export function buildToolSearchResultText(query, matches, source, group) {
+    const scope = [
+        source === undefined ? undefined : `source "${source}"`,
+        group === undefined ? undefined : `group "${group}"`,
+    ]
+        .filter((part) => part !== undefined)
+        .join(" in ");
+    const scopeText = scope.length === 0 ? "" : ` in ${scope}`;
+    if (matches.length === 0) {
+        return `No tools matched "${query}"${scopeText}. No tools were activated; try different keywords or run tool_search with a broader query. Your active tool set is unchanged.`;
+    }
+    const bullets = matches
+        .map((match) => `- ${match.name} — ${oneLine(match.doc.description) ?? "(no description)"}`)
+        .join("\n");
+    const marker = emitActivationMarker(matches.map((match) => ({ name: match.name, registrationId: match.doc.registrationId })));
+    return [
+        `Found ${matches.length} tool(s) matching "${query}"${scopeText}. Matched tools are now active and callable from your NEXT turn:`,
+        "",
+        bullets,
+        "",
+        marker,
+    ].join("\n");
+}
+function isRecord(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+function oneLine(text) {
+    if (text === undefined)
+        return undefined;
+    const collapsed = text.replace(/\s+/g, " ").trim();
+    if (collapsed.length === 0)
+        return undefined;
+    return collapsed.length <= 100 ? collapsed : `${collapsed.slice(0, 97)}...`;
+}
+//# sourceMappingURL=tool.js.map