@markusylisiurunen/tau 0.2.129 → 0.3.1

package/dist/core/session/core_session.js

@@ -7,6 +7,9 @@ export class CoreSession {
     reset() {
         this.engine.reset();
     }
+    restoreState(input) {
+        this.engine.restoreState(input);
+    }
     dispose() {
         this.engine.dispose();
     }
@@ -25,6 +28,9 @@ export class CoreSession {
     onEvent(handler) {
         return this.engine.onEvent(handler);
     }
+    onSubagentEvent(handler) {
+        return this.engine.onSubagentEvent(handler);
+    }
     async terminateSubagent(id) {
         return await this.engine.terminateSubagent(id);
     }
@@ -64,6 +70,9 @@ export class CoreSession {
     async compact(options) {
         return await this.engine.compact(options);
     }
+    async pruneToolResults(options) {
+        return await this.engine.pruneToolResults(options);
+    }
     async *events(signal, options) {
         return yield* this.engine.processTurn(signal, options);
     }

package/dist/core/session/core_session.js.map

@@ -1 +1 @@
-{"version":3,"file":"core_session.js","sourceRoot":"","sources":["../../../src/core/session/core_session.ts"],"names":[],"mappings":"AAMA,OAAO,EAOL,aAAa,GACd,MAAM,qBAAqB,CAAC;AA0B7B,MAAM,OAAO,WAAW;IACL,MAAM,CAAgB;IAEvC,YAAY,OAA2B;QACrC,IAAI,CAAC,MAAM,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED,OAAO;QACL,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IAED,UAAU,CACR,OAAgB,EAChB,YAAoB,EACpB,eAAuC;QAEvC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,YAAY,EAAE,eAAe,CAAC,CAAC;IACjE,CAAC;IAED,YAAY,CAAC,KAAgB;QAC3B,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,gBAAgB,CAAC,OAAuE;QACtF,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,CAAC,OAAmC;QACzC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,EAAU;QAChC,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,WAAW,CAAC,YAAoB,EAAE,OAAqC;QACrE,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAED,UAAU,CAAC,OAAgB,EAAE,OAAqC;QAChE,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,OAAgB;QAC5C,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED,kBAAkB,CAAC,cAAsB,EAAE,OAAgB;QACzD,OAAO,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACjE,CAAC;IAED,oBAAoB;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;IAC5C,CAAC;IAED,sBAAsB,CAAC,cAAsB;QAC3C,OAAO,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;IAC7B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;IAChC,CAAC;IAED,IAAI,iBAAiB;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;IAC/C,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC;IAC5C,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAiC;QAC7C,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,MAAmB,EACnB,OAAkD;QAElD,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzD,CAAC;CACF"}
+{"version":3,"file":"core_session.js","sourceRoot":"","sources":["../../../src/core/session/core_session.ts"],"names":[],"mappings":"AAMA,OAAO,EAOL,aAAa,GAGd,MAAM,qBAAqB,CAAC;AA4B7B,MAAM,OAAO,WAAW;IACL,MAAM,CAAgB;IAEvC,YAAY,OAA2B;QACrC,IAAI,CAAC,MAAM,GAAG,IAAI,aAAa,CAAC,OAAO,CAAC,CAAC;IAC3C,CAAC;IAED,KAAK;QACH,IAAI,CAAC,MAAM,CAAC,KAAK,EAAE,CAAC;IACtB,CAAC;IAED,YAAY,CAAC,KAAqE;QAChF,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,OAAO;QACL,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;IACxB,CAAC;IAED,UAAU,CACR,OAAgB,EAChB,YAAoB,EACpB,eAAuC;QAEvC,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,YAAY,EAAE,eAAe,CAAC,CAAC;IACjE,CAAC;IAED,YAAY,CAAC,KAAgB;QAC3B,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,KAAK,CAAC,CAAC;IAClC,CAAC;IAED,SAAS,CAAC,MAAc;QACtB,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,MAAM,CAAC,CAAC;IAChC,CAAC;IAED,gBAAgB,CAAC,OAAuE;QACtF,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACxC,CAAC;IAED,OAAO,CAAC,OAAmC;QACzC,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC;IAED,eAAe,CAAC,OAA6C;QAC3D,OAAO,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;IAC9C,CAAC;IAED,KAAK,CAAC,iBAAiB,CAAC,EAAU;QAChC,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;IACjD,CAAC;IAED,WAAW,CAAC,YAAoB,EAAE,OAAqC;QACrE,OAAO,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;IACxD,CAAC;IAED,UAAU,CAAC,OAAgB,EAAE,OAAqC;QAChE,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAClD,CAAC;IAED,cAAc,CAAC,KAAa,EAAE,OAAgB;QAC5C,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;IACpD,CAAC;IAED,kBAAkB,CAAC,cAAsB,EAAE,OAAgB;QACzD,OAAO,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IACjE,CAAC;IAED,oBAAoB;QAClB,OAAO,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;IAC5C,CAAC;IAED,sBAAsB,CAAC,cAAsB;QAC3C,OAAO,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC,cAAc,CAAC,CAAC;IAC5D,CAAC;IAED,IAAI,OAAO;QACT,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC;IAC7B,CAAC;IAED,IAAI,UAAU;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC;IAChC,CAAC;IAED,IAAI,iBAAiB;QACnB,OAAO,IAAI,CAAC,MAAM,CAAC,yBAAyB,CAAC;IAC/C,CAAC;IAED,IAAI,cAAc;QAChB,OAAO,IAAI,CAAC,MAAM,CAAC,sBAAsB,CAAC;IAC5C,CAAC;IAED,IAAI,SAAS;QACX,OAAO,IAAI,CAAC,MAAM,CAAC,cAAc,CAAC;IACpC,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,OAAiC;QAC7C,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC;IAC5C,CAAC;IAED,KAAK,CAAC,gBAAgB,CACpB,OAA+E;QAE/E,OAAO,MAAM,IAAI,CAAC,MAAM,CAAC,gBAAgB,CAAC,OAAO,CAAC,CAAC;IACrD,CAAC;IAED,KAAK,CAAC,CAAC,MAAM,CACX,MAAmB,EACnB,OAAkD;QAElD,OAAO,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IACzD,CAAC;CACF"}

package/dist/core/session/direct_bash.js

@@ -0,0 +1,34 @@
+import { BASH_DEFAULT_TIMEOUT_MS, buildBashUiText, formatBashUserMessageText, getBashOutputPolicy, prepareBashOutput, } from "../tools/bash.js";
+export async function runDirectBashCommand(options) {
+    const now = options.now ?? Date.now;
+    const startedAt = now();
+    const { output, exitCode, truncated: captureTruncated, } = await options.backend.runBash(options.command, {
+        signal: options.signal,
+        timeoutMs: BASH_DEFAULT_TIMEOUT_MS,
+    });
+    const durationMs = Math.max(0, now() - startedAt);
+    const truncationInfo = await prepareBashOutput(output, captureTruncated, getBashOutputPolicy({ mode: "user" }), options.backend);
+    const userMessageText = formatBashUserMessageText({
+        command: options.command,
+        truncationInfo,
+    });
+    const uiText = buildBashUiText({
+        truncationInfo,
+        exitCode,
+        durationMs,
+        previewLines: { head: 12, tail: 12 },
+        fullText: userMessageText,
+    });
+    const userHistoryEntryId = options.addToContext && options.addUserText
+        ? await options.addUserText(userMessageText)
+        : undefined;
+    return {
+        command: options.command,
+        exitCode,
+        truncationInfo,
+        uiText,
+        durationMs,
+        ...(userHistoryEntryId !== undefined ? { userHistoryEntryId } : {}),
+    };
+}
+//# sourceMappingURL=direct_bash.js.map

package/dist/core/session/direct_bash.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"direct_bash.js","sourceRoot":"","sources":["../../../src/core/session/direct_bash.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,uBAAuB,EAEvB,eAAe,EACf,yBAAyB,EACzB,mBAAmB,EACnB,iBAAiB,GAClB,MAAM,kBAAkB,CAAC;AAsB1B,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAAmC;IAEnC,MAAM,GAAG,GAAG,OAAO,CAAC,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC;IACpC,MAAM,SAAS,GAAG,GAAG,EAAE,CAAC;IACxB,MAAM,EACJ,MAAM,EACN,QAAQ,EACR,SAAS,EAAE,gBAAgB,GAC5B,GAAG,MAAM,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE;QACjD,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,SAAS,EAAE,uBAAuB;KACnC,CAAC,CAAC;IACH,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;IAClD,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAC5C,MAAM,EACN,gBAAgB,EAChB,mBAAmB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,EACrC,OAAO,CAAC,OAAO,CAChB,CAAC;IACF,MAAM,eAAe,GAAG,yBAAyB,CAAC;QAChD,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,cAAc;KACf,CAAC,CAAC;IACH,MAAM,MAAM,GAAG,eAAe,CAAC;QAC7B,cAAc;QACd,QAAQ;QACR,UAAU;QACV,YAAY,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE;QACpC,QAAQ,EAAE,eAAe;KAC1B,CAAC,CAAC;IACH,MAAM,kBAAkB,GACtB,OAAO,CAAC,YAAY,IAAI,OAAO,CAAC,WAAW;QACzC,CAAC,CAAC,MAAM,OAAO,CAAC,WAAW,CAAC,eAAe,CAAC;QAC5C,CAAC,CAAC,SAAS,CAAC;IAEhB,OAAO;QACL,OAAO,EAAE,OAAO,CAAC,OAAO;QACxB,QAAQ;QACR,cAAc;QACd,MAAM;QACN,UAAU;QACV,GAAG,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;KACpE,CAAC;AACJ,CAAC"}

package/dist/core/session/pruning.js

@@ -0,0 +1,595 @@
+import { TOOL_NAME_BASH, TOOL_NAME_EDIT } from "../tools/tool_names.js";
+import { buildLineDiff, collapseLongUnchangedDiffRuns } from "../utils/line_diff.js";
+import { extractAllFencedCodeBlocks } from "../utils/messages.js";
+import { bytesToTokens, formatTokenEstimate, tokensToBytes } from "../utils/token.js";
+import { truncateToBytesFromStart } from "../utils/truncate.js";
+const DEFAULT_PRUNE_FRACTION = 0.25;
+const PRUNED_TOOL_RESULT_PREFIX = "[Tool result pruned]";
+const PRUNED_EDIT_RESULT_PREFIX = "[Tool result pruned] Edit diff";
+const PRUNED_EDIT_ARGUMENT_MARKER = "[Content pruned]";
+const PRUNE_EDIT_UNCHANGED_CONTEXT_LINES = 4;
+const PRUNE_PREVIEW_MAX_TOKENS = 512;
+const PRUNE_MAX_OVERAGE_RATIO = 0.1;
+export function defaultSessionPruneFraction() {
+    return DEFAULT_PRUNE_FRACTION;
+}
+export function parseSessionPruneFraction(extra) {
+    if (!extra) {
+        return DEFAULT_PRUNE_FRACTION;
+    }
+    const parsed = Number(extra);
+    if (!Number.isFinite(parsed)) {
+        return null;
+    }
+    if (parsed < 0 || parsed > 1) {
+        return null;
+    }
+    return parsed;
+}
+export function parseSessionPruneFractionAndGuidance(extra) {
+    if (!extra) {
+        return { fraction: DEFAULT_PRUNE_FRACTION };
+    }
+    const trimmed = extra.trim();
+    if (!trimmed) {
+        return { fraction: DEFAULT_PRUNE_FRACTION };
+    }
+    const [firstToken, ...rest] = trimmed.split(/\s+/);
+    const parsed = Number(firstToken);
+    if (Number.isFinite(parsed)) {
+        if (parsed < 0 || parsed > 1) {
+            return { fraction: null };
+        }
+        const guidance = rest.join(" ").trim();
+        return guidance ? { fraction: parsed, guidance } : { fraction: parsed };
+    }
+    return { fraction: DEFAULT_PRUNE_FRACTION, guidance: trimmed };
+}
+export function prepareSessionSmartPrunePrompt(args) {
+    validatePruneFraction(args.fraction);
+    if (args.fraction === 0 || args.historyEntries.length === 0) {
+        return undefined;
+    }
+    const preparation = preparePruneCandidates(args.historyEntries, args.fraction);
+    if (preparation.candidates.length === 0 || preparation.totalTokens === 0) {
+        return undefined;
+    }
+    if (preparation.targetTokens <= 0) {
+        return undefined;
+    }
+    return {
+        prompt: buildSmartPrunePrompt({
+            history: args.historyEntries.map((entry) => entry.message),
+            targetTokens: preparation.targetTokens,
+            guidance: args.guidance,
+        }),
+        targetTokens: preparation.targetTokens,
+    };
+}
+export function buildSmartPruneSystemPrompt() {
+    return [
+        "You are a context pruning assistant.",
+        "Your task is to select which bash tool outputs should be pruned from the conversation history.",
+        "Analyze the conversation to understand what the user is working on and which tool outputs are most relevant.",
+        "Prioritize keeping outputs that contain important information, errors, or results that may be referenced later.",
+        "Prefer pruning outputs that are verbose, redundant, or contain routine information that can be regenerated if needed.",
+        "Follow the user's guidance carefully when provided.",
+    ].join(" ");
+}
+export function clampPruneReasoning(reasoning) {
+    switch (reasoning) {
+        case undefined:
+        case "none":
+            return undefined;
+        case "minimal":
+            return "low";
+        case "low":
+        case "medium":
+            return reasoning;
+        default:
+            return "medium";
+    }
+}
+export function parseSmartPruneResponse(raw) {
+    const trimmed = raw.trim();
+    if (!trimmed) {
+        return [];
+    }
+    const fenced = extractAllFencedCodeBlocks(trimmed);
+    const source = (fenced ?? trimmed).trim();
+    const start = source.indexOf("{");
+    const end = source.lastIndexOf("}");
+    if (start === -1 || end === -1 || end < start) {
+        return null;
+    }
+    const jsonText = source.slice(start, end + 1);
+    let parsed;
+    try {
+        parsed = JSON.parse(jsonText);
+    }
+    catch {
+        return null;
+    }
+    if (!parsed || typeof parsed !== "object" || Array.isArray(parsed)) {
+        return null;
+    }
+    const prune = parsed.prune;
+    if (!Array.isArray(prune) || prune.some((item) => typeof item !== "string")) {
+        return null;
+    }
+    return prune;
+}
+export function pruneSessionHistory(args) {
+    validatePruneFraction(args.options.fraction);
+    if (args.options.fraction === 0) {
+        return emptyPruneResult("prune fraction is 0, nothing to prune.");
+    }
+    if (args.historyEntries.length === 0) {
+        return emptyPruneResult("no conversation to prune.");
+    }
+    const editSummary = pruneEditToolHistory(args.historyEntries, args.replaceMessageById);
+    const preparation = preparePruneCandidates(args.historyEntries, args.options.fraction);
+    const fallbackMessage = args.options.strategy === "smart" && args.options.smartSelection?.length === 0
+        ? "model returned no prune candidates."
+        : "no bash tool results or edit tool calls to prune.";
+    if (preparation.candidates.length === 0 || preparation.totalTokens === 0) {
+        return buildNoopOrEditResult(editSummary, fallbackMessage);
+    }
+    if (preparation.targetTokens <= 0) {
+        return buildNoopOrEditResult(editSummary, "prune fraction is too small to remove anything.");
+    }
+    const toPrune = args.options.strategy === "smart"
+        ? selectSmartPruneCandidates(args.options.smartSelection ?? [], preparation.candidates, preparation.targetTokens)
+        : selectPruneCandidatesByStrategy(preparation, args.options.strategy);
+    if (toPrune.length === 0) {
+        return buildNoopOrEditResult(editSummary, fallbackMessage);
+    }
+    const applied = applyBashPruneCandidates(toPrune, args.replaceMessageById);
+    const message = buildPruneSummaryMessage(editSummary, applied.resultsPruned, applied.bytesPruned);
+    return {
+        message,
+        noop: false,
+        bashResultsPruned: applied.resultsPruned,
+        editCallsPruned: editSummary.callsPruned,
+        editResultsPruned: editSummary.resultsPruned,
+        bytesPruned: applied.bytesPruned + editSummary.bytesRemoved,
+        prunedToolResults: [...editSummary.prunedToolResults, ...applied.prunedToolResults],
+    };
+}
+function validatePruneFraction(fraction) {
+    if (!Number.isFinite(fraction) || fraction < 0 || fraction > 1) {
+        throw new Error("invalid prune fraction. use a number between 0 and 1.");
+    }
+}
+function emptyPruneResult(message) {
+    return {
+        message,
+        noop: true,
+        bashResultsPruned: 0,
+        editCallsPruned: 0,
+        editResultsPruned: 0,
+        bytesPruned: 0,
+        prunedToolResults: [],
+    };
+}
+function buildNoopOrEditResult(editSummary, fallbackMessage) {
+    if (!hasAnyPrunedEditChanges(editSummary)) {
+        return emptyPruneResult(fallbackMessage);
+    }
+    return {
+        message: buildPruneSummaryMessage(editSummary),
+        noop: false,
+        bashResultsPruned: 0,
+        editCallsPruned: editSummary.callsPruned,
+        editResultsPruned: editSummary.resultsPruned,
+        bytesPruned: editSummary.bytesRemoved,
+        prunedToolResults: editSummary.prunedToolResults,
+    };
+}
+function preparePruneCandidates(historyEntries, fraction) {
+    const candidates = [];
+    let totalTokens = 0;
+    for (const entry of historyEntries) {
+        const message = entry.message;
+        if (message.role !== "toolResult") {
+            continue;
+        }
+        const toolResult = message;
+        if (toolResult.toolName !== TOOL_NAME_BASH) {
+            continue;
+        }
+        const inspection = inspectToolResultContent(toolResult);
+        if (inspection.firstText?.startsWith(PRUNED_TOOL_RESULT_PREFIX) || inspection.hasImage) {
+            continue;
+        }
+        const tokens = bytesToTokens(inspection.textBytes);
+        candidates.push({
+            historyEntryId: entry.id,
+            toolResult,
+            bytes: inspection.textBytes,
+            tokens,
+            order: candidates.length,
+        });
+        totalTokens += tokens;
+    }
+    return {
+        candidates,
+        totalTokens,
+        targetTokens: Math.ceil(totalTokens * fraction),
+    };
+}
+function selectPruneCandidatesByStrategy(preparation, strategy) {
+    const ordered = strategy === "largest"
+        ? [...preparation.candidates].sort((a, b) => b.tokens - a.tokens || b.bytes - a.bytes || a.order - b.order)
+        : preparation.candidates;
+    const selected = [];
+    let selectedTokens = 0;
+    for (const candidate of ordered) {
+        if (selectedTokens >= preparation.targetTokens) {
+            break;
+        }
+        selected.push(candidate);
+        selectedTokens += candidate.tokens;
+    }
+    return selected;
+}
+function selectSmartPruneCandidates(ids, candidates, targetTokens) {
+    const candidatesById = new Map();
+    for (const candidate of candidates) {
+        candidatesById.set(candidate.toolResult.toolCallId, candidate);
+    }
+    const selected = [];
+    const seen = new Set();
+    let tokens = 0;
+    const maxTokens = Math.ceil(targetTokens * (1 + PRUNE_MAX_OVERAGE_RATIO));
+    for (const id of ids) {
+        if (seen.has(id))
+            continue;
+        const candidate = candidatesById.get(id);
+        if (!candidate)
+            continue;
+        if (tokens + candidate.tokens > maxTokens)
+            continue;
+        selected.push(candidate);
+        seen.add(id);
+        tokens += candidate.tokens;
+        if (tokens >= targetTokens) {
+            break;
+        }
+    }
+    return selected;
+}
+function applyBashPruneCandidates(candidates, replaceMessageById) {
+    let bytesPruned = 0;
+    const prunedToolResults = [];
+    for (const candidate of candidates) {
+        bytesPruned += candidate.bytes;
+        const noticeText = buildPrunedToolResultNotice(candidate.toolResult, candidate.bytes);
+        const prunedResult = {
+            ...candidate.toolResult,
+            content: [{ type: "text", text: noticeText }],
+        };
+        replacePrunedMessageOrThrow(replaceMessageById, candidate.historyEntryId, prunedResult, "bash tool result");
+        prunedToolResults.push({ toolCallId: prunedResult.toolCallId, content: noticeText });
+    }
+    return { resultsPruned: candidates.length, bytesPruned, prunedToolResults };
+}
+function hasAnyPrunedEditChanges(summary) {
+    return summary.callsPruned > 0 || summary.resultsPruned > 0;
+}
+function buildPruneSummaryMessage(editSummary, bashResultsPruned = 0, bashBytesPruned = 0) {
+    const editSummaryText = formatPrunedEditSummary(editSummary);
+    if (bashResultsPruned <= 0) {
+        if (!editSummaryText) {
+            return "no bash tool results or edit tool calls to prune.";
+        }
+        return `pruned ${editSummaryText}.`;
+    }
+    const bashLabel = formatTokenEstimate(bashBytesPruned);
+    const bashNoun = bashResultsPruned === 1 ? "result" : "results";
+    if (!editSummaryText) {
+        return `pruned ${bashResultsPruned} bash tool ${bashNoun} (${bashLabel}).`;
+    }
+    return `pruned ${bashResultsPruned} bash tool ${bashNoun} (${bashLabel}) and ${editSummaryText}.`;
+}
+function formatPrunedEditSummary(summary) {
+    if (!hasAnyPrunedEditChanges(summary)) {
+        return "";
+    }
+    const parts = [];
+    if (summary.callsPruned > 0) {
+        const noun = summary.callsPruned === 1 ? "call" : "calls";
+        parts.push(`${summary.callsPruned} edit tool ${noun}`);
+    }
+    if (summary.resultsPruned > 0) {
+        const noun = summary.resultsPruned === 1 ? "result" : "results";
+        parts.push(`${summary.resultsPruned} edit tool ${noun}`);
+    }
+    const tokenEstimate = formatTokenEstimate(summary.bytesRemoved);
+    return `${parts.join(" and ")} (${tokenEstimate})`;
+}
+function pruneEditToolHistory(historyEntries, replaceMessageById) {
+    const summary = {
+        callsPruned: 0,
+        resultsPruned: 0,
+        bytesRemoved: 0,
+        prunedToolResults: [],
+    };
+    const editCallsById = new Map();
+    for (const entry of historyEntries) {
+        const message = entry.message;
+        if (message.role !== "assistant") {
+            continue;
+        }
+        const assistant = message;
+        const assistantContent = getAssistantContentOrThrow(assistant, "pruning edit tool calls");
+        let changed = false;
+        const nextContent = assistantContent.map((block) => {
+            if (block.type !== "toolCall") {
+                return block;
+            }
+            const toolCall = block;
+            if (toolCall.name !== TOOL_NAME_EDIT) {
+                return block;
+            }
+            const args = getToolCallArgumentsObject(toolCall);
+            const oldText = getRequiredEditArgument(args, "oldText", toolCall.id);
+            const newText = getRequiredEditArgument(args, "newText", toolCall.id);
+            if (oldText === PRUNED_EDIT_ARGUMENT_MARKER && newText === PRUNED_EDIT_ARGUMENT_MARKER) {
+                return block;
+            }
+            changed = true;
+            summary.callsPruned += 1;
+            const oldBytes = Buffer.byteLength(oldText, "utf8");
+            const newBytes = Buffer.byteLength(newText, "utf8");
+            const markerBytes = Buffer.byteLength(PRUNED_EDIT_ARGUMENT_MARKER, "utf8") * 2;
+            summary.bytesRemoved += Math.max(0, oldBytes + newBytes - markerBytes);
+            editCallsById.set(toolCall.id, { oldText, newText });
+            return {
+                ...toolCall,
+                arguments: {
+                    ...args,
+                    oldText: PRUNED_EDIT_ARGUMENT_MARKER,
+                    newText: PRUNED_EDIT_ARGUMENT_MARKER,
+                },
+            };
+        });
+        if (changed) {
+            replacePrunedMessageOrThrow(replaceMessageById, entry.id, { ...assistant, content: nextContent }, "edit tool call");
+        }
+    }
+    if (editCallsById.size === 0) {
+        return summary;
+    }
+    for (const entry of historyEntries) {
+        const message = entry.message;
+        if (message.role !== "toolResult") {
+            continue;
+        }
+        const toolResult = message;
+        if (toolResult.toolName !== TOOL_NAME_EDIT) {
+            continue;
+        }
+        if (toolResult.isError) {
+            continue;
+        }
+        const callDiff = editCallsById.get(toolResult.toolCallId);
+        if (!callDiff) {
+            continue;
+        }
+        const inspection = inspectToolResultContent(toolResult);
+        if (inspection.firstText?.startsWith(PRUNED_EDIT_RESULT_PREFIX)) {
+            continue;
+        }
+        const prunedText = buildPrunedEditToolResult(callDiff);
+        const prunedResult = {
+            ...toolResult,
+            content: [{ type: "text", text: prunedText }],
+        };
+        replacePrunedMessageOrThrow(replaceMessageById, entry.id, prunedResult, "edit tool result");
+        summary.prunedToolResults.push({ toolCallId: prunedResult.toolCallId, content: prunedText });
+        summary.resultsPruned += 1;
+    }
+    return summary;
+}
+function replacePrunedMessageOrThrow(replaceMessageById, historyEntryId, message, label) {
+    if (!replaceMessageById(historyEntryId, message)) {
+        throw new Error(`failed to replace pruned ${label} '${historyEntryId}'`);
+    }
+}
+function getAssistantContentOrThrow(assistant, context) {
+    if (!Array.isArray(assistant.content)) {
+        throw new Error(`invalid assistant message content while ${context}`);
+    }
+    return assistant.content;
+}
+function getToolCallArgumentsObject(toolCall) {
+    const { arguments: argumentsValue } = toolCall;
+    if (!argumentsValue || typeof argumentsValue !== "object" || Array.isArray(argumentsValue)) {
+        throw new Error(`invalid edit tool call arguments (tool_call_id=${toolCall.id})`);
+    }
+    return argumentsValue;
+}
+function getRequiredEditArgument(args, key, toolCallId) {
+    const value = args[key];
+    if (typeof value !== "string") {
+        throw new Error(`missing edit argument '${key}' (tool_call_id=${toolCallId})`);
+    }
+    return value;
+}
+function buildPrunedEditToolResult(diff) {
+    const diffText = buildPrunedEditDiff(diff.oldText, diff.newText);
+    return `${PRUNED_EDIT_RESULT_PREFIX}\n${diffText}`;
+}
+function buildPrunedEditDiff(oldText, newText) {
+    const diff = buildLineDiff(oldText, newText);
+    if (diff.added === 0 && diff.removed === 0) {
+        return "(No textual changes)";
+    }
+    const collapsed = collapseLongUnchangedDiffRuns({
+        diffLines: diff.lines,
+        maxUnchangedLines: PRUNE_EDIT_UNCHANGED_CONTEXT_LINES,
+    });
+    return collapsed.length > 0 ? collapsed.join("\n") : "(No textual changes)";
+}
+function buildPrunedToolResultNotice(toolResult, bytes) {
+    const tokenEstimate = formatTokenEstimate(bytes);
+    return `${PRUNED_TOOL_RESULT_PREFIX} ${toolResult.toolName} output removed (${tokenEstimate}). Re-run the command if needed.`;
+}
+function inspectToolResultContent(toolResult) {
+    const parts = [];
+    let textBytes = 0;
+    let hasImage = false;
+    let firstText;
+    for (const block of toolResult.content) {
+        if (block.type === "text") {
+            if (firstText === undefined) {
+                firstText = block.text;
+            }
+            parts.push(block.text);
+            textBytes += Buffer.byteLength(block.text, "utf8");
+            continue;
+        }
+        if (block.type === "image") {
+            hasImage = true;
+        }
+    }
+    return {
+        text: parts.join("\n").trimEnd(),
+        textBytes,
+        hasImage,
+        firstText,
+    };
+}
+function buildSmartPrunePrompt(args) {
+    const lines = [];
+    lines.push(`Target pruning token budget: ${args.targetTokens}`);
+    lines.push([
+        "The budget is the total tokens to prune.",
+        "It is better to slightly exceed it than to be significantly below it.",
+    ].join(" "));
+    lines.push('Return only JSON: {"prune":[...]} with tool_call_id values from the conversation history.');
+    lines.push("Only include tool_call_id values for bash tool results.");
+    if (args.guidance?.trim()) {
+        lines.push(`Guidance (prioritize this): ${args.guidance.trim()}`);
+    }
+    else {
+        lines.push("Guidance: (none provided, use your best judgment)");
+    }
+    lines.push("");
+    lines.push("Conversation history:");
+    lines.push("<conversation>");
+    lines.push(...formatPruneConversationHistory(args.history));
+    lines.push("</conversation>");
+    return lines.join("\n");
+}
+function formatPruneConversationHistory(history) {
+    const lines = [];
+    for (const message of history) {
+        if (message.role === "user") {
+            lines.push("<user>");
+            appendPruneContentLines(lines, message.content);
+            lines.push("</user>");
+            lines.push("");
+            continue;
+        }
+        if (message.role === "assistant") {
+            lines.push("<assistant>");
+            const assistant = message;
+            const assistantContent = getAssistantContentOrThrow(assistant, "building smart prune prompt");
+            for (const block of assistantContent) {
+                if (block.type === "text") {
+                    appendPruneText(lines, block.text);
+                    continue;
+                }
+                if (block.type === "toolCall") {
+                    const toolCall = block;
+                    lines.push(buildPruneToolCallTag(toolCall));
+                }
+            }
+            lines.push("</assistant>");
+            lines.push("");
+            continue;
+        }
+        if (message.role === "toolResult") {
+            const toolResult = message;
+            lines.push(...buildPruneToolResultLines(toolResult));
+            lines.push("");
+        }
+    }
+    if (lines[lines.length - 1] === "") {
+        lines.pop();
+    }
+    return lines;
+}
+function appendPruneContentLines(lines, content) {
+    if (typeof content === "string") {
+        appendPruneText(lines, content);
+        return;
+    }
+    if (!Array.isArray(content)) {
+        return;
+    }
+    for (const block of content) {
+        if (block.type === "text") {
+            appendPruneText(lines, block.text);
+        }
+    }
+}
+function appendPruneText(lines, text) {
+    const escaped = escapeXml(text);
+    const parts = escaped.split(/\r?\n/);
+    for (const part of parts) {
+        lines.push(part);
+    }
+}
+function buildPruneToolCallTag(toolCall) {
+    const name = escapeXmlAttribute(toolCall.name);
+    const id = escapeXmlAttribute(toolCall.id);
+    const argsValue = JSON.stringify(toolCall.arguments);
+    const args = escapeXmlAttribute(argsValue);
+    return `<tool-call name="${name}" id="${id}" args="${args}" />`;
+}
+function buildPruneToolResultLines(toolResult) {
+    const lines = [];
+    const name = escapeXmlAttribute(toolResult.toolName);
+    const toolCallId = escapeXmlAttribute(toolResult.toolCallId);
+    const inspection = inspectToolResultContent(toolResult);
+    if (inspection.hasImage) {
+        lines.push(`<tool-result name="${name}" tool_call_id="${toolCallId}">`);
+        lines.push("<preview>[image omitted]</preview>");
+        lines.push("</tool-result>");
+        return lines;
+    }
+    const preview = buildPruneToolResultPreview(inspection.text);
+    lines.push(`<tool-result name="${name}" tool_call_id="${toolCallId}" total_tokens="${preview.totalTokens}">`);
+    lines.push(`<preview max_tokens="${PRUNE_PREVIEW_MAX_TOKENS}">`);
+    lines.push(...preview.lines);
+    lines.push("</preview>");
+    lines.push("</tool-result>");
+    return lines;
+}
+function buildPruneToolResultPreview(text) {
+    const normalized = text.trimEnd();
+    const totalBytes = Buffer.byteLength(normalized, "utf8");
+    const totalTokens = bytesToTokens(totalBytes);
+    if (totalTokens <= PRUNE_PREVIEW_MAX_TOKENS) {
+        const escaped = escapeXml(normalized);
+        const lines = escaped ? escaped.split(/\r?\n/) : [""];
+        return { lines, totalTokens };
+    }
+    const head = truncateToBytesFromStart(normalized, tokensToBytes(PRUNE_PREVIEW_MAX_TOKENS));
+    const remainingTokens = Math.max(0, totalTokens - PRUNE_PREVIEW_MAX_TOKENS);
+    const suffix = `...${remainingTokens} more tokens...`;
+    const previewText = head.endsWith("\n") || head === "" ? `${head}${suffix}` : `${head}\n${suffix}`;
+    const escaped = escapeXml(previewText);
+    const lines = escaped ? escaped.split(/\r?\n/) : [""];
+    return { lines, totalTokens };
+}
+function escapeXml(text) {
+    return text.replace(/&/g, "&amp;").replace(/</g, "&lt;").replace(/>/g, "&gt;");
+}
+function escapeXmlAttribute(text) {
+    return escapeXml(text).replace(/"/g, "&quot;").replace(/'/g, "&apos;");
+}
+//# sourceMappingURL=pruning.js.map