@astralyn/pi 0.87.6 → 1.0.0

package/dist/extensions/statusline/render.js

@@ -0,0 +1,163 @@
+import { truncateToWidth, visibleWidth } from "@earendil-works/pi-tui";
+const CONTEXT_WARNING_PERCENT = 40;
+const CONTEXT_ERROR_PERCENT = 80;
+const MIN_GAP = 1;
+/** Shorten a path to `~` relative to the user's home directory. */
+function formatWorkingDirectory(cwd, home) {
+    const homeDirectory = home.replace(/[\\/]+$/, "");
+    if (!homeDirectory)
+        return cwd;
+    const normalizePath = (filePath) => filePath.replace(/\\/g, "/").replace(/\/+$/, "").toLowerCase();
+    const normalizedHomeDirectory = normalizePath(homeDirectory);
+    const normalizedCwd = normalizePath(cwd);
+    if (normalizedCwd === normalizedHomeDirectory)
+        return "~";
+    if (normalizedCwd.startsWith(`${normalizedHomeDirectory}/`)) {
+        return `~${cwd.slice(homeDirectory.length).replace(/\\/g, "/")}`;
+    }
+    return cwd;
+}
+/** Collapse a display path to `~/basename` or bare basename when tight. */
+function pathBasename(displayPath) {
+    const normalized = displayPath.replace(/\\/g, "/").replace(/\/+$/, "");
+    if (!normalized || normalized === "~")
+        return normalized || displayPath;
+    const slash = normalized.lastIndexOf("/");
+    if (slash < 0)
+        return normalized;
+    const base = normalized.slice(slash + 1);
+    return normalized.startsWith("~/") ? `~/${base}` : base;
+}
+function formatTokenCount(count) {
+    if (count < 1000)
+        return `${count}`;
+    if (count < 10_000)
+        return `${(count / 1000).toFixed(1)}k`;
+    if (count < 1_000_000)
+        return `${Math.round(count / 1000)}k`;
+    if (count < 10_000_000)
+        return `${(count / 1_000_000).toFixed(1)}M`;
+    return `${Math.round(count / 1_000_000)}M`;
+}
+function joinSegments(segments, separator) {
+    return segments.filter(Boolean).join(separator);
+}
+function formatContextUsage(data, theme) {
+    const contextWindow = data.contextUsage?.contextWindow ?? data.model?.contextWindow;
+    const suffix = contextWindow ? `/${formatTokenCount(contextWindow)}` : "";
+    const percent = data.contextUsage?.percent;
+    const label = `CTX ${percent == null ? "?" : percent.toFixed(1)}%${suffix}`;
+    if (percent != null && percent > CONTEXT_ERROR_PERCENT)
+        return theme.fg("error", label);
+    if (percent != null && percent > CONTEXT_WARNING_PERCENT)
+        return theme.fg("warning", label);
+    return theme.fg("accent", label);
+}
+/** Format fields once, then offer full usage, without W, and without W/R. */
+function formatUsageCandidates(summary, subscription, theme) {
+    const tokens = [
+        summary.input > 0 ? `↑${formatTokenCount(summary.input)}` : "",
+        summary.output > 0 ? `↓${formatTokenCount(summary.output)}` : "",
+    ];
+    const read = summary.cacheRead > 0 ? `R${formatTokenCount(summary.cacheRead)}` : "";
+    const write = summary.cacheWrite > 0 ? `W${formatTokenCount(summary.cacheWrite)}` : "";
+    const suffix = [
+        summary.latestCacheHitPercent === undefined ? "" : `CH${summary.latestCacheHitPercent.toFixed(1)}%`,
+        summary.cost > 0 ? `$${summary.cost.toFixed(3)}${subscription ? " (sub)" : ""}` : "",
+    ];
+    const candidates = [[read, write], [read], []].map((cache) => joinSegments([...tokens, ...cache, ...suffix], " "));
+    return [...new Set(candidates.filter(Boolean))].map((text) => theme.fg("dim", text));
+}
+function formatExtensionStatuses(statuses, theme) {
+    return ([...statuses.entries()]
+        .sort(([first], [second]) => first.localeCompare(second))
+        .map(([, text]) => text
+        .replace(/[\r\n\t]/g, " ")
+        .replace(/ +/g, " ")
+        .trim())
+        .filter(Boolean)
+        // Preserve colors supplied by other extensions.
+        .map((text) => (text.includes("\x1b[") ? text : theme.fg("muted", text)))
+        .join("  "));
+}
+function fitsLeftRight(left, right, width) {
+    return visibleWidth(left) + visibleWidth(right) + (left && right ? MIN_GAP : 0) <= width;
+}
+/** Right-align the second cluster; when both compete, cap it at about 45%. */
+function layoutLeftRight(left, right, width, ellipsis) {
+    if (width <= 0)
+        return "";
+    if (!right)
+        return truncateToWidth(left, width, ellipsis);
+    let rightWidth = visibleWidth(right);
+    if (!left)
+        return `${" ".repeat(Math.max(0, width - rightWidth))}${truncateToWidth(right, width, ellipsis)}`;
+    let leftWidth = visibleWidth(left);
+    if (leftWidth + rightWidth + MIN_GAP > width) {
+        const maxRightWidth = Math.min(rightWidth, Math.max(8, Math.floor(width * 0.45)));
+        right = truncateToWidth(right, maxRightWidth, ellipsis);
+        rightWidth = visibleWidth(right);
+        const maxLeftWidth = width - rightWidth - MIN_GAP;
+        if (maxLeftWidth < 1)
+            return truncateToWidth(right, width, ellipsis);
+        left = truncateToWidth(left, maxLeftWidth, ellipsis);
+        leftWidth = visibleWidth(left);
+    }
+    return `${left}${" ".repeat(Math.max(MIN_GAP, width - leftWidth - rightWidth))}${right}`;
+}
+/** Center status absolutely when possible; otherwise balance the remaining gap. */
+function layoutSecondaryLine(left, status, right, width, ellipsis) {
+    const leftWidth = visibleWidth(left);
+    const statusWidth = visibleWidth(status);
+    const rightWidth = visibleWidth(right);
+    const free = width - leftWidth - statusWidth - rightWidth;
+    if (!status || free < MIN_GAP * 2)
+        return layoutLeftRight(left, right, width, ellipsis);
+    let leftPad = Math.floor((width - statusWidth) / 2) - leftWidth;
+    if (leftPad < MIN_GAP || free - leftPad < MIN_GAP)
+        leftPad = Math.floor(free / 2);
+    return `${left}${" ".repeat(leftPad)}${status}${" ".repeat(free - leftPad)}${right}`;
+}
+function renderPrimaryLine(data, theme, width, ellipsis) {
+    const { model, thinkingLevel } = data;
+    const name = theme.fg("toolTitle", theme.bold(model?.name ?? model?.id ?? "no-model"));
+    const provider = model?.provider ? theme.fg("muted", `(${model.provider})`) : "";
+    const effort = model?.reasoning && thinkingLevel !== "off" ? theme.getThinkingBorderColor(thinkingLevel)(thinkingLevel) : "";
+    const separator = theme.fg("dim", " · ");
+    const rich = joinSegments([joinSegments([name, provider], " "), effort], separator);
+    const plain = joinSegments([name, effort], separator);
+    const cwd = formatWorkingDirectory(data.cwd, data.home);
+    const full = theme.fg("success", cwd);
+    const short = theme.fg("success", pathBasename(cwd));
+    const branch = data.gitBranch ? theme.fg("accent", data.gitBranch) : "";
+    // Shortening the path or removing the provider can make room for the branch again.
+    const candidates = [
+        { left: rich, right: joinSegments([full, branch], separator) },
+        { left: rich, right: full },
+        { left: rich, right: joinSegments([short, branch], separator) },
+        { left: rich, right: short },
+        { left: plain, right: joinSegments([short, branch], separator) },
+        { left: plain, right: short },
+        { left: plain, right: "" },
+    ];
+    const chosen = candidates.find(({ left, right }) => fitsLeftRight(left, right, width)) ?? {
+        left: plain,
+        right: short,
+    };
+    return layoutLeftRight(chosen.left, chosen.right, width, ellipsis);
+}
+/** Fixed two-line presentation, independent of session access and footer lifecycle. */
+export function renderStatusline(data, theme, width) {
+    const ellipsis = theme.fg("dim", "...");
+    const primary = renderPrimaryLine(data, theme, width, ellipsis);
+    const context = formatContextUsage(data, theme);
+    const status = formatExtensionStatuses(data.statuses, theme);
+    const candidates = formatUsageCandidates(data.usage, data.usesSubscription, theme);
+    const usage = candidates.find((candidate) => fitsLeftRight(context, candidate, width));
+    // If no usage variant fits, retain the original full-usage truncation policy.
+    const secondary = candidates.length > 0 && usage === undefined
+        ? layoutLeftRight(context, candidates[0], width, ellipsis)
+        : layoutSecondaryLine(context, status, usage ?? "", width, ellipsis);
+    return [primary, secondary];
+}
+//# sourceMappingURL=render.js.map

package/dist/extensions/statusline/render.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"render.js","sourceRoot":"","sources":["../../../src/extensions/statusline/render.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,eAAe,EAAE,YAAY,EAAE,MAAM,wBAAwB,CAAC;AAiBvE,MAAM,uBAAuB,GAAG,EAAE,CAAC;AACnC,MAAM,qBAAqB,GAAG,EAAE,CAAC;AACjC,MAAM,OAAO,GAAG,CAAC,CAAC;AAElB,mEAAmE;AACnE,SAAS,sBAAsB,CAAC,GAAW,EAAE,IAAY;IACxD,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;IAClD,IAAI,CAAC,aAAa;QAAE,OAAO,GAAG,CAAC;IAC/B,MAAM,aAAa,GAAG,CAAC,QAAgB,EAAE,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,WAAW,EAAE,CAAC;IAC3G,MAAM,uBAAuB,GAAG,aAAa,CAAC,aAAa,CAAC,CAAC;IAC7D,MAAM,aAAa,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC;IACzC,IAAI,aAAa,KAAK,uBAAuB;QAAE,OAAO,GAAG,CAAC;IAC1D,IAAI,aAAa,CAAC,UAAU,CAAC,GAAG,uBAAuB,GAAG,CAAC,EAAE,CAAC;QAC7D,OAAO,IAAI,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,CAAC;IAClE,CAAC;IACD,OAAO,GAAG,CAAC;AACZ,CAAC;AAED,2EAA2E;AAC3E,SAAS,YAAY,CAAC,WAAmB;IACxC,MAAM,UAAU,GAAG,WAAW,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACvE,IAAI,CAAC,UAAU,IAAI,UAAU,KAAK,GAAG;QAAE,OAAO,UAAU,IAAI,WAAW,CAAC;IACxE,MAAM,KAAK,GAAG,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC;IAC1C,IAAI,KAAK,GAAG,CAAC;QAAE,OAAO,UAAU,CAAC;IACjC,MAAM,IAAI,GAAG,UAAU,CAAC,KAAK,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC;IACzC,OAAO,UAAU,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;AACzD,CAAC;AAED,SAAS,gBAAgB,CAAC,KAAa;IACtC,IAAI,KAAK,GAAG,IAAI;QAAE,OAAO,GAAG,KAAK,EAAE,CAAC;IACpC,IAAI,KAAK,GAAG,MAAM;QAAE,OAAO,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IAC3D,IAAI,KAAK,GAAG,SAAS;QAAE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,GAAG,CAAC;IAC7D,IAAI,KAAK,GAAG,UAAU;QAAE,OAAO,GAAG,CAAC,KAAK,GAAG,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,CAAC;IACpE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC,GAAG,CAAC;AAC5C,CAAC;AAED,SAAS,YAAY,CAAC,QAAkB,EAAE,SAAiB;IAC1D,OAAO,QAAQ,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;AACjD,CAAC;AAED,SAAS,kBAAkB,CAAC,IAAoB,EAAE,KAAY;IAC7D,MAAM,aAAa,GAAG,IAAI,CAAC,YAAY,EAAE,aAAa,IAAI,IAAI,CAAC,KAAK,EAAE,aAAa,CAAC;IACpF,MAAM,MAAM,GAAG,aAAa,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,OAAO,CAAC;IAC3C,MAAM,KAAK,GAAG,OAAO,OAAO,IAAI,IAAI,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,MAAM,EAAE,CAAC;IAC5E,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,GAAG,qBAAqB;QAAE,OAAO,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACxF,IAAI,OAAO,IAAI,IAAI,IAAI,OAAO,GAAG,uBAAuB;QAAE,OAAO,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAC5F,OAAO,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;AAClC,CAAC;AAED,6EAA6E;AAC7E,SAAS,qBAAqB,CAAC,OAAqB,EAAE,YAAqB,EAAE,KAAY;IACxF,MAAM,MAAM,GAAG;QACd,OAAO,CAAC,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;QAC9D,OAAO,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE;KAChE,CAAC;IACF,MAAM,IAAI,GAAG,OAAO,CAAC,SAAS,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACpF,MAAM,KAAK,GAAG,OAAO,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,gBAAgB,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;IACvF,MAAM,MAAM,GAAG;QACd,OAAO,CAAC,qBAAqB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,OAAO,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG;QACnG,OAAO,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,GAAG,YAAY,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE;KACpF,CAAC;IACF,MAAM,UAAU,GAAG,CAAC,CAAC,IAAI,EAAE,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,GAAG,MAAM,EAAE,GAAG,KAAK,EAAE,GAAG,MAAM,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;IACnH,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,UAAU,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC,CAAC;AACtF,CAAC;AAED,SAAS,uBAAuB,CAAC,QAAqC,EAAE,KAAY;IACnF,OAAO,CACN,CAAC,GAAG,QAAQ,CAAC,OAAO,EAAE,CAAC;SACrB,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,MAAM,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;SACxD,GAAG,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CACjB,IAAI;SACF,OAAO,CAAC,WAAW,EAAE,GAAG,CAAC;SACzB,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC;SACnB,IAAI,EAAE,CACR;SACA,MAAM,CAAC,OAAO,CAAC;QAChB,gDAAgD;SAC/C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC,CAAC;SACxE,IAAI,CAAC,IAAI,CAAC,CACZ,CAAC;AACH,CAAC;AAED,SAAS,aAAa,CAAC,IAAY,EAAE,KAAa,EAAE,KAAa;IAChE,OAAO,YAAY,CAAC,IAAI,CAAC,GAAG,YAAY,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,IAAI,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC;AAC1F,CAAC;AAED,8EAA8E;AAC9E,SAAS,eAAe,CAAC,IAAY,EAAE,KAAa,EAAE,KAAa,EAAE,QAAgB;IACpF,IAAI,KAAK,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC1B,IAAI,CAAC,KAAK;QAAE,OAAO,eAAe,CAAC,IAAI,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;IAC1D,IAAI,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACrC,IAAI,CAAC,IAAI;QAAE,OAAO,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,GAAG,UAAU,CAAC,CAAC,GAAG,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC;IAC7G,IAAI,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IACnC,IAAI,SAAS,GAAG,UAAU,GAAG,OAAO,GAAG,KAAK,EAAE,CAAC;QAC9C,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;QAClF,KAAK,GAAG,eAAe,CAAC,KAAK,EAAE,aAAa,EAAE,QAAQ,CAAC,CAAC;QACxD,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;QACjC,MAAM,YAAY,GAAG,KAAK,GAAG,UAAU,GAAG,OAAO,CAAC;QAClD,IAAI,YAAY,GAAG,CAAC;YAAE,OAAO,eAAe,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;QACrE,IAAI,GAAG,eAAe,CAAC,IAAI,EAAE,YAAY,EAAE,QAAQ,CAAC,CAAC;QACrD,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,GAAG,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,KAAK,GAAG,SAAS,GAAG,UAAU,CAAC,CAAC,GAAG,KAAK,EAAE,CAAC;AAC1F,CAAC;AAED,mFAAmF;AACnF,SAAS,mBAAmB,CAAC,IAAY,EAAE,MAAc,EAAE,KAAa,EAAE,KAAa,EAAE,QAAgB;IACxG,MAAM,SAAS,GAAG,YAAY,CAAC,IAAI,CAAC,CAAC;IACrC,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;IACzC,MAAM,UAAU,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC;IACvC,MAAM,IAAI,GAAG,KAAK,GAAG,SAAS,GAAG,WAAW,GAAG,UAAU,CAAC;IAC1D,IAAI,CAAC,MAAM,IAAI,IAAI,GAAG,OAAO,GAAG,CAAC;QAAE,OAAO,eAAe,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;IACxF,IAAI,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,GAAG,SAAS,CAAC;IAChE,IAAI,OAAO,GAAG,OAAO,IAAI,IAAI,GAAG,OAAO,GAAG,OAAO;QAAE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC;IAClF,OAAO,GAAG,IAAI,GAAG,GAAG,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,IAAI,GAAG,OAAO,CAAC,GAAG,KAAK,EAAE,CAAC;AACtF,CAAC;AAED,SAAS,iBAAiB,CAAC,IAAoB,EAAE,KAAY,EAAE,KAAa,EAAE,QAAgB;IAC7F,MAAM,EAAE,KAAK,EAAE,aAAa,EAAE,GAAG,IAAI,CAAC;IACtC,MAAM,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,KAAK,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,IAAI,KAAK,EAAE,EAAE,IAAI,UAAU,CAAC,CAAC,CAAC;IACvF,MAAM,QAAQ,GAAG,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,OAAO,EAAE,IAAI,KAAK,CAAC,QAAQ,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACjF,MAAM,MAAM,GACX,KAAK,EAAE,SAAS,IAAI,aAAa,KAAK,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,sBAAsB,CAAC,aAAa,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IAC/G,MAAM,SAAS,GAAG,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACzC,MAAM,IAAI,GAAG,YAAY,CAAC,CAAC,YAAY,CAAC,CAAC,IAAI,EAAE,QAAQ,CAAC,EAAE,GAAG,CAAC,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;IACpF,MAAM,KAAK,GAAG,YAAY,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,CAAC;IACtD,MAAM,GAAG,GAAG,sBAAsB,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,IAAI,CAAC,CAAC;IACxD,MAAM,IAAI,GAAG,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;IACtC,MAAM,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC,SAAS,EAAE,YAAY,CAAC,GAAG,CAAC,CAAC,CAAC;IACrD,MAAM,MAAM,GAAG,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC,QAAQ,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;IACxE,mFAAmF;IACnF,MAAM,UAAU,GAAG;QAClB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,IAAI,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,EAAE;QAC9D,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE;QAC3B,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,EAAE;QAC/D,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE;QAC5B,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,YAAY,CAAC,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,SAAS,CAAC,EAAE;QAChE,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE;QAC7B,EAAE,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;KAC1B,CAAC;IACF,MAAM,MAAM,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,CAAC,CAAC,IAAI;QACzF,IAAI,EAAE,KAAK;QACX,KAAK,EAAE,KAAK;KACZ,CAAC;IACF,OAAO,eAAe,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;AACpE,CAAC;AAED,uFAAuF;AACvF,MAAM,UAAU,gBAAgB,CAAC,IAAoB,EAAE,KAAY,EAAE,KAAa;IACjF,MAAM,QAAQ,GAAG,KAAK,CAAC,EAAE,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IACxC,MAAM,OAAO,GAAG,iBAAiB,CAAC,IAAI,EAAE,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;IAChE,MAAM,OAAO,GAAG,kBAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IAChD,MAAM,MAAM,GAAG,uBAAuB,CAAC,IAAI,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IAC7D,MAAM,UAAU,GAAG,qBAAqB,CAAC,IAAI,CAAC,KAAK,EAAE,IAAI,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;IACnF,MAAM,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,aAAa,CAAC,OAAO,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC;IACvF,8EAA8E;IAC9E,MAAM,SAAS,GACd,UAAU,CAAC,MAAM,GAAG,CAAC,IAAI,KAAK,KAAK,SAAS;QAC3C,CAAC,CAAC,eAAe,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,CAAC;QAC1D,CAAC,CAAC,mBAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,KAAK,IAAI,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,CAAC;IACvE,OAAO,CAAC,OAAO,EAAE,SAAS,CAAC,CAAC;AAC7B,CAAC","sourcesContent":["import type { ThinkingLevel } from \"@earendil-works/pi-agent-core\";\nimport type { Model } from \"@earendil-works/pi-ai\";\nimport { truncateToWidth, visibleWidth } from \"@earendil-works/pi-tui\";\nimport type { ContextUsage } from \"../../core/extensions/types.ts\";\nimport type { Theme } from \"../../modes/interactive/theme/theme.ts\";\nimport type { UsageSummary } from \"./usage.ts\";\n\nexport interface StatuslineData {\n\tmodel: Pick<Model<string>, \"id\" | \"name\" | \"provider\" | \"reasoning\" | \"contextWindow\"> | undefined;\n\tthinkingLevel: ThinkingLevel;\n\tcwd: string;\n\thome: string;\n\tgitBranch: string | null;\n\tcontextUsage: ContextUsage | undefined;\n\tusage: UsageSummary;\n\tusesSubscription: boolean;\n\tstatuses: ReadonlyMap<string, string>;\n}\n\nconst CONTEXT_WARNING_PERCENT = 40;\nconst CONTEXT_ERROR_PERCENT = 80;\nconst MIN_GAP = 1;\n\n/** Shorten a path to `~` relative to the user's home directory. */\nfunction formatWorkingDirectory(cwd: string, home: string): string {\n\tconst homeDirectory = home.replace(/[\\\\/]+$/, \"\");\n\tif (!homeDirectory) return cwd;\n\tconst normalizePath = (filePath: string) => filePath.replace(/\\\\/g, \"/\").replace(/\\/+$/, \"\").toLowerCase();\n\tconst normalizedHomeDirectory = normalizePath(homeDirectory);\n\tconst normalizedCwd = normalizePath(cwd);\n\tif (normalizedCwd === normalizedHomeDirectory) return \"~\";\n\tif (normalizedCwd.startsWith(`${normalizedHomeDirectory}/`)) {\n\t\treturn `~${cwd.slice(homeDirectory.length).replace(/\\\\/g, \"/\")}`;\n\t}\n\treturn cwd;\n}\n\n/** Collapse a display path to `~/basename` or bare basename when tight. */\nfunction pathBasename(displayPath: string): string {\n\tconst normalized = displayPath.replace(/\\\\/g, \"/\").replace(/\\/+$/, \"\");\n\tif (!normalized || normalized === \"~\") return normalized || displayPath;\n\tconst slash = normalized.lastIndexOf(\"/\");\n\tif (slash < 0) return normalized;\n\tconst base = normalized.slice(slash + 1);\n\treturn normalized.startsWith(\"~/\") ? `~/${base}` : base;\n}\n\nfunction formatTokenCount(count: number): string {\n\tif (count < 1000) return `${count}`;\n\tif (count < 10_000) return `${(count / 1000).toFixed(1)}k`;\n\tif (count < 1_000_000) return `${Math.round(count / 1000)}k`;\n\tif (count < 10_000_000) return `${(count / 1_000_000).toFixed(1)}M`;\n\treturn `${Math.round(count / 1_000_000)}M`;\n}\n\nfunction joinSegments(segments: string[], separator: string): string {\n\treturn segments.filter(Boolean).join(separator);\n}\n\nfunction formatContextUsage(data: StatuslineData, theme: Theme): string {\n\tconst contextWindow = data.contextUsage?.contextWindow ?? data.model?.contextWindow;\n\tconst suffix = contextWindow ? `/${formatTokenCount(contextWindow)}` : \"\";\n\tconst percent = data.contextUsage?.percent;\n\tconst label = `CTX ${percent == null ? \"?\" : percent.toFixed(1)}%${suffix}`;\n\tif (percent != null && percent > CONTEXT_ERROR_PERCENT) return theme.fg(\"error\", label);\n\tif (percent != null && percent > CONTEXT_WARNING_PERCENT) return theme.fg(\"warning\", label);\n\treturn theme.fg(\"accent\", label);\n}\n\n/** Format fields once, then offer full usage, without W, and without W/R. */\nfunction formatUsageCandidates(summary: UsageSummary, subscription: boolean, theme: Theme): string[] {\n\tconst tokens = [\n\t\tsummary.input > 0 ? `↑${formatTokenCount(summary.input)}` : \"\",\n\t\tsummary.output > 0 ? `↓${formatTokenCount(summary.output)}` : \"\",\n\t];\n\tconst read = summary.cacheRead > 0 ? `R${formatTokenCount(summary.cacheRead)}` : \"\";\n\tconst write = summary.cacheWrite > 0 ? `W${formatTokenCount(summary.cacheWrite)}` : \"\";\n\tconst suffix = [\n\t\tsummary.latestCacheHitPercent === undefined ? \"\" : `CH${summary.latestCacheHitPercent.toFixed(1)}%`,\n\t\tsummary.cost > 0 ? `$${summary.cost.toFixed(3)}${subscription ? \" (sub)\" : \"\"}` : \"\",\n\t];\n\tconst candidates = [[read, write], [read], []].map((cache) => joinSegments([...tokens, ...cache, ...suffix], \" \"));\n\treturn [...new Set(candidates.filter(Boolean))].map((text) => theme.fg(\"dim\", text));\n}\n\nfunction formatExtensionStatuses(statuses: ReadonlyMap<string, string>, theme: Theme): string {\n\treturn (\n\t\t[...statuses.entries()]\n\t\t\t.sort(([first], [second]) => first.localeCompare(second))\n\t\t\t.map(([, text]) =>\n\t\t\t\ttext\n\t\t\t\t\t.replace(/[\\r\\n\\t]/g, \" \")\n\t\t\t\t\t.replace(/ +/g, \" \")\n\t\t\t\t\t.trim(),\n\t\t\t)\n\t\t\t.filter(Boolean)\n\t\t\t// Preserve colors supplied by other extensions.\n\t\t\t.map((text) => (text.includes(\"\\x1b[\") ? text : theme.fg(\"muted\", text)))\n\t\t\t.join(\"  \")\n\t);\n}\n\nfunction fitsLeftRight(left: string, right: string, width: number): boolean {\n\treturn visibleWidth(left) + visibleWidth(right) + (left && right ? MIN_GAP : 0) <= width;\n}\n\n/** Right-align the second cluster; when both compete, cap it at about 45%. */\nfunction layoutLeftRight(left: string, right: string, width: number, ellipsis: string): string {\n\tif (width <= 0) return \"\";\n\tif (!right) return truncateToWidth(left, width, ellipsis);\n\tlet rightWidth = visibleWidth(right);\n\tif (!left) return `${\" \".repeat(Math.max(0, width - rightWidth))}${truncateToWidth(right, width, ellipsis)}`;\n\tlet leftWidth = visibleWidth(left);\n\tif (leftWidth + rightWidth + MIN_GAP > width) {\n\t\tconst maxRightWidth = Math.min(rightWidth, Math.max(8, Math.floor(width * 0.45)));\n\t\tright = truncateToWidth(right, maxRightWidth, ellipsis);\n\t\trightWidth = visibleWidth(right);\n\t\tconst maxLeftWidth = width - rightWidth - MIN_GAP;\n\t\tif (maxLeftWidth < 1) return truncateToWidth(right, width, ellipsis);\n\t\tleft = truncateToWidth(left, maxLeftWidth, ellipsis);\n\t\tleftWidth = visibleWidth(left);\n\t}\n\treturn `${left}${\" \".repeat(Math.max(MIN_GAP, width - leftWidth - rightWidth))}${right}`;\n}\n\n/** Center status absolutely when possible; otherwise balance the remaining gap. */\nfunction layoutSecondaryLine(left: string, status: string, right: string, width: number, ellipsis: string): string {\n\tconst leftWidth = visibleWidth(left);\n\tconst statusWidth = visibleWidth(status);\n\tconst rightWidth = visibleWidth(right);\n\tconst free = width - leftWidth - statusWidth - rightWidth;\n\tif (!status || free < MIN_GAP * 2) return layoutLeftRight(left, right, width, ellipsis);\n\tlet leftPad = Math.floor((width - statusWidth) / 2) - leftWidth;\n\tif (leftPad < MIN_GAP || free - leftPad < MIN_GAP) leftPad = Math.floor(free / 2);\n\treturn `${left}${\" \".repeat(leftPad)}${status}${\" \".repeat(free - leftPad)}${right}`;\n}\n\nfunction renderPrimaryLine(data: StatuslineData, theme: Theme, width: number, ellipsis: string): string {\n\tconst { model, thinkingLevel } = data;\n\tconst name = theme.fg(\"toolTitle\", theme.bold(model?.name ?? model?.id ?? \"no-model\"));\n\tconst provider = model?.provider ? theme.fg(\"muted\", `(${model.provider})`) : \"\";\n\tconst effort =\n\t\tmodel?.reasoning && thinkingLevel !== \"off\" ? theme.getThinkingBorderColor(thinkingLevel)(thinkingLevel) : \"\";\n\tconst separator = theme.fg(\"dim\", \" · \");\n\tconst rich = joinSegments([joinSegments([name, provider], \" \"), effort], separator);\n\tconst plain = joinSegments([name, effort], separator);\n\tconst cwd = formatWorkingDirectory(data.cwd, data.home);\n\tconst full = theme.fg(\"success\", cwd);\n\tconst short = theme.fg(\"success\", pathBasename(cwd));\n\tconst branch = data.gitBranch ? theme.fg(\"accent\", data.gitBranch) : \"\";\n\t// Shortening the path or removing the provider can make room for the branch again.\n\tconst candidates = [\n\t\t{ left: rich, right: joinSegments([full, branch], separator) },\n\t\t{ left: rich, right: full },\n\t\t{ left: rich, right: joinSegments([short, branch], separator) },\n\t\t{ left: rich, right: short },\n\t\t{ left: plain, right: joinSegments([short, branch], separator) },\n\t\t{ left: plain, right: short },\n\t\t{ left: plain, right: \"\" },\n\t];\n\tconst chosen = candidates.find(({ left, right }) => fitsLeftRight(left, right, width)) ?? {\n\t\tleft: plain,\n\t\tright: short,\n\t};\n\treturn layoutLeftRight(chosen.left, chosen.right, width, ellipsis);\n}\n\n/** Fixed two-line presentation, independent of session access and footer lifecycle. */\nexport function renderStatusline(data: StatuslineData, theme: Theme, width: number): string[] {\n\tconst ellipsis = theme.fg(\"dim\", \"...\");\n\tconst primary = renderPrimaryLine(data, theme, width, ellipsis);\n\tconst context = formatContextUsage(data, theme);\n\tconst status = formatExtensionStatuses(data.statuses, theme);\n\tconst candidates = formatUsageCandidates(data.usage, data.usesSubscription, theme);\n\tconst usage = candidates.find((candidate) => fitsLeftRight(context, candidate, width));\n\t// If no usage variant fits, retain the original full-usage truncation policy.\n\tconst secondary =\n\t\tcandidates.length > 0 && usage === undefined\n\t\t\t? layoutLeftRight(context, candidates[0], width, ellipsis)\n\t\t\t: layoutSecondaryLine(context, status, usage ?? \"\", width, ellipsis);\n\treturn [primary, secondary];\n}\n"]}

package/dist/extensions/statusline/usage.d.ts

@@ -0,0 +1,12 @@
+import type { ReadonlySessionManager } from "../../core/session-manager.ts";
+import { type UsageTotals } from "../../core/usage-totals.ts";
+export interface UsageSummary extends UsageTotals {
+    latestCacheHitPercent: number | undefined;
+}
+/** Per-footer accounting cache. Persisted entries are append-only; every append moves the leaf. */
+export declare class BranchUsage {
+    private cached?;
+    invalidate(): void;
+    read(manager: ReadonlySessionManager): UsageSummary;
+}
+//# sourceMappingURL=usage.d.ts.map

package/dist/extensions/statusline/usage.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.d.ts","sourceRoot":"","sources":["../../../src/extensions/statusline/usage.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AAC5E,OAAO,EAA2D,KAAK,WAAW,EAAE,MAAM,4BAA4B,CAAC;AAEvH,MAAM,WAAW,YAAa,SAAQ,WAAW;IAChD,qBAAqB,EAAE,MAAM,GAAG,SAAS,CAAC;CAC1C;AAED,mGAAmG;AACnG,qBAAa,WAAW;IACvB,OAAO,CAAC,MAAM,CAAC,CAKb;IAEF,UAAU,IAAI,IAAI;IAIlB,IAAI,CAAC,OAAO,EAAE,sBAAsB,GAAG,YAAY;CA6BnD"}

package/dist/extensions/statusline/usage.js

@@ -0,0 +1,36 @@
+import { addUsageToTotals, createUsageTotals, getAccountedUsages } from "../../core/usage-totals.js";
+/** Per-footer accounting cache. Persisted entries are append-only; every append moves the leaf. */
+export class BranchUsage {
+    invalidate() {
+        this.cached = undefined;
+    }
+    read(manager) {
+        const sessionId = manager.getSessionId();
+        const leafId = manager.getLeafId();
+        if (this.cached?.manager === manager && this.cached.sessionId === sessionId && this.cached.leafId === leafId) {
+            return this.cached.usage;
+        }
+        const entries = manager.getBranch();
+        const totals = createUsageTotals();
+        for (const usage of getAccountedUsages(entries))
+            addUsageToTotals(totals, usage);
+        // CH belongs to the latest assistant request, including failures. Other
+        // usage contributes to totals without replacing that request's cache rate.
+        let latestCacheHitPercent;
+        for (let index = entries.length - 1; index >= 0; index--) {
+            const entry = entries[index];
+            if (entry.type !== "message" || entry.message.role !== "assistant")
+                continue;
+            const { input, cacheRead, cacheWrite } = entry.message.usage;
+            const promptTokens = input + cacheRead + cacheWrite;
+            if ((cacheRead > 0 || cacheWrite > 0) && promptTokens > 0) {
+                latestCacheHitPercent = (cacheRead / promptTokens) * 100;
+            }
+            break;
+        }
+        const usage = { ...totals, latestCacheHitPercent };
+        this.cached = { manager, sessionId, leafId, usage };
+        return usage;
+    }
+}
+//# sourceMappingURL=usage.js.map

package/dist/extensions/statusline/usage.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"usage.js","sourceRoot":"","sources":["../../../src/extensions/statusline/usage.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,gBAAgB,EAAE,iBAAiB,EAAE,kBAAkB,EAAoB,MAAM,4BAA4B,CAAC;AAMvH,mGAAmG;AACnG,MAAM,OAAO,WAAW;IAQvB,UAAU;QACT,IAAI,CAAC,MAAM,GAAG,SAAS,CAAC;IACzB,CAAC;IAED,IAAI,CAAC,OAA+B;QACnC,MAAM,SAAS,GAAG,OAAO,CAAC,YAAY,EAAE,CAAC;QACzC,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACnC,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,KAAK,OAAO,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS,KAAK,SAAS,IAAI,IAAI,CAAC,MAAM,CAAC,MAAM,KAAK,MAAM,EAAE,CAAC;YAC9G,OAAO,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;QAC1B,CAAC;QAED,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;QACpC,MAAM,MAAM,GAAG,iBAAiB,EAAE,CAAC;QACnC,KAAK,MAAM,KAAK,IAAI,kBAAkB,CAAC,OAAO,CAAC;YAAE,gBAAgB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;QAEjF,wEAAwE;QACxE,2EAA2E;QAC3E,IAAI,qBAAyC,CAAC;QAC9C,KAAK,IAAI,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;YAC1D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;YAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,SAAS,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,KAAK,WAAW;gBAAE,SAAS;YAC7E,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC;YAC7D,MAAM,YAAY,GAAG,KAAK,GAAG,SAAS,GAAG,UAAU,CAAC;YACpD,IAAI,CAAC,SAAS,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC,IAAI,YAAY,GAAG,CAAC,EAAE,CAAC;gBAC3D,qBAAqB,GAAG,CAAC,SAAS,GAAG,YAAY,CAAC,GAAG,GAAG,CAAC;YAC1D,CAAC;YACD,MAAM;QACP,CAAC;QAED,MAAM,KAAK,GAAG,EAAE,GAAG,MAAM,EAAE,qBAAqB,EAAE,CAAC;QACnD,IAAI,CAAC,MAAM,GAAG,EAAE,OAAO,EAAE,SAAS,EAAE,MAAM,EAAE,KAAK,EAAE,CAAC;QACpD,OAAO,KAAK,CAAC;IACd,CAAC;CACD","sourcesContent":["import type { ReadonlySessionManager } from \"../../core/session-manager.ts\";\nimport { addUsageToTotals, createUsageTotals, getAccountedUsages, type UsageTotals } from \"../../core/usage-totals.ts\";\n\nexport interface UsageSummary extends UsageTotals {\n\tlatestCacheHitPercent: number | undefined;\n}\n\n/** Per-footer accounting cache. Persisted entries are append-only; every append moves the leaf. */\nexport class BranchUsage {\n\tprivate cached?: {\n\t\tmanager: ReadonlySessionManager;\n\t\tsessionId: string;\n\t\tleafId: string | null;\n\t\tusage: UsageSummary;\n\t};\n\n\tinvalidate(): void {\n\t\tthis.cached = undefined;\n\t}\n\n\tread(manager: ReadonlySessionManager): UsageSummary {\n\t\tconst sessionId = manager.getSessionId();\n\t\tconst leafId = manager.getLeafId();\n\t\tif (this.cached?.manager === manager && this.cached.sessionId === sessionId && this.cached.leafId === leafId) {\n\t\t\treturn this.cached.usage;\n\t\t}\n\n\t\tconst entries = manager.getBranch();\n\t\tconst totals = createUsageTotals();\n\t\tfor (const usage of getAccountedUsages(entries)) addUsageToTotals(totals, usage);\n\n\t\t// CH belongs to the latest assistant request, including failures. Other\n\t\t// usage contributes to totals without replacing that request's cache rate.\n\t\tlet latestCacheHitPercent: number | undefined;\n\t\tfor (let index = entries.length - 1; index >= 0; index--) {\n\t\t\tconst entry = entries[index];\n\t\t\tif (entry.type !== \"message\" || entry.message.role !== \"assistant\") continue;\n\t\t\tconst { input, cacheRead, cacheWrite } = entry.message.usage;\n\t\t\tconst promptTokens = input + cacheRead + cacheWrite;\n\t\t\tif ((cacheRead > 0 || cacheWrite > 0) && promptTokens > 0) {\n\t\t\t\tlatestCacheHitPercent = (cacheRead / promptTokens) * 100;\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\n\t\tconst usage = { ...totals, latestCacheHitPercent };\n\t\tthis.cached = { manager, sessionId, leafId, usage };\n\t\treturn usage;\n\t}\n}\n"]}

package/dist/extensions/tool-search/index.d.ts

@@ -0,0 +1,12 @@
+/**
+ * The `tool_search` tool as an extension. The CLI loads it as a built-in extension; SDK users add
+ * `createToolSearchExtension()` to their extension factories.
+ *
+ * `tool_search` is registered inactive. Activate it with `--tools`, the `defaultTools` setting, or
+ * `setActiveTools()`.
+ */
+import type { ExtensionFactory } from "../../core/extensions/types.ts";
+export declare function createToolSearchExtension(): ExtensionFactory;
+declare const _default: ExtensionFactory;
+export default _default;
+//# sourceMappingURL=index.d.ts.map

package/dist/extensions/tool-search/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/extensions/tool-search/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,gCAAgC,CAAC;AAGvE,wBAAgB,yBAAyB,IAAI,gBAAgB,CAS5D;;AAED,wBAA2C"}

package/dist/extensions/tool-search/index.js

@@ -0,0 +1,20 @@
+/**
+ * The `tool_search` tool as an extension. The CLI loads it as a built-in extension; SDK users add
+ * `createToolSearchExtension()` to their extension factories.
+ *
+ * `tool_search` is registered inactive. Activate it with `--tools`, the `defaultTools` setting, or
+ * `setActiveTools()`.
+ */
+import { createToolSearchToolDefinition, TOOL_SEARCH_TOOL_NAME, toolSearchSchema } from "./tool.js";
+export function createToolSearchExtension() {
+    return (pi) => {
+        pi.events.on("pi:discover-tool-orchestrators", (candidates) => {
+            if (candidates instanceof Map && candidates.get(TOOL_SEARCH_TOOL_NAME) === toolSearchSchema) {
+                candidates.set(TOOL_SEARCH_TOOL_NAME, true);
+            }
+        });
+        pi.registerTool({ ...createToolSearchToolDefinition({ tools: pi }), defaultActive: false });
+    };
+}
+export default createToolSearchExtension();
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/extensions/tool-search/index.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,EAAE,8BAA8B,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,MAAM,WAAW,CAAC;AAEpG,MAAM,UAAU,yBAAyB;IACxC,OAAO,CAAC,EAAE,EAAE,EAAE;QACb,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC,gCAAgC,EAAE,CAAC,UAAU,EAAE,EAAE;YAC7D,IAAI,UAAU,YAAY,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,qBAAqB,CAAC,KAAK,gBAAgB,EAAE,CAAC;gBAC7F,UAAU,CAAC,GAAG,CAAC,qBAAqB,EAAE,IAAI,CAAC,CAAC;YAC7C,CAAC;QACF,CAAC,CAAC,CAAC;QACH,EAAE,CAAC,YAAY,CAAC,EAAE,GAAG,8BAA8B,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,CAAC,CAAC;IAC7F,CAAC,CAAC;AACH,CAAC;AAED,eAAe,yBAAyB,EAAE,CAAC","sourcesContent":["/**\n * The `tool_search` tool as an extension. The CLI loads it as a built-in extension; SDK users add\n * `createToolSearchExtension()` to their extension factories.\n *\n * `tool_search` is registered inactive. Activate it with `--tools`, the `defaultTools` setting, or\n * `setActiveTools()`.\n */\n\nimport type { ExtensionFactory } from \"../../core/extensions/types.ts\";\nimport { createToolSearchToolDefinition, TOOL_SEARCH_TOOL_NAME, toolSearchSchema } from \"./tool.ts\";\n\nexport function createToolSearchExtension(): ExtensionFactory {\n\treturn (pi) => {\n\t\tpi.events.on(\"pi:discover-tool-orchestrators\", (candidates) => {\n\t\t\tif (candidates instanceof Map && candidates.get(TOOL_SEARCH_TOOL_NAME) === toolSearchSchema) {\n\t\t\t\tcandidates.set(TOOL_SEARCH_TOOL_NAME, true);\n\t\t\t}\n\t\t});\n\t\tpi.registerTool({ ...createToolSearchToolDefinition({ tools: pi }), defaultActive: false });\n\t};\n}\n\nexport default createToolSearchExtension();\n"]}

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

@@ -0,0 +1,73 @@
+/**
+ * Tool discovery: a BM25 ranker over tool metadata, shared by `searchTools()` in codemode scripts and
+ * the optional `tool_search` tool.
+ *
+ * `tool_search` searches tools that are not declared to
+ * the model (`codemode` and `deferred` exposure) and loads the matches, so they are declared for the
+ * next model call. Loading goes through the active tool set, so it is recorded in the transcript
+ * like any other tool change and survives `/tree`, resume, and fork on that branch.
+ */
+import { type Static, Type } from "typebox";
+import type { ExtensionAPI, ToolDefinition, ToolInfo, ToolNamespace } from "../../core/extensions/types.ts";
+export declare const TOOL_SEARCH_TOOL_NAME = "tool_search";
+export declare const DEFAULT_TOOL_SEARCH_LIMIT = 8;
+/** A tool as the ranker sees it: its name and the text built by {@link createToolSearchDocument}. */
+export interface ToolSearchDocument {
+    name: string;
+    text: string;
+}
+export interface ToolSearchMatch {
+    name: string;
+    score: number;
+}
+/** Ranks tools for a query. BM25 today; a hybrid ranker with embeddings can replace it. */
+export interface ToolRanker {
+    rank(query: string, documents: readonly ToolSearchDocument[], limit: number): ToolSearchMatch[];
+}
+/** Lowercase terms, split at camelCase boundaries and non-alphanumerics, without stop words. */
+export declare function tokenize(text: string): string[];
+/**
+ * Search text of a tool: the name, the name with `_`
+ * as spaces, the description, schema descriptions and property names, and the namespace with its
+ * description and instructions.
+ */
+export declare function createToolSearchDocument(tool: Pick<ToolInfo, "name" | "description" | "parameters">, namespace?: ToolNamespace): ToolSearchDocument;
+/** Okapi BM25 with the usual parameters. Ties keep document order. */
+export declare class Bm25Ranker implements ToolRanker {
+    private readonly k1;
+    private readonly b;
+    constructor(options?: {
+        k1?: number;
+        b?: number;
+    });
+    rank(query: string, documents: readonly ToolSearchDocument[], limit: number): ToolSearchMatch[];
+}
+export declare const toolSearchSchema: Type.TObject<{
+    query: Type.TString;
+    limit: Type.TOptional<Type.TNumber>;
+}>;
+export type ToolSearchInput = Static<typeof toolSearchSchema>;
+/** Whether the tool is this `tool_search`, not another extension's tool of the same name. */
+export declare function isToolSearchTool(tool: Pick<ToolInfo, "name" | "parameters">): boolean;
+export interface ToolSearchResultTool {
+    name: string;
+    description: string;
+}
+export interface ToolSearchToolDetails {
+    /** Tools loaded by this call. */
+    loaded: string[];
+}
+export interface ToolSearchToolOptions {
+    /**
+     * The session's tools. `tool_search` searches the tools that are not declared to the model and
+     * activates the matches. Without it, the tool finds nothing. An `ExtensionAPI` fits.
+     */
+    tools?: Pick<ExtensionAPI, "getAllTools" | "getActiveTools" | "setActiveTools">;
+}
+/**
+ * The `tool_search` description. It does not list the searchable tools or their namespaces, so it
+ * stays the same while tools are registered, for example when MCP servers connect.
+ */
+export declare const TOOL_SEARCH_DESCRIPTION = "# Tool discovery\n\nSearches over deferred tool metadata with BM25 and exposes matching tools for the next model call.\n\nSome of the tools, such as tools of MCP servers, may not have been provided to you upfront, and you should use this tool (`tool_search`) to search for the required tools. For MCP tool discovery, always use `tool_search`.";
+export declare function createToolSearchToolDefinition(options?: ToolSearchToolOptions): ToolDefinition<typeof toolSearchSchema, ToolSearchToolDetails>;
+//# sourceMappingURL=tool.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"tool.d.ts","sourceRoot":"","sources":["../../../src/extensions/tool-search/tool.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAE,KAAK,MAAM,EAAE,IAAI,EAAE,MAAM,SAAS,CAAC;AAC5C,OAAO,KAAK,EACX,YAAY,EACZ,cAAc,EAEd,QAAQ,EACR,aAAa,EACb,MAAM,gCAAgC,CAAC;AAExC,eAAO,MAAM,qBAAqB,gBAAgB,CAAC;AACnD,eAAO,MAAM,yBAAyB,IAAI,CAAC;AAE3C,qGAAqG;AACrG,MAAM,WAAW,kBAAkB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,IAAI,EAAE,MAAM,CAAC;CACb;AAED,MAAM,WAAW,eAAe;IAC/B,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,MAAM,CAAC;CACd;AAED,2FAA2F;AAC3F,MAAM,WAAW,UAAU;IAC1B,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,kBAAkB,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,eAAe,EAAE,CAAC;CAChG;AAkCD,gGAAgG;AAChG,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,EAAE,CAQ/C;AAuBD;;;;GAIG;AACH,wBAAgB,wBAAwB,CACvC,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,GAAG,aAAa,GAAG,YAAY,CAAC,EAC3D,SAAS,CAAC,EAAE,aAAa,GACvB,kBAAkB,CAKpB;AAED,sEAAsE;AACtE,qBAAa,UAAW,YAAW,UAAU;IAC5C,OAAO,CAAC,QAAQ,CAAC,EAAE,CAAS;IAC5B,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAS;gBAEf,OAAO,GAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,CAAC,EAAE,MAAM,CAAA;KAAO;IAKrD,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,SAAS,kBAAkB,EAAE,EAAE,KAAK,EAAE,MAAM,GAAG,eAAe,EAAE;CA6B/F;AAED,eAAO,MAAM,gBAAgB;;;EAK3B,CAAC;AAEH,MAAM,MAAM,eAAe,GAAG,MAAM,CAAC,OAAO,gBAAgB,CAAC,CAAC;AAE9D,6FAA6F;AAC7F,wBAAgB,gBAAgB,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,EAAE,MAAM,GAAG,YAAY,CAAC,GAAG,OAAO,CAErF;AAED,MAAM,WAAW,oBAAoB;IACpC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;CACpB;AAED,MAAM,WAAW,qBAAqB;IACrC,iCAAiC;IACjC,MAAM,EAAE,MAAM,EAAE,CAAC;CACjB;AAED,MAAM,WAAW,qBAAqB;IACrC;;;OAGG;IACH,KAAK,CAAC,EAAE,IAAI,CAAC,YAAY,EAAE,aAAa,GAAG,gBAAgB,GAAG,gBAAgB,CAAC,CAAC;CAChF;AA2BD;;;GAGG;AACH,eAAO,MAAM,uBAAuB,2VAAyX,CAAC;AAE9Z,wBAAgB,8BAA8B,CAC7C,OAAO,GAAE,qBAA0B,GACjC,cAAc,CAAC,OAAO,gBAAgB,EAAE,qBAAqB,CAAC,CAuBhE"}

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

@@ -0,0 +1,187 @@
+/**
+ * Tool discovery: a BM25 ranker over tool metadata, shared by `searchTools()` in codemode scripts and
+ * the optional `tool_search` tool.
+ *
+ * `tool_search` searches tools that are not declared to
+ * the model (`codemode` and `deferred` exposure) and loads the matches, so they are declared for the
+ * next model call. Loading goes through the active tool set, so it is recorded in the transcript
+ * like any other tool change and survives `/tree`, resume, and fork on that branch.
+ */
+import { Type } from "typebox";
+export const TOOL_SEARCH_TOOL_NAME = "tool_search";
+export const DEFAULT_TOOL_SEARCH_LIMIT = 8;
+const STOP_WORDS = new Set([
+    "a",
+    "an",
+    "and",
+    "are",
+    "as",
+    "at",
+    "be",
+    "by",
+    "for",
+    "from",
+    "in",
+    "is",
+    "it",
+    "of",
+    "on",
+    "or",
+    "that",
+    "the",
+    "this",
+    "to",
+    "with",
+]);
+/** Naive singular form, so `issues` matches `issue` and `searches` matches `search`. */
+function stem(term) {
+    if (term.length > 4 && term.endsWith("ies"))
+        return `${term.slice(0, -3)}y`;
+    if (term.length > 4 && /(ches|shes|sses|xes|zes)$/.test(term))
+        return term.slice(0, -2);
+    if (term.length > 3 && term.endsWith("s") && !term.endsWith("ss"))
+        return term.slice(0, -1);
+    return term;
+}
+/** Lowercase terms, split at camelCase boundaries and non-alphanumerics, without stop words. */
+export function tokenize(text) {
+    return text
+        .replace(/([a-z0-9])([A-Z])/g, "$1 $2")
+        .replace(/([A-Z]+)([A-Z][a-z])/g, "$1 $2")
+        .toLowerCase()
+        .split(/[^a-z0-9]+/)
+        .filter((term) => term.length > 0 && !STOP_WORDS.has(term))
+        .map(stem);
+}
+function isObject(value) {
+    return typeof value === "object" && value !== null && !Array.isArray(value);
+}
+/** Schema descriptions and property names, recursively. */
+function schemaText(schema, parts) {
+    if (!isObject(schema))
+        return;
+    if (typeof schema.description === "string")
+        parts.push(schema.description);
+    if (isObject(schema.properties)) {
+        for (const [name, property] of Object.entries(schema.properties)) {
+            parts.push(name);
+            schemaText(property, parts);
+        }
+    }
+    schemaText(schema.items, parts);
+    for (const key of ["anyOf", "oneOf", "allOf"]) {
+        const variants = schema[key];
+        if (Array.isArray(variants))
+            for (const variant of variants)
+                schemaText(variant, parts);
+    }
+}
+/**
+ * Search text of a tool: the name, the name with `_`
+ * as spaces, the description, schema descriptions and property names, and the namespace with its
+ * description and instructions.
+ */
+export function createToolSearchDocument(tool, namespace) {
+    const parts = [tool.name, tool.name.replaceAll("_", " "), tool.description];
+    schemaText(tool.parameters, parts);
+    if (namespace)
+        parts.push(namespace.name, namespace.description ?? "", namespace.instructions ?? "");
+    return { name: tool.name, text: parts.filter((part) => part.trim()).join(" ") };
+}
+/** Okapi BM25 with the usual parameters. Ties keep document order. */
+export class Bm25Ranker {
+    constructor(options = {}) {
+        this.k1 = options.k1 ?? 1.2;
+        this.b = options.b ?? 0.75;
+    }
+    rank(query, documents, limit) {
+        const queryTerms = [...new Set(tokenize(query))];
+        if (queryTerms.length === 0 || documents.length === 0 || limit <= 0)
+            return [];
+        const termCounts = documents.map((document) => {
+            const counts = new Map();
+            for (const term of tokenize(document.text))
+                counts.set(term, (counts.get(term) ?? 0) + 1);
+            return counts;
+        });
+        const lengths = termCounts.map((counts) => [...counts.values()].reduce((sum, count) => sum + count, 0));
+        const averageLength = lengths.reduce((sum, length) => sum + length, 0) / documents.length || 1;
+        const idf = new Map(queryTerms.map((term) => {
+            const frequency = termCounts.filter((counts) => counts.has(term)).length;
+            return [term, Math.log(1 + (documents.length - frequency + 0.5) / (frequency + 0.5))];
+        }));
+        const matches = [];
+        documents.forEach((document, index) => {
+            let score = 0;
+            for (const term of queryTerms) {
+                const count = termCounts[index].get(term);
+                if (!count)
+                    continue;
+                const norm = this.k1 * (1 - this.b + (this.b * lengths[index]) / averageLength);
+                score += (idf.get(term) ?? 0) * ((count * (this.k1 + 1)) / (count + norm));
+            }
+            if (score > 0)
+                matches.push({ name: document.name, score });
+        });
+        return matches.sort((a, b) => b.score - a.score).slice(0, limit);
+    }
+}
+export const toolSearchSchema = Type.Object({
+    query: Type.String({ description: "Search query for deferred tools." }),
+    limit: Type.Optional(Type.Number({ description: `Maximum number of tools to return. Defaults to ${DEFAULT_TOOL_SEARCH_LIMIT}.` })),
+});
+/** Whether the tool is this `tool_search`, not another extension's tool of the same name. */
+export function isToolSearchTool(tool) {
+    return tool.name === TOOL_SEARCH_TOOL_NAME && tool.parameters === toolSearchSchema;
+}
+/** Whether `tool_search` can load a tool with this exposure. */
+function isSearchable(exposure) {
+    return exposure === "codemode" || exposure === "deferred";
+}
+/**
+ * Rank the searchable tools that are not active yet and activate the matches, so the next model
+ * call declares them. Activation is recorded in the transcript like any tool change.
+ */
+function searchAndLoad(tools, query, limit) {
+    const active = tools.getActiveTools();
+    const candidates = tools.getAllTools().filter((tool) => isSearchable(tool.exposure) && !active.includes(tool.name));
+    const documents = candidates.map((tool) => createToolSearchDocument(tool, tool.namespace));
+    const matches = new Bm25Ranker().rank(query, documents, limit);
+    if (matches.length > 0)
+        tools.setActiveTools([...active, ...matches.map((match) => match.name)]);
+    return matches.map((match) => ({
+        name: match.name,
+        description: candidates.find((tool) => tool.name === match.name)?.description ?? "",
+    }));
+}
+/**
+ * The `tool_search` description. It does not list the searchable tools or their namespaces, so it
+ * stays the same while tools are registered, for example when MCP servers connect.
+ */
+export const TOOL_SEARCH_DESCRIPTION = `# Tool discovery\n\nSearches over deferred tool metadata with BM25 and exposes matching tools for the next model call.\n\nSome of the tools, such as tools of MCP servers, may not have been provided to you upfront, and you should use this tool (\`${TOOL_SEARCH_TOOL_NAME}\`) to search for the required tools. For MCP tool discovery, always use \`${TOOL_SEARCH_TOOL_NAME}\`.`;
+export function createToolSearchToolDefinition(options = {}) {
+    return {
+        name: TOOL_SEARCH_TOOL_NAME,
+        label: TOOL_SEARCH_TOOL_NAME,
+        description: TOOL_SEARCH_DESCRIPTION,
+        promptSnippet: "Search for tools that are not loaded yet and load the matches",
+        parameters: toolSearchSchema,
+        // Searching is not something scripts need; it changes what the model sees.
+        exposure: "model-only",
+        async execute(_toolCallId, { query, limit }) {
+            if (query.trim() === "")
+                throw new Error("query must not be empty");
+            const max = limit ?? DEFAULT_TOOL_SEARCH_LIMIT;
+            if (!Number.isInteger(max) || max <= 0)
+                throw new Error("limit must be a positive integer");
+            const tools = options.tools ? searchAndLoad(options.tools, query, max) : [];
+            const text = tools.length === 0
+                ? "No matching tools found."
+                : `Loaded ${tools.length} tool${tools.length === 1 ? "" : "s"}. They are available from your next call:\n${tools
+                    .map((tool) => `- ${tool.name}: ${tool.description.trim().split(/\r?\n/)[0]}`)
+                    .join("\n")}`;
+            return { content: [{ type: "text", text }], details: { loaded: tools.map((tool) => tool.name) } };
+        },
+    };
+}
+//# sourceMappingURL=tool.js.map

package/dist/extensions/tool-search/tool.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"tool.js","sourceRoot":"","sources":["../../../src/extensions/tool-search/tool.ts"],"names":[],"mappings":"AAAA;;;;;;;;GAQG;AAEH,OAAO,EAAe,IAAI,EAAE,MAAM,SAAS,CAAC;AAS5C,MAAM,CAAC,MAAM,qBAAqB,GAAG,aAAa,CAAC;AACnD,MAAM,CAAC,MAAM,yBAAyB,GAAG,CAAC,CAAC;AAkB3C,MAAM,UAAU,GAAwB,IAAI,GAAG,CAAC;IAC/C,GAAG;IACH,IAAI;IACJ,KAAK;IACL,KAAK;IACL,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,KAAK;IACL,MAAM;IACN,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,IAAI;IACJ,MAAM;IACN,KAAK;IACL,MAAM;IACN,IAAI;IACJ,MAAM;CACN,CAAC,CAAC;AAEH,wFAAwF;AACxF,SAAS,IAAI,CAAC,IAAY;IACzB,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,KAAK,CAAC;QAAE,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IAC5E,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,2BAA2B,CAAC,IAAI,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IACxF,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC;QAAE,OAAO,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAC5F,OAAO,IAAI,CAAC;AACb,CAAC;AAED,gGAAgG;AAChG,MAAM,UAAU,QAAQ,CAAC,IAAY;IACpC,OAAO,IAAI;SACT,OAAO,CAAC,oBAAoB,EAAE,OAAO,CAAC;SACtC,OAAO,CAAC,uBAAuB,EAAE,OAAO,CAAC;SACzC,WAAW,EAAE;SACb,KAAK,CAAC,YAAY,CAAC;SACnB,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;SAC1D,GAAG,CAAC,IAAI,CAAC,CAAC;AACb,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC/B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC;AAC7E,CAAC;AAED,2DAA2D;AAC3D,SAAS,UAAU,CAAC,MAAe,EAAE,KAAe;IACnD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC;QAAE,OAAO;IAC9B,IAAI,OAAO,MAAM,CAAC,WAAW,KAAK,QAAQ;QAAE,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC;IAC3E,IAAI,QAAQ,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;QACjC,KAAK,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC;YAClE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACjB,UAAU,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAC7B,CAAC;IACF,CAAC;IACD,UAAU,CAAC,MAAM,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAChC,KAAK,MAAM,GAAG,IAAI,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,CAAC;QAC/C,MAAM,QAAQ,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC;QAC7B,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC;YAAE,KAAK,MAAM,OAAO,IAAI,QAAQ;gBAAE,UAAU,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;IACzF,CAAC;AACF,CAAC;AAED;;;;GAIG;AACH,MAAM,UAAU,wBAAwB,CACvC,IAA2D,EAC3D,SAAyB;IAEzB,MAAM,KAAK,GAAG,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;IAC5E,UAAU,CAAC,IAAI,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;IACnC,IAAI,SAAS;QAAE,KAAK,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE,SAAS,CAAC,WAAW,IAAI,EAAE,EAAE,SAAS,CAAC,YAAY,IAAI,EAAE,CAAC,CAAC;IACrG,OAAO,EAAE,IAAI,EAAE,IAAI,CAAC,IAAI,EAAE,IAAI,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;AACjF,CAAC;AAED,sEAAsE;AACtE,MAAM,OAAO,UAAU;IAItB,YAAY,UAAuC,EAAE;QACpD,IAAI,CAAC,EAAE,GAAG,OAAO,CAAC,EAAE,IAAI,GAAG,CAAC;QAC5B,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,CAAC,IAAI,IAAI,CAAC;IAC5B,CAAC;IAED,IAAI,CAAC,KAAa,EAAE,SAAwC,EAAE,KAAa;QAC1E,MAAM,UAAU,GAAG,CAAC,GAAG,IAAI,GAAG,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACjD,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC;YAAE,OAAO,EAAE,CAAC;QAC/E,MAAM,UAAU,GAAG,SAAS,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC7C,MAAM,MAAM,GAAG,IAAI,GAAG,EAAkB,CAAC;YACzC,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,QAAQ,CAAC,IAAI,CAAC;gBAAE,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC1F,OAAO,MAAM,CAAC;QACf,CAAC,CAAC,CAAC;QACH,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC,GAAG,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,CAAC,GAAG,GAAG,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;QACxG,MAAM,aAAa,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,MAAM,EAAE,EAAE,CAAC,GAAG,GAAG,MAAM,EAAE,CAAC,CAAC,GAAG,SAAS,CAAC,MAAM,IAAI,CAAC,CAAC;QAC/F,MAAM,GAAG,GAAG,IAAI,GAAG,CAClB,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE;YACvB,MAAM,SAAS,GAAG,UAAU,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC;YACzE,OAAO,CAAC,IAAI,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,SAAS,CAAC,MAAM,GAAG,SAAS,GAAG,GAAG,CAAC,GAAG,CAAC,SAAS,GAAG,GAAG,CAAC,CAAC,CAAU,CAAC;QAChG,CAAC,CAAC,CACF,CAAC;QACF,MAAM,OAAO,GAAsB,EAAE,CAAC;QACtC,SAAS,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,KAAK,EAAE,EAAE;YACrC,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,KAAK,MAAM,IAAI,IAAI,UAAU,EAAE,CAAC;gBAC/B,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC;gBAC1C,IAAI,CAAC,KAAK;oBAAE,SAAS;gBACrB,MAAM,IAAI,GAAG,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,CAAC,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,GAAG,aAAa,CAAC,CAAC;gBAChF,KAAK,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,GAAG,CAAC,IAAI,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,CAAC;YAC5E,CAAC;YACD,IAAI,KAAK,GAAG,CAAC;gBAAE,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC;QAC7D,CAAC,CAAC,CAAC;QACH,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC;IAClE,CAAC;CACD;AAED,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,MAAM,CAAC;IAC3C,KAAK,EAAE,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,kCAAkC,EAAE,CAAC;IACvE,KAAK,EAAE,IAAI,CAAC,QAAQ,CACnB,IAAI,CAAC,MAAM,CAAC,EAAE,WAAW,EAAE,kDAAkD,yBAAyB,GAAG,EAAE,CAAC,CAC5G;CACD,CAAC,CAAC;AAIH,6FAA6F;AAC7F,MAAM,UAAU,gBAAgB,CAAC,IAA2C;IAC3E,OAAO,IAAI,CAAC,IAAI,KAAK,qBAAqB,IAAI,IAAI,CAAC,UAAU,KAAK,gBAAgB,CAAC;AACpF,CAAC;AAoBD,gEAAgE;AAChE,SAAS,YAAY,CAAC,QAAsB;IAC3C,OAAO,QAAQ,KAAK,UAAU,IAAI,QAAQ,KAAK,UAAU,CAAC;AAC3D,CAAC;AAED;;;GAGG;AACH,SAAS,aAAa,CACrB,KAAkD,EAClD,KAAa,EACb,KAAa;IAEb,MAAM,MAAM,GAAG,KAAK,CAAC,cAAc,EAAE,CAAC;IACtC,MAAM,UAAU,GAAG,KAAK,CAAC,WAAW,EAAE,CAAC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC;IACpH,MAAM,SAAS,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,wBAAwB,CAAC,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC;IAC3F,MAAM,OAAO,GAAG,IAAI,UAAU,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,SAAS,EAAE,KAAK,CAAC,CAAC;IAC/D,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC;QAAE,KAAK,CAAC,cAAc,CAAC,CAAC,GAAG,MAAM,EAAE,GAAG,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IACjG,OAAO,OAAO,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAC9B,IAAI,EAAE,KAAK,CAAC,IAAI;QAChB,WAAW,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,CAAC,EAAE,WAAW,IAAI,EAAE;KACnF,CAAC,CAAC,CAAC;AACL,CAAC;AAED;;;GAGG;AACH,MAAM,CAAC,MAAM,uBAAuB,GAAG,yPAAyP,qBAAqB,8EAA8E,qBAAqB,KAAK,CAAC;AAE9Z,MAAM,UAAU,8BAA8B,CAC7C,UAAiC,EAAE;IAEnC,OAAO;QACN,IAAI,EAAE,qBAAqB;QAC3B,KAAK,EAAE,qBAAqB;QAC5B,WAAW,EAAE,uBAAuB;QACpC,aAAa,EAAE,+DAA+D;QAC9E,UAAU,EAAE,gBAAgB;QAC5B,2EAA2E;QAC3E,QAAQ,EAAE,YAAY;QACtB,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,KAAK,EAAE,KAAK,EAAE;YAC1C,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,EAAE;gBAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;YACpE,MAAM,GAAG,GAAG,KAAK,IAAI,yBAAyB,CAAC;YAC/C,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,GAAG,CAAC,IAAI,GAAG,IAAI,CAAC;gBAAE,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;YAC5F,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,OAAO,CAAC,KAAK,EAAE,KAAK,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YAC5E,MAAM,IAAI,GACT,KAAK,CAAC,MAAM,KAAK,CAAC;gBACjB,CAAC,CAAC,0BAA0B;gBAC5B,CAAC,CAAC,UAAU,KAAK,CAAC,MAAM,QAAQ,KAAK,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,8CAA8C,KAAK;qBAC7G,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,KAAK,IAAI,CAAC,IAAI,KAAK,IAAI,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;qBAC7E,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;YAClB,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,MAAM,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,EAAE,CAAC;QACnG,CAAC;KACD,CAAC;AACH,CAAC","sourcesContent":["/**\n * Tool discovery: a BM25 ranker over tool metadata, shared by `searchTools()` in codemode scripts and\n * the optional `tool_search` tool.\n *\n * `tool_search` searches tools that are not declared to\n * the model (`codemode` and `deferred` exposure) and loads the matches, so they are declared for the\n * next model call. Loading goes through the active tool set, so it is recorded in the transcript\n * like any other tool change and survives `/tree`, resume, and fork on that branch.\n */\n\nimport { type Static, Type } from \"typebox\";\nimport type {\n\tExtensionAPI,\n\tToolDefinition,\n\tToolExposure,\n\tToolInfo,\n\tToolNamespace,\n} from \"../../core/extensions/types.ts\";\n\nexport const TOOL_SEARCH_TOOL_NAME = \"tool_search\";\nexport const DEFAULT_TOOL_SEARCH_LIMIT = 8;\n\n/** A tool as the ranker sees it: its name and the text built by {@link createToolSearchDocument}. */\nexport interface ToolSearchDocument {\n\tname: string;\n\ttext: string;\n}\n\nexport interface ToolSearchMatch {\n\tname: string;\n\tscore: number;\n}\n\n/** Ranks tools for a query. BM25 today; a hybrid ranker with embeddings can replace it. */\nexport interface ToolRanker {\n\trank(query: string, documents: readonly ToolSearchDocument[], limit: number): ToolSearchMatch[];\n}\n\nconst STOP_WORDS: ReadonlySet<string> = new Set([\n\t\"a\",\n\t\"an\",\n\t\"and\",\n\t\"are\",\n\t\"as\",\n\t\"at\",\n\t\"be\",\n\t\"by\",\n\t\"for\",\n\t\"from\",\n\t\"in\",\n\t\"is\",\n\t\"it\",\n\t\"of\",\n\t\"on\",\n\t\"or\",\n\t\"that\",\n\t\"the\",\n\t\"this\",\n\t\"to\",\n\t\"with\",\n]);\n\n/** Naive singular form, so `issues` matches `issue` and `searches` matches `search`. */\nfunction stem(term: string): string {\n\tif (term.length > 4 && term.endsWith(\"ies\")) return `${term.slice(0, -3)}y`;\n\tif (term.length > 4 && /(ches|shes|sses|xes|zes)$/.test(term)) return term.slice(0, -2);\n\tif (term.length > 3 && term.endsWith(\"s\") && !term.endsWith(\"ss\")) return term.slice(0, -1);\n\treturn term;\n}\n\n/** Lowercase terms, split at camelCase boundaries and non-alphanumerics, without stop words. */\nexport function tokenize(text: string): string[] {\n\treturn text\n\t\t.replace(/([a-z0-9])([A-Z])/g, \"$1 $2\")\n\t\t.replace(/([A-Z]+)([A-Z][a-z])/g, \"$1 $2\")\n\t\t.toLowerCase()\n\t\t.split(/[^a-z0-9]+/)\n\t\t.filter((term) => term.length > 0 && !STOP_WORDS.has(term))\n\t\t.map(stem);\n}\n\nfunction isObject(value: unknown): value is Record<string, unknown> {\n\treturn typeof value === \"object\" && value !== null && !Array.isArray(value);\n}\n\n/** Schema descriptions and property names, recursively. */\nfunction schemaText(schema: unknown, parts: string[]): void {\n\tif (!isObject(schema)) return;\n\tif (typeof schema.description === \"string\") parts.push(schema.description);\n\tif (isObject(schema.properties)) {\n\t\tfor (const [name, property] of Object.entries(schema.properties)) {\n\t\t\tparts.push(name);\n\t\t\tschemaText(property, parts);\n\t\t}\n\t}\n\tschemaText(schema.items, parts);\n\tfor (const key of [\"anyOf\", \"oneOf\", \"allOf\"]) {\n\t\tconst variants = schema[key];\n\t\tif (Array.isArray(variants)) for (const variant of variants) schemaText(variant, parts);\n\t}\n}\n\n/**\n * Search text of a tool: the name, the name with `_`\n * as spaces, the description, schema descriptions and property names, and the namespace with its\n * description and instructions.\n */\nexport function createToolSearchDocument(\n\ttool: Pick<ToolInfo, \"name\" | \"description\" | \"parameters\">,\n\tnamespace?: ToolNamespace,\n): ToolSearchDocument {\n\tconst parts = [tool.name, tool.name.replaceAll(\"_\", \" \"), tool.description];\n\tschemaText(tool.parameters, parts);\n\tif (namespace) parts.push(namespace.name, namespace.description ?? \"\", namespace.instructions ?? \"\");\n\treturn { name: tool.name, text: parts.filter((part) => part.trim()).join(\" \") };\n}\n\n/** Okapi BM25 with the usual parameters. Ties keep document order. */\nexport class Bm25Ranker implements ToolRanker {\n\tprivate readonly k1: number;\n\tprivate readonly b: number;\n\n\tconstructor(options: { k1?: number; b?: number } = {}) {\n\t\tthis.k1 = options.k1 ?? 1.2;\n\t\tthis.b = options.b ?? 0.75;\n\t}\n\n\trank(query: string, documents: readonly ToolSearchDocument[], limit: number): ToolSearchMatch[] {\n\t\tconst queryTerms = [...new Set(tokenize(query))];\n\t\tif (queryTerms.length === 0 || documents.length === 0 || limit <= 0) return [];\n\t\tconst termCounts = documents.map((document) => {\n\t\t\tconst counts = new Map<string, number>();\n\t\t\tfor (const term of tokenize(document.text)) counts.set(term, (counts.get(term) ?? 0) + 1);\n\t\t\treturn counts;\n\t\t});\n\t\tconst lengths = termCounts.map((counts) => [...counts.values()].reduce((sum, count) => sum + count, 0));\n\t\tconst averageLength = lengths.reduce((sum, length) => sum + length, 0) / documents.length || 1;\n\t\tconst idf = new Map(\n\t\t\tqueryTerms.map((term) => {\n\t\t\t\tconst frequency = termCounts.filter((counts) => counts.has(term)).length;\n\t\t\t\treturn [term, Math.log(1 + (documents.length - frequency + 0.5) / (frequency + 0.5))] as const;\n\t\t\t}),\n\t\t);\n\t\tconst matches: ToolSearchMatch[] = [];\n\t\tdocuments.forEach((document, index) => {\n\t\t\tlet score = 0;\n\t\t\tfor (const term of queryTerms) {\n\t\t\t\tconst count = termCounts[index].get(term);\n\t\t\t\tif (!count) continue;\n\t\t\t\tconst norm = this.k1 * (1 - this.b + (this.b * lengths[index]) / averageLength);\n\t\t\t\tscore += (idf.get(term) ?? 0) * ((count * (this.k1 + 1)) / (count + norm));\n\t\t\t}\n\t\t\tif (score > 0) matches.push({ name: document.name, score });\n\t\t});\n\t\treturn matches.sort((a, b) => b.score - a.score).slice(0, limit);\n\t}\n}\n\nexport const toolSearchSchema = Type.Object({\n\tquery: Type.String({ description: \"Search query for deferred tools.\" }),\n\tlimit: Type.Optional(\n\t\tType.Number({ description: `Maximum number of tools to return. Defaults to ${DEFAULT_TOOL_SEARCH_LIMIT}.` }),\n\t),\n});\n\nexport type ToolSearchInput = Static<typeof toolSearchSchema>;\n\n/** Whether the tool is this `tool_search`, not another extension's tool of the same name. */\nexport function isToolSearchTool(tool: Pick<ToolInfo, \"name\" | \"parameters\">): boolean {\n\treturn tool.name === TOOL_SEARCH_TOOL_NAME && tool.parameters === toolSearchSchema;\n}\n\nexport interface ToolSearchResultTool {\n\tname: string;\n\tdescription: string;\n}\n\nexport interface ToolSearchToolDetails {\n\t/** Tools loaded by this call. */\n\tloaded: string[];\n}\n\nexport interface ToolSearchToolOptions {\n\t/**\n\t * The session's tools. `tool_search` searches the tools that are not declared to the model and\n\t * activates the matches. Without it, the tool finds nothing. An `ExtensionAPI` fits.\n\t */\n\ttools?: Pick<ExtensionAPI, \"getAllTools\" | \"getActiveTools\" | \"setActiveTools\">;\n}\n\n/** Whether `tool_search` can load a tool with this exposure. */\nfunction isSearchable(exposure: ToolExposure): boolean {\n\treturn exposure === \"codemode\" || exposure === \"deferred\";\n}\n\n/**\n * Rank the searchable tools that are not active yet and activate the matches, so the next model\n * call declares them. Activation is recorded in the transcript like any tool change.\n */\nfunction searchAndLoad(\n\ttools: NonNullable<ToolSearchToolOptions[\"tools\"]>,\n\tquery: string,\n\tlimit: number,\n): ToolSearchResultTool[] {\n\tconst active = tools.getActiveTools();\n\tconst candidates = tools.getAllTools().filter((tool) => isSearchable(tool.exposure) && !active.includes(tool.name));\n\tconst documents = candidates.map((tool) => createToolSearchDocument(tool, tool.namespace));\n\tconst matches = new Bm25Ranker().rank(query, documents, limit);\n\tif (matches.length > 0) tools.setActiveTools([...active, ...matches.map((match) => match.name)]);\n\treturn matches.map((match) => ({\n\t\tname: match.name,\n\t\tdescription: candidates.find((tool) => tool.name === match.name)?.description ?? \"\",\n\t}));\n}\n\n/**\n * The `tool_search` description. It does not list the searchable tools or their namespaces, so it\n * stays the same while tools are registered, for example when MCP servers connect.\n */\nexport const TOOL_SEARCH_DESCRIPTION = `# Tool discovery\\n\\nSearches over deferred tool metadata with BM25 and exposes matching tools for the next model call.\\n\\nSome of the tools, such as tools of MCP servers, may not have been provided to you upfront, and you should use this tool (\\`${TOOL_SEARCH_TOOL_NAME}\\`) to search for the required tools. For MCP tool discovery, always use \\`${TOOL_SEARCH_TOOL_NAME}\\`.`;\n\nexport function createToolSearchToolDefinition(\n\toptions: ToolSearchToolOptions = {},\n): ToolDefinition<typeof toolSearchSchema, ToolSearchToolDetails> {\n\treturn {\n\t\tname: TOOL_SEARCH_TOOL_NAME,\n\t\tlabel: TOOL_SEARCH_TOOL_NAME,\n\t\tdescription: TOOL_SEARCH_DESCRIPTION,\n\t\tpromptSnippet: \"Search for tools that are not loaded yet and load the matches\",\n\t\tparameters: toolSearchSchema,\n\t\t// Searching is not something scripts need; it changes what the model sees.\n\t\texposure: \"model-only\",\n\t\tasync execute(_toolCallId, { query, limit }) {\n\t\t\tif (query.trim() === \"\") throw new Error(\"query must not be empty\");\n\t\t\tconst max = limit ?? DEFAULT_TOOL_SEARCH_LIMIT;\n\t\t\tif (!Number.isInteger(max) || max <= 0) throw new Error(\"limit must be a positive integer\");\n\t\t\tconst tools = options.tools ? searchAndLoad(options.tools, query, max) : [];\n\t\t\tconst text =\n\t\t\t\ttools.length === 0\n\t\t\t\t\t? \"No matching tools found.\"\n\t\t\t\t\t: `Loaded ${tools.length} tool${tools.length === 1 ? \"\" : \"s\"}. They are available from your next call:\\n${tools\n\t\t\t\t\t\t\t.map((tool) => `- ${tool.name}: ${tool.description.trim().split(/\\r?\\n/)[0]}`)\n\t\t\t\t\t\t\t.join(\"\\n\")}`;\n\t\t\treturn { content: [{ type: \"text\", text }], details: { loaded: tools.map((tool) => tool.name) } };\n\t\t},\n\t};\n}\n"]}