@akon-labs/nexus-cli 0.1.8 → 0.1.10

package/dist/extensions/mcp/sampling-handler.js

@@ -0,0 +1,229 @@
+import { complete } from "@earendil-works/pi-ai/compat";
+import { throwIfAborted } from "./abort.js";
+import { combineAbortSignals } from "./runtime-owner.js";
+import { truncateAtWord } from "./utils.js";
+export function registerSamplingHandler(client, options) {
+    client.setRequestHandler("sampling/createMessage", (request, context) => {
+        const requestSignal = context.signal;
+        return handleSamplingRequest({
+            ...options,
+            getSignal: () => combineAbortSignals(options.getSignal(), requestSignal),
+        }, request);
+    });
+}
+export async function handleSamplingRequest(options, request) {
+    const params = request.params;
+    const signal = options.getSignal();
+    throwIfAborted(signal);
+    if ("task" in params && params.task) {
+        throw new Error("MCP sampling tasks are not supported");
+    }
+    if (params.includeContext && params.includeContext !== "none") {
+        throw new Error("MCP sampling context inclusion is not supported");
+    }
+    if (params.tools?.length) {
+        throw new Error("MCP sampling tool use is not supported");
+    }
+    if (params.toolChoice) {
+        throw new Error("MCP sampling tool choice is not supported");
+    }
+    if (params.stopSequences?.length) {
+        throw new Error("MCP sampling stop sequences are not supported");
+    }
+    const messages = params.messages.map(convertSamplingMessage);
+    const { model, apiKey, headers } = await resolveSamplingModel(options, params.modelPreferences);
+    throwIfAborted(signal);
+    await confirmSampling(options, "Approve MCP sampling request", formatRequestApproval(options.serverName, `${model.provider}/${model.id}`, params.systemPrompt, messages));
+    throwIfAborted(signal);
+    const result = await complete(model, {
+        ...(params.systemPrompt !== undefined ? { systemPrompt: params.systemPrompt } : {}),
+        messages,
+    }, {
+        ...(apiKey !== undefined ? { apiKey } : {}),
+        ...(headers !== undefined ? { headers } : {}),
+        maxTokens: params.maxTokens,
+        ...(params.temperature !== undefined ? { temperature: params.temperature } : {}),
+        ...(params.metadata !== undefined ? { metadata: params.metadata } : {}),
+        ...(signal ? { signal } : {}),
+    });
+    const converted = convertAssistantResult(result);
+    throwIfAborted(signal);
+    await confirmSampling(options, "Return MCP sampling response", formatResponseApproval(options.serverName, converted));
+    return converted;
+}
+function formatRequestApproval(serverName, modelName, systemPrompt, messages) {
+    const lines = [
+        `${serverName} wants to sample ${messages.length} message${messages.length === 1 ? "" : "s"} with ${modelName}.`,
+    ];
+    if (systemPrompt) {
+        lines.push(`System: ${truncateAtWord(systemPrompt, 400)}`);
+    }
+    for (const [index, message] of messages.entries()) {
+        lines.push(`${index + 1}. ${message.role}: ${truncateAtWord(messageText(message), 400)}`);
+    }
+    return lines.join("\n\n");
+}
+function formatResponseApproval(serverName, response) {
+    const text = response.content.type === "text" ? response.content.text : `[${response.content.type} content]`;
+    return `${serverName} will receive this response from ${response.model}:\n\n${truncateAtWord(text, 1000)}`;
+}
+function messageText(message) {
+    if (typeof message.content === "string")
+        return message.content;
+    return message.content
+        .map((block) => {
+        if (block.type === "text")
+            return block.text;
+        if (block.type === "image")
+            return `[image: ${block.mimeType}]`;
+        if (block.type === "thinking")
+            return "[thinking]";
+        if (block.type === "toolCall")
+            return `[tool call: ${block.name}]`;
+        return "[content]";
+    })
+        .join("\n");
+}
+async function resolveSamplingModel(options, modelPreferences) {
+    const candidates = [];
+    const availableModels = options.modelRegistry.getAvailable();
+    for (const hint of modelPreferences?.hints ?? []) {
+        const normalizedHint = hint.name?.trim().toLowerCase();
+        if (!normalizedHint)
+            continue;
+        for (const model of availableModels) {
+            const searchableNames = [`${model.provider}/${model.id}`, model.id, model.name];
+            if (searchableNames.some((name) => name.toLowerCase().includes(normalizedHint))) {
+                addSamplingCandidate(candidates, model);
+            }
+        }
+    }
+    const currentModel = options.getCurrentModel();
+    if (currentModel)
+        addSamplingCandidate(candidates, currentModel);
+    for (const model of availableModels) {
+        addSamplingCandidate(candidates, model);
+    }
+    const errors = [];
+    const signal = options.getSignal();
+    for (const model of candidates) {
+        throwIfAborted(signal);
+        const auth = await options.modelRegistry.getApiKeyAndHeaders(model);
+        throwIfAborted(signal);
+        if (auth.ok === false) {
+            errors.push(`${model.provider}/${model.id}: ${auth.error}`);
+            continue;
+        }
+        return {
+            model,
+            ...(auth.apiKey !== undefined ? { apiKey: auth.apiKey } : {}),
+            ...(auth.headers !== undefined ? { headers: auth.headers } : {}),
+        };
+    }
+    if (errors.length > 0) {
+        throw new Error(`No configured auth for MCP sampling model. ${errors.join("; ")}`);
+    }
+    throw new Error("No Nexus model is available for MCP sampling");
+}
+function addSamplingCandidate(candidates, model) {
+    if (!candidates.some((candidate) => candidate.provider === model.provider && candidate.id === model.id)) {
+        candidates.push(model);
+    }
+}
+async function confirmSampling(options, title, message) {
+    if (options.autoApprove)
+        return;
+    if (!options.ui) {
+        throw new Error("MCP sampling requires interactive approval. Set settings.samplingAutoApprove to true to allow it without UI.");
+    }
+    const approved = await options.ui.confirm(title, message);
+    if (!approved) {
+        throw new Error("MCP sampling request was declined");
+    }
+}
+function convertSamplingMessage(message) {
+    const blocks = Array.isArray(message.content) ? message.content : [message.content];
+    if (message.role === "user") {
+        return {
+            role: "user",
+            content: blocks.map(convertUserContent),
+            timestamp: Date.now(),
+        };
+    }
+    return {
+        role: "assistant",
+        content: blocks.map(convertAssistantContent),
+        api: "mcp-sampling",
+        provider: "mcp",
+        model: "sampling-request",
+        usage: zeroUsage(),
+        stopReason: "stop",
+        timestamp: Date.now(),
+    };
+}
+function convertUserContent(block) {
+    if (block.type === "text") {
+        return { type: "text", text: block.text };
+    }
+    throw new Error(`MCP sampling ${block.type} content is not supported`);
+}
+function convertAssistantContent(block) {
+    if (block.type === "text") {
+        return { type: "text", text: block.text };
+    }
+    throw new Error(`MCP sampling assistant ${block.type} content is not supported`);
+}
+function convertAssistantResult(message) {
+    if (message.stopReason === "error") {
+        throw new Error(message.errorMessage ?? "MCP sampling model call failed");
+    }
+    if (message.stopReason === "aborted") {
+        throw new Error(message.errorMessage ?? "MCP sampling model call was aborted");
+    }
+    const text = message.content
+        .map((block) => {
+        if (block.type === "text")
+            return block.text;
+        if (block.type === "thinking")
+            return undefined;
+        throw new Error(`MCP sampling result ${block.type} content is not supported`);
+    })
+        .filter((value) => value !== undefined)
+        .join("\n\n")
+        .trim();
+    if (!text) {
+        throw new Error("MCP sampling result did not contain text content");
+    }
+    return {
+        role: "assistant",
+        content: { type: "text", text },
+        model: `${message.provider}/${message.model}`,
+        stopReason: mapStopReason(message.stopReason),
+    };
+}
+function mapStopReason(reason) {
+    if (reason === "stop")
+        return "endTurn";
+    if (reason === "length")
+        return "maxTokens";
+    if (reason === "toolUse")
+        return "toolUse";
+    return reason;
+}
+function zeroUsage() {
+    return {
+        input: 0,
+        output: 0,
+        cacheRead: 0,
+        cacheWrite: 0,
+        totalTokens: 0,
+        cost: {
+            input: 0,
+            output: 0,
+            cacheRead: 0,
+            cacheWrite: 0,
+            total: 0,
+        },
+    };
+}
+//# sourceMappingURL=sampling-handler.js.map

package/dist/extensions/mcp/sampling-handler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"sampling-handler.js","sourceRoot":"","sources":["../../../src/extensions/mcp/sampling-handler.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAE,MAAM,8BAA8B,CAAC;AAWxD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAC5C,OAAO,EAAE,mBAAmB,EAAE,MAAM,oBAAoB,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AAgB5C,MAAM,UAAU,uBAAuB,CAAC,MAAc,EAAE,OAA+B,EAAQ;IAC9F,MAAM,CAAC,iBAAiB,CAAC,wBAAwB,EAAE,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,CAAC;QACxE,MAAM,aAAa,GAAI,OAAoC,CAAC,MAAM,CAAC;QACnE,OAAO,qBAAqB,CAC3B;YACC,GAAG,OAAO;YACV,SAAS,EAAE,GAAG,EAAE,CAAC,mBAAmB,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,aAAa,CAAC;SACxE,EACD,OAAO,CACP,CAAC;IAAA,CACF,CAAC,CAAC;AAAA,CACH;AAED,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAC1C,OAA+B,EAC/B,OAA6B,EACE;IAC/B,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;IAC9B,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IACnC,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,IAAI,MAAM,IAAI,MAAM,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;QACrC,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;IACzD,CAAC;IACD,IAAI,MAAM,CAAC,cAAc,IAAI,MAAM,CAAC,cAAc,KAAK,MAAM,EAAE,CAAC;QAC/D,MAAM,IAAI,KAAK,CAAC,iDAAiD,CAAC,CAAC;IACpE,CAAC;IACD,IAAI,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,CAAC;QAC1B,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;IAC3D,CAAC;IACD,IAAI,MAAM,CAAC,UAAU,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,2CAA2C,CAAC,CAAC;IAC9D,CAAC;IACD,IAAI,MAAM,CAAC,aAAa,EAAE,MAAM,EAAE,CAAC;QAClC,MAAM,IAAI,KAAK,CAAC,+CAA+C,CAAC,CAAC;IAClE,CAAC;IAED,MAAM,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,sBAAsB,CAAC,CAAC;IAC7D,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;IAChG,cAAc,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,eAAe,CACpB,OAAO,EACP,8BAA8B,EAC9B,qBAAqB,CAAC,OAAO,CAAC,UAAU,EAAE,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,EAAE,EAAE,MAAM,CAAC,YAAY,EAAE,QAAQ,CAAC,CACzG,CAAC;IACF,cAAc,CAAC,MAAM,CAAC,CAAC;IAEvB,MAAM,MAAM,GAAG,MAAM,QAAQ,CAC5B,KAAK,EACL;QACC,GAAG,CAAC,MAAM,CAAC,YAAY,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,YAAY,EAAE,MAAM,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QACnF,QAAQ;KACR,EACD;QACC,GAAG,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC3C,GAAG,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAC7C,SAAS,EAAE,MAAM,CAAC,SAAS;QAC3B,GAAG,CAAC,MAAM,CAAC,WAAW,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,WAAW,EAAE,MAAM,CAAC,WAAW,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAChF,GAAG,CAAC,MAAM,CAAC,QAAQ,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,MAAM,CAAC,QAAmC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;QAClG,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KAC7B,CACD,CAAC;IAEF,MAAM,SAAS,GAAG,sBAAsB,CAAC,MAAM,CAAC,CAAC;IACjD,cAAc,CAAC,MAAM,CAAC,CAAC;IACvB,MAAM,eAAe,CACpB,OAAO,EACP,8BAA8B,EAC9B,sBAAsB,CAAC,OAAO,CAAC,UAAU,EAAE,SAAS,CAAC,CACrD,CAAC;IACF,OAAO,SAAS,CAAC;AAAA,CACjB;AAED,SAAS,qBAAqB,CAC7B,UAAkB,EAClB,SAAiB,EACjB,YAAgC,EAChC,QAAmB,EACV;IACT,MAAM,KAAK,GAAG;QACb,GAAG,UAAU,oBAAoB,QAAQ,CAAC,MAAM,WAAW,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,SAAS,SAAS,GAAG;KAChH,CAAC;IACF,IAAI,YAAY,EAAE,CAAC;QAClB,KAAK,CAAC,IAAI,CAAC,WAAW,cAAc,CAAC,YAAY,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IAC5D,CAAC;IACD,KAAK,MAAM,CAAC,KAAK,EAAE,OAAO,CAAC,IAAI,QAAQ,CAAC,OAAO,EAAE,EAAE,CAAC;QACnD,KAAK,CAAC,IAAI,CAAC,GAAG,KAAK,GAAG,CAAC,KAAK,OAAO,CAAC,IAAI,KAAK,cAAc,CAAC,WAAW,CAAC,OAAO,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC,CAAC;IAC3F,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;AAAA,CAC1B;AAED,SAAS,sBAAsB,CAAC,UAAkB,EAAE,QAA6B,EAAU;IAC1F,MAAM,IAAI,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,IAAI,WAAW,CAAC;IAC7G,OAAO,GAAG,UAAU,oCAAoC,QAAQ,CAAC,KAAK,QAAQ,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,EAAE,CAAC;AAAA,CAC3G;AAED,SAAS,WAAW,CAAC,OAAgB,EAAU;IAC9C,IAAI,OAAO,OAAO,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,OAAO,CAAC,OAAO,CAAC;IAChE,OAAO,OAAO,CAAC,OAAO;SACpB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;QACf,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;QAC7C,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;YAAE,OAAO,WAAW,KAAK,CAAC,QAAQ,GAAG,CAAC;QAChE,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,YAAY,CAAC;QACnD,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,eAAe,KAAK,CAAC,IAAI,GAAG,CAAC;QACnE,OAAO,WAAW,CAAC;IAAA,CACnB,CAAC;SACD,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACb;AAED,KAAK,UAAU,oBAAoB,CAClC,OAA+B,EAC/B,gBAA8C,EAK5C;IACF,MAAM,UAAU,GAAiB,EAAE,CAAC;IACpC,MAAM,eAAe,GAAG,OAAO,CAAC,aAAa,CAAC,YAAY,EAAE,CAAC;IAE7D,KAAK,MAAM,IAAI,IAAI,gBAAgB,EAAE,KAAK,IAAI,EAAE,EAAE,CAAC;QAClD,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QACvD,IAAI,CAAC,cAAc;YAAE,SAAS;QAC9B,KAAK,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;YACrC,MAAM,eAAe,GAAG,CAAC,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YAChF,IAAI,eAAe,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;gBACjF,oBAAoB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACzC,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,YAAY,GAAG,OAAO,CAAC,eAAe,EAAE,CAAC;IAC/C,IAAI,YAAY;QAAE,oBAAoB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IAEjE,KAAK,MAAM,KAAK,IAAI,eAAe,EAAE,CAAC;QACrC,oBAAoB,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACzC,CAAC;IAED,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IACnC,KAAK,MAAM,KAAK,IAAI,UAAU,EAAE,CAAC;QAChC,cAAc,CAAC,MAAM,CAAC,CAAC;QACvB,MAAM,IAAI,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;QACpE,cAAc,CAAC,MAAM,CAAC,CAAC;QACvB,IAAI,IAAI,CAAC,EAAE,KAAK,KAAK,EAAE,CAAC;YACvB,MAAM,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,QAAQ,IAAI,KAAK,CAAC,EAAE,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC,CAAC;YAC5D,SAAS;QACV,CAAC;QACD,OAAO;YACN,KAAK;YACL,GAAG,CAAC,IAAI,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;YAC7D,GAAG,CAAC,IAAI,CAAC,OAAO,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;SAChE,CAAC;IACH,CAAC;IAED,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACvB,MAAM,IAAI,KAAK,CAAC,8CAA8C,MAAM,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IACpF,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,8CAA8C,CAAC,CAAC;AAAA,CAChE;AAED,SAAS,oBAAoB,CAAC,UAAwB,EAAE,KAAiB,EAAQ;IAChF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,QAAQ,KAAK,KAAK,CAAC,QAAQ,IAAI,SAAS,CAAC,EAAE,KAAK,KAAK,CAAC,EAAE,CAAC,EAAE,CAAC;QACzG,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IACxB,CAAC;AAAA,CACD;AAED,KAAK,UAAU,eAAe,CAAC,OAA+B,EAAE,KAAa,EAAE,OAAe,EAAiB;IAC9G,IAAI,OAAO,CAAC,WAAW;QAAE,OAAO;IAChC,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CACd,8GAA8G,CAC9G,CAAC;IACH,CAAC;IACD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,EAAE,CAAC,OAAO,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IAC1D,IAAI,CAAC,QAAQ,EAAE,CAAC;QACf,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACtD,CAAC;AAAA,CACD;AAED,SAAS,sBAAsB,CAAC,OAAwB,EAAW;IAClE,MAAM,MAAM,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACpF,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC7B,OAAO;YACN,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,kBAAkB,CAAC;YACvC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB,CAAC;IACH,CAAC;IAED,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,MAAM,CAAC,GAAG,CAAC,uBAAuB,CAAC;QAC5C,GAAG,EAAE,cAAc;QACnB,QAAQ,EAAE,KAAK;QACf,KAAK,EAAE,kBAAkB;QACzB,KAAK,EAAE,SAAS,EAAE;QAClB,UAAU,EAAE,MAAM;QAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;KACrB,CAAC;AAAA,CACF;AAED,SAAS,kBAAkB,CAAC,KAAkC,EAAe;IAC5E,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC3B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,gBAAgB,KAAK,CAAC,IAAI,2BAA2B,CAAC,CAAC;AAAA,CACvE;AAED,SAAS,uBAAuB,CAAC,KAAkC,EAAe;IACjF,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC3B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC;IAC3C,CAAC;IACD,MAAM,IAAI,KAAK,CAAC,0BAA0B,KAAK,CAAC,IAAI,2BAA2B,CAAC,CAAC;AAAA,CACjF;AAED,SAAS,sBAAsB,CAAC,OAAyB,EAAuB;IAC/E,IAAI,OAAO,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;QACpC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,gCAAgC,CAAC,CAAC;IAC3E,CAAC;IACD,IAAI,OAAO,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACtC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,qCAAqC,CAAC,CAAC;IAChF,CAAC;IAED,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO;SAC1B,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC;QACf,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,KAAK,CAAC,IAAI,CAAC;QAC7C,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU;YAAE,OAAO,SAAS,CAAC;QAChD,MAAM,IAAI,KAAK,CAAC,uBAAuB,KAAK,CAAC,IAAI,2BAA2B,CAAC,CAAC;IAAA,CAC9E,CAAC;SACD,MAAM,CAAC,CAAC,KAAK,EAAmB,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC;SACvD,IAAI,CAAC,MAAM,CAAC;SACZ,IAAI,EAAE,CAAC;IAET,IAAI,CAAC,IAAI,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;IACrE,CAAC;IAED,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE;QAC/B,KAAK,EAAE,GAAG,OAAO,CAAC,QAAQ,IAAI,OAAO,CAAC,KAAK,EAAE;QAC7C,UAAU,EAAE,aAAa,CAAC,OAAO,CAAC,UAAU,CAAC;KAC7C,CAAC;AAAA,CACF;AAED,SAAS,aAAa,CAAC,MAAsC,EAAqC;IACjG,IAAI,MAAM,KAAK,MAAM;QAAE,OAAO,SAAS,CAAC;IACxC,IAAI,MAAM,KAAK,QAAQ;QAAE,OAAO,WAAW,CAAC;IAC5C,IAAI,MAAM,KAAK,SAAS;QAAE,OAAO,SAAS,CAAC;IAC3C,OAAO,MAAM,CAAC;AAAA,CACd;AAED,SAAS,SAAS,GAA8B;IAC/C,OAAO;QACN,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE;YACL,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,CAAC;SACR;KACD,CAAC;AAAA,CACF","sourcesContent":["import type { Api, AssistantMessage, Message, Model, ProviderHeaders, TextContent } from \"@earendil-works/pi-ai\";\nimport { complete } from \"@earendil-works/pi-ai/compat\";\nimport type {\n\tClient,\n\tCreateMessageRequest,\n\tCreateMessageResult,\n\tModelPreferences,\n\tSamplingMessage,\n\tSamplingMessageContentBlock,\n} from \"@modelcontextprotocol/client\";\nimport type { ExtensionUIContext } from \"../../core/extensions/types.ts\";\nimport type { ModelRegistry } from \"../../core/model-registry.ts\";\nimport { throwIfAborted } from \"./abort.ts\";\nimport { combineAbortSignals } from \"./runtime-owner.ts\";\nimport { truncateAtWord } from \"./utils.ts\";\n\nexport type SamplingUIContext = Pick<ExtensionUIContext, \"confirm\">;\nexport type SamplingModelRegistry = Pick<ModelRegistry, \"getAvailable\" | \"getApiKeyAndHeaders\">;\n\nexport interface SamplingHandlerOptions {\n\tserverName: string;\n\tautoApprove: boolean;\n\tui?: SamplingUIContext;\n\tmodelRegistry: SamplingModelRegistry;\n\tgetCurrentModel: () => Model<Api> | undefined;\n\tgetSignal: () => AbortSignal | undefined;\n}\n\nexport type ServerSamplingConfig = Omit<SamplingHandlerOptions, \"serverName\">;\n\nexport function registerSamplingHandler(client: Client, options: SamplingHandlerOptions): void {\n\tclient.setRequestHandler(\"sampling/createMessage\", (request, context) => {\n\t\tconst requestSignal = (context as { signal?: AbortSignal }).signal;\n\t\treturn handleSamplingRequest(\n\t\t\t{\n\t\t\t\t...options,\n\t\t\t\tgetSignal: () => combineAbortSignals(options.getSignal(), requestSignal),\n\t\t\t},\n\t\t\trequest,\n\t\t);\n\t});\n}\n\nexport async function handleSamplingRequest(\n\toptions: SamplingHandlerOptions,\n\trequest: CreateMessageRequest,\n): Promise<CreateMessageResult> {\n\tconst params = request.params;\n\tconst signal = options.getSignal();\n\tthrowIfAborted(signal);\n\n\tif (\"task\" in params && params.task) {\n\t\tthrow new Error(\"MCP sampling tasks are not supported\");\n\t}\n\tif (params.includeContext && params.includeContext !== \"none\") {\n\t\tthrow new Error(\"MCP sampling context inclusion is not supported\");\n\t}\n\tif (params.tools?.length) {\n\t\tthrow new Error(\"MCP sampling tool use is not supported\");\n\t}\n\tif (params.toolChoice) {\n\t\tthrow new Error(\"MCP sampling tool choice is not supported\");\n\t}\n\tif (params.stopSequences?.length) {\n\t\tthrow new Error(\"MCP sampling stop sequences are not supported\");\n\t}\n\n\tconst messages = params.messages.map(convertSamplingMessage);\n\tconst { model, apiKey, headers } = await resolveSamplingModel(options, params.modelPreferences);\n\tthrowIfAborted(signal);\n\tawait confirmSampling(\n\t\toptions,\n\t\t\"Approve MCP sampling request\",\n\t\tformatRequestApproval(options.serverName, `${model.provider}/${model.id}`, params.systemPrompt, messages),\n\t);\n\tthrowIfAborted(signal);\n\n\tconst result = await complete(\n\t\tmodel,\n\t\t{\n\t\t\t...(params.systemPrompt !== undefined ? { systemPrompt: params.systemPrompt } : {}),\n\t\t\tmessages,\n\t\t},\n\t\t{\n\t\t\t...(apiKey !== undefined ? { apiKey } : {}),\n\t\t\t...(headers !== undefined ? { headers } : {}),\n\t\t\tmaxTokens: params.maxTokens,\n\t\t\t...(params.temperature !== undefined ? { temperature: params.temperature } : {}),\n\t\t\t...(params.metadata !== undefined ? { metadata: params.metadata as Record<string, unknown> } : {}),\n\t\t\t...(signal ? { signal } : {}),\n\t\t},\n\t);\n\n\tconst converted = convertAssistantResult(result);\n\tthrowIfAborted(signal);\n\tawait confirmSampling(\n\t\toptions,\n\t\t\"Return MCP sampling response\",\n\t\tformatResponseApproval(options.serverName, converted),\n\t);\n\treturn converted;\n}\n\nfunction formatRequestApproval(\n\tserverName: string,\n\tmodelName: string,\n\tsystemPrompt: string | undefined,\n\tmessages: Message[],\n): string {\n\tconst lines = [\n\t\t`${serverName} wants to sample ${messages.length} message${messages.length === 1 ? \"\" : \"s\"} with ${modelName}.`,\n\t];\n\tif (systemPrompt) {\n\t\tlines.push(`System: ${truncateAtWord(systemPrompt, 400)}`);\n\t}\n\tfor (const [index, message] of messages.entries()) {\n\t\tlines.push(`${index + 1}. ${message.role}: ${truncateAtWord(messageText(message), 400)}`);\n\t}\n\treturn lines.join(\"\\n\\n\");\n}\n\nfunction formatResponseApproval(serverName: string, response: CreateMessageResult): string {\n\tconst text = response.content.type === \"text\" ? response.content.text : `[${response.content.type} content]`;\n\treturn `${serverName} will receive this response from ${response.model}:\\n\\n${truncateAtWord(text, 1000)}`;\n}\n\nfunction messageText(message: Message): string {\n\tif (typeof message.content === \"string\") return message.content;\n\treturn message.content\n\t\t.map((block) => {\n\t\t\tif (block.type === \"text\") return block.text;\n\t\t\tif (block.type === \"image\") return `[image: ${block.mimeType}]`;\n\t\t\tif (block.type === \"thinking\") return \"[thinking]\";\n\t\t\tif (block.type === \"toolCall\") return `[tool call: ${block.name}]`;\n\t\t\treturn \"[content]\";\n\t\t})\n\t\t.join(\"\\n\");\n}\n\nasync function resolveSamplingModel(\n\toptions: SamplingHandlerOptions,\n\tmodelPreferences: ModelPreferences | undefined,\n): Promise<{\n\tmodel: Model<Api>;\n\tapiKey?: string;\n\theaders?: ProviderHeaders;\n}> {\n\tconst candidates: Model<Api>[] = [];\n\tconst availableModels = options.modelRegistry.getAvailable();\n\n\tfor (const hint of modelPreferences?.hints ?? []) {\n\t\tconst normalizedHint = hint.name?.trim().toLowerCase();\n\t\tif (!normalizedHint) continue;\n\t\tfor (const model of availableModels) {\n\t\t\tconst searchableNames = [`${model.provider}/${model.id}`, model.id, model.name];\n\t\t\tif (searchableNames.some((name) => name.toLowerCase().includes(normalizedHint))) {\n\t\t\t\taddSamplingCandidate(candidates, model);\n\t\t\t}\n\t\t}\n\t}\n\n\tconst currentModel = options.getCurrentModel();\n\tif (currentModel) addSamplingCandidate(candidates, currentModel);\n\n\tfor (const model of availableModels) {\n\t\taddSamplingCandidate(candidates, model);\n\t}\n\n\tconst errors: string[] = [];\n\tconst signal = options.getSignal();\n\tfor (const model of candidates) {\n\t\tthrowIfAborted(signal);\n\t\tconst auth = await options.modelRegistry.getApiKeyAndHeaders(model);\n\t\tthrowIfAborted(signal);\n\t\tif (auth.ok === false) {\n\t\t\terrors.push(`${model.provider}/${model.id}: ${auth.error}`);\n\t\t\tcontinue;\n\t\t}\n\t\treturn {\n\t\t\tmodel,\n\t\t\t...(auth.apiKey !== undefined ? { apiKey: auth.apiKey } : {}),\n\t\t\t...(auth.headers !== undefined ? { headers: auth.headers } : {}),\n\t\t};\n\t}\n\n\tif (errors.length > 0) {\n\t\tthrow new Error(`No configured auth for MCP sampling model. ${errors.join(\"; \")}`);\n\t}\n\tthrow new Error(\"No Nexus model is available for MCP sampling\");\n}\n\nfunction addSamplingCandidate(candidates: Model<Api>[], model: Model<Api>): void {\n\tif (!candidates.some((candidate) => candidate.provider === model.provider && candidate.id === model.id)) {\n\t\tcandidates.push(model);\n\t}\n}\n\nasync function confirmSampling(options: SamplingHandlerOptions, title: string, message: string): Promise<void> {\n\tif (options.autoApprove) return;\n\tif (!options.ui) {\n\t\tthrow new Error(\n\t\t\t\"MCP sampling requires interactive approval. Set settings.samplingAutoApprove to true to allow it without UI.\",\n\t\t);\n\t}\n\tconst approved = await options.ui.confirm(title, message);\n\tif (!approved) {\n\t\tthrow new Error(\"MCP sampling request was declined\");\n\t}\n}\n\nfunction convertSamplingMessage(message: SamplingMessage): Message {\n\tconst blocks = Array.isArray(message.content) ? message.content : [message.content];\n\tif (message.role === \"user\") {\n\t\treturn {\n\t\t\trole: \"user\",\n\t\t\tcontent: blocks.map(convertUserContent),\n\t\t\ttimestamp: Date.now(),\n\t\t};\n\t}\n\n\treturn {\n\t\trole: \"assistant\",\n\t\tcontent: blocks.map(convertAssistantContent),\n\t\tapi: \"mcp-sampling\",\n\t\tprovider: \"mcp\",\n\t\tmodel: \"sampling-request\",\n\t\tusage: zeroUsage(),\n\t\tstopReason: \"stop\",\n\t\ttimestamp: Date.now(),\n\t};\n}\n\nfunction convertUserContent(block: SamplingMessageContentBlock): TextContent {\n\tif (block.type === \"text\") {\n\t\treturn { type: \"text\", text: block.text };\n\t}\n\tthrow new Error(`MCP sampling ${block.type} content is not supported`);\n}\n\nfunction convertAssistantContent(block: SamplingMessageContentBlock): TextContent {\n\tif (block.type === \"text\") {\n\t\treturn { type: \"text\", text: block.text };\n\t}\n\tthrow new Error(`MCP sampling assistant ${block.type} content is not supported`);\n}\n\nfunction convertAssistantResult(message: AssistantMessage): CreateMessageResult {\n\tif (message.stopReason === \"error\") {\n\t\tthrow new Error(message.errorMessage ?? \"MCP sampling model call failed\");\n\t}\n\tif (message.stopReason === \"aborted\") {\n\t\tthrow new Error(message.errorMessage ?? \"MCP sampling model call was aborted\");\n\t}\n\n\tconst text = message.content\n\t\t.map((block) => {\n\t\t\tif (block.type === \"text\") return block.text;\n\t\t\tif (block.type === \"thinking\") return undefined;\n\t\t\tthrow new Error(`MCP sampling result ${block.type} content is not supported`);\n\t\t})\n\t\t.filter((value): value is string => value !== undefined)\n\t\t.join(\"\\n\\n\")\n\t\t.trim();\n\n\tif (!text) {\n\t\tthrow new Error(\"MCP sampling result did not contain text content\");\n\t}\n\n\treturn {\n\t\trole: \"assistant\",\n\t\tcontent: { type: \"text\", text },\n\t\tmodel: `${message.provider}/${message.model}`,\n\t\tstopReason: mapStopReason(message.stopReason),\n\t};\n}\n\nfunction mapStopReason(reason: AssistantMessage[\"stopReason\"]): CreateMessageResult[\"stopReason\"] {\n\tif (reason === \"stop\") return \"endTurn\";\n\tif (reason === \"length\") return \"maxTokens\";\n\tif (reason === \"toolUse\") return \"toolUse\";\n\treturn reason;\n}\n\nfunction zeroUsage(): AssistantMessage[\"usage\"] {\n\treturn {\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\ttotal: 0,\n\t\t},\n\t};\n}\n"]}

package/dist/extensions/mcp/search-ranking.d.ts

@@ -0,0 +1,25 @@
+import type { McpExtensionState } from "./state.ts";
+import type { ServerEntry, ToolMetadata, ToolPrefix } from "./types.ts";
+export interface RankedToolMatch {
+    server: string;
+    tool: ToolMetadata;
+    score: number;
+}
+/**
+ * Resolve the configured searchKeywords entries that apply to a tool.
+ * Keys match by original name, prefixed name, or glob — the same candidate
+ * set includeTools/excludeTools use — and all matching entries are unioned.
+ */
+export declare function resolveSearchKeywords(definition: ServerEntry | undefined, toolOriginalName: string, serverName: string, globalPrefix: ToolPrefix): string[];
+export declare function normalizeSearchText(value: string): string;
+export declare function tokenize(value: string): string[];
+export declare function scoreToolMatch(tool: ToolMetadata, server: string, query: string, keywords?: string[]): number | null;
+export declare function rankToolMatches(state: McpExtensionState, query: string, server?: string, includeKeywords?: boolean): RankedToolMatch[];
+export declare function paginate<T>(items: T[], offset: number, limit: number): {
+    items: T[];
+    total: number;
+    hasMore: boolean;
+    nextOffset: number | null;
+};
+export declare function rankSuggestions(state: McpExtensionState, name: string, limit: number): string[];
+//# sourceMappingURL=search-ranking.d.ts.map

package/dist/extensions/mcp/search-ranking.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"search-ranking.d.ts","sourceRoot":"","sources":["../../../src/extensions/mcp/search-ranking.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AACpD,OAAO,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAwBxE,MAAM,WAAW,eAAe;IAC/B,MAAM,EAAE,MAAM,CAAC;IACf,IAAI,EAAE,YAAY,CAAC;IACnB,KAAK,EAAE,MAAM,CAAC;CACd;AAsBD;;;;GAIG;AACH,wBAAgB,qBAAqB,CACpC,UAAU,EAAE,WAAW,GAAG,SAAS,EACnC,gBAAgB,EAAE,MAAM,EACxB,UAAU,EAAE,MAAM,EAClB,YAAY,EAAE,UAAU,GACtB,MAAM,EAAE,CAkBV;AAED,wBAAgB,mBAAmB,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,CAKzD;AAED,wBAAgB,QAAQ,CAAC,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAIhD;AAqHD,wBAAgB,cAAc,CAAC,IAAI,EAAE,YAAY,EAAE,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,MAAM,EAAE,GAAG,MAAM,GAAG,IAAI,CAKpH;AA0CD,wBAAgB,eAAe,CAC9B,KAAK,EAAE,iBAAiB,EACxB,KAAK,EAAE,MAAM,EACb,MAAM,CAAC,EAAE,MAAM,EACf,eAAe,UAAO,GACpB,eAAe,EAAE,CAgBnB;AAED,wBAAgB,QAAQ,CAAC,CAAC,EACzB,KAAK,EAAE,CAAC,EAAE,EACV,MAAM,EAAE,MAAM,EACd,KAAK,EAAE,MAAM,GACX;IAAE,KAAK,EAAE,CAAC,EAAE,CAAC;IAAC,KAAK,EAAE,MAAM,CAAC;IAAC,OAAO,EAAE,OAAO,CAAC;IAAC,UAAU,EAAE,MAAM,GAAG,IAAI,CAAA;CAAE,CAY5E;AAED,wBAAgB,eAAe,CAAC,KAAK,EAAE,iBAAiB,EAAE,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,GAAG,MAAM,EAAE,CAU/F","sourcesContent":["import type { McpExtensionState } from \"./state.ts\";\nimport type { ServerEntry, ToolMetadata, ToolPrefix } from \"./types.ts\";\nimport {\n\tgetServerPrefix,\n\tgetToolNameCandidates,\n\tisServerDisabled,\n\tmatchesToolPattern,\n\tresolveToolPrefix,\n} from \"./types.ts\";\n\n/**\n * Shortest field token allowed to stem-match a longer query token.\n * Real descriptions tokenize possessives into single letters (\"project's\" -> [\"project\", \"s\"]),\n * which would otherwise make every query starting with that letter a match.\n */\nconst MIN_STEM_LENGTH = 4;\n\nconst FIELD_WEIGHTS = {\n\tname: 12,\n\toriginalName: 10,\n\tserver: 8,\n\tdescription: 5,\n\tkeywords: 5,\n} as const;\n\nexport interface RankedToolMatch {\n\tserver: string;\n\ttool: ToolMetadata;\n\tscore: number;\n}\n\ntype SearchField = Exclude<keyof typeof FIELD_WEIGHTS, \"keywords\">;\n\ninterface PreparedToolSearch {\n\ttool: ToolMetadata;\n\tfields: Array<[SearchField, string, string[]]>;\n\tnameTokens: string[];\n\tkeywordPhrases: string[];\n\tkeywordTokens: string[];\n}\n\ninterface CachedServerSearch {\n\tmetadata: ToolMetadata[];\n\tsearchKeywords: ServerEntry[\"searchKeywords\"];\n\ttoolPrefix: ReturnType<typeof resolveToolPrefix>;\n\twithKeywords?: PreparedToolSearch[];\n\twithoutKeywords?: PreparedToolSearch[];\n}\n\nconst searchCache = new WeakMap<McpExtensionState, Map<string, CachedServerSearch>>();\n\n/**\n * Resolve the configured searchKeywords entries that apply to a tool.\n * Keys match by original name, prefixed name, or glob — the same candidate\n * set includeTools/excludeTools use — and all matching entries are unioned.\n */\nexport function resolveSearchKeywords(\n\tdefinition: ServerEntry | undefined,\n\ttoolOriginalName: string,\n\tserverName: string,\n\tglobalPrefix: ToolPrefix,\n): string[] {\n\tconst map = definition?.searchKeywords;\n\tif (!map || typeof map !== \"object\" || Array.isArray(map)) return [];\n\tconst candidates = getToolNameCandidates(toolOriginalName, serverName, resolveToolPrefix(definition, globalPrefix));\n\tconst keywords: string[] = [];\n\tconst seen = new Set<string>();\n\tfor (const [pattern, values] of Object.entries(map)) {\n\t\tif (!Array.isArray(values)) continue;\n\t\tif (!matchesToolPattern(candidates, [pattern])) continue;\n\t\tfor (const value of values) {\n\t\t\tif (typeof value !== \"string\") continue;\n\t\t\tconst trimmed = value.trim();\n\t\t\tif (!trimmed || seen.has(trimmed)) continue;\n\t\t\tseen.add(trimmed);\n\t\t\tkeywords.push(trimmed);\n\t\t}\n\t}\n\treturn keywords;\n}\n\nexport function normalizeSearchText(value: string): string {\n\treturn value\n\t\t.replace(/([a-z0-9])([A-Z])/g, \"$1 $2\")\n\t\t.replace(/[_./:-]+/g, \" \")\n\t\t.toLowerCase();\n}\n\nexport function tokenize(value: string): string[] {\n\treturn normalizeSearchText(value)\n\t\t.split(/[^a-z0-9]+/)\n\t\t.filter(Boolean);\n}\n\nfunction prepareToolSearch(tool: ToolMetadata, server: string, keywords?: string[]): PreparedToolSearch {\n\tconst fields = {\n\t\tname: normalizeSearchText(tool.name),\n\t\toriginalName: normalizeSearchText(tool.originalName),\n\t\tserver: normalizeSearchText(server),\n\t\tdescription: normalizeSearchText(tool.description),\n\t};\n\tconst preparedFields = (Object.entries(fields) as Array<[SearchField, string]>).map(\n\t\t([field, value]): [SearchField, string, string[]] => [field, value, tokenize(value)],\n\t);\n\tconst keywordPhrases = keywords?.map((keyword) => normalizeSearchText(keyword).trim()).filter(Boolean) ?? [];\n\treturn {\n\t\ttool,\n\t\tfields: preparedFields,\n\t\tnameTokens: preparedFields[0]![2],\n\t\tkeywordPhrases,\n\t\tkeywordTokens: keywordPhrases.flatMap(tokenize),\n\t};\n}\n\nfunction scorePreparedToolMatch(\n\tprepared: PreparedToolSearch,\n\tnormalizedQuery: string,\n\tqueryTokens: string[],\n): number | null {\n\tlet score = 0;\n\tlet phraseMatched = false;\n\tlet wholeFieldExact = false;\n\tconst matchedTokens = new Set<string>();\n\n\tfor (const [field, value, fieldTokens] of prepared.fields) {\n\t\tconst weight = FIELD_WEIGHTS[field];\n\t\tif (value === normalizedQuery) {\n\t\t\tscore += weight * 14;\n\t\t\tphraseMatched = true;\n\t\t\twholeFieldExact = true;\n\t\t} else if (value.startsWith(normalizedQuery)) {\n\t\t\tscore += weight * 9;\n\t\t\tphraseMatched = true;\n\t\t} else if (value.includes(normalizedQuery)) {\n\t\t\tscore += weight * 6;\n\t\t\tphraseMatched = true;\n\t\t}\n\n\t\tfor (const token of queryTokens) {\n\t\t\tif (fieldTokens.includes(token)) {\n\t\t\t\tscore += weight * 4;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (\n\t\t\t\tfieldTokens.some(\n\t\t\t\t\t(fieldToken) =>\n\t\t\t\t\t\tfieldToken.startsWith(token) ||\n\t\t\t\t\t\t(fieldToken.length >= MIN_STEM_LENGTH && token.startsWith(fieldToken)),\n\t\t\t\t)\n\t\t\t) {\n\t\t\t\tscore += weight * 2;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (value.includes(token)) {\n\t\t\t\tscore += weight;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t}\n\t\t}\n\t}\n\n\t// Configured keywords are discrete phrases, so the phrase-level bonus is\n\t// computed per phrase (best match wins) rather than on a joined string,\n\t// which would phrase-match queries spanning two unrelated keywords.\n\tif (prepared.keywordPhrases.length > 0) {\n\t\tconst weight = FIELD_WEIGHTS.keywords;\n\t\tlet phraseScore = 0;\n\t\tfor (const phrase of prepared.keywordPhrases) {\n\t\t\tif (phrase === normalizedQuery) {\n\t\t\t\tphraseScore = Math.max(phraseScore, weight * 14);\n\t\t\t\tphraseMatched = true;\n\t\t\t\twholeFieldExact = true;\n\t\t\t} else if (phrase.startsWith(normalizedQuery)) {\n\t\t\t\tphraseScore = Math.max(phraseScore, weight * 9);\n\t\t\t\tphraseMatched = true;\n\t\t\t} else if (phrase.includes(normalizedQuery)) {\n\t\t\t\tphraseScore = Math.max(phraseScore, weight * 6);\n\t\t\t\tphraseMatched = true;\n\t\t\t}\n\t\t}\n\t\tscore += phraseScore;\n\n\t\tfor (const token of queryTokens) {\n\t\t\tif (prepared.keywordTokens.includes(token)) {\n\t\t\t\tscore += weight * 4;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (\n\t\t\t\tprepared.keywordTokens.some(\n\t\t\t\t\t(keywordToken) =>\n\t\t\t\t\t\tkeywordToken.startsWith(token) ||\n\t\t\t\t\t\t(keywordToken.length >= MIN_STEM_LENGTH && token.startsWith(keywordToken)),\n\t\t\t\t)\n\t\t\t) {\n\t\t\t\tscore += weight * 2;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (prepared.keywordPhrases.some((phrase) => phrase.includes(token))) {\n\t\t\t\tscore += weight;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t}\n\t\t}\n\t}\n\n\tconst coverage = matchedTokens.size / queryTokens.length;\n\tif (!phraseMatched && (queryTokens.length <= 2 ? coverage !== 1 : coverage < 0.6)) return null;\n\n\tscore += coverage === 1 ? 25 : Math.round(coverage * 10);\n\tconst firstQueryToken = queryTokens[0];\n\tif (firstQueryToken !== undefined && prepared.nameTokens.includes(firstQueryToken)) score += 8;\n\tif (wholeFieldExact) score += 20;\n\treturn score;\n}\n\nexport function scoreToolMatch(tool: ToolMetadata, server: string, query: string, keywords?: string[]): number | null {\n\tconst normalizedQuery = normalizeSearchText(query).trim();\n\tconst queryTokens = tokenize(query);\n\tif (queryTokens.length === 0) return null;\n\treturn scorePreparedToolMatch(prepareToolSearch(tool, server, keywords), normalizedQuery, queryTokens);\n}\n\nfunction getPreparedTools(\n\tstate: McpExtensionState,\n\tserverName: string,\n\tmetadata: ToolMetadata[],\n\tdefinition: ServerEntry | undefined,\n\tglobalPrefix: ToolPrefix,\n\tincludeKeywords: boolean,\n): PreparedToolSearch[] {\n\tlet stateCache = searchCache.get(state);\n\tif (!stateCache) {\n\t\tstateCache = new Map();\n\t\tsearchCache.set(state, stateCache);\n\t}\n\tconst toolPrefix = resolveToolPrefix(definition, globalPrefix);\n\tlet cached = stateCache.get(serverName);\n\tif (\n\t\tcached?.metadata !== metadata ||\n\t\tcached.searchKeywords !== definition?.searchKeywords ||\n\t\tcached.toolPrefix !== toolPrefix\n\t) {\n\t\tcached = { metadata, searchKeywords: definition?.searchKeywords, toolPrefix };\n\t\tstateCache.set(serverName, cached);\n\t}\n\tconst cacheKey = includeKeywords ? \"withKeywords\" : \"withoutKeywords\";\n\tlet prepared = cached[cacheKey];\n\tif (!prepared) {\n\t\tprepared = metadata.map((tool) =>\n\t\t\tprepareToolSearch(\n\t\t\t\ttool,\n\t\t\t\tserverName,\n\t\t\t\tincludeKeywords && definition?.searchKeywords !== undefined\n\t\t\t\t\t? resolveSearchKeywords(definition, tool.originalName, serverName, globalPrefix)\n\t\t\t\t\t: undefined,\n\t\t\t),\n\t\t);\n\t\tcached[cacheKey] = prepared;\n\t}\n\treturn prepared;\n}\n\nexport function rankToolMatches(\n\tstate: McpExtensionState,\n\tquery: string,\n\tserver?: string,\n\tincludeKeywords = true,\n): RankedToolMatch[] {\n\tconst matches: RankedToolMatch[] = [];\n\tconst normalizedQuery = normalizeSearchText(query).trim();\n\tconst queryTokens = tokenize(query);\n\tif (queryTokens.length === 0) return matches;\n\tconst globalPrefix = state.config.settings?.toolPrefix ?? \"server\";\n\tfor (const [serverName, metadata] of state.toolMetadata.entries()) {\n\t\tif (server && serverName !== server) continue;\n\t\tconst definition = state.config.mcpServers[serverName];\n\t\tif (isServerDisabled(definition)) continue;\n\t\tfor (const prepared of getPreparedTools(state, serverName, metadata, definition, globalPrefix, includeKeywords)) {\n\t\t\tconst score = scorePreparedToolMatch(prepared, normalizedQuery, queryTokens);\n\t\t\tif (score !== null) matches.push({ server: serverName, tool: prepared.tool, score });\n\t\t}\n\t}\n\treturn matches.sort((a, b) => b.score - a.score || a.tool.name.localeCompare(b.tool.name));\n}\n\nexport function paginate<T>(\n\titems: T[],\n\toffset: number,\n\tlimit: number,\n): { items: T[]; total: number; hasMore: boolean; nextOffset: number | null } {\n\tconst safeOffset = Number.isFinite(offset) ? Math.max(0, Math.trunc(offset)) : 0;\n\tconst safeLimit = Number.isFinite(limit) ? Math.max(1, Math.trunc(limit)) : 1;\n\tconst total = items.length;\n\tconst page = items.slice(safeOffset, safeOffset + safeLimit);\n\tconst nextOffset = safeOffset + page.length;\n\treturn {\n\t\titems: page,\n\t\ttotal,\n\t\thasMore: nextOffset < total,\n\t\tnextOffset: nextOffset < total ? nextOffset : null,\n\t};\n}\n\nexport function rankSuggestions(state: McpExtensionState, name: string, limit: number): string[] {\n\tconst stripped = Object.keys(state.config.mcpServers)\n\t\t.flatMap((server) => ([\"server\", \"short\", \"mcp\"] as const).map((prefix) => getServerPrefix(server, prefix)))\n\t\t.filter((candidate): candidate is string => Boolean(candidate) && name.startsWith(`${candidate}_`))\n\t\t.sort((a, b) => b.length - a.length)\n\t\t.map((candidate) => name.slice(candidate.length + 1));\n\tconst query = stripped[0] ?? name;\n\treturn rankToolMatches(state, query, undefined, false)\n\t\t.slice(0, limit)\n\t\t.map((match) => match.tool.name);\n}\n"]}

package/dist/extensions/mcp/search-ranking.js

@@ -0,0 +1,234 @@
+import { getServerPrefix, getToolNameCandidates, isServerDisabled, matchesToolPattern, resolveToolPrefix, } from "./types.js";
+/**
+ * Shortest field token allowed to stem-match a longer query token.
+ * Real descriptions tokenize possessives into single letters ("project's" -> ["project", "s"]),
+ * which would otherwise make every query starting with that letter a match.
+ */
+const MIN_STEM_LENGTH = 4;
+const FIELD_WEIGHTS = {
+    name: 12,
+    originalName: 10,
+    server: 8,
+    description: 5,
+    keywords: 5,
+};
+const searchCache = new WeakMap();
+/**
+ * Resolve the configured searchKeywords entries that apply to a tool.
+ * Keys match by original name, prefixed name, or glob — the same candidate
+ * set includeTools/excludeTools use — and all matching entries are unioned.
+ */
+export function resolveSearchKeywords(definition, toolOriginalName, serverName, globalPrefix) {
+    const map = definition?.searchKeywords;
+    if (!map || typeof map !== "object" || Array.isArray(map))
+        return [];
+    const candidates = getToolNameCandidates(toolOriginalName, serverName, resolveToolPrefix(definition, globalPrefix));
+    const keywords = [];
+    const seen = new Set();
+    for (const [pattern, values] of Object.entries(map)) {
+        if (!Array.isArray(values))
+            continue;
+        if (!matchesToolPattern(candidates, [pattern]))
+            continue;
+        for (const value of values) {
+            if (typeof value !== "string")
+                continue;
+            const trimmed = value.trim();
+            if (!trimmed || seen.has(trimmed))
+                continue;
+            seen.add(trimmed);
+            keywords.push(trimmed);
+        }
+    }
+    return keywords;
+}
+export function normalizeSearchText(value) {
+    return value
+        .replace(/([a-z0-9])([A-Z])/g, "$1 $2")
+        .replace(/[_./:-]+/g, " ")
+        .toLowerCase();
+}
+export function tokenize(value) {
+    return normalizeSearchText(value)
+        .split(/[^a-z0-9]+/)
+        .filter(Boolean);
+}
+function prepareToolSearch(tool, server, keywords) {
+    const fields = {
+        name: normalizeSearchText(tool.name),
+        originalName: normalizeSearchText(tool.originalName),
+        server: normalizeSearchText(server),
+        description: normalizeSearchText(tool.description),
+    };
+    const preparedFields = Object.entries(fields).map(([field, value]) => [field, value, tokenize(value)]);
+    const keywordPhrases = keywords?.map((keyword) => normalizeSearchText(keyword).trim()).filter(Boolean) ?? [];
+    return {
+        tool,
+        fields: preparedFields,
+        nameTokens: preparedFields[0][2],
+        keywordPhrases,
+        keywordTokens: keywordPhrases.flatMap(tokenize),
+    };
+}
+function scorePreparedToolMatch(prepared, normalizedQuery, queryTokens) {
+    let score = 0;
+    let phraseMatched = false;
+    let wholeFieldExact = false;
+    const matchedTokens = new Set();
+    for (const [field, value, fieldTokens] of prepared.fields) {
+        const weight = FIELD_WEIGHTS[field];
+        if (value === normalizedQuery) {
+            score += weight * 14;
+            phraseMatched = true;
+            wholeFieldExact = true;
+        }
+        else if (value.startsWith(normalizedQuery)) {
+            score += weight * 9;
+            phraseMatched = true;
+        }
+        else if (value.includes(normalizedQuery)) {
+            score += weight * 6;
+            phraseMatched = true;
+        }
+        for (const token of queryTokens) {
+            if (fieldTokens.includes(token)) {
+                score += weight * 4;
+                matchedTokens.add(token);
+            }
+            else if (fieldTokens.some((fieldToken) => fieldToken.startsWith(token) ||
+                (fieldToken.length >= MIN_STEM_LENGTH && token.startsWith(fieldToken)))) {
+                score += weight * 2;
+                matchedTokens.add(token);
+            }
+            else if (value.includes(token)) {
+                score += weight;
+                matchedTokens.add(token);
+            }
+        }
+    }
+    // Configured keywords are discrete phrases, so the phrase-level bonus is
+    // computed per phrase (best match wins) rather than on a joined string,
+    // which would phrase-match queries spanning two unrelated keywords.
+    if (prepared.keywordPhrases.length > 0) {
+        const weight = FIELD_WEIGHTS.keywords;
+        let phraseScore = 0;
+        for (const phrase of prepared.keywordPhrases) {
+            if (phrase === normalizedQuery) {
+                phraseScore = Math.max(phraseScore, weight * 14);
+                phraseMatched = true;
+                wholeFieldExact = true;
+            }
+            else if (phrase.startsWith(normalizedQuery)) {
+                phraseScore = Math.max(phraseScore, weight * 9);
+                phraseMatched = true;
+            }
+            else if (phrase.includes(normalizedQuery)) {
+                phraseScore = Math.max(phraseScore, weight * 6);
+                phraseMatched = true;
+            }
+        }
+        score += phraseScore;
+        for (const token of queryTokens) {
+            if (prepared.keywordTokens.includes(token)) {
+                score += weight * 4;
+                matchedTokens.add(token);
+            }
+            else if (prepared.keywordTokens.some((keywordToken) => keywordToken.startsWith(token) ||
+                (keywordToken.length >= MIN_STEM_LENGTH && token.startsWith(keywordToken)))) {
+                score += weight * 2;
+                matchedTokens.add(token);
+            }
+            else if (prepared.keywordPhrases.some((phrase) => phrase.includes(token))) {
+                score += weight;
+                matchedTokens.add(token);
+            }
+        }
+    }
+    const coverage = matchedTokens.size / queryTokens.length;
+    if (!phraseMatched && (queryTokens.length <= 2 ? coverage !== 1 : coverage < 0.6))
+        return null;
+    score += coverage === 1 ? 25 : Math.round(coverage * 10);
+    const firstQueryToken = queryTokens[0];
+    if (firstQueryToken !== undefined && prepared.nameTokens.includes(firstQueryToken))
+        score += 8;
+    if (wholeFieldExact)
+        score += 20;
+    return score;
+}
+export function scoreToolMatch(tool, server, query, keywords) {
+    const normalizedQuery = normalizeSearchText(query).trim();
+    const queryTokens = tokenize(query);
+    if (queryTokens.length === 0)
+        return null;
+    return scorePreparedToolMatch(prepareToolSearch(tool, server, keywords), normalizedQuery, queryTokens);
+}
+function getPreparedTools(state, serverName, metadata, definition, globalPrefix, includeKeywords) {
+    let stateCache = searchCache.get(state);
+    if (!stateCache) {
+        stateCache = new Map();
+        searchCache.set(state, stateCache);
+    }
+    const toolPrefix = resolveToolPrefix(definition, globalPrefix);
+    let cached = stateCache.get(serverName);
+    if (cached?.metadata !== metadata ||
+        cached.searchKeywords !== definition?.searchKeywords ||
+        cached.toolPrefix !== toolPrefix) {
+        cached = { metadata, searchKeywords: definition?.searchKeywords, toolPrefix };
+        stateCache.set(serverName, cached);
+    }
+    const cacheKey = includeKeywords ? "withKeywords" : "withoutKeywords";
+    let prepared = cached[cacheKey];
+    if (!prepared) {
+        prepared = metadata.map((tool) => prepareToolSearch(tool, serverName, includeKeywords && definition?.searchKeywords !== undefined
+            ? resolveSearchKeywords(definition, tool.originalName, serverName, globalPrefix)
+            : undefined));
+        cached[cacheKey] = prepared;
+    }
+    return prepared;
+}
+export function rankToolMatches(state, query, server, includeKeywords = true) {
+    const matches = [];
+    const normalizedQuery = normalizeSearchText(query).trim();
+    const queryTokens = tokenize(query);
+    if (queryTokens.length === 0)
+        return matches;
+    const globalPrefix = state.config.settings?.toolPrefix ?? "server";
+    for (const [serverName, metadata] of state.toolMetadata.entries()) {
+        if (server && serverName !== server)
+            continue;
+        const definition = state.config.mcpServers[serverName];
+        if (isServerDisabled(definition))
+            continue;
+        for (const prepared of getPreparedTools(state, serverName, metadata, definition, globalPrefix, includeKeywords)) {
+            const score = scorePreparedToolMatch(prepared, normalizedQuery, queryTokens);
+            if (score !== null)
+                matches.push({ server: serverName, tool: prepared.tool, score });
+        }
+    }
+    return matches.sort((a, b) => b.score - a.score || a.tool.name.localeCompare(b.tool.name));
+}
+export function paginate(items, offset, limit) {
+    const safeOffset = Number.isFinite(offset) ? Math.max(0, Math.trunc(offset)) : 0;
+    const safeLimit = Number.isFinite(limit) ? Math.max(1, Math.trunc(limit)) : 1;
+    const total = items.length;
+    const page = items.slice(safeOffset, safeOffset + safeLimit);
+    const nextOffset = safeOffset + page.length;
+    return {
+        items: page,
+        total,
+        hasMore: nextOffset < total,
+        nextOffset: nextOffset < total ? nextOffset : null,
+    };
+}
+export function rankSuggestions(state, name, limit) {
+    const stripped = Object.keys(state.config.mcpServers)
+        .flatMap((server) => ["server", "short", "mcp"].map((prefix) => getServerPrefix(server, prefix)))
+        .filter((candidate) => Boolean(candidate) && name.startsWith(`${candidate}_`))
+        .sort((a, b) => b.length - a.length)
+        .map((candidate) => name.slice(candidate.length + 1));
+    const query = stripped[0] ?? name;
+    return rankToolMatches(state, query, undefined, false)
+        .slice(0, limit)
+        .map((match) => match.tool.name);
+}
+//# sourceMappingURL=search-ranking.js.map

package/dist/extensions/mcp/search-ranking.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"search-ranking.js","sourceRoot":"","sources":["../../../src/extensions/mcp/search-ranking.ts"],"names":[],"mappings":"AAEA,OAAO,EACN,eAAe,EACf,qBAAqB,EACrB,gBAAgB,EAChB,kBAAkB,EAClB,iBAAiB,GACjB,MAAM,YAAY,CAAC;AAEpB;;;;GAIG;AACH,MAAM,eAAe,GAAG,CAAC,CAAC;AAE1B,MAAM,aAAa,GAAG;IACrB,IAAI,EAAE,EAAE;IACR,YAAY,EAAE,EAAE;IAChB,MAAM,EAAE,CAAC;IACT,WAAW,EAAE,CAAC;IACd,QAAQ,EAAE,CAAC;CACF,CAAC;AA0BX,MAAM,WAAW,GAAG,IAAI,OAAO,EAAsD,CAAC;AAEtF;;;;GAIG;AACH,MAAM,UAAU,qBAAqB,CACpC,UAAmC,EACnC,gBAAwB,EACxB,UAAkB,EAClB,YAAwB,EACb;IACX,MAAM,GAAG,GAAG,UAAU,EAAE,cAAc,CAAC;IACvC,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC;QAAE,OAAO,EAAE,CAAC;IACrE,MAAM,UAAU,GAAG,qBAAqB,CAAC,gBAAgB,EAAE,UAAU,EAAE,iBAAiB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC;IACpH,MAAM,QAAQ,GAAa,EAAE,CAAC;IAC9B,MAAM,IAAI,GAAG,IAAI,GAAG,EAAU,CAAC;IAC/B,KAAK,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,EAAE,CAAC;QACrD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC;YAAE,SAAS;QACrC,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC,OAAO,CAAC,CAAC;YAAE,SAAS;QACzD,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YAC5B,IAAI,OAAO,KAAK,KAAK,QAAQ;gBAAE,SAAS;YACxC,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;YAC7B,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC;gBAAE,SAAS;YAC5C,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;YAClB,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QACxB,CAAC;IACF,CAAC;IACD,OAAO,QAAQ,CAAC;AAAA,CAChB;AAED,MAAM,UAAU,mBAAmB,CAAC,KAAa,EAAU;IAC1D,OAAO,KAAK;SACV,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC;SACtC,OAAO,CAAC,WAAW,EAAE,GAAG,CAAC;SACzB,WAAW,EAAE,CAAC;AAAA,CAChB;AAED,MAAM,UAAU,QAAQ,CAAC,KAAa,EAAY;IACjD,OAAO,mBAAmB,CAAC,KAAK,CAAC;SAC/B,KAAK,CAAC,YAAY,CAAC;SACnB,MAAM,CAAC,OAAO,CAAC,CAAC;AAAA,CAClB;AAED,SAAS,iBAAiB,CAAC,IAAkB,EAAE,MAAc,EAAE,QAAmB,EAAsB;IACvG,MAAM,MAAM,GAAG;QACd,IAAI,EAAE,mBAAmB,CAAC,IAAI,CAAC,IAAI,CAAC;QACpC,YAAY,EAAE,mBAAmB,CAAC,IAAI,CAAC,YAAY,CAAC;QACpD,MAAM,EAAE,mBAAmB,CAAC,MAAM,CAAC;QACnC,WAAW,EAAE,mBAAmB,CAAC,IAAI,CAAC,WAAW,CAAC;KAClD,CAAC;IACF,MAAM,cAAc,GAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAkC,CAAC,GAAG,CAClF,CAAC,CAAC,KAAK,EAAE,KAAK,CAAC,EAAmC,EAAE,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,CAAC,CACpF,CAAC;IACF,MAAM,cAAc,GAAG,QAAQ,EAAE,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,mBAAmB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;IAC7G,OAAO;QACN,IAAI;QACJ,MAAM,EAAE,cAAc;QACtB,UAAU,EAAE,cAAc,CAAC,CAAC,CAAE,CAAC,CAAC,CAAC;QACjC,cAAc;QACd,aAAa,EAAE,cAAc,CAAC,OAAO,CAAC,QAAQ,CAAC;KAC/C,CAAC;AAAA,CACF;AAED,SAAS,sBAAsB,CAC9B,QAA4B,EAC5B,eAAuB,EACvB,WAAqB,EACL;IAChB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,aAAa,GAAG,KAAK,CAAC;IAC1B,IAAI,eAAe,GAAG,KAAK,CAAC;IAC5B,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IAExC,KAAK,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,WAAW,CAAC,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;QAC3D,MAAM,MAAM,GAAG,aAAa,CAAC,KAAK,CAAC,CAAC;QACpC,IAAI,KAAK,KAAK,eAAe,EAAE,CAAC;YAC/B,KAAK,IAAI,MAAM,GAAG,EAAE,CAAC;YACrB,aAAa,GAAG,IAAI,CAAC;YACrB,eAAe,GAAG,IAAI,CAAC;QACxB,CAAC;aAAM,IAAI,KAAK,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;YAC9C,KAAK,IAAI,MAAM,GAAG,CAAC,CAAC;YACpB,aAAa,GAAG,IAAI,CAAC;QACtB,CAAC;aAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;YAC5C,KAAK,IAAI,MAAM,GAAG,CAAC,CAAC;YACpB,aAAa,GAAG,IAAI,CAAC;QACtB,CAAC;QAED,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACjC,IAAI,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,KAAK,IAAI,MAAM,GAAG,CAAC,CAAC;gBACpB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;iBAAM,IACN,WAAW,CAAC,IAAI,CACf,CAAC,UAAU,EAAE,EAAE,CACd,UAAU,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC5B,CAAC,UAAU,CAAC,MAAM,IAAI,eAAe,IAAI,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC,CACvE,EACA,CAAC;gBACF,KAAK,IAAI,MAAM,GAAG,CAAC,CAAC;gBACpB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;iBAAM,IAAI,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAClC,KAAK,IAAI,MAAM,CAAC;gBAChB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC;IACF,CAAC;IAED,yEAAyE;IACzE,wEAAwE;IACxE,oEAAoE;IACpE,IAAI,QAAQ,CAAC,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACxC,MAAM,MAAM,GAAG,aAAa,CAAC,QAAQ,CAAC;QACtC,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC9C,IAAI,MAAM,KAAK,eAAe,EAAE,CAAC;gBAChC,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,GAAG,EAAE,CAAC,CAAC;gBACjD,aAAa,GAAG,IAAI,CAAC;gBACrB,eAAe,GAAG,IAAI,CAAC;YACxB,CAAC;iBAAM,IAAI,MAAM,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC/C,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC;gBAChD,aAAa,GAAG,IAAI,CAAC;YACtB,CAAC;iBAAM,IAAI,MAAM,CAAC,QAAQ,CAAC,eAAe,CAAC,EAAE,CAAC;gBAC7C,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC;gBAChD,aAAa,GAAG,IAAI,CAAC;YACtB,CAAC;QACF,CAAC;QACD,KAAK,IAAI,WAAW,CAAC;QAErB,KAAK,MAAM,KAAK,IAAI,WAAW,EAAE,CAAC;YACjC,IAAI,QAAQ,CAAC,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC5C,KAAK,IAAI,MAAM,GAAG,CAAC,CAAC;gBACpB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;iBAAM,IACN,QAAQ,CAAC,aAAa,CAAC,IAAI,CAC1B,CAAC,YAAY,EAAE,EAAE,CAChB,YAAY,CAAC,UAAU,CAAC,KAAK,CAAC;gBAC9B,CAAC,YAAY,CAAC,MAAM,IAAI,eAAe,IAAI,KAAK,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAC3E,EACA,CAAC;gBACF,KAAK,IAAI,MAAM,GAAG,CAAC,CAAC;gBACpB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;iBAAM,IAAI,QAAQ,CAAC,cAAc,CAAC,IAAI,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC;gBAC7E,KAAK,IAAI,MAAM,CAAC;gBAChB,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC;IACF,CAAC;IAED,MAAM,QAAQ,GAAG,aAAa,CAAC,IAAI,GAAG,WAAW,CAAC,MAAM,CAAC;IACzD,IAAI,CAAC,aAAa,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,CAAC,CAAC,CAAC,CAAC,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,QAAQ,GAAG,GAAG,CAAC;QAAE,OAAO,IAAI,CAAC;IAE/F,KAAK,IAAI,QAAQ,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC,CAAC;IACzD,MAAM,eAAe,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC;IACvC,IAAI,eAAe,KAAK,SAAS,IAAI,QAAQ,CAAC,UAAU,CAAC,QAAQ,CAAC,eAAe,CAAC;QAAE,KAAK,IAAI,CAAC,CAAC;IAC/F,IAAI,eAAe;QAAE,KAAK,IAAI,EAAE,CAAC;IACjC,OAAO,KAAK,CAAC;AAAA,CACb;AAED,MAAM,UAAU,cAAc,CAAC,IAAkB,EAAE,MAAc,EAAE,KAAa,EAAE,QAAmB,EAAiB;IACrH,MAAM,eAAe,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1D,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACpC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,IAAI,CAAC;IAC1C,OAAO,sBAAsB,CAAC,iBAAiB,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC;AAAA,CACvG;AAED,SAAS,gBAAgB,CACxB,KAAwB,EACxB,UAAkB,EAClB,QAAwB,EACxB,UAAmC,EACnC,YAAwB,EACxB,eAAwB,EACD;IACvB,IAAI,UAAU,GAAG,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;IACxC,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,UAAU,GAAG,IAAI,GAAG,EAAE,CAAC;QACvB,WAAW,CAAC,GAAG,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;IACpC,CAAC;IACD,MAAM,UAAU,GAAG,iBAAiB,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;IAC/D,IAAI,MAAM,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;IACxC,IACC,MAAM,EAAE,QAAQ,KAAK,QAAQ;QAC7B,MAAM,CAAC,cAAc,KAAK,UAAU,EAAE,cAAc;QACpD,MAAM,CAAC,UAAU,KAAK,UAAU,EAC/B,CAAC;QACF,MAAM,GAAG,EAAE,QAAQ,EAAE,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,UAAU,EAAE,CAAC;QAC9E,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IACpC,CAAC;IACD,MAAM,QAAQ,GAAG,eAAe,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,iBAAiB,CAAC;IACtE,IAAI,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,CAAC;IAChC,IAAI,CAAC,QAAQ,EAAE,CAAC;QACf,QAAQ,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAChC,iBAAiB,CAChB,IAAI,EACJ,UAAU,EACV,eAAe,IAAI,UAAU,EAAE,cAAc,KAAK,SAAS;YAC1D,CAAC,CAAC,qBAAqB,CAAC,UAAU,EAAE,IAAI,CAAC,YAAY,EAAE,UAAU,EAAE,YAAY,CAAC;YAChF,CAAC,CAAC,SAAS,CACZ,CACD,CAAC;QACF,MAAM,CAAC,QAAQ,CAAC,GAAG,QAAQ,CAAC;IAC7B,CAAC;IACD,OAAO,QAAQ,CAAC;AAAA,CAChB;AAED,MAAM,UAAU,eAAe,CAC9B,KAAwB,EACxB,KAAa,EACb,MAAe,EACf,eAAe,GAAG,IAAI,EACF;IACpB,MAAM,OAAO,GAAsB,EAAE,CAAC;IACtC,MAAM,eAAe,GAAG,mBAAmB,CAAC,KAAK,CAAC,CAAC,IAAI,EAAE,CAAC;IAC1D,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,CAAC;IACpC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC;QAAE,OAAO,OAAO,CAAC;IAC7C,MAAM,YAAY,GAAG,KAAK,CAAC,MAAM,CAAC,QAAQ,EAAE,UAAU,IAAI,QAAQ,CAAC;IACnE,KAAK,MAAM,CAAC,UAAU,EAAE,QAAQ,CAAC,IAAI,KAAK,CAAC,YAAY,CAAC,OAAO,EAAE,EAAE,CAAC;QACnE,IAAI,MAAM,IAAI,UAAU,KAAK,MAAM;YAAE,SAAS;QAC9C,MAAM,UAAU,GAAG,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;QACvD,IAAI,gBAAgB,CAAC,UAAU,CAAC;YAAE,SAAS;QAC3C,KAAK,MAAM,QAAQ,IAAI,gBAAgB,CAAC,KAAK,EAAE,UAAU,EAAE,QAAQ,EAAE,UAAU,EAAE,YAAY,EAAE,eAAe,CAAC,EAAE,CAAC;YACjH,MAAM,KAAK,GAAG,sBAAsB,CAAC,QAAQ,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC;YAC7E,IAAI,KAAK,KAAK,IAAI;gBAAE,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QACtF,CAAC;IACF,CAAC;IACD,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;AAAA,CAC3F;AAED,MAAM,UAAU,QAAQ,CACvB,KAAU,EACV,MAAc,EACd,KAAa,EACgE;IAC7E,MAAM,UAAU,GAAG,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IACjF,MAAM,SAAS,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9E,MAAM,KAAK,GAAG,KAAK,CAAC,MAAM,CAAC;IAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,KAAK,CAAC,UAAU,EAAE,UAAU,GAAG,SAAS,CAAC,CAAC;IAC7D,MAAM,UAAU,GAAG,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC;IAC5C,OAAO;QACN,KAAK,EAAE,IAAI;QACX,KAAK;QACL,OAAO,EAAE,UAAU,GAAG,KAAK;QAC3B,UAAU,EAAE,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,IAAI;KAClD,CAAC;AAAA,CACF;AAED,MAAM,UAAU,eAAe,CAAC,KAAwB,EAAE,IAAY,EAAE,KAAa,EAAY;IAChG,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,UAAU,CAAC;SACnD,OAAO,CAAC,CAAC,MAAM,EAAE,EAAE,CAAE,CAAC,QAAQ,EAAE,OAAO,EAAE,KAAK,CAAW,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;SAC3G,MAAM,CAAC,CAAC,SAAS,EAAuB,EAAE,CAAC,OAAO,CAAC,SAAS,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,GAAG,SAAS,GAAG,CAAC,CAAC;SAClG,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,MAAM,CAAC;SACnC,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IACvD,MAAM,KAAK,GAAG,QAAQ,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;IAClC,OAAO,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC;SACpD,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC;SACf,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CAClC","sourcesContent":["import type { McpExtensionState } from \"./state.ts\";\nimport type { ServerEntry, ToolMetadata, ToolPrefix } from \"./types.ts\";\nimport {\n\tgetServerPrefix,\n\tgetToolNameCandidates,\n\tisServerDisabled,\n\tmatchesToolPattern,\n\tresolveToolPrefix,\n} from \"./types.ts\";\n\n/**\n * Shortest field token allowed to stem-match a longer query token.\n * Real descriptions tokenize possessives into single letters (\"project's\" -> [\"project\", \"s\"]),\n * which would otherwise make every query starting with that letter a match.\n */\nconst MIN_STEM_LENGTH = 4;\n\nconst FIELD_WEIGHTS = {\n\tname: 12,\n\toriginalName: 10,\n\tserver: 8,\n\tdescription: 5,\n\tkeywords: 5,\n} as const;\n\nexport interface RankedToolMatch {\n\tserver: string;\n\ttool: ToolMetadata;\n\tscore: number;\n}\n\ntype SearchField = Exclude<keyof typeof FIELD_WEIGHTS, \"keywords\">;\n\ninterface PreparedToolSearch {\n\ttool: ToolMetadata;\n\tfields: Array<[SearchField, string, string[]]>;\n\tnameTokens: string[];\n\tkeywordPhrases: string[];\n\tkeywordTokens: string[];\n}\n\ninterface CachedServerSearch {\n\tmetadata: ToolMetadata[];\n\tsearchKeywords: ServerEntry[\"searchKeywords\"];\n\ttoolPrefix: ReturnType<typeof resolveToolPrefix>;\n\twithKeywords?: PreparedToolSearch[];\n\twithoutKeywords?: PreparedToolSearch[];\n}\n\nconst searchCache = new WeakMap<McpExtensionState, Map<string, CachedServerSearch>>();\n\n/**\n * Resolve the configured searchKeywords entries that apply to a tool.\n * Keys match by original name, prefixed name, or glob — the same candidate\n * set includeTools/excludeTools use — and all matching entries are unioned.\n */\nexport function resolveSearchKeywords(\n\tdefinition: ServerEntry | undefined,\n\ttoolOriginalName: string,\n\tserverName: string,\n\tglobalPrefix: ToolPrefix,\n): string[] {\n\tconst map = definition?.searchKeywords;\n\tif (!map || typeof map !== \"object\" || Array.isArray(map)) return [];\n\tconst candidates = getToolNameCandidates(toolOriginalName, serverName, resolveToolPrefix(definition, globalPrefix));\n\tconst keywords: string[] = [];\n\tconst seen = new Set<string>();\n\tfor (const [pattern, values] of Object.entries(map)) {\n\t\tif (!Array.isArray(values)) continue;\n\t\tif (!matchesToolPattern(candidates, [pattern])) continue;\n\t\tfor (const value of values) {\n\t\t\tif (typeof value !== \"string\") continue;\n\t\t\tconst trimmed = value.trim();\n\t\t\tif (!trimmed || seen.has(trimmed)) continue;\n\t\t\tseen.add(trimmed);\n\t\t\tkeywords.push(trimmed);\n\t\t}\n\t}\n\treturn keywords;\n}\n\nexport function normalizeSearchText(value: string): string {\n\treturn value\n\t\t.replace(/([a-z0-9])([A-Z])/g, \"$1 $2\")\n\t\t.replace(/[_./:-]+/g, \" \")\n\t\t.toLowerCase();\n}\n\nexport function tokenize(value: string): string[] {\n\treturn normalizeSearchText(value)\n\t\t.split(/[^a-z0-9]+/)\n\t\t.filter(Boolean);\n}\n\nfunction prepareToolSearch(tool: ToolMetadata, server: string, keywords?: string[]): PreparedToolSearch {\n\tconst fields = {\n\t\tname: normalizeSearchText(tool.name),\n\t\toriginalName: normalizeSearchText(tool.originalName),\n\t\tserver: normalizeSearchText(server),\n\t\tdescription: normalizeSearchText(tool.description),\n\t};\n\tconst preparedFields = (Object.entries(fields) as Array<[SearchField, string]>).map(\n\t\t([field, value]): [SearchField, string, string[]] => [field, value, tokenize(value)],\n\t);\n\tconst keywordPhrases = keywords?.map((keyword) => normalizeSearchText(keyword).trim()).filter(Boolean) ?? [];\n\treturn {\n\t\ttool,\n\t\tfields: preparedFields,\n\t\tnameTokens: preparedFields[0]![2],\n\t\tkeywordPhrases,\n\t\tkeywordTokens: keywordPhrases.flatMap(tokenize),\n\t};\n}\n\nfunction scorePreparedToolMatch(\n\tprepared: PreparedToolSearch,\n\tnormalizedQuery: string,\n\tqueryTokens: string[],\n): number | null {\n\tlet score = 0;\n\tlet phraseMatched = false;\n\tlet wholeFieldExact = false;\n\tconst matchedTokens = new Set<string>();\n\n\tfor (const [field, value, fieldTokens] of prepared.fields) {\n\t\tconst weight = FIELD_WEIGHTS[field];\n\t\tif (value === normalizedQuery) {\n\t\t\tscore += weight * 14;\n\t\t\tphraseMatched = true;\n\t\t\twholeFieldExact = true;\n\t\t} else if (value.startsWith(normalizedQuery)) {\n\t\t\tscore += weight * 9;\n\t\t\tphraseMatched = true;\n\t\t} else if (value.includes(normalizedQuery)) {\n\t\t\tscore += weight * 6;\n\t\t\tphraseMatched = true;\n\t\t}\n\n\t\tfor (const token of queryTokens) {\n\t\t\tif (fieldTokens.includes(token)) {\n\t\t\t\tscore += weight * 4;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (\n\t\t\t\tfieldTokens.some(\n\t\t\t\t\t(fieldToken) =>\n\t\t\t\t\t\tfieldToken.startsWith(token) ||\n\t\t\t\t\t\t(fieldToken.length >= MIN_STEM_LENGTH && token.startsWith(fieldToken)),\n\t\t\t\t)\n\t\t\t) {\n\t\t\t\tscore += weight * 2;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (value.includes(token)) {\n\t\t\t\tscore += weight;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t}\n\t\t}\n\t}\n\n\t// Configured keywords are discrete phrases, so the phrase-level bonus is\n\t// computed per phrase (best match wins) rather than on a joined string,\n\t// which would phrase-match queries spanning two unrelated keywords.\n\tif (prepared.keywordPhrases.length > 0) {\n\t\tconst weight = FIELD_WEIGHTS.keywords;\n\t\tlet phraseScore = 0;\n\t\tfor (const phrase of prepared.keywordPhrases) {\n\t\t\tif (phrase === normalizedQuery) {\n\t\t\t\tphraseScore = Math.max(phraseScore, weight * 14);\n\t\t\t\tphraseMatched = true;\n\t\t\t\twholeFieldExact = true;\n\t\t\t} else if (phrase.startsWith(normalizedQuery)) {\n\t\t\t\tphraseScore = Math.max(phraseScore, weight * 9);\n\t\t\t\tphraseMatched = true;\n\t\t\t} else if (phrase.includes(normalizedQuery)) {\n\t\t\t\tphraseScore = Math.max(phraseScore, weight * 6);\n\t\t\t\tphraseMatched = true;\n\t\t\t}\n\t\t}\n\t\tscore += phraseScore;\n\n\t\tfor (const token of queryTokens) {\n\t\t\tif (prepared.keywordTokens.includes(token)) {\n\t\t\t\tscore += weight * 4;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (\n\t\t\t\tprepared.keywordTokens.some(\n\t\t\t\t\t(keywordToken) =>\n\t\t\t\t\t\tkeywordToken.startsWith(token) ||\n\t\t\t\t\t\t(keywordToken.length >= MIN_STEM_LENGTH && token.startsWith(keywordToken)),\n\t\t\t\t)\n\t\t\t) {\n\t\t\t\tscore += weight * 2;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t} else if (prepared.keywordPhrases.some((phrase) => phrase.includes(token))) {\n\t\t\t\tscore += weight;\n\t\t\t\tmatchedTokens.add(token);\n\t\t\t}\n\t\t}\n\t}\n\n\tconst coverage = matchedTokens.size / queryTokens.length;\n\tif (!phraseMatched && (queryTokens.length <= 2 ? coverage !== 1 : coverage < 0.6)) return null;\n\n\tscore += coverage === 1 ? 25 : Math.round(coverage * 10);\n\tconst firstQueryToken = queryTokens[0];\n\tif (firstQueryToken !== undefined && prepared.nameTokens.includes(firstQueryToken)) score += 8;\n\tif (wholeFieldExact) score += 20;\n\treturn score;\n}\n\nexport function scoreToolMatch(tool: ToolMetadata, server: string, query: string, keywords?: string[]): number | null {\n\tconst normalizedQuery = normalizeSearchText(query).trim();\n\tconst queryTokens = tokenize(query);\n\tif (queryTokens.length === 0) return null;\n\treturn scorePreparedToolMatch(prepareToolSearch(tool, server, keywords), normalizedQuery, queryTokens);\n}\n\nfunction getPreparedTools(\n\tstate: McpExtensionState,\n\tserverName: string,\n\tmetadata: ToolMetadata[],\n\tdefinition: ServerEntry | undefined,\n\tglobalPrefix: ToolPrefix,\n\tincludeKeywords: boolean,\n): PreparedToolSearch[] {\n\tlet stateCache = searchCache.get(state);\n\tif (!stateCache) {\n\t\tstateCache = new Map();\n\t\tsearchCache.set(state, stateCache);\n\t}\n\tconst toolPrefix = resolveToolPrefix(definition, globalPrefix);\n\tlet cached = stateCache.get(serverName);\n\tif (\n\t\tcached?.metadata !== metadata ||\n\t\tcached.searchKeywords !== definition?.searchKeywords ||\n\t\tcached.toolPrefix !== toolPrefix\n\t) {\n\t\tcached = { metadata, searchKeywords: definition?.searchKeywords, toolPrefix };\n\t\tstateCache.set(serverName, cached);\n\t}\n\tconst cacheKey = includeKeywords ? \"withKeywords\" : \"withoutKeywords\";\n\tlet prepared = cached[cacheKey];\n\tif (!prepared) {\n\t\tprepared = metadata.map((tool) =>\n\t\t\tprepareToolSearch(\n\t\t\t\ttool,\n\t\t\t\tserverName,\n\t\t\t\tincludeKeywords && definition?.searchKeywords !== undefined\n\t\t\t\t\t? resolveSearchKeywords(definition, tool.originalName, serverName, globalPrefix)\n\t\t\t\t\t: undefined,\n\t\t\t),\n\t\t);\n\t\tcached[cacheKey] = prepared;\n\t}\n\treturn prepared;\n}\n\nexport function rankToolMatches(\n\tstate: McpExtensionState,\n\tquery: string,\n\tserver?: string,\n\tincludeKeywords = true,\n): RankedToolMatch[] {\n\tconst matches: RankedToolMatch[] = [];\n\tconst normalizedQuery = normalizeSearchText(query).trim();\n\tconst queryTokens = tokenize(query);\n\tif (queryTokens.length === 0) return matches;\n\tconst globalPrefix = state.config.settings?.toolPrefix ?? \"server\";\n\tfor (const [serverName, metadata] of state.toolMetadata.entries()) {\n\t\tif (server && serverName !== server) continue;\n\t\tconst definition = state.config.mcpServers[serverName];\n\t\tif (isServerDisabled(definition)) continue;\n\t\tfor (const prepared of getPreparedTools(state, serverName, metadata, definition, globalPrefix, includeKeywords)) {\n\t\t\tconst score = scorePreparedToolMatch(prepared, normalizedQuery, queryTokens);\n\t\t\tif (score !== null) matches.push({ server: serverName, tool: prepared.tool, score });\n\t\t}\n\t}\n\treturn matches.sort((a, b) => b.score - a.score || a.tool.name.localeCompare(b.tool.name));\n}\n\nexport function paginate<T>(\n\titems: T[],\n\toffset: number,\n\tlimit: number,\n): { items: T[]; total: number; hasMore: boolean; nextOffset: number | null } {\n\tconst safeOffset = Number.isFinite(offset) ? Math.max(0, Math.trunc(offset)) : 0;\n\tconst safeLimit = Number.isFinite(limit) ? Math.max(1, Math.trunc(limit)) : 1;\n\tconst total = items.length;\n\tconst page = items.slice(safeOffset, safeOffset + safeLimit);\n\tconst nextOffset = safeOffset + page.length;\n\treturn {\n\t\titems: page,\n\t\ttotal,\n\t\thasMore: nextOffset < total,\n\t\tnextOffset: nextOffset < total ? nextOffset : null,\n\t};\n}\n\nexport function rankSuggestions(state: McpExtensionState, name: string, limit: number): string[] {\n\tconst stripped = Object.keys(state.config.mcpServers)\n\t\t.flatMap((server) => ([\"server\", \"short\", \"mcp\"] as const).map((prefix) => getServerPrefix(server, prefix)))\n\t\t.filter((candidate): candidate is string => Boolean(candidate) && name.startsWith(`${candidate}_`))\n\t\t.sort((a, b) => b.length - a.length)\n\t\t.map((candidate) => name.slice(candidate.length + 1));\n\tconst query = stripped[0] ?? name;\n\treturn rankToolMatches(state, query, undefined, false)\n\t\t.slice(0, limit)\n\t\t.map((match) => match.tool.name);\n}\n"]}