@greppleai/grip 0.0.0-stage → 0.0.1

package/dist/core/bash-executor.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"bash-executor.d.ts","sourceRoot":"","sources":["../../src/core/bash-executor.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAMH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,4BAA4B,CAAC;AAGrE,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAO5D,MAAM,WAAW,mBAAmB;IACnC,+DAA+D;IAC/D,OAAO,CAAC,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,IAAI,CAAC;IAClC,mCAAmC;IACnC,MAAM,CAAC,EAAE,WAAW,CAAC;IACrB,yBAAyB;IACzB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,6DAA6D;IAC7D,QAAQ,CAAC,EAAE,OAAO,CAAC;CACnB;AAED,MAAM,MAAM,UAAU,GAAG,kBAAkB,CAAC;AAqE5C;;;GAGG;AACH,wBAAsB,yBAAyB,CAC9C,OAAO,EAAE,MAAM,EACf,GAAG,EAAE,MAAM,EACX,UAAU,EAAE,cAAc,EAC1B,OAAO,CAAC,EAAE,mBAAmB,GAC3B,OAAO,CAAC,UAAU,CAAC,CAiBrB","sourcesContent":["/**\n * Bash command execution with streaming support and cancellation.\n *\n * This module provides a unified bash execution implementation used by:\n * - AgentSession.executeBash() for interactive and RPC modes\n * - Direct calls from modes that need bash execution\n */\n\nimport { randomBytes } from \"node:crypto\";\nimport { createWriteStream, type WriteStream } from \"node:fs\";\nimport { tmpdir } from \"node:os\";\nimport { join } from \"node:path\";\nimport type { ShellCaptureResult } from \"@greppleai/grip-agent-core\";\nimport { stripAnsi } from \"../utils/ansi.ts\";\nimport { sanitizeBinaryOutput } from \"../utils/shell.ts\";\nimport type { ToolOperations } from \"./tools/operations.ts\";\nimport { DEFAULT_MAX_BYTES, truncateTail } from \"./tools/truncate.ts\";\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface BashExecutorOptions {\n\t/** Callback for streaming output chunks (already sanitized) */\n\tonChunk?: (chunk: string) => void;\n\t/** AbortSignal for cancellation */\n\tsignal?: AbortSignal;\n\t/** Timeout in seconds */\n\ttimeout?: number;\n\t/** Whether to truncate returned output. Defaults to true. */\n\ttruncate?: boolean;\n}\n\nexport type BashResult = ShellCaptureResult;\n\n// ============================================================================\n// Implementation\n// ============================================================================\n\nclass BashOutputCapture {\n\tprivate outputChunks: string[] = [];\n\tprivate outputBytes = 0;\n\tprivate readonly maxOutputBytes = DEFAULT_MAX_BYTES * 2;\n\tprivate readonly truncateOutput: boolean;\n\tprivate tempFilePath: string | undefined;\n\tprivate tempFileStream: WriteStream | undefined;\n\tprivate totalBytes = 0;\n\tprivate readonly decoder = new TextDecoder();\n\tprivate readonly options: BashExecutorOptions;\n\n\tconstructor(options: BashExecutorOptions = {}) {\n\t\tthis.options = options;\n\t\tthis.truncateOutput = options.truncate !== false;\n\t}\n\n\tprivate ensureTempFile(): void {\n\t\tif (this.tempFilePath) return;\n\t\tconst id = randomBytes(8).toString(\"hex\");\n\t\tthis.tempFilePath = join(tmpdir(), `grip-bash-${id}.log`);\n\t\tthis.tempFileStream = createWriteStream(this.tempFilePath);\n\t\tfor (const chunk of this.outputChunks) {\n\t\t\tthis.tempFileStream.write(chunk);\n\t\t}\n\t}\n\n\tonData = (data: Buffer): void => {\n\t\tthis.totalBytes += data.length;\n\t\tconst text = sanitizeBinaryOutput(stripAnsi(this.decoder.decode(data, { stream: true }))).replace(/\\r/g, \"\");\n\n\t\tif (this.totalBytes > DEFAULT_MAX_BYTES) this.ensureTempFile();\n\t\tthis.tempFileStream?.write(text);\n\n\t\tthis.outputChunks.push(text);\n\t\tthis.outputBytes += text.length;\n\t\twhile (this.truncateOutput && this.outputBytes > this.maxOutputBytes && this.outputChunks.length > 1) {\n\t\t\tconst removed = this.outputChunks.shift()!;\n\t\t\tthis.outputBytes -= removed.length;\n\t\t}\n\n\t\tthis.options.onChunk?.(text);\n\t};\n\n\tfinish(exitCode: number | null | undefined, cancelled: boolean): BashResult {\n\t\tconst fullOutput = this.outputChunks.join(\"\");\n\t\tconst truncationResult = this.truncateOutput\n\t\t\t? truncateTail(fullOutput)\n\t\t\t: { content: fullOutput, truncated: false };\n\t\tif (truncationResult.truncated) this.ensureTempFile();\n\t\tthis.close();\n\t\treturn {\n\t\t\toutput: truncationResult.content,\n\t\t\texitCode: cancelled ? undefined : (exitCode ?? undefined),\n\t\t\tcancelled,\n\t\t\ttruncated: truncationResult.truncated,\n\t\t\tfullOutputPath: this.tempFilePath,\n\t\t};\n\t}\n\n\tclose(): void {\n\t\tthis.tempFileStream?.end();\n\t}\n}\n/**\n * Execute a bash command using custom ToolOperations.\n * Used for remote daemon, container, and host-provided execution backends.\n */\nexport async function executeBashWithOperations(\n\tcommand: string,\n\tcwd: string,\n\toperations: ToolOperations,\n\toptions?: BashExecutorOptions,\n): Promise<BashResult> {\n\tconst outputCapture = new BashOutputCapture(options);\n\n\ttry {\n\t\tconst result = await operations.exec(command, {\n\t\t\tcwd,\n\t\t\tonData: outputCapture.onData,\n\t\t\tsignal: options?.signal,\n\t\t\ttimeout: options?.timeout,\n\t\t});\n\t\tconst cancelled = options?.signal?.aborted ?? false;\n\t\treturn outputCapture.finish(result.exitCode, cancelled);\n\t} catch (error) {\n\t\tif (options?.signal?.aborted) return outputCapture.finish(undefined, true);\n\t\toutputCapture.close();\n\t\tthrow error;\n\t}\n}\n"]}

package/dist/core/bash-executor.js

@@ -0,0 +1,99 @@
+/**
+ * Bash command execution with streaming support and cancellation.
+ *
+ * This module provides a unified bash execution implementation used by:
+ * - AgentSession.executeBash() for interactive and RPC modes
+ * - Direct calls from modes that need bash execution
+ */
+import { randomBytes } from "node:crypto";
+import { createWriteStream } from "node:fs";
+import { tmpdir } from "node:os";
+import { join } from "node:path";
+import { stripAnsi } from "../utils/ansi.js";
+import { sanitizeBinaryOutput } from "../utils/shell.js";
+import { DEFAULT_MAX_BYTES, truncateTail } from "./tools/truncate.js";
+// ============================================================================
+// Implementation
+// ============================================================================
+class BashOutputCapture {
+    outputChunks = [];
+    outputBytes = 0;
+    maxOutputBytes = DEFAULT_MAX_BYTES * 2;
+    truncateOutput;
+    tempFilePath;
+    tempFileStream;
+    totalBytes = 0;
+    decoder = new TextDecoder();
+    options;
+    constructor(options = {}) {
+        this.options = options;
+        this.truncateOutput = options.truncate !== false;
+    }
+    ensureTempFile() {
+        if (this.tempFilePath)
+            return;
+        const id = randomBytes(8).toString("hex");
+        this.tempFilePath = join(tmpdir(), `grip-bash-${id}.log`);
+        this.tempFileStream = createWriteStream(this.tempFilePath);
+        for (const chunk of this.outputChunks) {
+            this.tempFileStream.write(chunk);
+        }
+    }
+    onData = (data) => {
+        this.totalBytes += data.length;
+        const text = sanitizeBinaryOutput(stripAnsi(this.decoder.decode(data, { stream: true }))).replace(/\r/g, "");
+        if (this.totalBytes > DEFAULT_MAX_BYTES)
+            this.ensureTempFile();
+        this.tempFileStream?.write(text);
+        this.outputChunks.push(text);
+        this.outputBytes += text.length;
+        while (this.truncateOutput && this.outputBytes > this.maxOutputBytes && this.outputChunks.length > 1) {
+            const removed = this.outputChunks.shift();
+            this.outputBytes -= removed.length;
+        }
+        this.options.onChunk?.(text);
+    };
+    finish(exitCode, cancelled) {
+        const fullOutput = this.outputChunks.join("");
+        const truncationResult = this.truncateOutput
+            ? truncateTail(fullOutput)
+            : { content: fullOutput, truncated: false };
+        if (truncationResult.truncated)
+            this.ensureTempFile();
+        this.close();
+        return {
+            output: truncationResult.content,
+            exitCode: cancelled ? undefined : (exitCode ?? undefined),
+            cancelled,
+            truncated: truncationResult.truncated,
+            fullOutputPath: this.tempFilePath,
+        };
+    }
+    close() {
+        this.tempFileStream?.end();
+    }
+}
+/**
+ * Execute a bash command using custom ToolOperations.
+ * Used for remote daemon, container, and host-provided execution backends.
+ */
+export async function executeBashWithOperations(command, cwd, operations, options) {
+    const outputCapture = new BashOutputCapture(options);
+    try {
+        const result = await operations.exec(command, {
+            cwd,
+            onData: outputCapture.onData,
+            signal: options?.signal,
+            timeout: options?.timeout,
+        });
+        const cancelled = options?.signal?.aborted ?? false;
+        return outputCapture.finish(result.exitCode, cancelled);
+    }
+    catch (error) {
+        if (options?.signal?.aborted)
+            return outputCapture.finish(undefined, true);
+        outputCapture.close();
+        throw error;
+    }
+}
+//# sourceMappingURL=bash-executor.js.map

package/dist/core/bash-executor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bash-executor.js","sourceRoot":"","sources":["../../src/core/bash-executor.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAEH,OAAO,EAAE,WAAW,EAAE,MAAM,aAAa,CAAC;AAC1C,OAAO,EAAE,iBAAiB,EAAoB,MAAM,SAAS,CAAC;AAC9D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAEjC,OAAO,EAAE,SAAS,EAAE,MAAM,kBAAkB,CAAC;AAC7C,OAAO,EAAE,oBAAoB,EAAE,MAAM,mBAAmB,CAAC;AAEzD,OAAO,EAAE,iBAAiB,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAmBtE,+EAA+E;AAC/E,iBAAiB;AACjB,+EAA+E;AAE/E,MAAM,iBAAiB;IACd,YAAY,GAAa,EAAE,CAAC;IAC5B,WAAW,GAAG,CAAC,CAAC;IACP,cAAc,GAAG,iBAAiB,GAAG,CAAC,CAAC;IACvC,cAAc,CAAU;IACjC,YAAY,CAAqB;IACjC,cAAc,CAA0B;IACxC,UAAU,GAAG,CAAC,CAAC;IACN,OAAO,GAAG,IAAI,WAAW,EAAE,CAAC;IAC5B,OAAO,CAAsB;IAE9C,YAAY,OAAO,GAAwB,EAAE,EAAE;QAC9C,IAAI,CAAC,OAAO,GAAG,OAAO,CAAC;QACvB,IAAI,CAAC,cAAc,GAAG,OAAO,CAAC,QAAQ,KAAK,KAAK,CAAC;IAAA,CACjD;IAEO,cAAc,GAAS;QAC9B,IAAI,IAAI,CAAC,YAAY;YAAE,OAAO;QAC9B,MAAM,EAAE,GAAG,WAAW,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;QAC1C,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,MAAM,EAAE,EAAE,aAAa,EAAE,MAAM,CAAC,CAAC;QAC1D,IAAI,CAAC,cAAc,GAAG,iBAAiB,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;QAC3D,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;YACvC,IAAI,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAClC,CAAC;IAAA,CACD;IAED,MAAM,GAAG,CAAC,IAAY,EAAQ,EAAE,CAAC;QAChC,IAAI,CAAC,UAAU,IAAI,IAAI,CAAC,MAAM,CAAC;QAC/B,MAAM,IAAI,GAAG,oBAAoB,CAAC,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;QAE7G,IAAI,IAAI,CAAC,UAAU,GAAG,iBAAiB;YAAE,IAAI,CAAC,cAAc,EAAE,CAAC;QAC/D,IAAI,CAAC,cAAc,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAEjC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QAC7B,IAAI,CAAC,WAAW,IAAI,IAAI,CAAC,MAAM,CAAC;QAChC,OAAO,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACtG,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,CAAC,KAAK,EAAG,CAAC;YAC3C,IAAI,CAAC,WAAW,IAAI,OAAO,CAAC,MAAM,CAAC;QACpC,CAAC;QAED,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;IAAA,CAC7B,CAAC;IAEF,MAAM,CAAC,QAAmC,EAAE,SAAkB,EAAc;QAC3E,MAAM,UAAU,GAAG,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAC9C,MAAM,gBAAgB,GAAG,IAAI,CAAC,cAAc;YAC3C,CAAC,CAAC,YAAY,CAAC,UAAU,CAAC;YAC1B,CAAC,CAAC,EAAE,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC;QAC7C,IAAI,gBAAgB,CAAC,SAAS;YAAE,IAAI,CAAC,cAAc,EAAE,CAAC;QACtD,IAAI,CAAC,KAAK,EAAE,CAAC;QACb,OAAO;YACN,MAAM,EAAE,gBAAgB,CAAC,OAAO;YAChC,QAAQ,EAAE,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,QAAQ,IAAI,SAAS,CAAC;YACzD,SAAS;YACT,SAAS,EAAE,gBAAgB,CAAC,SAAS;YACrC,cAAc,EAAE,IAAI,CAAC,YAAY;SACjC,CAAC;IAAA,CACF;IAED,KAAK,GAAS;QACb,IAAI,CAAC,cAAc,EAAE,GAAG,EAAE,CAAC;IAAA,CAC3B;CACD;AACD;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,yBAAyB,CAC9C,OAAe,EACf,GAAW,EACX,UAA0B,EAC1B,OAA6B,EACP;IACtB,MAAM,aAAa,GAAG,IAAI,iBAAiB,CAAC,OAAO,CAAC,CAAC;IAErD,IAAI,CAAC;QACJ,MAAM,MAAM,GAAG,MAAM,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE;YAC7C,GAAG;YACH,MAAM,EAAE,aAAa,CAAC,MAAM;YAC5B,MAAM,EAAE,OAAO,EAAE,MAAM;YACvB,OAAO,EAAE,OAAO,EAAE,OAAO;SACzB,CAAC,CAAC;QACH,MAAM,SAAS,GAAG,OAAO,EAAE,MAAM,EAAE,OAAO,IAAI,KAAK,CAAC;QACpD,OAAO,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,QAAQ,EAAE,SAAS,CAAC,CAAC;IACzD,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QAChB,IAAI,OAAO,EAAE,MAAM,EAAE,OAAO;YAAE,OAAO,aAAa,CAAC,MAAM,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC3E,aAAa,CAAC,KAAK,EAAE,CAAC;QACtB,MAAM,KAAK,CAAC;IACb,CAAC;AAAA,CACD","sourcesContent":["/**\n * Bash command execution with streaming support and cancellation.\n *\n * This module provides a unified bash execution implementation used by:\n * - AgentSession.executeBash() for interactive and RPC modes\n * - Direct calls from modes that need bash execution\n */\n\nimport { randomBytes } from \"node:crypto\";\nimport { createWriteStream, type WriteStream } from \"node:fs\";\nimport { tmpdir } from \"node:os\";\nimport { join } from \"node:path\";\nimport type { ShellCaptureResult } from \"@greppleai/grip-agent-core\";\nimport { stripAnsi } from \"../utils/ansi.ts\";\nimport { sanitizeBinaryOutput } from \"../utils/shell.ts\";\nimport type { ToolOperations } from \"./tools/operations.ts\";\nimport { DEFAULT_MAX_BYTES, truncateTail } from \"./tools/truncate.ts\";\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface BashExecutorOptions {\n\t/** Callback for streaming output chunks (already sanitized) */\n\tonChunk?: (chunk: string) => void;\n\t/** AbortSignal for cancellation */\n\tsignal?: AbortSignal;\n\t/** Timeout in seconds */\n\ttimeout?: number;\n\t/** Whether to truncate returned output. Defaults to true. */\n\ttruncate?: boolean;\n}\n\nexport type BashResult = ShellCaptureResult;\n\n// ============================================================================\n// Implementation\n// ============================================================================\n\nclass BashOutputCapture {\n\tprivate outputChunks: string[] = [];\n\tprivate outputBytes = 0;\n\tprivate readonly maxOutputBytes = DEFAULT_MAX_BYTES * 2;\n\tprivate readonly truncateOutput: boolean;\n\tprivate tempFilePath: string | undefined;\n\tprivate tempFileStream: WriteStream | undefined;\n\tprivate totalBytes = 0;\n\tprivate readonly decoder = new TextDecoder();\n\tprivate readonly options: BashExecutorOptions;\n\n\tconstructor(options: BashExecutorOptions = {}) {\n\t\tthis.options = options;\n\t\tthis.truncateOutput = options.truncate !== false;\n\t}\n\n\tprivate ensureTempFile(): void {\n\t\tif (this.tempFilePath) return;\n\t\tconst id = randomBytes(8).toString(\"hex\");\n\t\tthis.tempFilePath = join(tmpdir(), `grip-bash-${id}.log`);\n\t\tthis.tempFileStream = createWriteStream(this.tempFilePath);\n\t\tfor (const chunk of this.outputChunks) {\n\t\t\tthis.tempFileStream.write(chunk);\n\t\t}\n\t}\n\n\tonData = (data: Buffer): void => {\n\t\tthis.totalBytes += data.length;\n\t\tconst text = sanitizeBinaryOutput(stripAnsi(this.decoder.decode(data, { stream: true }))).replace(/\\r/g, \"\");\n\n\t\tif (this.totalBytes > DEFAULT_MAX_BYTES) this.ensureTempFile();\n\t\tthis.tempFileStream?.write(text);\n\n\t\tthis.outputChunks.push(text);\n\t\tthis.outputBytes += text.length;\n\t\twhile (this.truncateOutput && this.outputBytes > this.maxOutputBytes && this.outputChunks.length > 1) {\n\t\t\tconst removed = this.outputChunks.shift()!;\n\t\t\tthis.outputBytes -= removed.length;\n\t\t}\n\n\t\tthis.options.onChunk?.(text);\n\t};\n\n\tfinish(exitCode: number | null | undefined, cancelled: boolean): BashResult {\n\t\tconst fullOutput = this.outputChunks.join(\"\");\n\t\tconst truncationResult = this.truncateOutput\n\t\t\t? truncateTail(fullOutput)\n\t\t\t: { content: fullOutput, truncated: false };\n\t\tif (truncationResult.truncated) this.ensureTempFile();\n\t\tthis.close();\n\t\treturn {\n\t\t\toutput: truncationResult.content,\n\t\t\texitCode: cancelled ? undefined : (exitCode ?? undefined),\n\t\t\tcancelled,\n\t\t\ttruncated: truncationResult.truncated,\n\t\t\tfullOutputPath: this.tempFilePath,\n\t\t};\n\t}\n\n\tclose(): void {\n\t\tthis.tempFileStream?.end();\n\t}\n}\n/**\n * Execute a bash command using custom ToolOperations.\n * Used for remote daemon, container, and host-provided execution backends.\n */\nexport async function executeBashWithOperations(\n\tcommand: string,\n\tcwd: string,\n\toperations: ToolOperations,\n\toptions?: BashExecutorOptions,\n): Promise<BashResult> {\n\tconst outputCapture = new BashOutputCapture(options);\n\n\ttry {\n\t\tconst result = await operations.exec(command, {\n\t\t\tcwd,\n\t\t\tonData: outputCapture.onData,\n\t\t\tsignal: options?.signal,\n\t\t\ttimeout: options?.timeout,\n\t\t});\n\t\tconst cancelled = options?.signal?.aborted ?? false;\n\t\treturn outputCapture.finish(result.exitCode, cancelled);\n\t} catch (error) {\n\t\tif (options?.signal?.aborted) return outputCapture.finish(undefined, true);\n\t\toutputCapture.close();\n\t\tthrow error;\n\t}\n}\n"]}

package/dist/core/compaction/branch-summarization.d.ts

@@ -0,0 +1,83 @@
+/**
+ * Branch summarization for tree navigation.
+ *
+ * When navigating to a different point in the session tree, this generates
+ * a summary of the branch being left so context isn't lost.
+ */
+import type { BranchPreparation as AgentBranchPreparation, BranchSummaryDetails as AgentBranchSummaryDetails, StreamFn } from "@greppleai/grip-agent-core";
+import type { Model, ProviderHeaders, RetryCallbacks, RetryPolicy } from "@greppleai/grip-ai";
+import type { ReadonlySession } from "../session/session.ts";
+import type { SessionEntry } from "../session/types.ts";
+export interface BranchSummaryResult {
+    summary?: string;
+    readFiles?: string[];
+    modifiedFiles?: string[];
+    aborted?: boolean;
+    error?: string;
+}
+/** Details stored in BranchSummaryEntry.details for file tracking. */
+export type BranchSummaryDetails = AgentBranchSummaryDetails;
+/** Messages, file operations, and token count prepared for branch summarization. */
+export type BranchPreparation = AgentBranchPreparation;
+export interface CollectEntriesResult {
+    /** Entries to summarize, in chronological order */
+    entries: SessionEntry[];
+    /** Common ancestor between old and new position, if any */
+    commonAncestorId: string | null;
+}
+export interface GenerateBranchSummaryOptions {
+    /** Model to use for summarization */
+    model: Model<any>;
+    /** API key for the model */
+    apiKey?: string;
+    /** Request headers for the model */
+    headers?: ProviderHeaders;
+    /** Abort signal for cancellation */
+    signal: AbortSignal;
+    /** Optional custom instructions for summarization */
+    customInstructions?: string;
+    /** If true, customInstructions replaces the default prompt instead of being appended */
+    replaceInstructions?: boolean;
+    /** Tokens reserved for prompt + LLM response (default 16384) */
+    reserveTokens?: number;
+    /** Session stream function, preserving custom provider behavior without Agent state/events. */
+    streamFn?: StreamFn;
+    /** Retry transient summary failures using the session retry policy. */
+    retry?: RetryPolicy;
+    callbacks?: RetryCallbacks;
+}
+/**
+ * Collect entries that should be summarized when navigating from one position to another.
+ *
+ * Walks from oldLeafId back to the common ancestor with targetId, collecting entries
+ * along the way. Does NOT stop at compaction boundaries - those are included and their
+ * summaries become context.
+ *
+ * @param session - Session manager (read-only access)
+ * @param oldLeafId - Current position (where we're navigating from)
+ * @param targetId - Target position (where we're navigating to)
+ * @returns Entries to summarize and the common ancestor
+ */
+export declare function collectEntriesForBranchSummary(session: ReadonlySession, oldLeafId: string | null, targetId: string): CollectEntriesResult;
+/**
+ * Prepare entries for summarization with token budget.
+ *
+ * Walks entries from NEWEST to OLDEST, adding messages until we hit the token budget.
+ * This ensures we keep the most recent context when the branch is too long.
+ *
+ * Also collects file operations from:
+ * - Tool calls in assistant messages
+ * - Existing branch_summary entries' details (for cumulative tracking)
+ *
+ * @param entries - Entries in chronological order
+ * @param tokenBudget - Maximum tokens to include (0 = no limit)
+ */
+export declare function prepareBranchEntries(entries: SessionEntry[], tokenBudget?: number): BranchPreparation;
+/**
+ * Generate a summary of abandoned branch entries.
+ *
+ * @param entries - Session entries to summarize (chronological order)
+ * @param options - Generation options
+ */
+export declare function generateBranchSummary(entries: SessionEntry[], options: GenerateBranchSummaryOptions): Promise<BranchSummaryResult>;
+//# sourceMappingURL=branch-summarization.d.ts.map

package/dist/core/compaction/branch-summarization.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"branch-summarization.d.ts","sourceRoot":"","sources":["../../../src/core/compaction/branch-summarization.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EACX,iBAAiB,IAAI,sBAAsB,EAC3C,oBAAoB,IAAI,yBAAyB,EAGjD,QAAQ,EACR,MAAM,4BAA4B,CAAC;AACpC,OAAO,KAAK,EACX,KAAK,EACL,eAAe,EACf,cAAc,EACd,WAAW,EAGX,MAAM,oBAAoB,CAAC;AAG5B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,uBAAuB,CAAC;AAC7D,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,qBAAqB,CAAC;AAexD,MAAM,WAAW,mBAAmB;IACnC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,SAAS,CAAC,EAAE,MAAM,EAAE,CAAC;IACrB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;IACzB,OAAO,CAAC,EAAE,OAAO,CAAC;IAClB,KAAK,CAAC,EAAE,MAAM,CAAC;CACf;AAED,sEAAsE;AACtE,MAAM,MAAM,oBAAoB,GAAG,yBAAyB,CAAC;AAE7D,oFAAoF;AACpF,MAAM,MAAM,iBAAiB,GAAG,sBAAsB,CAAC;AAEvD,MAAM,WAAW,oBAAoB;IACpC,mDAAmD;IACnD,OAAO,EAAE,YAAY,EAAE,CAAC;IACxB,2DAA2D;IAC3D,gBAAgB,EAAE,MAAM,GAAG,IAAI,CAAC;CAChC;AAED,MAAM,WAAW,4BAA4B;IAC5C,qCAAqC;IACrC,KAAK,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC;IAClB,4BAA4B;IAC5B,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,oCAAoC;IACpC,OAAO,CAAC,EAAE,eAAe,CAAC;IAC1B,oCAAoC;IACpC,MAAM,EAAE,WAAW,CAAC;IACpB,qDAAqD;IACrD,kBAAkB,CAAC,EAAE,MAAM,CAAC;IAC5B,wFAAwF;IACxF,mBAAmB,CAAC,EAAE,OAAO,CAAC;IAC9B,gEAAgE;IAChE,aAAa,CAAC,EAAE,MAAM,CAAC;IACvB,+FAA+F;IAC/F,QAAQ,CAAC,EAAE,QAAQ,CAAC;IACpB,uEAAuE;IACvE,KAAK,CAAC,EAAE,WAAW,CAAC;IACpB,SAAS,CAAC,EAAE,cAAc,CAAC;CAC3B;AAMD;;;;;;;;;;;GAWG;AACH,wBAAgB,8BAA8B,CAC7C,OAAO,EAAE,eAAe,EACxB,SAAS,EAAE,MAAM,GAAG,IAAI,EACxB,QAAQ,EAAE,MAAM,GACd,oBAAoB,CAkCtB;AAsDD;;;;;;;;;;;;GAYG;AACH,wBAAgB,oBAAoB,CAAC,OAAO,EAAE,YAAY,EAAE,EAAE,WAAW,GAAE,MAAU,GAAG,iBAAiB,CAKxG;AAwCD;;;;;GAKG;AACH,wBAAsB,qBAAqB,CAC1C,OAAO,EAAE,YAAY,EAAE,EACvB,OAAO,EAAE,4BAA4B,GACnC,OAAO,CAAC,mBAAmB,CAAC,CA8E9B","sourcesContent":["/**\n * Branch summarization for tree navigation.\n *\n * When navigating to a different point in the session tree, this generates\n * a summary of the branch being left so context isn't lost.\n */\n\nimport type {\n\tBranchPreparation as AgentBranchPreparation,\n\tBranchSummaryDetails as AgentBranchSummaryDetails,\n\tAgentMessage,\n\tFileOperations,\n\tStreamFn,\n} from \"@greppleai/grip-agent-core\";\nimport type {\n\tModel,\n\tProviderHeaders,\n\tRetryCallbacks,\n\tRetryPolicy,\n\tSimpleStreamOptions,\n\tTextContent,\n} from \"@greppleai/grip-ai\";\nimport { convertToLlm } from \"../messages.ts\";\nimport { sessionEntryToContextMessages } from \"../session/context.ts\";\nimport type { ReadonlySession } from \"../session/session.ts\";\nimport type { SessionEntry } from \"../session/types.ts\";\nimport { completeSummarization, estimateTokens } from \"./compaction.ts\";\nimport {\n\tcomputeFileLists,\n\tcreateFileOps,\n\textractFileOpsFromMessage,\n\tformatFileOperations,\n\tSUMMARIZATION_SYSTEM_PROMPT,\n\tserializeConversation,\n} from \"./utils.ts\";\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface BranchSummaryResult {\n\tsummary?: string;\n\treadFiles?: string[];\n\tmodifiedFiles?: string[];\n\taborted?: boolean;\n\terror?: string;\n}\n\n/** Details stored in BranchSummaryEntry.details for file tracking. */\nexport type BranchSummaryDetails = AgentBranchSummaryDetails;\n\n/** Messages, file operations, and token count prepared for branch summarization. */\nexport type BranchPreparation = AgentBranchPreparation;\n\nexport interface CollectEntriesResult {\n\t/** Entries to summarize, in chronological order */\n\tentries: SessionEntry[];\n\t/** Common ancestor between old and new position, if any */\n\tcommonAncestorId: string | null;\n}\n\nexport interface GenerateBranchSummaryOptions {\n\t/** Model to use for summarization */\n\tmodel: Model<any>;\n\t/** API key for the model */\n\tapiKey?: string;\n\t/** Request headers for the model */\n\theaders?: ProviderHeaders;\n\t/** Abort signal for cancellation */\n\tsignal: AbortSignal;\n\t/** Optional custom instructions for summarization */\n\tcustomInstructions?: string;\n\t/** If true, customInstructions replaces the default prompt instead of being appended */\n\treplaceInstructions?: boolean;\n\t/** Tokens reserved for prompt + LLM response (default 16384) */\n\treserveTokens?: number;\n\t/** Session stream function, preserving custom provider behavior without Agent state/events. */\n\tstreamFn?: StreamFn;\n\t/** Retry transient summary failures using the session retry policy. */\n\tretry?: RetryPolicy;\n\tcallbacks?: RetryCallbacks;\n}\n\n// ============================================================================\n// Entry Collection\n// ============================================================================\n\n/**\n * Collect entries that should be summarized when navigating from one position to another.\n *\n * Walks from oldLeafId back to the common ancestor with targetId, collecting entries\n * along the way. Does NOT stop at compaction boundaries - those are included and their\n * summaries become context.\n *\n * @param session - Session manager (read-only access)\n * @param oldLeafId - Current position (where we're navigating from)\n * @param targetId - Target position (where we're navigating to)\n * @returns Entries to summarize and the common ancestor\n */\nexport function collectEntriesForBranchSummary(\n\tsession: ReadonlySession,\n\toldLeafId: string | null,\n\ttargetId: string,\n): CollectEntriesResult {\n\t// If no old position, nothing to summarize\n\tif (!oldLeafId) {\n\t\treturn { entries: [], commonAncestorId: null };\n\t}\n\n\t// Find common ancestor (deepest node that's on both paths)\n\tconst oldPath = new Set(session.getBranch(oldLeafId).map((e) => e.id));\n\tconst targetPath = session.getBranch(targetId);\n\n\t// targetPath is root-first, so iterate backwards to find deepest common ancestor\n\tlet commonAncestorId: string | null = null;\n\tfor (let i = targetPath.length - 1; i >= 0; i--) {\n\t\tif (oldPath.has(targetPath[i].id)) {\n\t\t\tcommonAncestorId = targetPath[i].id;\n\t\t\tbreak;\n\t\t}\n\t}\n\n\t// Collect entries from old leaf back to common ancestor\n\tconst entries: SessionEntry[] = [];\n\tlet current: string | null = oldLeafId;\n\n\twhile (current && current !== commonAncestorId) {\n\t\tconst entry = session.getEntry(current);\n\t\tif (!entry) break;\n\t\tentries.push(entry);\n\t\tcurrent = entry.parentId;\n\t}\n\n\t// Reverse to get chronological order\n\tentries.reverse();\n\n\treturn { entries, commonAncestorId };\n}\n\n// ============================================================================\n// Entry to Message Conversion\n// ============================================================================\n\n/** Project an entry into messages visible to the summary provider. */\nfunction getVisibleContextMessages(entry: SessionEntry): AgentMessage[] {\n\treturn sessionEntryToContextMessages(entry).filter((message) => convertToLlm([message]).length > 0);\n}\ninterface BranchMessageSelection {\n\tmessages: AgentMessage[];\n\ttotalTokens: number;\n}\n\nfunction collectCumulativeFileOperations(entries: SessionEntry[], fileOps: FileOperations): void {\n\tfor (const entry of entries) {\n\t\tif (entry.type !== \"branch_summary\" || entry.fromHook || !entry.details) continue;\n\t\tconst details = entry.details as BranchSummaryDetails;\n\t\tif (Array.isArray(details.readFiles)) {\n\t\t\tfor (const filePath of details.readFiles) fileOps.read.add(filePath);\n\t\t}\n\t\tif (Array.isArray(details.modifiedFiles)) {\n\t\t\tfor (const filePath of details.modifiedFiles) fileOps.edited.add(filePath);\n\t\t}\n\t}\n}\n\nfunction selectBranchMessages(\n\tentries: SessionEntry[],\n\ttokenBudget: number,\n\tfileOps: FileOperations,\n): BranchMessageSelection {\n\tconst messages: AgentMessage[] = [];\n\tlet totalTokens = 0;\n\tfor (let index = entries.length - 1; index >= 0; index--) {\n\t\tconst entry = entries[index];\n\t\tconst entryMessages = getVisibleContextMessages(entry);\n\t\tif (entryMessages.length === 0) continue;\n\t\tfor (const message of entryMessages) extractFileOpsFromMessage(message, fileOps);\n\t\tconst tokens = entryMessages.reduce((sum, message) => sum + estimateTokens(message), 0);\n\t\tif (tokenBudget > 0 && totalTokens + tokens > tokenBudget) {\n\t\t\tconst isExistingSummary = entry.type === \"compaction\" || entry.type === \"branch_summary\";\n\t\t\tif (isExistingSummary && totalTokens < tokenBudget * 0.9) {\n\t\t\t\tmessages.unshift(...entryMessages);\n\t\t\t\ttotalTokens += tokens;\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\t\tmessages.unshift(...entryMessages);\n\t\ttotalTokens += tokens;\n\t}\n\treturn { messages, totalTokens };\n}\n/**\n * Prepare entries for summarization with token budget.\n *\n * Walks entries from NEWEST to OLDEST, adding messages until we hit the token budget.\n * This ensures we keep the most recent context when the branch is too long.\n *\n * Also collects file operations from:\n * - Tool calls in assistant messages\n * - Existing branch_summary entries' details (for cumulative tracking)\n *\n * @param entries - Entries in chronological order\n * @param tokenBudget - Maximum tokens to include (0 = no limit)\n */\nexport function prepareBranchEntries(entries: SessionEntry[], tokenBudget: number = 0): BranchPreparation {\n\tconst fileOps = createFileOps();\n\tcollectCumulativeFileOperations(entries, fileOps);\n\tconst { messages, totalTokens } = selectBranchMessages(entries, tokenBudget, fileOps);\n\treturn { messages, fileOps, totalTokens };\n}\n\n// ============================================================================\n// Summary Generation\n// ============================================================================\n\nconst BRANCH_SUMMARY_PREAMBLE = `The user explored a different conversation branch before returning here.\nSummary of that exploration:\n\n`;\n\nconst BRANCH_SUMMARY_PROMPT = `Create a structured summary of this conversation branch for context when returning later.\n\nUse this EXACT format:\n\n## Goal\n[What was the user trying to accomplish in this branch?]\n\n## Constraints & Preferences\n- [Any constraints, preferences, or requirements mentioned]\n- [Or \"(none)\" if none were mentioned]\n\n## Progress\n### Done\n- [x] [Completed tasks/changes]\n\n### In Progress\n- [ ] [Work that was started but not finished]\n\n### Blocked\n- [Issues preventing progress, if any]\n\n## Key Decisions\n- **[Decision]**: [Brief rationale]\n\n## Next Steps\n1. [What should happen next to continue this work]\n\nKeep each section concise. Preserve exact file paths, function names, and error messages.`;\n\n/**\n * Generate a summary of abandoned branch entries.\n *\n * @param entries - Session entries to summarize (chronological order)\n * @param options - Generation options\n */\nexport async function generateBranchSummary(\n\tentries: SessionEntry[],\n\toptions: GenerateBranchSummaryOptions,\n): Promise<BranchSummaryResult> {\n\tconst {\n\t\tmodel,\n\t\tapiKey,\n\t\theaders,\n\t\tsignal,\n\t\tcustomInstructions,\n\t\treplaceInstructions,\n\t\treserveTokens = 16384,\n\t\tstreamFn,\n\t\tretry,\n\t\tcallbacks,\n\t} = options;\n\n\t// Token budget = context window minus reserved space for prompt + response\n\tconst contextWindow = model.contextWindow || 128000;\n\tconst tokenBudget = contextWindow - reserveTokens;\n\n\tconst { messages, fileOps } = prepareBranchEntries(entries, tokenBudget);\n\n\tif (messages.length === 0) {\n\t\treturn { summary: \"No content to summarize\" };\n\t}\n\n\t// Transform to LLM-compatible messages, then serialize to text\n\t// Serialization prevents the model from treating it as a conversation to continue\n\tconst llmMessages = convertToLlm(messages);\n\tconst conversationText = serializeConversation(llmMessages);\n\n\t// Build prompt\n\tlet instructions: string;\n\tif (replaceInstructions && customInstructions) {\n\t\tinstructions = customInstructions;\n\t} else if (customInstructions) {\n\t\tinstructions = `${BRANCH_SUMMARY_PROMPT}\\n\\nAdditional focus: ${customInstructions}`;\n\t} else {\n\t\tinstructions = BRANCH_SUMMARY_PROMPT;\n\t}\n\tconst promptText = `<conversation>\\n${conversationText}\\n</conversation>\\n\\n${instructions}`;\n\n\tconst summarizationMessages = [\n\t\t{\n\t\t\trole: \"user\" as const,\n\t\t\tcontent: [{ type: \"text\" as const, text: promptText }],\n\t\t\ttimestamp: Date.now(),\n\t\t},\n\t];\n\n\t// Keep summary calls detached from Agent state/events while preserving the active stream implementation.\n\tconst context = { systemPrompt: SUMMARIZATION_SYSTEM_PROMPT, messages: summarizationMessages };\n\tconst requestOptions: SimpleStreamOptions = { apiKey, headers, signal, maxTokens: 2048 };\n\tconst response = await completeSummarization(model, context, requestOptions, streamFn, retry, callbacks);\n\n\t// Check if aborted or errored\n\tif (response.stopReason === \"aborted\") {\n\t\treturn { aborted: true };\n\t}\n\tif (response.stopReason === \"error\") {\n\t\treturn { error: response.errorMessage || \"Summarization failed\" };\n\t}\n\n\tlet summary = response.content\n\t\t.filter((c): c is TextContent => c.type === \"text\")\n\t\t.map((c) => c.text)\n\t\t.join(\"\\n\");\n\n\t// Prepend preamble to provide context about the branch summary\n\tsummary = BRANCH_SUMMARY_PREAMBLE + summary;\n\n\t// Compute file lists and append to summary\n\tconst { readFiles, modifiedFiles } = computeFileLists(fileOps);\n\tsummary += formatFileOperations(readFiles, modifiedFiles);\n\n\treturn {\n\t\tsummary: summary || \"No summary generated\",\n\t\treadFiles,\n\t\tmodifiedFiles,\n\t};\n}\n"]}

package/dist/core/compaction/branch-summarization.js

@@ -0,0 +1,220 @@
+/**
+ * Branch summarization for tree navigation.
+ *
+ * When navigating to a different point in the session tree, this generates
+ * a summary of the branch being left so context isn't lost.
+ */
+import { convertToLlm } from "../messages.js";
+import { sessionEntryToContextMessages } from "../session/context.js";
+import { completeSummarization, estimateTokens } from "./compaction.js";
+import { computeFileLists, createFileOps, extractFileOpsFromMessage, formatFileOperations, SUMMARIZATION_SYSTEM_PROMPT, serializeConversation, } from "./utils.js";
+// ============================================================================
+// Entry Collection
+// ============================================================================
+/**
+ * Collect entries that should be summarized when navigating from one position to another.
+ *
+ * Walks from oldLeafId back to the common ancestor with targetId, collecting entries
+ * along the way. Does NOT stop at compaction boundaries - those are included and their
+ * summaries become context.
+ *
+ * @param session - Session manager (read-only access)
+ * @param oldLeafId - Current position (where we're navigating from)
+ * @param targetId - Target position (where we're navigating to)
+ * @returns Entries to summarize and the common ancestor
+ */
+export function collectEntriesForBranchSummary(session, oldLeafId, targetId) {
+    // If no old position, nothing to summarize
+    if (!oldLeafId) {
+        return { entries: [], commonAncestorId: null };
+    }
+    // Find common ancestor (deepest node that's on both paths)
+    const oldPath = new Set(session.getBranch(oldLeafId).map((e) => e.id));
+    const targetPath = session.getBranch(targetId);
+    // targetPath is root-first, so iterate backwards to find deepest common ancestor
+    let commonAncestorId = null;
+    for (let i = targetPath.length - 1; i >= 0; i--) {
+        if (oldPath.has(targetPath[i].id)) {
+            commonAncestorId = targetPath[i].id;
+            break;
+        }
+    }
+    // Collect entries from old leaf back to common ancestor
+    const entries = [];
+    let current = oldLeafId;
+    while (current && current !== commonAncestorId) {
+        const entry = session.getEntry(current);
+        if (!entry)
+            break;
+        entries.push(entry);
+        current = entry.parentId;
+    }
+    // Reverse to get chronological order
+    entries.reverse();
+    return { entries, commonAncestorId };
+}
+// ============================================================================
+// Entry to Message Conversion
+// ============================================================================
+/** Project an entry into messages visible to the summary provider. */
+function getVisibleContextMessages(entry) {
+    return sessionEntryToContextMessages(entry).filter((message) => convertToLlm([message]).length > 0);
+}
+function collectCumulativeFileOperations(entries, fileOps) {
+    for (const entry of entries) {
+        if (entry.type !== "branch_summary" || entry.fromHook || !entry.details)
+            continue;
+        const details = entry.details;
+        if (Array.isArray(details.readFiles)) {
+            for (const filePath of details.readFiles)
+                fileOps.read.add(filePath);
+        }
+        if (Array.isArray(details.modifiedFiles)) {
+            for (const filePath of details.modifiedFiles)
+                fileOps.edited.add(filePath);
+        }
+    }
+}
+function selectBranchMessages(entries, tokenBudget, fileOps) {
+    const messages = [];
+    let totalTokens = 0;
+    for (let index = entries.length - 1; index >= 0; index--) {
+        const entry = entries[index];
+        const entryMessages = getVisibleContextMessages(entry);
+        if (entryMessages.length === 0)
+            continue;
+        for (const message of entryMessages)
+            extractFileOpsFromMessage(message, fileOps);
+        const tokens = entryMessages.reduce((sum, message) => sum + estimateTokens(message), 0);
+        if (tokenBudget > 0 && totalTokens + tokens > tokenBudget) {
+            const isExistingSummary = entry.type === "compaction" || entry.type === "branch_summary";
+            if (isExistingSummary && totalTokens < tokenBudget * 0.9) {
+                messages.unshift(...entryMessages);
+                totalTokens += tokens;
+            }
+            break;
+        }
+        messages.unshift(...entryMessages);
+        totalTokens += tokens;
+    }
+    return { messages, totalTokens };
+}
+/**
+ * Prepare entries for summarization with token budget.
+ *
+ * Walks entries from NEWEST to OLDEST, adding messages until we hit the token budget.
+ * This ensures we keep the most recent context when the branch is too long.
+ *
+ * Also collects file operations from:
+ * - Tool calls in assistant messages
+ * - Existing branch_summary entries' details (for cumulative tracking)
+ *
+ * @param entries - Entries in chronological order
+ * @param tokenBudget - Maximum tokens to include (0 = no limit)
+ */
+export function prepareBranchEntries(entries, tokenBudget = 0) {
+    const fileOps = createFileOps();
+    collectCumulativeFileOperations(entries, fileOps);
+    const { messages, totalTokens } = selectBranchMessages(entries, tokenBudget, fileOps);
+    return { messages, fileOps, totalTokens };
+}
+// ============================================================================
+// Summary Generation
+// ============================================================================
+const BRANCH_SUMMARY_PREAMBLE = `The user explored a different conversation branch before returning here.
+Summary of that exploration:
+
+`;
+const BRANCH_SUMMARY_PROMPT = `Create a structured summary of this conversation branch for context when returning later.
+
+Use this EXACT format:
+
+## Goal
+[What was the user trying to accomplish in this branch?]
+
+## Constraints & Preferences
+- [Any constraints, preferences, or requirements mentioned]
+- [Or "(none)" if none were mentioned]
+
+## Progress
+### Done
+- [x] [Completed tasks/changes]
+
+### In Progress
+- [ ] [Work that was started but not finished]
+
+### Blocked
+- [Issues preventing progress, if any]
+
+## Key Decisions
+- **[Decision]**: [Brief rationale]
+
+## Next Steps
+1. [What should happen next to continue this work]
+
+Keep each section concise. Preserve exact file paths, function names, and error messages.`;
+/**
+ * Generate a summary of abandoned branch entries.
+ *
+ * @param entries - Session entries to summarize (chronological order)
+ * @param options - Generation options
+ */
+export async function generateBranchSummary(entries, options) {
+    const { model, apiKey, headers, signal, customInstructions, replaceInstructions, reserveTokens = 16384, streamFn, retry, callbacks, } = options;
+    // Token budget = context window minus reserved space for prompt + response
+    const contextWindow = model.contextWindow || 128000;
+    const tokenBudget = contextWindow - reserveTokens;
+    const { messages, fileOps } = prepareBranchEntries(entries, tokenBudget);
+    if (messages.length === 0) {
+        return { summary: "No content to summarize" };
+    }
+    // Transform to LLM-compatible messages, then serialize to text
+    // Serialization prevents the model from treating it as a conversation to continue
+    const llmMessages = convertToLlm(messages);
+    const conversationText = serializeConversation(llmMessages);
+    // Build prompt
+    let instructions;
+    if (replaceInstructions && customInstructions) {
+        instructions = customInstructions;
+    }
+    else if (customInstructions) {
+        instructions = `${BRANCH_SUMMARY_PROMPT}\n\nAdditional focus: ${customInstructions}`;
+    }
+    else {
+        instructions = BRANCH_SUMMARY_PROMPT;
+    }
+    const promptText = `<conversation>\n${conversationText}\n</conversation>\n\n${instructions}`;
+    const summarizationMessages = [
+        {
+            role: "user",
+            content: [{ type: "text", text: promptText }],
+            timestamp: Date.now(),
+        },
+    ];
+    // Keep summary calls detached from Agent state/events while preserving the active stream implementation.
+    const context = { systemPrompt: SUMMARIZATION_SYSTEM_PROMPT, messages: summarizationMessages };
+    const requestOptions = { apiKey, headers, signal, maxTokens: 2048 };
+    const response = await completeSummarization(model, context, requestOptions, streamFn, retry, callbacks);
+    // Check if aborted or errored
+    if (response.stopReason === "aborted") {
+        return { aborted: true };
+    }
+    if (response.stopReason === "error") {
+        return { error: response.errorMessage || "Summarization failed" };
+    }
+    let summary = response.content
+        .filter((c) => c.type === "text")
+        .map((c) => c.text)
+        .join("\n");
+    // Prepend preamble to provide context about the branch summary
+    summary = BRANCH_SUMMARY_PREAMBLE + summary;
+    // Compute file lists and append to summary
+    const { readFiles, modifiedFiles } = computeFileLists(fileOps);
+    summary += formatFileOperations(readFiles, modifiedFiles);
+    return {
+        summary: summary || "No summary generated",
+        readFiles,
+        modifiedFiles,
+    };
+}
+//# sourceMappingURL=branch-summarization.js.map

package/dist/core/compaction/branch-summarization.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"branch-summarization.js","sourceRoot":"","sources":["../../../src/core/compaction/branch-summarization.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAiBH,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,6BAA6B,EAAE,MAAM,uBAAuB,CAAC;AAGtE,OAAO,EAAE,qBAAqB,EAAE,cAAc,EAAE,MAAM,iBAAiB,CAAC;AACxE,OAAO,EACN,gBAAgB,EAChB,aAAa,EACb,yBAAyB,EACzB,oBAAoB,EACpB,2BAA2B,EAC3B,qBAAqB,GACrB,MAAM,YAAY,CAAC;AAiDpB,+EAA+E;AAC/E,mBAAmB;AACnB,+EAA+E;AAE/E;;;;;;;;;;;GAWG;AACH,MAAM,UAAU,8BAA8B,CAC7C,OAAwB,EACxB,SAAwB,EACxB,QAAgB,EACO;IACvB,2CAA2C;IAC3C,IAAI,CAAC,SAAS,EAAE,CAAC;QAChB,OAAO,EAAE,OAAO,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,CAAC;IAChD,CAAC;IAED,2DAA2D;IAC3D,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;IACvE,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,QAAQ,CAAC,CAAC;IAE/C,iFAAiF;IACjF,IAAI,gBAAgB,GAAkB,IAAI,CAAC;IAC3C,KAAK,IAAI,CAAC,GAAG,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACjD,IAAI,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;YACnC,gBAAgB,GAAG,UAAU,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACpC,MAAM;QACP,CAAC;IACF,CAAC;IAED,wDAAwD;IACxD,MAAM,OAAO,GAAmB,EAAE,CAAC;IACnC,IAAI,OAAO,GAAkB,SAAS,CAAC;IAEvC,OAAO,OAAO,IAAI,OAAO,KAAK,gBAAgB,EAAE,CAAC;QAChD,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;QACxC,IAAI,CAAC,KAAK;YAAE,MAAM;QAClB,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QACpB,OAAO,GAAG,KAAK,CAAC,QAAQ,CAAC;IAC1B,CAAC;IAED,qCAAqC;IACrC,OAAO,CAAC,OAAO,EAAE,CAAC;IAElB,OAAO,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;AAAA,CACrC;AAED,+EAA+E;AAC/E,8BAA8B;AAC9B,+EAA+E;AAE/E,sEAAsE;AACtE,SAAS,yBAAyB,CAAC,KAAmB,EAAkB;IACvE,OAAO,6BAA6B,CAAC,KAAK,CAAC,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,YAAY,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;AAAA,CACpG;AAMD,SAAS,+BAA+B,CAAC,OAAuB,EAAE,OAAuB,EAAQ;IAChG,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC7B,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,KAAK,CAAC,QAAQ,IAAI,CAAC,KAAK,CAAC,OAAO;YAAE,SAAS;QAClF,MAAM,OAAO,GAAG,KAAK,CAAC,OAA+B,CAAC;QACtD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,SAAS,CAAC,EAAE,CAAC;YACtC,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,SAAS;gBAAE,OAAO,CAAC,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YAC1C,KAAK,MAAM,QAAQ,IAAI,OAAO,CAAC,aAAa;gBAAE,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAC5E,CAAC;IACF,CAAC;AAAA,CACD;AAED,SAAS,oBAAoB,CAC5B,OAAuB,EACvB,WAAmB,EACnB,OAAuB,EACE;IACzB,MAAM,QAAQ,GAAmB,EAAE,CAAC;IACpC,IAAI,WAAW,GAAG,CAAC,CAAC;IACpB,KAAK,IAAI,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QAC1D,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC;QAC7B,MAAM,aAAa,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;QACvD,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC;YAAE,SAAS;QACzC,KAAK,MAAM,OAAO,IAAI,aAAa;YAAE,yBAAyB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACjF,MAAM,MAAM,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,OAAO,EAAE,EAAE,CAAC,GAAG,GAAG,cAAc,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC,CAAC;QACxF,IAAI,WAAW,GAAG,CAAC,IAAI,WAAW,GAAG,MAAM,GAAG,WAAW,EAAE,CAAC;YAC3D,MAAM,iBAAiB,GAAG,KAAK,CAAC,IAAI,KAAK,YAAY,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB,CAAC;YACzF,IAAI,iBAAiB,IAAI,WAAW,GAAG,WAAW,GAAG,GAAG,EAAE,CAAC;gBAC1D,QAAQ,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC;gBACnC,WAAW,IAAI,MAAM,CAAC;YACvB,CAAC;YACD,MAAM;QACP,CAAC;QACD,QAAQ,CAAC,OAAO,CAAC,GAAG,aAAa,CAAC,CAAC;QACnC,WAAW,IAAI,MAAM,CAAC;IACvB,CAAC;IACD,OAAO,EAAE,QAAQ,EAAE,WAAW,EAAE,CAAC;AAAA,CACjC;AACD;;;;;;;;;;;;GAYG;AACH,MAAM,UAAU,oBAAoB,CAAC,OAAuB,EAAE,WAAW,GAAW,CAAC,EAAqB;IACzG,MAAM,OAAO,GAAG,aAAa,EAAE,CAAC;IAChC,+BAA+B,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;IAClD,MAAM,EAAE,QAAQ,EAAE,WAAW,EAAE,GAAG,oBAAoB,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,CAAC;IACtF,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,WAAW,EAAE,CAAC;AAAA,CAC1C;AAED,+EAA+E;AAC/E,qBAAqB;AACrB,+EAA+E;AAE/E,MAAM,uBAAuB,GAAG;;;CAG/B,CAAC;AAEF,MAAM,qBAAqB,GAAG;;;;;;;;;;;;;;;;;;;;;;;;;;;0FA2B4D,CAAC;AAE3F;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,qBAAqB,CAC1C,OAAuB,EACvB,OAAqC,EACN;IAC/B,MAAM,EACL,KAAK,EACL,MAAM,EACN,OAAO,EACP,MAAM,EACN,kBAAkB,EAClB,mBAAmB,EACnB,aAAa,GAAG,KAAK,EACrB,QAAQ,EACR,KAAK,EACL,SAAS,GACT,GAAG,OAAO,CAAC;IAEZ,2EAA2E;IAC3E,MAAM,aAAa,GAAG,KAAK,CAAC,aAAa,IAAI,MAAM,CAAC;IACpD,MAAM,WAAW,GAAG,aAAa,GAAG,aAAa,CAAC;IAElD,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,GAAG,oBAAoB,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAEzE,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,EAAE,OAAO,EAAE,yBAAyB,EAAE,CAAC;IAC/C,CAAC;IAED,+DAA+D;IAC/D,kFAAkF;IAClF,MAAM,WAAW,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC3C,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAE5D,eAAe;IACf,IAAI,YAAoB,CAAC;IACzB,IAAI,mBAAmB,IAAI,kBAAkB,EAAE,CAAC;QAC/C,YAAY,GAAG,kBAAkB,CAAC;IACnC,CAAC;SAAM,IAAI,kBAAkB,EAAE,CAAC;QAC/B,YAAY,GAAG,GAAG,qBAAqB,yBAAyB,kBAAkB,EAAE,CAAC;IACtF,CAAC;SAAM,CAAC;QACP,YAAY,GAAG,qBAAqB,CAAC;IACtC,CAAC;IACD,MAAM,UAAU,GAAG,mBAAmB,gBAAgB,wBAAwB,YAAY,EAAE,CAAC;IAE7F,MAAM,qBAAqB,GAAG;QAC7B;YACC,IAAI,EAAE,MAAe;YACrB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAe,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;YACtD,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB;KACD,CAAC;IAEF,yGAAyG;IACzG,MAAM,OAAO,GAAG,EAAE,YAAY,EAAE,2BAA2B,EAAE,QAAQ,EAAE,qBAAqB,EAAE,CAAC;IAC/F,MAAM,cAAc,GAAwB,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC;IACzF,MAAM,QAAQ,GAAG,MAAM,qBAAqB,CAAC,KAAK,EAAE,OAAO,EAAE,cAAc,EAAE,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;IAEzG,8BAA8B;IAC9B,IAAI,QAAQ,CAAC,UAAU,KAAK,SAAS,EAAE,CAAC;QACvC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;IAC1B,CAAC;IACD,IAAI,QAAQ,CAAC,UAAU,KAAK,OAAO,EAAE,CAAC;QACrC,OAAO,EAAE,KAAK,EAAE,QAAQ,CAAC,YAAY,IAAI,sBAAsB,EAAE,CAAC;IACnE,CAAC;IAED,IAAI,OAAO,GAAG,QAAQ,CAAC,OAAO;SAC5B,MAAM,CAAC,CAAC,CAAC,EAAoB,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,MAAM,CAAC;SAClD,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAClB,IAAI,CAAC,IAAI,CAAC,CAAC;IAEb,+DAA+D;IAC/D,OAAO,GAAG,uBAAuB,GAAG,OAAO,CAAC;IAE5C,2CAA2C;IAC3C,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC,OAAO,CAAC,CAAC;IAC/D,OAAO,IAAI,oBAAoB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IAE1D,OAAO;QACN,OAAO,EAAE,OAAO,IAAI,sBAAsB;QAC1C,SAAS;QACT,aAAa;KACb,CAAC;AAAA,CACF","sourcesContent":["/**\n * Branch summarization for tree navigation.\n *\n * When navigating to a different point in the session tree, this generates\n * a summary of the branch being left so context isn't lost.\n */\n\nimport type {\n\tBranchPreparation as AgentBranchPreparation,\n\tBranchSummaryDetails as AgentBranchSummaryDetails,\n\tAgentMessage,\n\tFileOperations,\n\tStreamFn,\n} from \"@greppleai/grip-agent-core\";\nimport type {\n\tModel,\n\tProviderHeaders,\n\tRetryCallbacks,\n\tRetryPolicy,\n\tSimpleStreamOptions,\n\tTextContent,\n} from \"@greppleai/grip-ai\";\nimport { convertToLlm } from \"../messages.ts\";\nimport { sessionEntryToContextMessages } from \"../session/context.ts\";\nimport type { ReadonlySession } from \"../session/session.ts\";\nimport type { SessionEntry } from \"../session/types.ts\";\nimport { completeSummarization, estimateTokens } from \"./compaction.ts\";\nimport {\n\tcomputeFileLists,\n\tcreateFileOps,\n\textractFileOpsFromMessage,\n\tformatFileOperations,\n\tSUMMARIZATION_SYSTEM_PROMPT,\n\tserializeConversation,\n} from \"./utils.ts\";\n\n// ============================================================================\n// Types\n// ============================================================================\n\nexport interface BranchSummaryResult {\n\tsummary?: string;\n\treadFiles?: string[];\n\tmodifiedFiles?: string[];\n\taborted?: boolean;\n\terror?: string;\n}\n\n/** Details stored in BranchSummaryEntry.details for file tracking. */\nexport type BranchSummaryDetails = AgentBranchSummaryDetails;\n\n/** Messages, file operations, and token count prepared for branch summarization. */\nexport type BranchPreparation = AgentBranchPreparation;\n\nexport interface CollectEntriesResult {\n\t/** Entries to summarize, in chronological order */\n\tentries: SessionEntry[];\n\t/** Common ancestor between old and new position, if any */\n\tcommonAncestorId: string | null;\n}\n\nexport interface GenerateBranchSummaryOptions {\n\t/** Model to use for summarization */\n\tmodel: Model<any>;\n\t/** API key for the model */\n\tapiKey?: string;\n\t/** Request headers for the model */\n\theaders?: ProviderHeaders;\n\t/** Abort signal for cancellation */\n\tsignal: AbortSignal;\n\t/** Optional custom instructions for summarization */\n\tcustomInstructions?: string;\n\t/** If true, customInstructions replaces the default prompt instead of being appended */\n\treplaceInstructions?: boolean;\n\t/** Tokens reserved for prompt + LLM response (default 16384) */\n\treserveTokens?: number;\n\t/** Session stream function, preserving custom provider behavior without Agent state/events. */\n\tstreamFn?: StreamFn;\n\t/** Retry transient summary failures using the session retry policy. */\n\tretry?: RetryPolicy;\n\tcallbacks?: RetryCallbacks;\n}\n\n// ============================================================================\n// Entry Collection\n// ============================================================================\n\n/**\n * Collect entries that should be summarized when navigating from one position to another.\n *\n * Walks from oldLeafId back to the common ancestor with targetId, collecting entries\n * along the way. Does NOT stop at compaction boundaries - those are included and their\n * summaries become context.\n *\n * @param session - Session manager (read-only access)\n * @param oldLeafId - Current position (where we're navigating from)\n * @param targetId - Target position (where we're navigating to)\n * @returns Entries to summarize and the common ancestor\n */\nexport function collectEntriesForBranchSummary(\n\tsession: ReadonlySession,\n\toldLeafId: string | null,\n\ttargetId: string,\n): CollectEntriesResult {\n\t// If no old position, nothing to summarize\n\tif (!oldLeafId) {\n\t\treturn { entries: [], commonAncestorId: null };\n\t}\n\n\t// Find common ancestor (deepest node that's on both paths)\n\tconst oldPath = new Set(session.getBranch(oldLeafId).map((e) => e.id));\n\tconst targetPath = session.getBranch(targetId);\n\n\t// targetPath is root-first, so iterate backwards to find deepest common ancestor\n\tlet commonAncestorId: string | null = null;\n\tfor (let i = targetPath.length - 1; i >= 0; i--) {\n\t\tif (oldPath.has(targetPath[i].id)) {\n\t\t\tcommonAncestorId = targetPath[i].id;\n\t\t\tbreak;\n\t\t}\n\t}\n\n\t// Collect entries from old leaf back to common ancestor\n\tconst entries: SessionEntry[] = [];\n\tlet current: string | null = oldLeafId;\n\n\twhile (current && current !== commonAncestorId) {\n\t\tconst entry = session.getEntry(current);\n\t\tif (!entry) break;\n\t\tentries.push(entry);\n\t\tcurrent = entry.parentId;\n\t}\n\n\t// Reverse to get chronological order\n\tentries.reverse();\n\n\treturn { entries, commonAncestorId };\n}\n\n// ============================================================================\n// Entry to Message Conversion\n// ============================================================================\n\n/** Project an entry into messages visible to the summary provider. */\nfunction getVisibleContextMessages(entry: SessionEntry): AgentMessage[] {\n\treturn sessionEntryToContextMessages(entry).filter((message) => convertToLlm([message]).length > 0);\n}\ninterface BranchMessageSelection {\n\tmessages: AgentMessage[];\n\ttotalTokens: number;\n}\n\nfunction collectCumulativeFileOperations(entries: SessionEntry[], fileOps: FileOperations): void {\n\tfor (const entry of entries) {\n\t\tif (entry.type !== \"branch_summary\" || entry.fromHook || !entry.details) continue;\n\t\tconst details = entry.details as BranchSummaryDetails;\n\t\tif (Array.isArray(details.readFiles)) {\n\t\t\tfor (const filePath of details.readFiles) fileOps.read.add(filePath);\n\t\t}\n\t\tif (Array.isArray(details.modifiedFiles)) {\n\t\t\tfor (const filePath of details.modifiedFiles) fileOps.edited.add(filePath);\n\t\t}\n\t}\n}\n\nfunction selectBranchMessages(\n\tentries: SessionEntry[],\n\ttokenBudget: number,\n\tfileOps: FileOperations,\n): BranchMessageSelection {\n\tconst messages: AgentMessage[] = [];\n\tlet totalTokens = 0;\n\tfor (let index = entries.length - 1; index >= 0; index--) {\n\t\tconst entry = entries[index];\n\t\tconst entryMessages = getVisibleContextMessages(entry);\n\t\tif (entryMessages.length === 0) continue;\n\t\tfor (const message of entryMessages) extractFileOpsFromMessage(message, fileOps);\n\t\tconst tokens = entryMessages.reduce((sum, message) => sum + estimateTokens(message), 0);\n\t\tif (tokenBudget > 0 && totalTokens + tokens > tokenBudget) {\n\t\t\tconst isExistingSummary = entry.type === \"compaction\" || entry.type === \"branch_summary\";\n\t\t\tif (isExistingSummary && totalTokens < tokenBudget * 0.9) {\n\t\t\t\tmessages.unshift(...entryMessages);\n\t\t\t\ttotalTokens += tokens;\n\t\t\t}\n\t\t\tbreak;\n\t\t}\n\t\tmessages.unshift(...entryMessages);\n\t\ttotalTokens += tokens;\n\t}\n\treturn { messages, totalTokens };\n}\n/**\n * Prepare entries for summarization with token budget.\n *\n * Walks entries from NEWEST to OLDEST, adding messages until we hit the token budget.\n * This ensures we keep the most recent context when the branch is too long.\n *\n * Also collects file operations from:\n * - Tool calls in assistant messages\n * - Existing branch_summary entries' details (for cumulative tracking)\n *\n * @param entries - Entries in chronological order\n * @param tokenBudget - Maximum tokens to include (0 = no limit)\n */\nexport function prepareBranchEntries(entries: SessionEntry[], tokenBudget: number = 0): BranchPreparation {\n\tconst fileOps = createFileOps();\n\tcollectCumulativeFileOperations(entries, fileOps);\n\tconst { messages, totalTokens } = selectBranchMessages(entries, tokenBudget, fileOps);\n\treturn { messages, fileOps, totalTokens };\n}\n\n// ============================================================================\n// Summary Generation\n// ============================================================================\n\nconst BRANCH_SUMMARY_PREAMBLE = `The user explored a different conversation branch before returning here.\nSummary of that exploration:\n\n`;\n\nconst BRANCH_SUMMARY_PROMPT = `Create a structured summary of this conversation branch for context when returning later.\n\nUse this EXACT format:\n\n## Goal\n[What was the user trying to accomplish in this branch?]\n\n## Constraints & Preferences\n- [Any constraints, preferences, or requirements mentioned]\n- [Or \"(none)\" if none were mentioned]\n\n## Progress\n### Done\n- [x] [Completed tasks/changes]\n\n### In Progress\n- [ ] [Work that was started but not finished]\n\n### Blocked\n- [Issues preventing progress, if any]\n\n## Key Decisions\n- **[Decision]**: [Brief rationale]\n\n## Next Steps\n1. [What should happen next to continue this work]\n\nKeep each section concise. Preserve exact file paths, function names, and error messages.`;\n\n/**\n * Generate a summary of abandoned branch entries.\n *\n * @param entries - Session entries to summarize (chronological order)\n * @param options - Generation options\n */\nexport async function generateBranchSummary(\n\tentries: SessionEntry[],\n\toptions: GenerateBranchSummaryOptions,\n): Promise<BranchSummaryResult> {\n\tconst {\n\t\tmodel,\n\t\tapiKey,\n\t\theaders,\n\t\tsignal,\n\t\tcustomInstructions,\n\t\treplaceInstructions,\n\t\treserveTokens = 16384,\n\t\tstreamFn,\n\t\tretry,\n\t\tcallbacks,\n\t} = options;\n\n\t// Token budget = context window minus reserved space for prompt + response\n\tconst contextWindow = model.contextWindow || 128000;\n\tconst tokenBudget = contextWindow - reserveTokens;\n\n\tconst { messages, fileOps } = prepareBranchEntries(entries, tokenBudget);\n\n\tif (messages.length === 0) {\n\t\treturn { summary: \"No content to summarize\" };\n\t}\n\n\t// Transform to LLM-compatible messages, then serialize to text\n\t// Serialization prevents the model from treating it as a conversation to continue\n\tconst llmMessages = convertToLlm(messages);\n\tconst conversationText = serializeConversation(llmMessages);\n\n\t// Build prompt\n\tlet instructions: string;\n\tif (replaceInstructions && customInstructions) {\n\t\tinstructions = customInstructions;\n\t} else if (customInstructions) {\n\t\tinstructions = `${BRANCH_SUMMARY_PROMPT}\\n\\nAdditional focus: ${customInstructions}`;\n\t} else {\n\t\tinstructions = BRANCH_SUMMARY_PROMPT;\n\t}\n\tconst promptText = `<conversation>\\n${conversationText}\\n</conversation>\\n\\n${instructions}`;\n\n\tconst summarizationMessages = [\n\t\t{\n\t\t\trole: \"user\" as const,\n\t\t\tcontent: [{ type: \"text\" as const, text: promptText }],\n\t\t\ttimestamp: Date.now(),\n\t\t},\n\t];\n\n\t// Keep summary calls detached from Agent state/events while preserving the active stream implementation.\n\tconst context = { systemPrompt: SUMMARIZATION_SYSTEM_PROMPT, messages: summarizationMessages };\n\tconst requestOptions: SimpleStreamOptions = { apiKey, headers, signal, maxTokens: 2048 };\n\tconst response = await completeSummarization(model, context, requestOptions, streamFn, retry, callbacks);\n\n\t// Check if aborted or errored\n\tif (response.stopReason === \"aborted\") {\n\t\treturn { aborted: true };\n\t}\n\tif (response.stopReason === \"error\") {\n\t\treturn { error: response.errorMessage || \"Summarization failed\" };\n\t}\n\n\tlet summary = response.content\n\t\t.filter((c): c is TextContent => c.type === \"text\")\n\t\t.map((c) => c.text)\n\t\t.join(\"\\n\");\n\n\t// Prepend preamble to provide context about the branch summary\n\tsummary = BRANCH_SUMMARY_PREAMBLE + summary;\n\n\t// Compute file lists and append to summary\n\tconst { readFiles, modifiedFiles } = computeFileLists(fileOps);\n\tsummary += formatFileOperations(readFiles, modifiedFiles);\n\n\treturn {\n\t\tsummary: summary || \"No summary generated\",\n\t\treadFiles,\n\t\tmodifiedFiles,\n\t};\n}\n"]}

package/dist/core/compaction/compaction.d.ts

@@ -0,0 +1,65 @@
+/**
+ * Context compaction for long sessions.
+ *
+ * Pure functions for compaction logic. The session manager handles I/O,
+ * and after compaction the session is reloaded.
+ */
+import { type AgentMessage, type CompactionPreparation, type CompactionResult, type CompactionSettings, type ContextUsageEstimate, type CutPointResult, type StreamFn, type ThinkingLevel } from "@greppleai/grip-agent-core";
+import type { AssistantMessage, Context, Model, ProviderHeaders, RetryCallbacks, RetryPolicy, SimpleStreamOptions, Usage } from "@greppleai/grip-ai";
+import type { SessionEntry } from "../session/types.ts";
+export type { CompactionPreparation, CompactionResult, CompactionSettings, ContextUsageEstimate, CutPointResult, } from "@greppleai/grip-agent-core";
+export declare const DEFAULT_COMPACTION_SETTINGS: CompactionSettings;
+/**
+ * Calculate total context tokens from usage.
+ * Uses the native totalTokens field when available, falls back to computing from components.
+ */
+export declare function calculateContextTokens(usage: Usage): number;
+/**
+ * Find the last non-aborted assistant message usage from session entries.
+ */
+export declare function getLastAssistantUsage(entries: SessionEntry[]): Usage | undefined;
+/**
+ * Estimate context tokens from messages, using the last assistant usage when available.
+ * If there are messages after the last usage, estimate their tokens with estimateTokens.
+ */
+export declare function estimateContextTokens(messages: AgentMessage[]): ContextUsageEstimate;
+/**
+ * Check if compaction should trigger based on context usage.
+ */
+export declare function shouldCompact(contextTokens: number, contextWindow: number, settings: CompactionSettings): boolean;
+export declare function estimateTokens(message: AgentMessage): number;
+/** Find the context-visible user-like message that starts the containing turn. */
+export declare function findTurnStartIndex(entries: SessionEntry[], entryIndex: number, startIndex: number): number;
+/**
+ * Find the cut point in session entries that keeps approximately `keepRecentTokens`.
+ *
+ * Algorithm: Walk backwards from newest, accumulating estimated message sizes.
+ * Stop when we've accumulated >= keepRecentTokens. Cut at that point.
+ *
+ * Can cut at user OR assistant messages (never tool results). When cutting at an
+ * assistant message with tool calls, its tool results come after and will be kept.
+ *
+ * Returns CutPointResult with:
+ * - firstKeptEntryIndex: the entry index to start keeping from
+ * - turnStartIndex: if cutting mid-turn, the user message that started that turn
+ * - isSplitTurn: whether we're cutting in the middle of a turn
+ *
+ * Only considers entries between `startIndex` and `endIndex` (exclusive).
+ */
+export declare function findCutPoint(entries: SessionEntry[], startIndex: number, endIndex: number, keepRecentTokens: number): CutPointResult;
+export declare function completeSummarization(model: Model<any>, context: Context, options: SimpleStreamOptions, streamFn?: StreamFn, retry?: RetryPolicy, callbacks?: RetryCallbacks): Promise<AssistantMessage>;
+/**
+ * Generate a summary of the conversation using the LLM.
+ * If previousSummary is provided, uses the update prompt to merge.
+ */
+export declare function generateSummary(currentMessages: AgentMessage[], model: Model<any>, reserveTokens: number, apiKey: string | undefined, headers?: ProviderHeaders, signal?: AbortSignal, customInstructions?: string, previousSummary?: string, thinkingLevel?: ThinkingLevel, streamFn?: StreamFn, retry?: RetryPolicy, callbacks?: RetryCallbacks): Promise<string>;
+export declare function prepareCompaction(pathEntries: SessionEntry[], settings: CompactionSettings): CompactionPreparation | undefined;
+/**
+ * Generate summaries for compaction using prepared data.
+ * Returns CompactionResult - Session adds uuid/parentUuid when saving.
+ *
+ * @param preparation - Pre-calculated preparation from prepareCompaction()
+ * @param customInstructions - Optional custom focus for the summary
+ */
+export declare function compact(preparation: CompactionPreparation, model: Model<any>, apiKey: string | undefined, headers?: ProviderHeaders, customInstructions?: string, signal?: AbortSignal, thinkingLevel?: ThinkingLevel, streamFn?: StreamFn, retry?: RetryPolicy, callbacks?: RetryCallbacks): Promise<CompactionResult>;
+//# sourceMappingURL=compaction.d.ts.map