kagent-ts 0.1.3 → 0.1.5

package/dist/compression/progressive-compressor.js

@@ -0,0 +1,367 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ProgressiveCompressor = void 0;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const types_1 = require("../messages/types");
+const token_counter_1 = require("../utils/token-counter");
+const logger_1 = require("../logging/logger");
+// ─── Configuration ──────────────────────────────────────────────────────────
+const TOOL_TOTAL_BYTE_LIMIT = 200 * 1024; // Step 1: 200 KB
+const KEEP_BYTES = 2048; // Step 1: keep 2 KB
+const TRUNCATED_DIR = ".kagent-context"; // Step 1: save full output here
+/** Tools whose results are "read-type" — can be re-run to get the same data. */
+const READ_TOOLS = new Set([
+    "read_file",
+    "grep_search",
+    "glob_search",
+]);
+// ─── Main ───────────────────────────────────────────────────────────────────
+/**
+ * Progressive 4-step compression.
+ *
+ * Each step is applied in order. After each step the token count is
+ * re-checked; if the context now fits comfortably, later steps are skipped.
+ */
+class ProgressiveCompressor {
+    config;
+    logger;
+    constructor(config, logger) {
+        this.config = config;
+        this.logger = logger ?? new logger_1.ConsoleLogger();
+    }
+    /**
+     * Run progressive compression.
+     *
+     * @param messages       The full message list (without system message).
+     * @param systemMessage  The system message (preserved).
+     * @param llm            Optional LLM provider for Step 4 summarization.
+     *                       If omitted, Step 4 is skipped.
+     * @returns Compressed message list + removal count.
+     */
+    async compress(messages, systemMessage, llm, model) {
+        const originalCount = messages.length;
+        let result = messages;
+        let applied = false;
+        const t = this.config.compressionThreshold;
+        const triggerToken = t < 1
+            ? Math.round(this.config.maxTokens * (1 - t))
+            : this.config.maxTokens - t;
+        // ── Step 1: Truncate large tool results ────────────────────────────
+        const after1 = this.step1TruncateToolResults(result);
+        if (after1 !== result) {
+            applied = true;
+            result = after1;
+        }
+        if (this.tokenCount(result, systemMessage, model) < triggerToken) {
+            return this.finalize(result, originalCount, applied);
+        }
+        // ── Step 2: Drop old turns beyond keepTurns ────────────────────────
+        const turnStartsForStep2 = computeTurnStarts(result);
+        const after2 = this.step2DropOldTurns(result, turnStartsForStep2);
+        if (after2 !== result) {
+            applied = true;
+            result = after2;
+        }
+        if (this.tokenCount(result, systemMessage, model) < triggerToken) {
+            return this.finalize(result, originalCount, applied);
+        }
+        // ── Step 3: Drop stale tool results ────────────────────────────────
+        const after3 = this.step3DropStaleToolResults(result);
+        if (after3 !== result) {
+            applied = true;
+            result = after3;
+        }
+        if (this.tokenCount(result, systemMessage, model) < triggerToken) {
+            return this.finalize(result, originalCount, applied);
+        }
+        // ── Step 4: LLM summarization ──────────────────────────────────────
+        if (llm) {
+            try {
+                const turnStartsForStep4 = computeTurnStarts(result);
+                const after4 = await this.step4LlmSummarize(result, systemMessage, llm, turnStartsForStep4);
+                if (after4 !== result) {
+                    applied = true;
+                    result = after4;
+                }
+            }
+            catch (err) {
+                this.logger.warn("Compression", `Step 4 (LLM summarization) failed: ` +
+                    `${err instanceof Error ? err.message : String(err)}. ` +
+                    `Falling back to truncation.`);
+            }
+        }
+        return this.finalize(result, originalCount, applied);
+    }
+    finalize(messages, originalCount, applied) {
+        return {
+            messages,
+            removedCount: applied ? Math.max(0, originalCount - messages.length) : 0,
+            applied,
+        };
+    }
+    // ─── Token Counting ───────────────────────────────────────────────────────
+    tokenCount(messages, sys, model) {
+        let total = 0;
+        if (sys)
+            total += 3 + (0, token_counter_1.countTokens)(sys.content, model);
+        for (const m of messages) {
+            total += 3 + (0, token_counter_1.countTokens)(m.content, model);
+        }
+        return total;
+    }
+    // ─── Step 1: Truncate large tool results ──────────────────────────────────
+    /**
+     * If total tool-result bytes exceed 200 KB, truncate the largest
+     * results (keep 2 KB + save to disk). Returns a new array if modified.
+     */
+    step1TruncateToolResults(messages) {
+        // Collect tool messages with their byte sizes
+        const toolIndices = [];
+        let totalBytes = 0;
+        for (let i = 0; i < messages.length; i++) {
+            if (messages[i].role === types_1.Role.Tool) {
+                const bytes = Buffer.byteLength(messages[i].content, "utf-8");
+                totalBytes += bytes;
+                toolIndices.push({ idx: i, bytes });
+            }
+        }
+        if (totalBytes <= TOOL_TOTAL_BYTE_LIMIT)
+            return messages;
+        // Sort largest first, truncate until under limit
+        toolIndices.sort((a, b) => b.bytes - a.bytes);
+        const result = [...messages];
+        let excess = totalBytes - TOOL_TOTAL_BYTE_LIMIT;
+        fs.mkdirSync(TRUNCATED_DIR, { recursive: true });
+        for (const { idx, bytes } of toolIndices) {
+            if (excess <= 0)
+                break;
+            // Keep KEEP_BYTES, save rest to file
+            const content = result[idx].content;
+            const truncatedBytes = bytes - KEEP_BYTES;
+            if (truncatedBytes <= 0)
+                continue;
+            const now = Date.now();
+            const hash = simpleHash(content.slice(0, 128) + now);
+            const name = result[idx].name ?? "unknown_tool";
+            const filename = `ctx_trunc_${sanitize(name)}_${now}_${hash}.txt`;
+            const filePath = path.join(TRUNCATED_DIR, filename);
+            fs.writeFileSync(filePath, content, "utf-8");
+            result[idx] = {
+                ...result[idx],
+                content: content.slice(0, KEEP_BYTES) +
+                    `\n\n---\n[Context truncated: ${(bytes / 1024).toFixed(1)} KB → ${(KEEP_BYTES / 1024).toFixed(1)} KB. ` +
+                    `Full output saved to: ${filePath}]`,
+            };
+            excess -= truncatedBytes;
+        }
+        return result;
+    }
+    // ─── Step 2: Drop old turns ───────────────────────────────────────────────
+    /**
+     * Remove messages older than `keepTurns` conversation turns.
+     * A "turn" is a user message + all subsequent assistant/tool messages
+     * until the next user message.
+     */
+    step2DropOldTurns(messages, turnStarts) {
+        if (turnStarts.length <= this.config.keepTurns)
+            return messages;
+        const cutoffIdx = turnStarts[turnStarts.length - this.config.keepTurns];
+        if (cutoffIdx <= 0)
+            return messages;
+        const dropped = turnStarts.length - this.config.keepTurns;
+        const marker = {
+            role: types_1.Role.User,
+            content: `[Earlier conversation (${dropped} turn(s)) removed to save context space.]`,
+            timestamp: Date.now(),
+        };
+        return [marker, ...messages.slice(cutoffIdx)];
+    }
+    // ─── Step 3: Drop stale tool results ──────────────────────────────────────
+    /**
+     * Remove tool results older than `toolResultMaxAgeMs` for read-type
+     * tools (results that can be reproduced by re-running the tool).
+     * Sub-agent results (`<subagent-result>`) are preserved.
+     */
+    step3DropStaleToolResults(messages) {
+        const now = Date.now();
+        const maxAge = this.config.toolResultMaxAgeMs;
+        let changed = false;
+        const result = messages.map((m) => {
+            if (m.role !== types_1.Role.Tool)
+                return m;
+            if (!m.timestamp)
+                return m;
+            // Only target read-type tools (results that can be reproduced by re-running)
+            if (!READ_TOOLS.has(m.name ?? ""))
+                return m;
+            // Preserve sub-agent results (injected as user messages, but guard anyway)
+            if (m.content.includes("<subagent-result"))
+                return m;
+            const age = now - m.timestamp;
+            if (age < maxAge)
+                return m;
+            changed = true;
+            return {
+                ...m,
+                content: `[Tool "${m.name ?? "unknown"}" result (${Math.round(age / 60000)} min ago) removed. ` +
+                    `Re-run the tool if this information is still needed.]`,
+            };
+        });
+        return changed ? result : messages;
+    }
+    // ─── Step 4: LLM full-conversation summarization ────────────────────────
+    /**
+     * Generate a structured summary of ALL messages via the LLM.
+     *
+     * The entire conversation (from the first user message onward) is
+     * compressed into a single summary. No recent turns are kept verbatim —
+     * the summary is the sole source of history after compression.
+     *
+     * A continuity hint is appended so the LLM knows it is resuming, not
+     * starting from scratch.
+     */
+    async step4LlmSummarize(messages, _systemMessage, llm, _turnStarts) {
+        const prompt = buildSummaryPrompt(messages);
+        try {
+            const response = await llm.chat([{ role: types_1.Role.User, content: prompt }], []);
+            const summary = {
+                role: types_1.Role.User,
+                content: `[Context Summary — conversation compressed to save space]\n\n` +
+                    response.content,
+                timestamp: Date.now(),
+            };
+            const hint = {
+                role: types_1.Role.User,
+                content: "[System] The conversation above has been compressed into a summary. " +
+                    "You are NOT starting from scratch — continue your work from where you left off. " +
+                    "Use the summary above to understand what happened before and pick up right where you were.",
+                timestamp: Date.now(),
+            };
+            return [summary, hint];
+        }
+        catch {
+            // If summarization fails, fall back to simple truncation with a marker
+            this.logger.warn("Compression", "Step 4 LLM call failed; falling back to simple truncation.");
+            const marker = {
+                role: types_1.Role.User,
+                content: `[Context compressed — LLM summarization was unavailable. ` +
+                    `Some earlier context may be missing. Continue to the best of your ability.]`,
+                timestamp: Date.now(),
+            };
+            return [marker];
+        }
+    }
+}
+exports.ProgressiveCompressor = ProgressiveCompressor;
+// ─── Summary Prompt Builder ─────────────────────────────────────────────────
+const MAX_CHARS_PER_MESSAGE = 3000;
+function buildSummaryPrompt(messages) {
+    const transcriptLines = messages.map((m) => {
+        const roleLabel = m.role.toUpperCase();
+        let content = m.content;
+        if (content.length > MAX_CHARS_PER_MESSAGE) {
+            content = content.slice(0, MAX_CHARS_PER_MESSAGE) + "\n... [truncated for summarization]";
+        }
+        return `[${roleLabel}]${m.name ? ` (${m.name})` : ""}\n${content}`;
+    });
+    const transcript = transcriptLines.join("\n\n---\n\n");
+    return [
+        "You are a context compression assistant. Read the following conversation transcript ",
+        "and generate a structured summary document. The summary MUST include ALL of these ",
+        "sections (use them as headings, do not skip any):",
+        "",
+        "## 1. Primary Request and Intent",
+        "Every request the user made and their intent/goal.",
+        "",
+        "## 2. Key Technical Concepts",
+        "Technologies, libraries, patterns, and concepts discussed or used.",
+        "",
+        "## 3. Files and Code Sections",
+        "Every file path mentioned or edited, with relevant code snippets where available.",
+        "",
+        "## 4. Errors and Fixes",
+        "Every error encountered and how it was resolved (or if still unresolved).",
+        "",
+        "## 5. Problem Solving",
+        "Chronological walkthrough of how problems were approached and resolved.",
+        "",
+        "## 6. ALL User Messages",
+        "Copy every single user message verbatim. Do NOT summarize, omit, or paraphrase any.",
+        "",
+        "## 7. Pending Tasks",
+        "Tasks that were requested but not yet started or completed.",
+        "",
+        "## 8. Current Work",
+        "What was being worked on right before this compression happened — ",
+        "the exact state and context of the ongoing task.",
+        "",
+        "## 9. Optional Next Step",
+        "If you can infer the logical next action, suggest it briefly. " +
+            "Otherwise say 'Continue from where you left off.'",
+        "",
+        "---",
+        "",
+        "=== CONVERSATION TRANSCRIPT ===",
+        "",
+        transcript,
+        "",
+        "=== END ===",
+        "",
+        "Now generate the structured summary document.",
+    ].join("\n");
+}
+// ─── Helpers ────────────────────────────────────────────────────────────────
+function computeTurnStarts(messages) {
+    const starts = [];
+    for (let i = 0; i < messages.length; i++) {
+        if (messages[i].role === types_1.Role.User) {
+            starts.push(i);
+        }
+    }
+    return starts;
+}
+function sanitize(name) {
+    return name.replace(/[^a-zA-Z0-9_\-.]/g, "_");
+}
+function simpleHash(str) {
+    let hash = 5381;
+    for (let i = 0; i < str.length; i++) {
+        hash = ((hash << 5) + hash + str.charCodeAt(i)) & 0xffffffff;
+    }
+    return (hash >>> 0).toString(16).padStart(8, "0");
+}
+//# sourceMappingURL=progressive-compressor.js.map

package/dist/compression/progressive-compressor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"progressive-compressor.js","sourceRoot":"","sources":["../../src/compression/progressive-compressor.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA,uCAAyB;AACzB,2CAA6B;AAC7B,6CAAsD;AACtD,0DAAqD;AAErD,8CAA0D;AAE1D,+EAA+E;AAE/E,MAAM,qBAAqB,GAAG,GAAG,GAAG,IAAI,CAAC,CAAE,iBAAiB;AAC5D,MAAM,UAAU,GAAG,IAAI,CAAC,CAAoB,oBAAoB;AAChE,MAAM,aAAa,GAAG,iBAAiB,CAAC,CAAI,gCAAgC;AAE5E,gFAAgF;AAChF,MAAM,UAAU,GAAG,IAAI,GAAG,CAAC;IACzB,WAAW;IACX,aAAa;IACb,aAAa;CACd,CAAC,CAAC;AASH,+EAA+E;AAE/E;;;;;GAKG;AACH,MAAa,qBAAqB;IACxB,MAAM,CAAoB;IAC1B,MAAM,CAAS;IAEvB,YAAY,MAAyB,EAAE,MAAe;QACpD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;QACrB,IAAI,CAAC,MAAM,GAAG,MAAM,IAAI,IAAI,sBAAa,EAAE,CAAC;IAC9C,CAAC;IAED;;;;;;;;OAQG;IACH,KAAK,CAAC,QAAQ,CACZ,QAAuB,EACvB,aAAiC,EACjC,GAAiB,EACjB,KAAc;QAEd,MAAM,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC;QACtC,IAAI,MAAM,GAAG,QAAQ,CAAC;QACtB,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,CAAC,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,CAAC;QAC3C,MAAM,YAAY,GAAG,CAAC,GAAG,CAAC;YACxB,CAAC,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;YAC7C,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,CAAC,CAAC;QAE9B,sEAAsE;QACtE,MAAM,MAAM,GAAG,IAAI,CAAC,wBAAwB,CAAC,MAAM,CAAC,CAAC;QACrD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YAAC,OAAO,GAAG,IAAI,CAAC;YAAC,MAAM,GAAG,MAAM,CAAC;QAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,CAAC,GAAG,YAAY,EAAE,CAAC;YACjE,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACvD,CAAC;QAED,sEAAsE;QACtE,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;QAClE,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YAAC,OAAO,GAAG,IAAI,CAAC;YAAC,MAAM,GAAG,MAAM,CAAC;QAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,CAAC,GAAG,YAAY,EAAE,CAAC;YACjE,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACvD,CAAC;QAED,sEAAsE;QACtE,MAAM,MAAM,GAAG,IAAI,CAAC,yBAAyB,CAAC,MAAM,CAAC,CAAC;QACtD,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;YAAC,OAAO,GAAG,IAAI,CAAC;YAAC,MAAM,GAAG,MAAM,CAAC;QAAC,CAAC;QAC3D,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,CAAC,GAAG,YAAY,EAAE,CAAC;YACjE,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;QACvD,CAAC;QAED,sEAAsE;QACtE,IAAI,GAAG,EAAE,CAAC;YACR,IAAI,CAAC;gBACH,MAAM,kBAAkB,GAAG,iBAAiB,CAAC,MAAM,CAAC,CAAC;gBACrD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,MAAM,EAAE,aAAa,EAAE,GAAG,EAAE,kBAAkB,CAAC,CAAC;gBAC5F,IAAI,MAAM,KAAK,MAAM,EAAE,CAAC;oBAAC,OAAO,GAAG,IAAI,CAAC;oBAAC,MAAM,GAAG,MAAM,CAAC;gBAAC,CAAC;YAC7D,CAAC;YAAC,OAAO,GAAY,EAAE,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,aAAa,EACb,qCAAqC;oBACrC,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI;oBACvD,6BAA6B,CAC9B,CAAC;YACJ,CAAC;QACH,CAAC;QAED,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,CAAC;IACvD,CAAC;IAEO,QAAQ,CACd,QAAuB,EACvB,aAAqB,EACrB,OAAgB;QAEhB,OAAO;YACL,QAAQ;YACR,YAAY,EAAE,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,aAAa,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;YACxE,OAAO;SACR,CAAC;IACJ,CAAC;IAED,6EAA6E;IAErE,UAAU,CAAC,QAAuB,EAAE,GAAuB,EAAE,KAAc;QACjF,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,GAAG;YAAE,KAAK,IAAI,CAAC,GAAG,IAAA,2BAAW,EAAC,GAAG,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QACtD,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;YACzB,KAAK,IAAI,CAAC,GAAG,IAAA,2BAAW,EAAC,CAAC,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC;QAC7C,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,6EAA6E;IAE7E;;;OAGG;IACK,wBAAwB,CAAC,QAAuB;QACtD,8CAA8C;QAC9C,MAAM,WAAW,GAA0C,EAAE,CAAC;QAC9D,IAAI,UAAU,GAAG,CAAC,CAAC;QAEnB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,YAAI,CAAC,IAAI,EAAE,CAAC;gBACnC,MAAM,KAAK,GAAG,MAAM,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC9D,UAAU,IAAI,KAAK,CAAC;gBACpB,WAAW,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC;YACtC,CAAC;QACH,CAAC;QAED,IAAI,UAAU,IAAI,qBAAqB;YAAE,OAAO,QAAQ,CAAC;QAEzD,iDAAiD;QACjD,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAE9C,MAAM,MAAM,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC7B,IAAI,MAAM,GAAG,UAAU,GAAG,qBAAqB,CAAC;QAChD,EAAE,CAAC,SAAS,CAAC,aAAa,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;QAEjD,KAAK,MAAM,EAAE,GAAG,EAAE,KAAK,EAAE,IAAI,WAAW,EAAE,CAAC;YACzC,IAAI,MAAM,IAAI,CAAC;gBAAE,MAAM;YAEvB,qCAAqC;YACrC,MAAM,OAAO,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,OAAO,CAAC;YACpC,MAAM,cAAc,GAAG,KAAK,GAAG,UAAU,CAAC;YAC1C,IAAI,cAAc,IAAI,CAAC;gBAAE,SAAS;YAElC,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;YACvB,MAAM,IAAI,GAAG,UAAU,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,GAAG,CAAC,GAAG,GAAG,CAAC,CAAC;YACrD,MAAM,IAAI,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,IAAI,cAAc,CAAC;YAChD,MAAM,QAAQ,GAAG,aAAa,QAAQ,CAAC,IAAI,CAAC,IAAI,GAAG,IAAI,IAAI,MAAM,CAAC;YAClE,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,aAAa,EAAE,QAAQ,CAAC,CAAC;YAEpD,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;YAE7C,MAAM,CAAC,GAAG,CAAC,GAAG;gBACZ,GAAG,MAAM,CAAC,GAAG,CAAC;gBACd,OAAO,EACL,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,UAAU,CAAC;oBAC5B,gCAAgC,CAAC,KAAK,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,UAAU,GAAG,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO;oBACvG,yBAAyB,QAAQ,GAAG;aACvC,CAAC;YAEF,MAAM,IAAI,cAAc,CAAC;QAC3B,CAAC;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,6EAA6E;IAE7E;;;;OAIG;IACK,iBAAiB,CAAC,QAAuB,EAAE,UAAoB;QACrE,IAAI,UAAU,CAAC,MAAM,IAAI,IAAI,CAAC,MAAM,CAAC,SAAS;YAAE,OAAO,QAAQ,CAAC;QAEhE,MAAM,SAAS,GAAG,UAAU,CAAC,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QACxE,IAAI,SAAS,IAAI,CAAC;YAAE,OAAO,QAAQ,CAAC;QAEpC,MAAM,OAAO,GAAG,UAAU,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,SAAS,CAAC;QAC1D,MAAM,MAAM,GAAgB;YAC1B,IAAI,EAAE,YAAI,CAAC,IAAI;YACf,OAAO,EAAE,0BAA0B,OAAO,2CAA2C;YACrF,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,OAAO,CAAC,MAAM,EAAE,GAAG,QAAQ,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC;IAChD,CAAC;IAED,6EAA6E;IAE7E;;;;OAIG;IACK,yBAAyB,CAAC,QAAuB;QACvD,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC;QAC9C,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,MAAM,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;YAChC,IAAI,CAAC,CAAC,IAAI,KAAK,YAAI,CAAC,IAAI;gBAAE,OAAO,CAAC,CAAC;YACnC,IAAI,CAAC,CAAC,CAAC,SAAS;gBAAE,OAAO,CAAC,CAAC;YAE3B,6EAA6E;YAC7E,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,EAAE,CAAC;gBAAE,OAAO,CAAC,CAAC;YAE5C,2EAA2E;YAC3E,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAC;gBAAE,OAAO,CAAC,CAAC;YAErD,MAAM,GAAG,GAAG,GAAG,GAAG,CAAC,CAAC,SAAS,CAAC;YAC9B,IAAI,GAAG,GAAG,MAAM;gBAAE,OAAO,CAAC,CAAC;YAE3B,OAAO,GAAG,IAAI,CAAC;YACf,OAAO;gBACL,GAAG,CAAC;gBACJ,OAAO,EAAE,UAAU,CAAC,CAAC,IAAI,IAAI,SAAS,aAAa,IAAI,CAAC,KAAK,CAAC,GAAG,GAAG,KAAK,CAAC,qBAAqB;oBAC7F,uDAAuD;aAC1D,CAAC;QACJ,CAAC,CAAC,CAAC;QAEH,OAAO,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,QAAQ,CAAC;IACrC,CAAC;IAED,2EAA2E;IAE3E;;;;;;;;;OASG;IACK,KAAK,CAAC,iBAAiB,CAC7B,QAAuB,EACvB,cAAkC,EAClC,GAAgB,EAChB,WAAqB;QAErB,MAAM,MAAM,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;QAE5C,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,MAAM,GAAG,CAAC,IAAI,CAC7B,CAAC,EAAE,IAAI,EAAE,YAAI,CAAC,IAAI,EAAE,OAAO,EAAE,MAAM,EAAE,CAAC,EACtC,EAAE,CACH,CAAC;YAEF,MAAM,OAAO,GAAgB;gBAC3B,IAAI,EAAE,YAAI,CAAC,IAAI;gBACf,OAAO,EACL,+DAA+D;oBAC/D,QAAQ,CAAC,OAAO;gBAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,IAAI,GAAgB;gBACxB,IAAI,EAAE,YAAI,CAAC,IAAI;gBACf,OAAO,EACL,sEAAsE;oBACtE,kFAAkF;oBAClF,4FAA4F;gBAC9F,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,OAAO,CAAC,OAAO,EAAE,IAAI,CAAC,CAAC;QACzB,CAAC;QAAC,MAAM,CAAC;YACP,uEAAuE;YACvE,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,aAAa,EAAE,4DAA4D,CAAC,CAAC;YAC9F,MAAM,MAAM,GAAgB;gBAC1B,IAAI,EAAE,YAAI,CAAC,IAAI;gBACf,OAAO,EACL,2DAA2D;oBAC3D,6EAA6E;gBAC/E,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YACF,OAAO,CAAC,MAAM,CAAC,CAAC;QAClB,CAAC;IACH,CAAC;CACF;AA7QD,sDA6QC;AAED,+EAA+E;AAE/E,MAAM,qBAAqB,GAAG,IAAI,CAAC;AAEnC,SAAS,kBAAkB,CAAC,QAAuB;IACjD,MAAM,eAAe,GAAG,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;QACzC,MAAM,SAAS,GAAG,CAAC,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACvC,IAAI,OAAO,GAAG,CAAC,CAAC,OAAO,CAAC;QACxB,IAAI,OAAO,CAAC,MAAM,GAAG,qBAAqB,EAAE,CAAC;YAC3C,OAAO,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,qBAAqB,CAAC,GAAG,qCAAqC,CAAC;QAC5F,CAAC;QACD,OAAO,IAAI,SAAS,IAAI,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,EAAE,KAAK,OAAO,EAAE,CAAC;IACrE,CAAC,CAAC,CAAC;IAEH,MAAM,UAAU,GAAG,eAAe,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;IAEvD,OAAO;QACL,sFAAsF;QACtF,oFAAoF;QACpF,mDAAmD;QACnD,EAAE;QACF,kCAAkC;QAClC,oDAAoD;QACpD,EAAE;QACF,8BAA8B;QAC9B,oEAAoE;QACpE,EAAE;QACF,+BAA+B;QAC/B,mFAAmF;QACnF,EAAE;QACF,wBAAwB;QACxB,2EAA2E;QAC3E,EAAE;QACF,uBAAuB;QACvB,yEAAyE;QACzE,EAAE;QACF,yBAAyB;QACzB,qFAAqF;QACrF,EAAE;QACF,qBAAqB;QACrB,6DAA6D;QAC7D,EAAE;QACF,oBAAoB;QACpB,oEAAoE;QACpE,kDAAkD;QAClD,EAAE;QACF,0BAA0B;QAC1B,gEAAgE;YAChE,mDAAmD;QACnD,EAAE;QACF,KAAK;QACL,EAAE;QACF,iCAAiC;QACjC,EAAE;QACF,UAAU;QACV,EAAE;QACF,aAAa;QACb,EAAE;QACF,+CAA+C;KAChD,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,+EAA+E;AAE/E,SAAS,iBAAiB,CAAC,QAAuB;IAChD,MAAM,MAAM,GAAa,EAAE,CAAC;IAC5B,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,YAAI,CAAC,IAAI,EAAE,CAAC;YACnC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,QAAQ,CAAC,IAAY;IAC5B,OAAO,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,GAAG,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,UAAU,CAAC,GAAW;IAC7B,IAAI,IAAI,GAAG,IAAI,CAAC;IAChB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,GAAG,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,IAAI,GAAG,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,GAAG,IAAI,GAAG,GAAG,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,CAAC;IAC/D,CAAC;IACD,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,CAAC,EAAE,GAAG,CAAC,CAAC;AACpD,CAAC"}

package/dist/compression/types.d.ts

@@ -3,10 +3,6 @@
  */
 export interface CompressionConfig {
     /** The compression strategy to use. */
-    strategy: "sliding_window";
-    /** Number of messages to preserve when compression is triggered. */
-    keepLastN: number;
-    /** Whether to always keep system messages. */
-    keepSystemMessages: boolean;
+    strategy: "progressive";
 }
 //# sourceMappingURL=types.d.ts.map

package/dist/compression/types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/compression/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,uCAAuC;IACvC,QAAQ,EAAE,gBAAgB,CAAC;IAE3B,oEAAoE;IACpE,SAAS,EAAE,MAAM,CAAC;IAElB,8CAA8C;IAC9C,kBAAkB,EAAE,OAAO,CAAC;CAC7B"}
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../src/compression/types.ts"],"names":[],"mappings":"AAAA;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,uCAAuC;IACvC,QAAQ,EAAE,aAAa,CAAC;CACzB"}

package/dist/context/context-manager.d.ts

@@ -1,47 +1,66 @@
 import { MessageData } from "../messages/types";
 import { ContextConfig, ContextState } from "./types";
+import { LLMProvider } from "../llm/interface";
+import { Logger } from "../logging/logger";
 /**
  * ContextManager maintains the message window that will be sent to the LLM.
  *
  * Responsibilities:
- * - Accept new messages and update the running token count.
+ * - Accept new messages with automatic timestamps.
  * - Detect when the token threshold is crossed and trigger compression.
+ * - Run progressive 4-step compression with optional LLM summarization.
  * - Provide the current message list to the caller (e.g., Agent).
  */
 export declare class ContextManager {
     private config;
     private messages;
     private systemMessage;
-    private compressionStrategy;
     private _isCompressed;
-    constructor(config?: Partial<ContextConfig>);
+    private compressor;
+    private logger;
+    constructor(config?: Partial<ContextConfig>, logger?: Logger);
+    /**
+     * Compute the trigger token count based on the compressionThreshold mode.
+     * Absolute: maxTokens - threshold
+     * Ratio:    maxTokens * (1 - threshold)
+     */
+    private triggerTokens;
     /**
      * Set or update the system message (always preserved in the window).
      */
     setSystemMessage(content: string): void;
     /**
-     * Add a message to the context window.
+     * Add a message to the context window. Timestamps are auto-set if not
+     * already present.
      */
     addMessage(message: MessageData): void;
     /**
-     * Check whether the context window has exceeded the compression threshold.
+     * Check whether compression should trigger.
+     * @param model Optional model name for accurate tiktoken encoding.
+     * @returns true when remaining free tokens < compressionThreshold.
      */
-    shouldCompress(): boolean;
+    shouldCompress(model?: string): boolean;
     /**
-     * The token count at which compression triggers.
+     * Token count of all messages in the window.
+     * @param model Optional model name for accurate tiktoken encoding.
      */
-    private getCompressionThreshold;
+    getCurrentTokens(model?: string): number;
     /**
-     * Approximate token count of all messages in the window.
+     * Run progressive 4-step compression.
+     *
+     * @param llm Optional LLM provider for Step 4 (summarization).
+     *            If omitted, only steps 1-3 are applied.
      */
-    getCurrentTokens(): number;
+    compress(llm?: LLMProvider): Promise<{
+        removedCount: number;
+    }>;
     /**
-     * Run the compression strategy to reduce the window size.
-     * Resets the message list to the compressed output.
+     * Convenience: check + compress if needed.
+     *
+     * @param llm LLM provider for Step 4 summarization and accurate token counting.
+     * @returns true if compression was applied.
      */
-    compress(): {
-        removedCount: number;
-    };
+    checkAndCompress(llm?: LLMProvider): Promise<boolean>;
     /**
      * Get the current context messages (ready to send to the LLM).
      * Includes the system message as the first element if set.

package/dist/context/context-manager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"context-manager.d.ts","sourceRoot":"","sources":["../../src/context/context-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAQ,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAKtD;;;;;;;GAOG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,QAAQ,CAAqB;IACrC,OAAO,CAAC,aAAa,CAA4B;IACjD,OAAO,CAAC,mBAAmB,CAAsB;IACjD,OAAO,CAAC,aAAa,CAAS;gBAElB,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC;IAW3C;;OAEG;IACH,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAIvC;;OAEG;IACH,UAAU,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI;IAItC;;OAEG;IACH,cAAc,IAAI,OAAO;IAIzB;;OAEG;IACH,OAAO,CAAC,uBAAuB;IAM/B;;OAEG;IACH,gBAAgB,IAAI,MAAM;IAc1B;;;OAGG;IACH,QAAQ,IAAI;QAAE,YAAY,EAAE,MAAM,CAAA;KAAE;IAUpC;;;OAGG;IACH,kBAAkB,IAAI,WAAW,EAAE;IAOnC;;OAEG;IACH,WAAW,IAAI,WAAW,EAAE;IAI5B;;OAEG;IACH,IAAI,YAAY,IAAI,OAAO,CAE1B;IAED;;OAEG;IACH,QAAQ,IAAI,YAAY;IASxB;;OAEG;IACH,KAAK,IAAI,IAAI;CAId"}
+{"version":3,"file":"context-manager.d.ts","sourceRoot":"","sources":["../../src/context/context-manager.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAQ,MAAM,mBAAmB,CAAC;AACtD,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAGtD,OAAO,EAAE,WAAW,EAAE,MAAM,kBAAkB,CAAC;AAC/C,OAAO,EAAE,MAAM,EAAiB,MAAM,mBAAmB,CAAC;AAE1D;;;;;;;;GAQG;AACH,qBAAa,cAAc;IACzB,OAAO,CAAC,MAAM,CAAgB;IAC9B,OAAO,CAAC,QAAQ,CAAqB;IACrC,OAAO,CAAC,aAAa,CAA4B;IACjD,OAAO,CAAC,aAAa,CAAS;IAC9B,OAAO,CAAC,UAAU,CAAwB;IAC1C,OAAO,CAAC,MAAM,CAAS;gBAEX,MAAM,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,EAAE,MAAM,CAAC,EAAE,MAAM;IAqB5D;;;;OAIG;IACH,OAAO,CAAC,aAAa;IAOrB;;OAEG;IACH,gBAAgB,CAAC,OAAO,EAAE,MAAM,GAAG,IAAI;IAIvC;;;OAGG;IACH,UAAU,CAAC,OAAO,EAAE,WAAW,GAAG,IAAI;IAOtC;;;;OAIG;IACH,cAAc,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,OAAO;IASvC;;;OAGG;IACH,gBAAgB,CAAC,KAAK,CAAC,EAAE,MAAM,GAAG,MAAM;IAWxC;;;;;OAKG;IACG,QAAQ,CAAC,GAAG,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC;QAAE,YAAY,EAAE,MAAM,CAAA;KAAE,CAAC;IAepE;;;;;OAKG;IACG,gBAAgB,CAAC,GAAG,CAAC,EAAE,WAAW,GAAG,OAAO,CAAC,OAAO,CAAC;IAuB3D;;;OAGG;IACH,kBAAkB,IAAI,WAAW,EAAE;IAOnC;;OAEG;IACH,WAAW,IAAI,WAAW,EAAE;IAI5B;;OAEG;IACH,IAAI,YAAY,IAAI,OAAO,CAE1B;IAED;;OAEG;IACH,QAAQ,IAAI,YAAY;IASxB;;OAEG;IACH,KAAK,IAAI,IAAI;CAId"}

package/dist/context/context-manager.js

@@ -2,29 +2,51 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ContextManager = void 0;
 const types_1 = require("../messages/types");
-const sliding_window_1 = require("../compression/sliding-window");
 const token_counter_1 = require("../utils/token-counter");
+const progressive_compressor_1 = require("../compression/progressive-compressor");
+const logger_1 = require("../logging/logger");
 /**
  * ContextManager maintains the message window that will be sent to the LLM.
  *
  * Responsibilities:
- * - Accept new messages and update the running token count.
+ * - Accept new messages with automatic timestamps.
  * - Detect when the token threshold is crossed and trigger compression.
+ * - Run progressive 4-step compression with optional LLM summarization.
  * - Provide the current message list to the caller (e.g., Agent).
  */
 class ContextManager {
     config;
     messages = [];
     systemMessage = null;
-    compressionStrategy;
     _isCompressed = false;
-    constructor(config) {
+    compressor;
+    logger;
+    constructor(config, logger) {
         this.config = {
             maxTokens: config?.maxTokens ?? 128000,
-            compressionThresholdRatio: config?.compressionThresholdRatio ?? 0.75,
+            compressionThreshold: config?.compressionThreshold ?? 20000,
+            keepTurns: config?.keepTurns ?? 40,
+            toolResultMaxAgeMs: config?.toolResultMaxAgeMs ?? 60 * 60 * 1000,
             compression: config?.compression,
         };
-        this.compressionStrategy = new sliding_window_1.SlidingWindowCompression(this.config.compression);
+        // Validate ratio mode
+        if (this.config.compressionThreshold < 1 && this.config.compressionThreshold > 0.25) {
+            throw new Error(`compressionThreshold ratio must be ≤ 0.25 (trigger at ≥ 75% of context). ` +
+                `Got ${this.config.compressionThreshold} (trigger at ${Math.round((1 - this.config.compressionThreshold) * 100)}%).`);
+        }
+        this.logger = logger ?? new logger_1.ConsoleLogger();
+        this.compressor = new progressive_compressor_1.ProgressiveCompressor(this.config, this.logger);
+    }
+    /**
+     * Compute the trigger token count based on the compressionThreshold mode.
+     * Absolute: maxTokens - threshold
+     * Ratio:    maxTokens * (1 - threshold)
+     */
+    triggerTokens() {
+        const t = this.config.compressionThreshold;
+        return t < 1
+            ? Math.round(this.config.maxTokens * (1 - t))
+            : this.config.maxTokens - t;
     }
     /**
      * Set or update the system message (always preserved in the window).
@@ -33,49 +55,77 @@ class ContextManager {
         this.systemMessage = { role: types_1.Role.System, content };
     }
     /**
-     * Add a message to the context window.
+     * Add a message to the context window. Timestamps are auto-set if not
+     * already present.
      */
     addMessage(message) {
+        if (!message.timestamp) {
+            message.timestamp = Date.now();
+        }
         this.messages.push(message);
     }
     /**
-     * Check whether the context window has exceeded the compression threshold.
-     */
-    shouldCompress() {
-        return this.getCurrentTokens() >= this.getCompressionThreshold();
-    }
-    /**
-     * The token count at which compression triggers.
+     * Check whether compression should trigger.
+     * @param model Optional model name for accurate tiktoken encoding.
+     * @returns true when remaining free tokens < compressionThreshold.
      */
-    getCompressionThreshold() {
-        return Math.floor(this.config.maxTokens * this.config.compressionThresholdRatio);
+    shouldCompress(model) {
+        // Short-circuit: too few messages to ever hit the compression threshold.
+        // The system message + a handful of conversation turns won't breach
+        // typical thresholds (64K+). Avoids unnecessary tiktoken calls.
+        const totalMsgs = (this.systemMessage ? 1 : 0) + this.messages.length;
+        if (totalMsgs < 20)
+            return false;
+        return this.getCurrentTokens(model) >= this.triggerTokens();
     }
     /**
-     * Approximate token count of all messages in the window.
+     * Token count of all messages in the window.
+     * @param model Optional model name for accurate tiktoken encoding.
      */
-    getCurrentTokens() {
+    getCurrentTokens(model) {
         let total = 0;
-        // System message overhead
         if (this.systemMessage) {
-            total += 3 + (0, token_counter_1.countTokens)(this.systemMessage.content);
+            total += 3 + (0, token_counter_1.countTokens)(this.systemMessage.content, model);
         }
-        // Per-message overhead + content
         for (const msg of this.messages) {
-            total += 3; // role overhead
-            total += (0, token_counter_1.countTokens)(msg.content);
+            total += 3 + (0, token_counter_1.countTokens)(msg.content, model);
         }
         return total;
     }
     /**
-     * Run the compression strategy to reduce the window size.
-     * Resets the message list to the compressed output.
+     * Run progressive 4-step compression.
+     *
+     * @param llm Optional LLM provider for Step 4 (summarization).
+     *            If omitted, only steps 1-3 are applied.
      */
-    compress() {
-        const result = this.compressionStrategy.compress(this.messages, this.systemMessage ?? undefined);
+    async compress(llm) {
+        const model = llm?.model;
+        const result = await this.compressor.compress(this.messages, this.systemMessage, llm, model);
         this.messages = result.messages;
-        this._isCompressed = result.applied;
+        if (result.applied) {
+            this._isCompressed = true;
+        }
         return { removedCount: result.removedCount };
     }
+    /**
+     * Convenience: check + compress if needed.
+     *
+     * @param llm LLM provider for Step 4 summarization and accurate token counting.
+     * @returns true if compression was applied.
+     */
+    async checkAndCompress(llm) {
+        const model = llm?.model;
+        if (!this.shouldCompress(model))
+            return false;
+        const beforeTokens = this.getCurrentTokens(model);
+        this.logger.info("Context", `Compression triggered: ${beforeTokens} tokens, ` +
+            `threshold at ${this.triggerTokens()}.`);
+        const { removedCount } = await this.compress(llm);
+        const afterTokens = this.getCurrentTokens(model);
+        this.logger.info("Context", `Compression done: ${beforeTokens} → ${afterTokens} tokens ` +
+            `(${removedCount > 0 ? `removed ~${removedCount} messages` : "no messages removed"}).`);
+        return removedCount > 0;
+    }
     /**
      * Get the current context messages (ready to send to the LLM).
      * Includes the system message as the first element if set.