@antaif3ng/til-work 0.1.0

package/dist/core/agent.js

@@ -0,0 +1,372 @@
+/**
+ * Agent runtime: drives the LLM ↔ tool execution loop.
+ * Dual-loop architecture matching pi-mono:
+ *   Outer loop — processes follow-up messages after agent would stop
+ *   Inner loop — processes tool calls + steering (interrupt) messages
+ */
+import { streamLLM } from "./llm.js";
+import { compact, calculateContextTokens, isContextOverflow, shouldCompact, estimateAllTokens, DEFAULT_COMPACTION_SETTINGS, } from "./compaction.js";
+export class Agent {
+    state;
+    listeners = new Set();
+    abortController;
+    maxTurns;
+    contextWindow;
+    compactionSettings;
+    compactionSummary;
+    getSteeringMessages;
+    getFollowUpMessages;
+    onToolCallRequest;
+    constructor(opts) {
+        this.maxTurns = opts.maxTurns ?? 200;
+        this.contextWindow = opts.contextWindow ?? opts.model.contextWindow ?? 128000;
+        this.compactionSettings = opts.compactionSettings ?? DEFAULT_COMPACTION_SETTINGS;
+        this.getSteeringMessages = opts.getSteeringMessages;
+        this.getFollowUpMessages = opts.getFollowUpMessages;
+        this.onToolCallRequest = opts.onToolCallRequest;
+        this.state = {
+            systemPrompt: opts.systemPrompt,
+            model: opts.model,
+            thinkingLevel: opts.thinkingLevel ?? "off",
+            tools: opts.tools ?? [],
+            messages: [],
+            isStreaming: false,
+        };
+    }
+    get currentState() {
+        return this.state;
+    }
+    subscribe(fn) {
+        this.listeners.add(fn);
+        return () => this.listeners.delete(fn);
+    }
+    setSystemPrompt(prompt) {
+        this.state.systemPrompt = prompt;
+    }
+    setModel(model) {
+        this.state.model = model;
+    }
+    setTools(tools) {
+        this.state.tools = tools;
+    }
+    clearMessages() {
+        this.state.messages = [];
+    }
+    /**
+     * Replace the message history (used when restoring a persisted session).
+     */
+    setMessages(messages) {
+        this.state.messages = [...messages];
+    }
+    abort() {
+        this.abortController?.abort();
+    }
+    /**
+     * Start a new agent loop with a user prompt.
+     */
+    async prompt(input) {
+        if (this.state.isStreaming) {
+            throw new Error("Agent is already processing. Wait for completion.");
+        }
+        const userMsg = {
+            role: "user",
+            content: input,
+            timestamp: Date.now(),
+        };
+        this.state.messages.push(userMsg);
+        await this.runLoop([userMsg]);
+    }
+    /**
+     * Continue the agent loop from the current context without adding a
+     * new message. Used for retry / resume scenarios.
+     */
+    async continue() {
+        if (this.state.isStreaming) {
+            throw new Error("Agent is already processing. Wait for completion.");
+        }
+        if (this.state.messages.length === 0) {
+            throw new Error("Cannot continue: no messages in context.");
+        }
+        const last = this.state.messages[this.state.messages.length - 1];
+        if (last.role === "assistant") {
+            throw new Error("Cannot continue from an assistant message.");
+        }
+        await this.runLoop([]);
+    }
+    // ── Core dual-loop ──
+    async runLoop(newMessages) {
+        this.abortController = new AbortController();
+        this.state.isStreaming = true;
+        this.state.error = undefined;
+        this.emit({ type: "agent_start" });
+        // Emit message events for the prompt(s) that started this loop
+        for (const msg of newMessages) {
+            this.emit({ type: "message_start", message: msg });
+            this.emit({ type: "message_end", message: msg });
+        }
+        try {
+            let turns = 0;
+            let pendingMessages = (await this.getSteeringMessages?.()) ?? [];
+            // Outer loop: continues when follow-up messages arrive after agent would stop
+            outer: while (true) {
+                let hasMoreToolCalls = true;
+                // Inner loop: process tool calls + steering messages
+                while (hasMoreToolCalls || pendingMessages.length > 0) {
+                    if (turns >= this.maxTurns)
+                        break outer;
+                    turns++;
+                    this.emit({ type: "turn_start" });
+                    // Inject any pending (steering / follow-up) messages
+                    if (pendingMessages.length > 0) {
+                        for (const msg of pendingMessages) {
+                            this.emit({ type: "message_start", message: msg });
+                            this.emit({ type: "message_end", message: msg });
+                            this.state.messages.push(msg);
+                        }
+                        pendingMessages = [];
+                    }
+                    // Stream LLM response
+                    const assistantMsg = await this.streamAssistantResponse();
+                    this.state.messages.push(assistantMsg);
+                    this.emit({ type: "message_end", message: assistantMsg });
+                    // Overflow → compact and retry this turn
+                    if (assistantMsg.stopReason === "error" &&
+                        isContextOverflow(assistantMsg, this.contextWindow)) {
+                        // Remove the failed error message
+                        this.state.messages.pop();
+                        const retried = await this.runCompaction("overflow");
+                        if (retried)
+                            continue;
+                        // If compaction can't help, fall through to error handling
+                        this.state.messages.push(assistantMsg);
+                    }
+                    // Abort / error → stop immediately (like pi-mono agent-loop:144-148)
+                    if (assistantMsg.stopReason === "error" ||
+                        assistantMsg.stopReason === "aborted") {
+                        this.emit({ type: "turn_end", message: assistantMsg, toolResults: [] });
+                        this.emit({ type: "agent_end", messages: this.state.messages });
+                        return;
+                    }
+                    // Extract tool calls
+                    const toolCalls = assistantMsg.content.filter((c) => c.type === "toolCall");
+                    hasMoreToolCalls = toolCalls.length > 0;
+                    const toolResults = [];
+                    let steeringAfterTools = null;
+                    if (hasMoreToolCalls) {
+                        const execution = await this.executeToolCalls(toolCalls);
+                        toolResults.push(...execution.toolResults);
+                        steeringAfterTools = execution.steeringMessages ?? null;
+                        for (const result of toolResults) {
+                            this.state.messages.push(result);
+                        }
+                    }
+                    this.emit({ type: "turn_end", message: assistantMsg, toolResults });
+                    // Threshold compaction check after each turn
+                    await this.checkThresholdCompaction(assistantMsg);
+                    // Gather pending messages for the next inner-loop iteration
+                    if (steeringAfterTools && steeringAfterTools.length > 0) {
+                        pendingMessages = steeringAfterTools;
+                    }
+                    else {
+                        pendingMessages =
+                            (await this.getSteeringMessages?.()) ?? [];
+                    }
+                }
+                // Agent would stop here. Check for follow-up messages.
+                const followUp = (await this.getFollowUpMessages?.()) ?? [];
+                if (followUp.length > 0) {
+                    pendingMessages = followUp;
+                    continue;
+                }
+                break;
+            }
+            this.emit({ type: "agent_end", messages: this.state.messages });
+        }
+        catch (err) {
+            if (err.message === "aborted" ||
+                err.name === "AbortError" ||
+                this.abortController?.signal.aborted) {
+                this.state.error = "Request was aborted";
+            }
+            else {
+                this.state.error = err.message || String(err);
+            }
+            const errorMsg = {
+                role: "assistant",
+                content: [{ type: "text", text: `Error: ${this.state.error}` }],
+                stopReason: "error",
+                errorMessage: this.state.error,
+                timestamp: Date.now(),
+            };
+            this.state.messages.push(errorMsg);
+            this.emit({ type: "agent_end", messages: [errorMsg] });
+        }
+        finally {
+            this.state.isStreaming = false;
+            this.abortController = undefined;
+        }
+    }
+    // ── LLM streaming ──
+    async streamAssistantResponse() {
+        const toolDefs = this.state.tools.map((t) => ({
+            name: t.name,
+            description: t.description,
+            parameters: t.parameters,
+        }));
+        return new Promise((resolve, reject) => {
+            const callbacks = {
+                onToken: (text) => {
+                    this.emit({
+                        type: "message_update",
+                        message: {
+                            role: "assistant",
+                            content: [{ type: "text", text }],
+                            timestamp: Date.now(),
+                        },
+                        delta: text,
+                    });
+                },
+                onDone: (msg) => resolve(msg),
+                onError: (err) => reject(err),
+            };
+            streamLLM(this.state.model, this.state.systemPrompt, this.state.messages, toolDefs, callbacks, this.abortController?.signal).catch(reject);
+        });
+    }
+    // ── Tool execution with steering support ──
+    async executeToolCalls(toolCalls) {
+        const results = [];
+        let steeringMessages;
+        for (let i = 0; i < toolCalls.length; i++) {
+            const tc = toolCalls[i];
+            const tool = this.state.tools.find((t) => t.name === tc.name);
+            if (!tool) {
+                const notFound = this.makeToolResult(tc, [{ type: "text", text: `Tool not found: ${tc.name}` }], true);
+                results.push(notFound);
+                this.emit({ type: "tool_execution_start", toolCallId: tc.id, toolName: tc.name, args: tc.arguments });
+                this.emit({ type: "tool_execution_end", toolCallId: tc.id, toolName: tc.name, result: notFound.content, isError: true });
+                continue;
+            }
+            // Permission check before execution
+            if (this.onToolCallRequest) {
+                try {
+                    const perm = await this.onToolCallRequest(tc.name, tc.arguments);
+                    if (perm.block) {
+                        const reason = perm.reason || "Tool execution blocked by permission check";
+                        const errorContent = [{ type: "text", text: reason }];
+                        results.push(this.makeToolResult(tc, errorContent, true));
+                        this.emit({ type: "tool_execution_start", toolCallId: tc.id, toolName: tc.name, args: tc.arguments });
+                        this.emit({ type: "tool_execution_end", toolCallId: tc.id, toolName: tc.name, result: errorContent, isError: true });
+                        continue;
+                    }
+                }
+                catch { /* permission check failed — allow execution */ }
+            }
+            this.emit({ type: "tool_execution_start", toolCallId: tc.id, toolName: tc.name, args: tc.arguments });
+            try {
+                const result = await tool.execute(tc.id, tc.arguments, this.abortController?.signal, (partial) => {
+                    this.emit({
+                        type: "tool_execution_update",
+                        toolCallId: tc.id,
+                        toolName: tc.name,
+                        partialResult: partial,
+                    });
+                });
+                results.push(this.makeToolResult(tc, result.content, false));
+                this.emit({
+                    type: "tool_execution_end",
+                    toolCallId: tc.id,
+                    toolName: tc.name,
+                    result: result.content,
+                    isError: false,
+                });
+            }
+            catch (err) {
+                const errorContent = [{ type: "text", text: err.message || String(err) }];
+                results.push(this.makeToolResult(tc, errorContent, true));
+                this.emit({
+                    type: "tool_execution_end",
+                    toolCallId: tc.id,
+                    toolName: tc.name,
+                    result: errorContent,
+                    isError: true,
+                });
+            }
+            // Check for steering messages after each tool (skip remaining if interrupted)
+            if (this.getSteeringMessages) {
+                const steering = await this.getSteeringMessages();
+                if (steering.length > 0) {
+                    steeringMessages = steering;
+                    // Skip remaining tool calls
+                    for (const skipped of toolCalls.slice(i + 1)) {
+                        results.push(this.skipToolCall(skipped));
+                    }
+                    break;
+                }
+            }
+        }
+        return { toolResults: results, steeringMessages };
+    }
+    skipToolCall(tc) {
+        const content = [{ type: "text", text: "Skipped due to queued user message." }];
+        this.emit({ type: "tool_execution_start", toolCallId: tc.id, toolName: tc.name, args: tc.arguments });
+        this.emit({ type: "tool_execution_end", toolCallId: tc.id, toolName: tc.name, result: content, isError: true });
+        return this.makeToolResult(tc, content, true);
+    }
+    makeToolResult(tc, content, isError) {
+        return {
+            role: "toolResult",
+            toolCallId: tc.id,
+            toolName: tc.name,
+            content,
+            isError,
+            timestamp: Date.now(),
+        };
+    }
+    // ── Compaction ──
+    async checkThresholdCompaction(lastAssistant) {
+        if (!this.compactionSettings.enabled)
+            return;
+        const contextTokens = lastAssistant.usage
+            ? calculateContextTokens(lastAssistant.usage)
+            : estimateAllTokens(this.state.messages);
+        if (shouldCompact(contextTokens, this.contextWindow, this.compactionSettings)) {
+            await this.runCompaction("threshold");
+        }
+    }
+    async runCompaction(reason) {
+        if (this.state.messages.length < 4)
+            return false;
+        this.emit({ type: "compaction_start", reason });
+        try {
+            const result = await compact(this.state.messages, this.state.model, this.compactionSettings, this.compactionSummary);
+            this.compactionSummary = result.summary;
+            // Replace messages: inject summary as a system-like user message, then kept messages
+            const summaryMsg = {
+                role: "user",
+                content: `[Previous conversation summary]\n\n${result.summary}`,
+                timestamp: Date.now(),
+            };
+            this.state.messages = [summaryMsg, ...result.keptMessages];
+            this.emit({
+                type: "compaction_end",
+                tokensBefore: result.tokensBefore,
+                tokensAfter: result.tokensAfter,
+            });
+            return true;
+        }
+        catch {
+            this.emit({
+                type: "compaction_end",
+                tokensBefore: estimateAllTokens(this.state.messages),
+                tokensAfter: estimateAllTokens(this.state.messages),
+            });
+            return false;
+        }
+    }
+    emit(event) {
+        for (const listener of this.listeners) {
+            listener(event);
+        }
+    }
+}
+//# sourceMappingURL=agent.js.map

package/dist/core/agent.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"agent.js","sourceRoot":"","sources":["../../src/core/agent.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,SAAS,EAAwB,MAAM,UAAU,CAAC;AAC3D,OAAO,EACN,OAAO,EACP,sBAAsB,EACtB,iBAAiB,EACjB,aAAa,EACb,iBAAiB,EAEjB,2BAA2B,GAC3B,MAAM,iBAAiB,CAAC;AAwCzB,MAAM,OAAO,KAAK;IACT,KAAK,CAAa;IAClB,SAAS,GAAG,IAAI,GAAG,EAA2B,CAAC;IAC/C,eAAe,CAAmB;IAClC,QAAQ,CAAS;IACjB,aAAa,CAAS;IACtB,kBAAkB,CAAqB;IACvC,iBAAiB,CAAU;IAC3B,mBAAmB,CAA4B;IAC/C,mBAAmB,CAA4B;IAC/C,iBAAiB,CAAiG;IAE1H,YAAY,IAAkB;QAC7B,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,QAAQ,IAAI,GAAG,CAAC;QACrC,IAAI,CAAC,aAAa,GAAG,IAAI,CAAC,aAAa,IAAI,IAAI,CAAC,KAAK,CAAC,aAAa,IAAI,MAAM,CAAC;QAC9E,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,kBAAkB,IAAI,2BAA2B,CAAC;QACjF,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,CAAC;QACpD,IAAI,CAAC,mBAAmB,GAAG,IAAI,CAAC,mBAAmB,CAAC;QACpD,IAAI,CAAC,iBAAiB,GAAG,IAAI,CAAC,iBAAiB,CAAC;QAChD,IAAI,CAAC,KAAK,GAAG;YACZ,YAAY,EAAE,IAAI,CAAC,YAAY;YAC/B,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,aAAa,EAAE,IAAI,CAAC,aAAa,IAAI,KAAK;YAC1C,KAAK,EAAE,IAAI,CAAC,KAAK,IAAI,EAAE;YACvB,QAAQ,EAAE,EAAE;YACZ,WAAW,EAAE,KAAK;SAClB,CAAC;IACH,CAAC;IAED,IAAI,YAAY;QACf,OAAO,IAAI,CAAC,KAAK,CAAC;IACnB,CAAC;IAED,SAAS,CAAC,EAA2B;QACpC,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACvB,OAAO,GAAG,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;IACxC,CAAC;IAED,eAAe,CAAC,MAAc;QAC7B,IAAI,CAAC,KAAK,CAAC,YAAY,GAAG,MAAM,CAAC;IAClC,CAAC;IAED,QAAQ,CAAC,KAAkB;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC;IAC1B,CAAC;IAED,QAAQ,CAAC,KAAkB;QAC1B,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC;IAC1B,CAAC;IAED,aAAa;QACZ,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC;IAC1B,CAAC;IAED;;OAEG;IACH,WAAW,CAAC,QAAmB;QAC9B,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;IACrC,CAAC;IAED,KAAK;QACJ,IAAI,CAAC,eAAe,EAAE,KAAK,EAAE,CAAC;IAC/B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,MAAM,CAAC,KAAa;QACzB,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACtE,CAAC;QAED,MAAM,OAAO,GAAY;YACxB,IAAI,EAAE,MAAM;YACZ,OAAO,EAAE,KAAK;YACd,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB,CAAC;QAEF,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,MAAM,IAAI,CAAC,OAAO,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC;IAC/B,CAAC;IAED;;;OAGG;IACH,KAAK,CAAC,QAAQ;QACb,IAAI,IAAI,CAAC,KAAK,CAAC,WAAW,EAAE,CAAC;YAC5B,MAAM,IAAI,KAAK,CAAC,mDAAmD,CAAC,CAAC;QACtE,CAAC;QACD,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACtC,MAAM,IAAI,KAAK,CAAC,0CAA0C,CAAC,CAAC;QAC7D,CAAC;QACD,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QACjE,IAAI,IAAI,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;YAC/B,MAAM,IAAI,KAAK,CAAC,4CAA4C,CAAC,CAAC;QAC/D,CAAC;QACD,MAAM,IAAI,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;IACxB,CAAC;IAED,uBAAuB;IAEf,KAAK,CAAC,OAAO,CAAC,WAAsB;QAC3C,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,EAAE,CAAC;QAC7C,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,IAAI,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,SAAS,CAAC;QAE7B,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC,CAAC;QAEnC,+DAA+D;QAC/D,KAAK,MAAM,GAAG,IAAI,WAAW,EAAE,CAAC;YAC/B,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;YACnD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;QAClD,CAAC;QAED,IAAI,CAAC;YACJ,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,eAAe,GAClB,CAAC,MAAM,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;YAE5C,8EAA8E;YAC9E,KAAK,EAAE,OAAO,IAAI,EAAE,CAAC;gBACpB,IAAI,gBAAgB,GAAG,IAAI,CAAC;gBAE5B,qDAAqD;gBACrD,OAAO,gBAAgB,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACvD,IAAI,KAAK,IAAI,IAAI,CAAC,QAAQ;wBAAE,MAAM,KAAK,CAAC;oBACxC,KAAK,EAAE,CAAC;oBAER,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC,CAAC;oBAElC,qDAAqD;oBACrD,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBAChC,KAAK,MAAM,GAAG,IAAI,eAAe,EAAE,CAAC;4BACnC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;4BACnD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,GAAG,EAAE,CAAC,CAAC;4BACjD,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;wBAC/B,CAAC;wBACD,eAAe,GAAG,EAAE,CAAC;oBACtB,CAAC;oBAED,sBAAsB;oBACtB,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;oBAC1D,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBACvC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,aAAa,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;oBAE3D,yCAAyC;oBACzC,IACC,YAAY,CAAC,UAAU,KAAK,OAAO;wBACnC,iBAAiB,CAAC,YAAY,EAAE,IAAI,CAAC,aAAa,CAAC,EAClD,CAAC;wBACF,kCAAkC;wBAClC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC;wBAC1B,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;wBACrD,IAAI,OAAO;4BAAE,SAAS;wBACtB,2DAA2D;wBAC3D,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBACxC,CAAC;oBAED,qEAAqE;oBACrE,IACC,YAAY,CAAC,UAAU,KAAK,OAAO;wBACnC,YAAY,CAAC,UAAU,KAAK,SAAS,EACpC,CAAC;wBACF,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE,CAAC,CAAC;wBACxE,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;wBAChE,OAAO;oBACR,CAAC;oBAEA,qBAAqB;oBACrB,MAAM,SAAS,GAAG,YAAY,CAAC,OAAO,CAAC,MAAM,CAC5C,CAAC,CAAC,EAAiB,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAC3C,CAAC;oBACF,gBAAgB,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;oBAExC,MAAM,WAAW,GAAwB,EAAE,CAAC;oBAC5C,IAAI,kBAAkB,GAAqB,IAAI,CAAC;oBAEhD,IAAI,gBAAgB,EAAE,CAAC;wBACtB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,SAAS,CAAC,CAAC;wBACzD,WAAW,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,WAAW,CAAC,CAAC;wBAC3C,kBAAkB,GAAG,SAAS,CAAC,gBAAgB,IAAI,IAAI,CAAC;wBAExD,KAAK,MAAM,MAAM,IAAI,WAAW,EAAE,CAAC;4BAClC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;wBAClC,CAAC;oBACF,CAAC;oBAEF,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC;oBAEpE,6CAA6C;oBAC7C,MAAM,IAAI,CAAC,wBAAwB,CAAC,YAAY,CAAC,CAAC;oBAElD,4DAA4D;oBAC5D,IAAI,kBAAkB,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;wBACxD,eAAe,GAAG,kBAAkB,CAAC;oBACtC,CAAC;yBAAM,CAAC;wBACP,eAAe;4BACd,CAAC,MAAM,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;oBAC7C,CAAC;gBACF,CAAC;gBAED,uDAAuD;gBACvD,MAAM,QAAQ,GACb,CAAC,MAAM,IAAI,CAAC,mBAAmB,EAAE,EAAE,CAAC,IAAI,EAAE,CAAC;gBAC5C,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,eAAe,GAAG,QAAQ,CAAC;oBAC3B,SAAS;gBACV,CAAC;gBAED,MAAM;YACP,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;QACjE,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YACnB,IACC,GAAG,CAAC,OAAO,KAAK,SAAS;gBACzB,GAAG,CAAC,IAAI,KAAK,YAAY;gBACzB,IAAI,CAAC,eAAe,EAAE,MAAM,CAAC,OAAO,EACnC,CAAC;gBACF,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,qBAAqB,CAAC;YAC1C,CAAC;iBAAM,CAAC;gBACP,IAAI,CAAC,KAAK,CAAC,KAAK,GAAG,GAAG,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC;YAC/C,CAAC;YAED,MAAM,QAAQ,GAAqB;gBAClC,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,EAAE,CAAC;gBAC/D,UAAU,EAAE,OAAO;gBACnB,YAAY,EAAE,IAAI,CAAC,KAAK,CAAC,KAAK;gBAC9B,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACrB,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACnC,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACxD,CAAC;gBAAS,CAAC;YACV,IAAI,CAAC,KAAK,CAAC,WAAW,GAAG,KAAK,CAAC;YAC/B,IAAI,CAAC,eAAe,GAAG,SAAS,CAAC;QAClC,CAAC;IACF,CAAC;IAED,sBAAsB;IAEd,KAAK,CAAC,uBAAuB;QACpC,MAAM,QAAQ,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;YAC7C,IAAI,EAAE,CAAC,CAAC,IAAI;YACZ,WAAW,EAAE,CAAC,CAAC,WAAW;YAC1B,UAAU,EAAE,CAAC,CAAC,UAAU;SACxB,CAAC,CAAC,CAAC;QAEJ,OAAO,IAAI,OAAO,CAAmB,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;YACxD,MAAM,SAAS,GAAoB;gBAClC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;oBACjB,IAAI,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,gBAAgB;wBACtB,OAAO,EAAE;4BACR,IAAI,EAAE,WAAW;4BACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;4BACjC,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;yBACrB;wBACD,KAAK,EAAE,IAAI;qBACX,CAAC,CAAC;gBACJ,CAAC;gBACD,MAAM,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC;gBAC7B,OAAO,EAAE,CAAC,GAAG,EAAE,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC;aAC7B,CAAC;YAEF,SAAS,CACR,IAAI,CAAC,KAAK,CAAC,KAAK,EAChB,IAAI,CAAC,KAAK,CAAC,YAAY,EACvB,IAAI,CAAC,KAAK,CAAC,QAAQ,EACnB,QAAQ,EACR,SAAS,EACT,IAAI,CAAC,eAAe,EAAE,MAAM,CAC5B,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;QACjB,CAAC,CAAC,CAAC;IACJ,CAAC;IAED,6CAA6C;IAErC,KAAK,CAAC,gBAAgB,CAC7B,SAAqB;QAErB,MAAM,OAAO,GAAwB,EAAE,CAAC;QACxC,IAAI,gBAAuC,CAAC;QAE5C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC3C,MAAM,EAAE,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC;YACxB,MAAM,IAAI,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,EAAE,CAAC,IAAI,CAAC,CAAC;YAE9D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACX,MAAM,QAAQ,GAAG,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,mBAAmB,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC;gBACvG,OAAO,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;gBACvB,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;gBACtG,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,CAAC,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;gBACzH,SAAS;YACV,CAAC;YAED,oCAAoC;YACpC,IAAI,IAAI,CAAC,iBAAiB,EAAE,CAAC;gBAC5B,IAAI,CAAC;oBACJ,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;oBACjE,IAAI,IAAI,CAAC,KAAK,EAAE,CAAC;wBAChB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,IAAI,4CAA4C,CAAC;wBAC3E,MAAM,YAAY,GAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC;wBACrE,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC;wBAC1D,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;wBACtG,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;wBACrH,SAAS;oBACV,CAAC;gBACF,CAAC;gBAAC,MAAM,CAAC,CAAC,+CAA+C,CAAC,CAAC;YAC5D,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;YAEtG,IAAI,CAAC;gBACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAChC,EAAE,CAAC,EAAE,EACL,EAAE,CAAC,SAAS,EACZ,IAAI,CAAC,eAAe,EAAE,MAAM,EAC5B,CAAC,OAAO,EAAE,EAAE;oBACX,IAAI,CAAC,IAAI,CAAC;wBACT,IAAI,EAAE,uBAAuB;wBAC7B,UAAU,EAAE,EAAE,CAAC,EAAE;wBACjB,QAAQ,EAAE,EAAE,CAAC,IAAI;wBACjB,aAAa,EAAE,OAAO;qBACtB,CAAC,CAAC;gBACJ,CAAC,CACD,CAAC;gBAEF,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,MAAM,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;gBAC7D,IAAI,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,oBAAoB;oBAC1B,UAAU,EAAE,EAAE,CAAC,EAAE;oBACjB,QAAQ,EAAE,EAAE,CAAC,IAAI;oBACjB,MAAM,EAAE,MAAM,CAAC,OAAO;oBACtB,OAAO,EAAE,KAAK;iBACd,CAAC,CAAC;YACJ,CAAC;YAAC,OAAO,GAAQ,EAAE,CAAC;gBACnB,MAAM,YAAY,GAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,CAAC,OAAO,IAAI,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;gBACzF,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC,CAAC;gBAC1D,IAAI,CAAC,IAAI,CAAC;oBACT,IAAI,EAAE,oBAAoB;oBAC1B,UAAU,EAAE,EAAE,CAAC,EAAE;oBACjB,QAAQ,EAAE,EAAE,CAAC,IAAI;oBACjB,MAAM,EAAE,YAAY;oBACpB,OAAO,EAAE,IAAI;iBACb,CAAC,CAAC;YACJ,CAAC;YAED,8EAA8E;YAC9E,IAAI,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAC9B,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,mBAAmB,EAAE,CAAC;gBAClD,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;oBACzB,gBAAgB,GAAG,QAAQ,CAAC;oBAC5B,4BAA4B;oBAC5B,KAAK,MAAM,OAAO,IAAI,SAAS,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;wBAC9C,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC;oBAC1C,CAAC;oBACD,MAAM;gBACP,CAAC;YACF,CAAC;QACF,CAAC;QAED,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC;IACnD,CAAC;IAEO,YAAY,CAAC,EAAY;QAChC,MAAM,OAAO,GAAkB,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,qCAAqC,EAAE,CAAC,CAAC;QAC/F,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,sBAAsB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,IAAI,EAAE,EAAE,CAAC,SAAS,EAAE,CAAC,CAAC;QACtG,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,UAAU,EAAE,EAAE,CAAC,EAAE,EAAE,QAAQ,EAAE,EAAE,CAAC,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC,CAAC;QAChH,OAAO,IAAI,CAAC,cAAc,CAAC,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;IAC/C,CAAC;IAEO,cAAc,CACrB,EAAY,EACZ,OAA4E,EAC5E,OAAgB;QAEhB,OAAO;YACN,IAAI,EAAE,YAAY;YAClB,UAAU,EAAE,EAAE,CAAC,EAAE;YACjB,QAAQ,EAAE,EAAE,CAAC,IAAI;YACjB,OAAO;YACP,OAAO;YACP,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACrB,CAAC;IACH,CAAC;IAED,mBAAmB;IAEX,KAAK,CAAC,wBAAwB,CAAC,aAA+B;QACrE,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,OAAO;YAAE,OAAO;QAE7C,MAAM,aAAa,GAAG,aAAa,CAAC,KAAK;YACxC,CAAC,CAAC,sBAAsB,CAAC,aAAa,CAAC,KAAK,CAAC;YAC7C,CAAC,CAAC,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QAE1C,IAAI,aAAa,CAAC,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC;YAC/E,MAAM,IAAI,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;QACvC,CAAC;IACF,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,MAAgC;QAC3D,IAAI,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC;YAAE,OAAO,KAAK,CAAC;QAEjD,IAAI,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,MAAM,EAAE,CAAC,CAAC;QAEhD,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,OAAO,CAC3B,IAAI,CAAC,KAAK,CAAC,QAAQ,EACnB,IAAI,CAAC,KAAK,CAAC,KAAK,EAChB,IAAI,CAAC,kBAAkB,EACvB,IAAI,CAAC,iBAAiB,CACtB,CAAC;YAEF,IAAI,CAAC,iBAAiB,GAAG,MAAM,CAAC,OAAO,CAAC;YAExC,qFAAqF;YACrF,MAAM,UAAU,GAAY;gBAC3B,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,sCAAsC,MAAM,CAAC,OAAO,EAAE;gBAC/D,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACrB,CAAC;YACF,IAAI,CAAC,KAAK,CAAC,QAAQ,GAAG,CAAC,UAAU,EAAE,GAAG,MAAM,CAAC,YAAY,CAAC,CAAC;YAE3D,IAAI,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,gBAAgB;gBACtB,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,WAAW,EAAE,MAAM,CAAC,WAAW;aAC/B,CAAC,CAAC;YAEH,OAAO,IAAI,CAAC;QACb,CAAC;QAAC,MAAM,CAAC;YACR,IAAI,CAAC,IAAI,CAAC;gBACT,IAAI,EAAE,gBAAgB;gBACtB,YAAY,EAAE,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;gBACpD,WAAW,EAAE,iBAAiB,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC;aACnD,CAAC,CAAC;YACH,OAAO,KAAK,CAAC;QACd,CAAC;IACF,CAAC;IAEO,IAAI,CAAC,KAAiB;QAC7B,KAAK,MAAM,QAAQ,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACvC,QAAQ,CAAC,KAAK,CAAC,CAAC;QACjB,CAAC;IACF,CAAC;CACD"}

package/dist/core/compaction.d.ts

@@ -0,0 +1,40 @@
+/**
+ * Context compaction engine.
+ * When the conversation grows beyond the model's context window, older messages
+ * are summarised by the LLM and replaced with a compact summary.
+ *
+ * Architecture mirrors pi-mono packages/coding-agent/src/core/compaction/.
+ */
+import type { AssistantMessage, Message, ModelConfig, Usage } from "./types.js";
+export interface CompactionSettings {
+    enabled: boolean;
+    /** Tokens reserved for the next response (threshold gap) */
+    reserveTokens: number;
+    /** Minimum tokens to keep from the recent end of the conversation */
+    keepRecentTokens: number;
+}
+export declare const DEFAULT_COMPACTION_SETTINGS: CompactionSettings;
+export declare function estimateTokens(message: Message): number;
+export declare function estimateAllTokens(messages: Message[]): number;
+export declare function calculateContextTokens(usage: Usage): number;
+export declare function shouldCompact(contextTokens: number, contextWindow: number, settings: CompactionSettings): boolean;
+export declare function isContextOverflow(message: AssistantMessage, contextWindow?: number): boolean;
+export interface CutResult {
+    /** Index of the first message to keep */
+    keepFromIndex: number;
+    /** Messages to be summarised (before keepFromIndex) */
+    summariseMessages: Message[];
+    /** Messages to keep (from keepFromIndex onwards) */
+    keptMessages: Message[];
+}
+export declare function findCutPoint(messages: Message[], keepRecentTokens: number): CutResult;
+export declare function generateSummary(messages: Message[], model: ModelConfig, previousSummary?: string): Promise<string>;
+export interface CompactionResult {
+    summary: string;
+    keptMessages: Message[];
+    tokensBefore: number;
+    tokensAfter: number;
+    previousSummary?: string;
+}
+export declare function compact(messages: Message[], model: ModelConfig, settings: CompactionSettings, previousSummary?: string): Promise<CompactionResult>;
+//# sourceMappingURL=compaction.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"compaction.d.ts","sourceRoot":"","sources":["../../src/core/compaction.ts"],"names":[],"mappings":"AAAA;;;;;;GAMG;AAGH,OAAO,KAAK,EAAE,gBAAgB,EAAE,OAAO,EAAE,WAAW,EAAe,KAAK,EAAE,MAAM,YAAY,CAAC;AAI7F,MAAM,WAAW,kBAAkB;IAClC,OAAO,EAAE,OAAO,CAAC;IACjB,4DAA4D;IAC5D,aAAa,EAAE,MAAM,CAAC;IACtB,qEAAqE;IACrE,gBAAgB,EAAE,MAAM,CAAC;CACzB;AAED,eAAO,MAAM,2BAA2B,EAAE,kBAIzC,CAAC;AAIF,wBAAgB,cAAc,CAAC,OAAO,EAAE,OAAO,GAAG,MAAM,CAmCvD;AAED,wBAAgB,iBAAiB,CAAC,QAAQ,EAAE,OAAO,EAAE,GAAG,MAAM,CAE7D;AAID,wBAAgB,sBAAsB,CAAC,KAAK,EAAE,KAAK,GAAG,MAAM,CAE3D;AAID,wBAAgB,aAAa,CAC5B,aAAa,EAAE,MAAM,EACrB,aAAa,EAAE,MAAM,EACrB,QAAQ,EAAE,kBAAkB,GAC1B,OAAO,CAGT;AAuBD,wBAAgB,iBAAiB,CAChC,OAAO,EAAE,gBAAgB,EACzB,aAAa,CAAC,EAAE,MAAM,GACpB,OAAO,CAgBT;AAID,MAAM,WAAW,SAAS;IACzB,yCAAyC;IACzC,aAAa,EAAE,MAAM,CAAC;IACtB,uDAAuD;IACvD,iBAAiB,EAAE,OAAO,EAAE,CAAC;IAC7B,oDAAoD;IACpD,YAAY,EAAE,OAAO,EAAE,CAAC;CACxB;AAED,wBAAgB,YAAY,CAC3B,QAAQ,EAAE,OAAO,EAAE,EACnB,gBAAgB,EAAE,MAAM,GACtB,SAAS,CA2CX;AAsED,wBAAsB,eAAe,CACpC,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,WAAW,EAClB,eAAe,CAAC,EAAE,MAAM,GACtB,OAAO,CAAC,MAAM,CAAC,CAWjB;AAID,MAAM,WAAW,gBAAgB;IAChC,OAAO,EAAE,MAAM,CAAC;IAChB,YAAY,EAAE,OAAO,EAAE,CAAC;IACxB,YAAY,EAAE,MAAM,CAAC;IACrB,WAAW,EAAE,MAAM,CAAC;IACpB,eAAe,CAAC,EAAE,MAAM,CAAC;CACzB;AAED,wBAAsB,OAAO,CAC5B,QAAQ,EAAE,OAAO,EAAE,EACnB,KAAK,EAAE,WAAW,EAClB,QAAQ,EAAE,kBAAkB,EAC5B,eAAe,CAAC,EAAE,MAAM,GACtB,OAAO,CAAC,gBAAgB,CAAC,CAa3B"}

package/dist/core/compaction.js

@@ -0,0 +1,228 @@
+/**
+ * Context compaction engine.
+ * When the conversation grows beyond the model's context window, older messages
+ * are summarised by the LLM and replaced with a compact summary.
+ *
+ * Architecture mirrors pi-mono packages/coding-agent/src/core/compaction/.
+ */
+import { completeLLM } from "./llm.js";
+export const DEFAULT_COMPACTION_SETTINGS = {
+    enabled: true,
+    reserveTokens: 16384,
+    keepRecentTokens: 20000,
+};
+// ── Token estimation (chars / 4 heuristic, same as pi-mono) ──
+export function estimateTokens(message) {
+    let chars = 0;
+    switch (message.role) {
+        case "user": {
+            if (typeof message.content === "string") {
+                chars = message.content.length;
+            }
+            else {
+                for (const c of message.content) {
+                    if (c.type === "text")
+                        chars += c.text.length;
+                    else
+                        chars += 4800; // image placeholder
+                }
+            }
+            break;
+        }
+        case "assistant": {
+            for (const c of message.content) {
+                if (c.type === "text")
+                    chars += c.text.length;
+                else if (c.type === "thinking")
+                    chars += c.thinking.length;
+                else if (c.type === "toolCall") {
+                    chars += c.name.length + JSON.stringify(c.arguments).length;
+                }
+            }
+            break;
+        }
+        case "toolResult": {
+            for (const c of message.content) {
+                if (c.type === "text")
+                    chars += c.text.length;
+                else
+                    chars += 4800;
+            }
+            break;
+        }
+    }
+    return Math.ceil(chars / 4);
+}
+export function estimateAllTokens(messages) {
+    return messages.reduce((sum, m) => sum + estimateTokens(m), 0);
+}
+// ── Context token calculation from real usage data ──
+export function calculateContextTokens(usage) {
+    return usage.totalTokens || (usage.input + usage.output + usage.cacheRead + usage.cacheWrite);
+}
+// ── Threshold check ──
+export function shouldCompact(contextTokens, contextWindow, settings) {
+    if (!settings.enabled)
+        return false;
+    return contextTokens > contextWindow - settings.reserveTokens;
+}
+// ── Overflow detection ──
+const OVERFLOW_PATTERNS = [
+    /prompt is too long/i,
+    /maximum context length/i,
+    /context.length.*exceeded/i,
+    /token limit/i,
+    /context window/i,
+    /too many tokens/i,
+    /input.*too long/i,
+    /request too large/i,
+    /max_tokens.*exceeded/i,
+    /content.*length.*limit/i,
+    /exceeds.*model.*maximum/i,
+    /reduce.*prompt/i,
+    /Please reduce/i,
+    /context_length_exceeded/i,
+    // Cerebras-style: 400/413 with no body
+    /^4(00|13)\s*(status code)?\s*\(no body\)/i,
+];
+export function isContextOverflow(message, contextWindow) {
+    if (message.stopReason === "error" && message.errorMessage) {
+        if (OVERFLOW_PATTERNS.some((p) => p.test(message.errorMessage))) {
+            return true;
+        }
+    }
+    // Silent overflow: successful response but usage exceeds context window
+    if (contextWindow && message.stopReason !== "error" && message.usage) {
+        const inputTokens = message.usage.input + message.usage.cacheRead;
+        if (inputTokens > contextWindow) {
+            return true;
+        }
+    }
+    return false;
+}
+export function findCutPoint(messages, keepRecentTokens) {
+    let accumulatedTokens = 0;
+    let cutIndex = 0;
+    // Walk backwards accumulating tokens
+    for (let i = messages.length - 1; i >= 0; i--) {
+        accumulatedTokens += estimateTokens(messages[i]);
+        if (accumulatedTokens >= keepRecentTokens) {
+            cutIndex = i;
+            break;
+        }
+    }
+    // Don't cut on a toolResult — move forward to include the full turn
+    while (cutIndex < messages.length && messages[cutIndex].role === "toolResult") {
+        cutIndex++;
+    }
+    // Don't cut inside an assistant+toolResult pair — if we land on an assistant
+    // message that has tool calls, include its tool results too
+    if (cutIndex > 0 && cutIndex < messages.length) {
+        const prev = messages[cutIndex - 1];
+        if (prev.role === "assistant") {
+            const hasToolCalls = prev.content.some((c) => c.type === "toolCall");
+            if (hasToolCalls) {
+                while (cutIndex < messages.length && messages[cutIndex].role === "toolResult") {
+                    cutIndex++;
+                }
+            }
+        }
+    }
+    // Ensure we're keeping at least 1 message
+    if (cutIndex >= messages.length)
+        cutIndex = Math.max(0, messages.length - 1);
+    // Ensure we're actually cutting something
+    if (cutIndex <= 0)
+        cutIndex = Math.min(1, messages.length);
+    return {
+        keepFromIndex: cutIndex,
+        summariseMessages: messages.slice(0, cutIndex),
+        keptMessages: messages.slice(cutIndex),
+    };
+}
+// ── Summary generation ──
+const SUMMARIZATION_SYSTEM_PROMPT = `You are a conversation summariser. Given a conversation, produce a structured summary that captures all important context needed for the AI assistant to continue effectively.
+
+Use this format:
+
+## Goal
+Brief description of what the user is trying to accomplish.
+
+## Progress
+- Done: what has been completed
+- In Progress: what is currently being worked on
+- Blocked: any blockers or issues
+
+## Key Decisions
+Important decisions or preferences expressed during the conversation.
+
+## Next Steps
+What should happen next based on the conversation.
+
+## Critical Context
+Any file paths, variable names, error messages, or technical details that must not be lost.`;
+const SUMMARIZATION_PROMPT = `Please summarise the following conversation. Preserve all technical details, file paths, decisions, and context needed to continue the task.
+
+<conversation>
+{CONVERSATION}
+</conversation>`;
+const UPDATE_SUMMARIZATION_PROMPT = `Please update the following summary with the new conversation content. Merge and deduplicate information while preserving all important context.
+
+<previous-summary>
+{PREVIOUS_SUMMARY}
+</previous-summary>
+
+<new-conversation>
+{CONVERSATION}
+</new-conversation>`;
+function serializeMessages(messages) {
+    const parts = [];
+    for (const msg of messages) {
+        if (msg.role === "user") {
+            const text = typeof msg.content === "string"
+                ? msg.content
+                : msg.content.filter((c) => c.type === "text").map((c) => c.text).join("\n");
+            parts.push(`[User]: ${text}`);
+        }
+        else if (msg.role === "assistant") {
+            const texts = [];
+            for (const c of msg.content) {
+                if (c.type === "text")
+                    texts.push(c.text);
+                else if (c.type === "toolCall")
+                    texts.push(`[Tool Call: ${c.name}(${JSON.stringify(c.arguments).slice(0, 200)})]`);
+            }
+            parts.push(`[Assistant]: ${texts.join("\n")}`);
+        }
+        else if (msg.role === "toolResult") {
+            const text = msg.content
+                .filter((c) => c.type === "text")
+                .map((c) => c.text)
+                .join("\n")
+                .slice(0, 500);
+            parts.push(`[Tool Result (${msg.toolName})]: ${text}`);
+        }
+    }
+    return parts.join("\n\n");
+}
+export async function generateSummary(messages, model, previousSummary) {
+    const conversation = serializeMessages(messages);
+    const maxTokens = Math.floor(0.8 * (DEFAULT_COMPACTION_SETTINGS.reserveTokens));
+    const prompt = previousSummary
+        ? UPDATE_SUMMARIZATION_PROMPT
+            .replace("{PREVIOUS_SUMMARY}", previousSummary)
+            .replace("{CONVERSATION}", conversation)
+        : SUMMARIZATION_PROMPT.replace("{CONVERSATION}", conversation);
+    return completeLLM(model, SUMMARIZATION_SYSTEM_PROMPT, prompt, maxTokens);
+}
+export async function compact(messages, model, settings, previousSummary) {
+    const tokensBefore = estimateAllTokens(messages);
+    const { summariseMessages, keptMessages } = findCutPoint(messages, settings.keepRecentTokens);
+    if (summariseMessages.length === 0) {
+        return { summary: previousSummary || "", keptMessages: messages, tokensBefore, tokensAfter: tokensBefore };
+    }
+    const summary = await generateSummary(summariseMessages, model, previousSummary);
+    const tokensAfter = estimateAllTokens(keptMessages) + Math.ceil(summary.length / 4);
+    return { summary, keptMessages, tokensBefore, tokensAfter, previousSummary };
+}
+//# sourceMappingURL=compaction.js.map