0xkobold 0.0.1

package/dist/src/extensions/core/context-pruning/pruner.js

@@ -0,0 +1,312 @@
+/**
+ * Context Pruning - Message Pruner
+ *
+ * Adapted from OpenClaw - intelligent message pruning to fit context windows
+ */
+const CHARS_PER_TOKEN_ESTIMATE = 4;
+// We currently skip pruning tool results that contain images
+const IMAGE_CHAR_ESTIMATE = 8_000;
+function asText(text) {
+    return { type: "text", text };
+}
+function collectTextSegments(content) {
+    const parts = [];
+    for (const block of content) {
+        if (block.type === "text") {
+            parts.push(block.text);
+        }
+    }
+    return parts;
+}
+function estimateJoinedTextLength(parts) {
+    if (parts.length === 0) {
+        return 0;
+    }
+    let len = 0;
+    for (const p of parts) {
+        len += p.length;
+    }
+    len += Math.max(0, parts.length - 1);
+    return len;
+}
+function takeHeadFromJoinedText(parts, maxChars) {
+    if (maxChars <= 0 || parts.length === 0) {
+        return "";
+    }
+    let remaining = maxChars;
+    let out = "";
+    for (let i = 0; i < parts.length && remaining > 0; i++) {
+        if (i > 0) {
+            out += "\n";
+            remaining -= 1;
+            if (remaining <= 0) {
+                break;
+            }
+        }
+        const p = parts[i];
+        if (p.length <= remaining) {
+            out += p;
+            remaining -= p.length;
+        }
+        else {
+            out += p.slice(0, remaining);
+            remaining = 0;
+        }
+    }
+    return out;
+}
+function takeTailFromJoinedText(parts, maxChars) {
+    if (maxChars <= 0 || parts.length === 0) {
+        return "";
+    }
+    let remaining = maxChars;
+    const out = [];
+    for (let i = parts.length - 1; i >= 0 && remaining > 0; i--) {
+        const p = parts[i];
+        if (p.length <= remaining) {
+            out.push(p);
+            remaining -= p.length;
+        }
+        else {
+            out.push(p.slice(p.length - remaining));
+            remaining = 0;
+            break;
+        }
+        if (remaining > 0 && i > 0) {
+            out.push("\n");
+            remaining -= 1;
+        }
+    }
+    out.reverse();
+    return out.join("");
+}
+function hasImageBlocks(content) {
+    for (const block of content) {
+        if (block.type === "image") {
+            return true;
+        }
+    }
+    return false;
+}
+function estimateMessageChars(message) {
+    if (message.role === "user") {
+        const content = message.content;
+        if (typeof content === "string") {
+            return content.length;
+        }
+        let chars = 0;
+        for (const b of content) {
+            if (b.type === "text") {
+                chars += b.text.length;
+            }
+            if (b.type === "image") {
+                chars += IMAGE_CHAR_ESTIMATE;
+            }
+        }
+        return chars;
+    }
+    if (message.role === "assistant") {
+        let chars = 0;
+        for (const b of message.content) {
+            if (b.type === "text") {
+                chars += b.text.length;
+            }
+            if (b.type === "thinking") {
+                chars += b.thinking.length;
+            }
+            if (b.type === "toolCall") {
+                try {
+                    chars += JSON.stringify(b.arguments ?? {}).length;
+                }
+                catch {
+                    chars += 128;
+                }
+            }
+        }
+        return chars;
+    }
+    if (message.role === "toolResult") {
+        let chars = 0;
+        for (const b of message.content) {
+            if (b.type === "text") {
+                chars += b.text.length;
+            }
+            if (b.type === "image") {
+                chars += IMAGE_CHAR_ESTIMATE;
+            }
+        }
+        return chars;
+    }
+    return 256;
+}
+function estimateContextChars(messages) {
+    return messages.reduce((sum, m) => sum + estimateMessageChars(m), 0);
+}
+function findAssistantCutoffIndex(messages, keepLastAssistants) {
+    if (keepLastAssistants <= 0) {
+        return messages.length;
+    }
+    let remaining = keepLastAssistants;
+    for (let i = messages.length - 1; i >= 0; i--) {
+        if (messages[i]?.role !== "assistant") {
+            continue;
+        }
+        remaining--;
+        if (remaining === 0) {
+            return i;
+        }
+    }
+    return null;
+}
+function findFirstUserIndex(messages) {
+    for (let i = 0; i < messages.length; i++) {
+        if (messages[i]?.role === "user") {
+            return i;
+        }
+    }
+    return null;
+}
+function softTrimToolResultMessage(params) {
+    const { msg, settings } = params;
+    if (hasImageBlocks(msg.content)) {
+        return null;
+    }
+    const parts = collectTextSegments(msg.content);
+    const rawLen = estimateJoinedTextLength(parts);
+    if (rawLen <= settings.softTrim.maxChars) {
+        return null;
+    }
+    const headChars = Math.max(0, settings.softTrim.headChars);
+    const tailChars = Math.max(0, settings.softTrim.tailChars);
+    if (headChars + tailChars >= rawLen) {
+        return null;
+    }
+    const head = takeHeadFromJoinedText(parts, headChars);
+    const tail = takeTailFromJoinedText(parts, tailChars);
+    const trimmed = `${head}\n...\n${tail}`;
+    const note = `\n\n[Tool result trimmed: kept first ${headChars} chars and last ${tailChars} chars of ${rawLen} chars.]`;
+    return { ...msg, content: [asText(trimmed + note)] };
+}
+function makeToolPrunablePredicate(match) {
+    const allowList = match.allow ?? [];
+    const denyList = match.deny ?? [];
+    const hasAllowList = allowList.length > 0;
+    return (toolName) => {
+        if (denyList.length > 0) {
+            for (const d of denyList) {
+                if (toolName === d || (d.includes("*") && new RegExp("^" + d.replace(/\*/g, ".*") + "$").test(toolName))) {
+                    return false;
+                }
+            }
+        }
+        if (!hasAllowList) {
+            return true;
+        }
+        for (const a of allowList) {
+            if (toolName === a || (a.includes("*") && new RegExp("^" + a.replace(/\*/g, ".*") + "$").test(toolName))) {
+                return true;
+            }
+        }
+        return false;
+    };
+}
+export function pruneContextMessages(params) {
+    const { messages, settings, ctx } = params;
+    const contextWindowTokens = typeof params.contextWindowTokensOverride === "number" &&
+        Number.isFinite(params.contextWindowTokensOverride) &&
+        params.contextWindowTokensOverride > 0
+        ? params.contextWindowTokensOverride
+        : ctx.model?.contextWindow;
+    if (!contextWindowTokens || contextWindowTokens <= 0) {
+        return messages;
+    }
+    const charWindow = contextWindowTokens * CHARS_PER_TOKEN_ESTIMATE;
+    if (charWindow <= 0) {
+        return messages;
+    }
+    const cutoffIndex = findAssistantCutoffIndex(messages, settings.keepLastAssistants);
+    if (cutoffIndex === null) {
+        return messages;
+    }
+    // Protect initial identity reads (SOUL.md, USER.md, etc.)
+    const firstUserIndex = findFirstUserIndex(messages);
+    const pruneStartIndex = firstUserIndex === null ? messages.length : firstUserIndex;
+    const isToolPrunable = params.isToolPrunable ?? makeToolPrunablePredicate(settings.tools);
+    const totalCharsBefore = estimateContextChars(messages);
+    let totalChars = totalCharsBefore;
+    let ratio = totalChars / charWindow;
+    if (ratio < settings.softTrimRatio) {
+        return messages;
+    }
+    const prunableToolIndexes = [];
+    let next = null;
+    for (let i = pruneStartIndex; i < cutoffIndex; i++) {
+        const msg = messages[i];
+        if (!msg || msg.role !== "toolResult") {
+            continue;
+        }
+        if (!isToolPrunable(msg.toolName)) {
+            continue;
+        }
+        if (hasImageBlocks(msg.content)) {
+            continue;
+        }
+        prunableToolIndexes.push(i);
+        const updated = softTrimToolResultMessage({
+            msg: msg,
+            settings,
+        });
+        if (!updated) {
+            continue;
+        }
+        const beforeChars = estimateMessageChars(msg);
+        const afterChars = estimateMessageChars(updated);
+        totalChars += afterChars - beforeChars;
+        if (!next) {
+            next = messages.slice();
+        }
+        next[i] = updated;
+    }
+    const outputAfterSoftTrim = next ?? messages;
+    ratio = totalChars / charWindow;
+    if (ratio < settings.hardClearRatio) {
+        return outputAfterSoftTrim;
+    }
+    if (!settings.hardClear.enabled) {
+        return outputAfterSoftTrim;
+    }
+    let prunableToolChars = 0;
+    for (const i of prunableToolIndexes) {
+        const msg = outputAfterSoftTrim[i];
+        if (!msg || msg.role !== "toolResult") {
+            continue;
+        }
+        prunableToolChars += estimateMessageChars(msg);
+    }
+    if (prunableToolChars < settings.minPrunableToolChars) {
+        return outputAfterSoftTrim;
+    }
+    for (const i of prunableToolIndexes) {
+        if (ratio < settings.hardClearRatio) {
+            break;
+        }
+        const msg = (next ?? messages)[i];
+        if (!msg || msg.role !== "toolResult") {
+            continue;
+        }
+        const beforeChars = estimateMessageChars(msg);
+        const cleared = {
+            ...msg,
+            content: [asText(settings.hardClear.placeholder)],
+        };
+        if (!next) {
+            next = messages.slice();
+        }
+        next[i] = cleared;
+        const afterChars = estimateMessageChars(cleared);
+        totalChars += afterChars - beforeChars;
+        ratio = totalChars / charWindow;
+    }
+    return next ?? messages;
+}
+//# sourceMappingURL=pruner.js.map

package/dist/src/extensions/core/context-pruning/pruner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"pruner.js","sourceRoot":"","sources":["../../../../../src/extensions/core/context-pruning/pruner.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAOH,MAAM,wBAAwB,GAAG,CAAC,CAAC;AACnC,6DAA6D;AAC7D,MAAM,mBAAmB,GAAG,KAAK,CAAC;AAElC,SAAS,MAAM,CAAC,IAAY;IAC1B,OAAO,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;AAChC,CAAC;AAED,SAAS,mBAAmB,CAAC,OAAkD;IAC7E,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;YAC1B,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACzB,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,wBAAwB,CAAC,KAAe;IAC/C,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACvB,OAAO,CAAC,CAAC;IACX,CAAC;IACD,IAAI,GAAG,GAAG,CAAC,CAAC;IACZ,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;QACtB,GAAG,IAAI,CAAC,CAAC,MAAM,CAAC;IAClB,CAAC;IACD,GAAG,IAAI,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IACrC,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAe,EAAE,QAAgB;IAC/D,IAAI,QAAQ,IAAI,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,SAAS,GAAG,QAAQ,CAAC;IACzB,IAAI,GAAG,GAAG,EAAE,CAAC;IACb,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QACvD,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YACV,GAAG,IAAI,IAAI,CAAC;YACZ,SAAS,IAAI,CAAC,CAAC;YACf,IAAI,SAAS,IAAI,CAAC,EAAE,CAAC;gBACnB,MAAM;YACR,CAAC;QACH,CAAC;QACD,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,IAAI,CAAC,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;YAC1B,GAAG,IAAI,CAAC,CAAC;YACT,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,GAAG,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC7B,SAAS,GAAG,CAAC,CAAC;QAChB,CAAC;IACH,CAAC;IACD,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,sBAAsB,CAAC,KAAe,EAAE,QAAgB;IAC/D,IAAI,QAAQ,IAAI,CAAC,IAAI,KAAK,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACxC,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,IAAI,SAAS,GAAG,QAAQ,CAAC;IACzB,MAAM,GAAG,GAAa,EAAE,CAAC;IACzB,KAAK,IAAI,CAAC,GAAG,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,SAAS,GAAG,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC5D,MAAM,CAAC,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACnB,IAAI,CAAC,CAAC,MAAM,IAAI,SAAS,EAAE,CAAC;YAC1B,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YACZ,SAAS,IAAI,CAAC,CAAC,MAAM,CAAC;QACxB,CAAC;aAAM,CAAC;YACN,GAAG,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC;YACxC,SAAS,GAAG,CAAC,CAAC;YACd,MAAM;QACR,CAAC;QACD,IAAI,SAAS,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YAC3B,GAAG,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACf,SAAS,IAAI,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,GAAG,CAAC,OAAO,EAAE,CAAC;IACd,OAAO,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;AACtB,CAAC;AAED,SAAS,cAAc,CAAC,OAAkD;IACxE,KAAK,MAAM,KAAK,IAAI,OAAO,EAAE,CAAC;QAC5B,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;YAC3B,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,oBAAoB,CAAC,OAAqB;IACjD,IAAI,OAAO,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;QAC5B,MAAM,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;QAChC,IAAI,OAAO,OAAO,KAAK,QAAQ,EAAE,CAAC;YAChC,OAAO,OAAO,CAAC,MAAM,CAAC;QACxB,CAAC;QACD,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;YACxB,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtB,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YACzB,CAAC;YACD,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACvB,KAAK,IAAI,mBAAmB,CAAC;YAC/B,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW,EAAE,CAAC;QACjC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtB,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YACzB,CAAC;YACD,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC1B,KAAK,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC;YAC7B,CAAC;YACD,IAAI,CAAC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,KAAK,IAAI,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;gBACpD,CAAC;gBAAC,MAAM,CAAC;oBACP,KAAK,IAAI,GAAG,CAAC;gBACf,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,IAAI,OAAO,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;QAClC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,KAAK,MAAM,CAAC,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAChC,IAAI,CAAC,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBACtB,KAAK,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC;YACzB,CAAC;YACD,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBACvB,KAAK,IAAI,mBAAmB,CAAC;YAC/B,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAED,OAAO,GAAG,CAAC;AACb,CAAC;AAED,SAAS,oBAAoB,CAAC,QAAwB;IACpD,OAAO,QAAQ,CAAC,MAAM,CAAC,CAAC,GAAG,EAAE,CAAC,EAAE,EAAE,CAAC,GAAG,GAAG,oBAAoB,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC;AACvE,CAAC;AAED,SAAS,wBAAwB,CAC/B,QAAwB,EACxB,kBAA0B;IAE1B,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;QAC5B,OAAO,QAAQ,CAAC,MAAM,CAAC;IACzB,CAAC;IAED,IAAI,SAAS,GAAG,kBAAkB,CAAC;IACnC,KAAK,IAAI,CAAC,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC9C,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,WAAW,EAAE,CAAC;YACtC,SAAS;QACX,CAAC;QACD,SAAS,EAAE,CAAC;QACZ,IAAI,SAAS,KAAK,CAAC,EAAE,CAAC;YACpB,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,kBAAkB,CAAC,QAAwB;IAClD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACzC,IAAI,QAAQ,CAAC,CAAC,CAAC,EAAE,IAAI,KAAK,MAAM,EAAE,CAAC;YACjC,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IACD,OAAO,IAAI,CAAC;AACd,CAAC;AAED,SAAS,yBAAyB,CAAC,MAGlC;IACC,MAAM,EAAE,GAAG,EAAE,QAAQ,EAAE,GAAG,MAAM,CAAC;IACjC,IAAI,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QAChC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,KAAK,GAAG,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IAC/C,MAAM,MAAM,GAAG,wBAAwB,CAAC,KAAK,CAAC,CAAC;IAC/C,IAAI,MAAM,IAAI,QAAQ,CAAC,QAAQ,CAAC,QAAQ,EAAE,CAAC;QACzC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC3D,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;IAC3D,IAAI,SAAS,GAAG,SAAS,IAAI,MAAM,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,IAAI,GAAG,sBAAsB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACtD,MAAM,IAAI,GAAG,sBAAsB,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;IACtD,MAAM,OAAO,GAAG,GAAG,IAAI,UAAU,IAAI,EAAE,CAAC;IAExC,MAAM,IAAI,GAAG,wCAAwC,SAAS,mBAAmB,SAAS,aAAa,MAAM,UAAU,CAAC;IAExH,OAAO,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,CAAC,MAAM,CAAC,OAAO,GAAG,IAAI,CAAC,CAAC,EAAE,CAAC;AACvD,CAAC;AAED,SAAS,yBAAyB,CAAC,KAA4C;IAC7E,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC;IACpC,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC;IAClC,MAAM,YAAY,GAAG,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC;IAE1C,OAAO,CAAC,QAAgB,EAAE,EAAE;QAC1B,IAAI,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;YACxB,KAAK,MAAM,CAAC,IAAI,QAAQ,EAAE,CAAC;gBACzB,IAAI,QAAQ,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;oBACzG,OAAO,KAAK,CAAC;gBACf,CAAC;YACH,CAAC;QACH,CAAC;QACD,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,OAAO,IAAI,CAAC;QACd,CAAC;QACD,KAAK,MAAM,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1B,IAAI,QAAQ,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,GAAG,CAAC,IAAI,IAAI,MAAM,CAAC,GAAG,GAAG,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,IAAI,CAAC,GAAG,GAAG,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE,CAAC;gBACzG,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,MAMpC;IACC,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,GAAG,MAAM,CAAC;IAC3C,MAAM,mBAAmB,GACvB,OAAO,MAAM,CAAC,2BAA2B,KAAK,QAAQ;QACtD,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,2BAA2B,CAAC;QACnD,MAAM,CAAC,2BAA2B,GAAG,CAAC;QACpC,CAAC,CAAC,MAAM,CAAC,2BAA2B;QACpC,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,aAAa,CAAC;IAC/B,IAAI,CAAC,mBAAmB,IAAI,mBAAmB,IAAI,CAAC,EAAE,CAAC;QACrD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,UAAU,GAAG,mBAAmB,GAAG,wBAAwB,CAAC;IAClE,IAAI,UAAU,IAAI,CAAC,EAAE,CAAC;QACpB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,WAAW,GAAG,wBAAwB,CAAC,QAAQ,EAAE,QAAQ,CAAC,kBAAkB,CAAC,CAAC;IACpF,IAAI,WAAW,KAAK,IAAI,EAAE,CAAC;QACzB,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,0DAA0D;IAC1D,MAAM,cAAc,GAAG,kBAAkB,CAAC,QAAQ,CAAC,CAAC;IACpD,MAAM,eAAe,GAAG,cAAc,KAAK,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,cAAc,CAAC;IAEnF,MAAM,cAAc,GAAG,MAAM,CAAC,cAAc,IAAI,yBAAyB,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAE1F,MAAM,gBAAgB,GAAG,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IACxD,IAAI,UAAU,GAAG,gBAAgB,CAAC;IAClC,IAAI,KAAK,GAAG,UAAU,GAAG,UAAU,CAAC;IACpC,IAAI,KAAK,GAAG,QAAQ,CAAC,aAAa,EAAE,CAAC;QACnC,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,MAAM,mBAAmB,GAAa,EAAE,CAAC;IACzC,IAAI,IAAI,GAA0B,IAAI,CAAC;IAEvC,KAAK,IAAI,CAAC,GAAG,eAAe,EAAE,CAAC,GAAG,WAAW,EAAE,CAAC,EAAE,EAAE,CAAC;QACnD,MAAM,GAAG,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACtC,SAAS;QACX,CAAC;QACD,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC;YAClC,SAAS;QACX,CAAC;QACD,IAAI,cAAc,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;YAChC,SAAS;QACX,CAAC;QACD,mBAAmB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;QAE5B,MAAM,OAAO,GAAG,yBAAyB,CAAC;YACxC,GAAG,EAAE,GAAmC;YACxC,QAAQ;SACT,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,SAAS;QACX,CAAC;QAED,MAAM,WAAW,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC9C,MAAM,UAAU,GAAG,oBAAoB,CAAC,OAAkC,CAAC,CAAC;QAC5E,UAAU,IAAI,UAAU,GAAG,WAAW,CAAC;QACvC,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,CAAC,CAAC,GAAG,OAAkC,CAAC;IAC/C,CAAC;IAED,MAAM,mBAAmB,GAAG,IAAI,IAAI,QAAQ,CAAC;IAC7C,KAAK,GAAG,UAAU,GAAG,UAAU,CAAC;IAChC,IAAI,KAAK,GAAG,QAAQ,CAAC,cAAc,EAAE,CAAC;QACpC,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IACD,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,EAAE,CAAC;QAChC,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,IAAI,iBAAiB,GAAG,CAAC,CAAC;IAC1B,KAAK,MAAM,CAAC,IAAI,mBAAmB,EAAE,CAAC;QACpC,MAAM,GAAG,GAAG,mBAAmB,CAAC,CAAC,CAAC,CAAC;QACnC,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACtC,SAAS;QACX,CAAC;QACD,iBAAiB,IAAI,oBAAoB,CAAC,GAAG,CAAC,CAAC;IACjD,CAAC;IACD,IAAI,iBAAiB,GAAG,QAAQ,CAAC,oBAAoB,EAAE,CAAC;QACtD,OAAO,mBAAmB,CAAC;IAC7B,CAAC;IAED,KAAK,MAAM,CAAC,IAAI,mBAAmB,EAAE,CAAC;QACpC,IAAI,KAAK,GAAG,QAAQ,CAAC,cAAc,EAAE,CAAC;YACpC,MAAM;QACR,CAAC;QACD,MAAM,GAAG,GAAG,CAAC,IAAI,IAAI,QAAQ,CAAC,CAAC,CAAC,CAAC,CAAC;QAClC,IAAI,CAAC,GAAG,IAAI,GAAG,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YACtC,SAAS;QACX,CAAC;QAED,MAAM,WAAW,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC;QAC9C,MAAM,OAAO,GAAsB;YACjC,GAAG,GAAG;YACN,OAAO,EAAE,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC;SAClD,CAAC;QACF,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,QAAQ,CAAC,KAAK,EAAE,CAAC;QAC1B,CAAC;QACD,IAAI,CAAC,CAAC,CAAC,GAAG,OAAkC,CAAC;QAC7C,MAAM,UAAU,GAAG,oBAAoB,CAAC,OAAkC,CAAC,CAAC;QAC5E,UAAU,IAAI,UAAU,GAAG,WAAW,CAAC;QACvC,KAAK,GAAG,UAAU,GAAG,UAAU,CAAC;IAClC,CAAC;IAED,OAAO,IAAI,IAAI,QAAQ,CAAC;AAC1B,CAAC"}

package/dist/src/extensions/core/context-pruning/runtime.js

@@ -0,0 +1,48 @@
+/**
+ * Context Pruning Runtime State
+ *
+ * Manages per-session pruning settings and cache timestamps
+ *
+ * NOTE: This module uses @ts-ignore for SessionManager methods that are
+ * not yet in the TypeScript definitions but exist at runtime:
+ * - getCurrentSession(): Returns the current active session
+ * - getExtensionRuntime(key): Gets extension-specific runtime state
+ * - setExtensionRuntime(key, value): Sets extension-specific runtime state
+ *
+ * These methods are available in the actual pi-coding-agent runtime but
+ * the type definitions are incomplete. This is a known issue tracked
+ * in the extension development docs.
+ */
+const RUNTIME_KEY = "contextPruning:v1";
+// @ts-ignore - getCurrentSession doesn't exist on SessionManager type
+export function getContextPruningRuntime(sessionManager) {
+    try {
+        // @ts-ignore - getCurrentSession doesn't exist
+        const session = sessionManager.getCurrentSession?.();
+        if (!session) {
+            return null;
+        }
+        // @ts-ignore - getExtensionRuntime doesn't exist
+        const runtime = session.getExtensionRuntime?.(RUNTIME_KEY);
+        return runtime ?? null;
+    }
+    catch {
+        return null;
+    }
+}
+// @ts-ignore - sessionManager type
+export function setContextPruningRuntime(sessionManager, runtime) {
+    try {
+        // @ts-ignore - getCurrentSession doesn't exist
+        const session = sessionManager.getCurrentSession?.();
+        if (!session) {
+            return;
+        }
+        // @ts-ignore - setExtensionRuntime doesn't exist  
+        session.setExtensionRuntime?.(RUNTIME_KEY, runtime);
+    }
+    catch {
+        // ignore
+    }
+}
+//# sourceMappingURL=runtime.js.map

package/dist/src/extensions/core/context-pruning/runtime.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"runtime.js","sourceRoot":"","sources":["../../../../../src/extensions/core/context-pruning/runtime.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAKH,MAAM,WAAW,GAAG,mBAAmB,CAAC;AAQxC,sEAAsE;AACtE,MAAM,UAAU,wBAAwB,CAAC,cAA8B;IACrE,IAAI,CAAC;QACH,+CAA+C;QAC/C,MAAM,OAAO,GAAG,cAAc,CAAC,iBAAiB,EAAE,EAAE,CAAC;QACrD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO,IAAI,CAAC;QACd,CAAC;QACD,iDAAiD;QACjD,MAAM,OAAO,GAAG,OAAO,CAAC,mBAAmB,EAAE,CAAC,WAAW,CAAsC,CAAC;QAChG,OAAO,OAAO,IAAI,IAAI,CAAC;IACzB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,IAAI,CAAC;IACd,CAAC;AACH,CAAC;AAED,mCAAmC;AACnC,MAAM,UAAU,wBAAwB,CACtC,cAA8B,EAC9B,OAA8B;IAE9B,IAAI,CAAC;QACH,+CAA+C;QAC/C,MAAM,OAAO,GAAG,cAAc,CAAC,iBAAiB,EAAE,EAAE,CAAC;QACrD,IAAI,CAAC,OAAO,EAAE,CAAC;YACb,OAAO;QACT,CAAC;QACD,mDAAmD;QACnD,OAAO,CAAC,mBAAmB,EAAE,CAAC,WAAW,EAAE,OAAO,CAAC,CAAC;IACtD,CAAC;IAAC,MAAM,CAAC;QACP,SAAS;IACX,CAAC;AACH,CAAC"}

package/dist/src/extensions/core/context-pruning/settings.js

@@ -0,0 +1,105 @@
+/**
+ * Context Pruning Settings
+ *
+ * Adapted from OpenClaw - manages context window limits intelligently
+ */
+// Parse duration string (e.g., "5m", "1h", "30s")
+function parseDurationMs(input, { defaultUnit = "m" } = {}) {
+    const trimmed = input.trim();
+    const match = trimmed.match(/^([0-9.]+)\s*([a-z]*)$/i);
+    if (!match) {
+        throw new Error(`Invalid duration: "${input}"`);
+    }
+    const value = parseFloat(match[1]);
+    if (!Number.isFinite(value) || value < 0) {
+        throw new Error(`Invalid duration value: "${input}"`);
+    }
+    let unit = match[2].toLowerCase() || defaultUnit;
+    switch (unit) {
+        case "ms":
+            return value;
+        case "s":
+            return value * 1000;
+        case "m":
+            return value * 60 * 1000;
+        case "h":
+            return value * 60 * 60 * 1000;
+        case "d":
+            return value * 24 * 60 * 60 * 1000;
+        default:
+            throw new Error(`Unknown time unit: "${unit}" in "${input}"`);
+    }
+}
+export const DEFAULT_CONTEXT_PRUNING_SETTINGS = {
+    mode: "cache-ttl",
+    ttlMs: 5 * 60 * 1000, // 5 minutes
+    keepLastAssistants: 3,
+    softTrimRatio: 0.3,
+    hardClearRatio: 0.5,
+    minPrunableToolChars: 50_000,
+    tools: {},
+    softTrim: {
+        maxChars: 4_000,
+        headChars: 1_500,
+        tailChars: 1_500,
+    },
+    hardClear: {
+        enabled: true,
+        placeholder: "[Old tool result content cleared]",
+    },
+};
+export function computeEffectiveSettings(raw) {
+    if (!raw || typeof raw !== "object") {
+        return null;
+    }
+    const cfg = raw;
+    if (cfg.mode === "off") {
+        return null;
+    }
+    const s = structuredClone(DEFAULT_CONTEXT_PRUNING_SETTINGS);
+    s.mode = cfg.mode ?? "cache-ttl";
+    if (typeof cfg.ttl === "string") {
+        try {
+            s.ttlMs = parseDurationMs(cfg.ttl, { defaultUnit: "m" });
+        }
+        catch {
+            // keep default ttl
+        }
+    }
+    if (typeof cfg.keepLastAssistants === "number" && Number.isFinite(cfg.keepLastAssistants)) {
+        s.keepLastAssistants = Math.max(0, Math.floor(cfg.keepLastAssistants));
+    }
+    if (typeof cfg.softTrimRatio === "number" && Number.isFinite(cfg.softTrimRatio)) {
+        s.softTrimRatio = Math.min(1, Math.max(0, cfg.softTrimRatio));
+    }
+    if (typeof cfg.hardClearRatio === "number" && Number.isFinite(cfg.hardClearRatio)) {
+        s.hardClearRatio = Math.min(1, Math.max(0, cfg.hardClearRatio));
+    }
+    if (typeof cfg.minPrunableToolChars === "number" && Number.isFinite(cfg.minPrunableToolChars)) {
+        s.minPrunableToolChars = Math.max(0, Math.floor(cfg.minPrunableToolChars));
+    }
+    if (cfg.tools) {
+        s.tools = cfg.tools;
+    }
+    if (cfg.softTrim) {
+        if (typeof cfg.softTrim.maxChars === "number" && Number.isFinite(cfg.softTrim.maxChars)) {
+            s.softTrim.maxChars = Math.max(0, Math.floor(cfg.softTrim.maxChars));
+        }
+        if (typeof cfg.softTrim.headChars === "number" && Number.isFinite(cfg.softTrim.headChars)) {
+            s.softTrim.headChars = Math.max(0, Math.floor(cfg.softTrim.headChars));
+        }
+        if (typeof cfg.softTrim.tailChars === "number" && Number.isFinite(cfg.softTrim.tailChars)) {
+            s.softTrim.tailChars = Math.max(0, Math.floor(cfg.softTrim.tailChars));
+        }
+    }
+    if (cfg.hardClear) {
+        if (typeof cfg.hardClear.enabled === "boolean") {
+            s.hardClear.enabled = cfg.hardClear.enabled;
+        }
+        if (typeof cfg.hardClear.placeholder === "string" && cfg.hardClear.placeholder.trim()) {
+            s.hardClear.placeholder = cfg.hardClear.placeholder.trim();
+        }
+    }
+    return s;
+}
+//# sourceMappingURL=settings.js.map

package/dist/src/extensions/core/context-pruning/settings.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"settings.js","sourceRoot":"","sources":["../../../../../src/extensions/core/context-pruning/settings.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAgDH,kDAAkD;AAClD,SAAS,eAAe,CAAC,KAAa,EAAE,EAAE,WAAW,GAAG,GAAG,KAAqD,EAAE;IAChH,MAAM,OAAO,GAAG,KAAK,CAAC,IAAI,EAAE,CAAC;IAC7B,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,yBAAyB,CAAC,CAAC;IACvD,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,sBAAsB,KAAK,GAAG,CAAC,CAAC;IAClD,CAAC;IACD,MAAM,KAAK,GAAG,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACnC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QACzC,MAAM,IAAI,KAAK,CAAC,4BAA4B,KAAK,GAAG,CAAC,CAAC;IACxD,CAAC;IACD,IAAI,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,IAAI,WAAW,CAAC;IACjD,QAAQ,IAAI,EAAE,CAAC;QACb,KAAK,IAAI;YACP,OAAO,KAAK,CAAC;QACf,KAAK,GAAG;YACN,OAAO,KAAK,GAAG,IAAI,CAAC;QACtB,KAAK,GAAG;YACN,OAAO,KAAK,GAAG,EAAE,GAAG,IAAI,CAAC;QAC3B,KAAK,GAAG;YACN,OAAO,KAAK,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QAChC,KAAK,GAAG;YACN,OAAO,KAAK,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;QACrC;YACE,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,SAAS,KAAK,GAAG,CAAC,CAAC;IAClE,CAAC;AACH,CAAC;AAED,MAAM,CAAC,MAAM,gCAAgC,GAAoC;IAC/E,IAAI,EAAE,WAAW;IACjB,KAAK,EAAE,CAAC,GAAG,EAAE,GAAG,IAAI,EAAE,YAAY;IAClC,kBAAkB,EAAE,CAAC;IACrB,aAAa,EAAE,GAAG;IAClB,cAAc,EAAE,GAAG;IACnB,oBAAoB,EAAE,MAAM;IAC5B,KAAK,EAAE,EAAE;IACT,QAAQ,EAAE;QACR,QAAQ,EAAE,KAAK;QACf,SAAS,EAAE,KAAK;QAChB,SAAS,EAAE,KAAK;KACjB;IACD,SAAS,EAAE;QACT,OAAO,EAAE,IAAI;QACb,WAAW,EAAE,mCAAmC;KACjD;CACF,CAAC;AAEF,MAAM,UAAU,wBAAwB,CAAC,GAAY;IACnD,IAAI,CAAC,GAAG,IAAI,OAAO,GAAG,KAAK,QAAQ,EAAE,CAAC;QACpC,OAAO,IAAI,CAAC;IACd,CAAC;IACD,MAAM,GAAG,GAAG,GAA2B,CAAC;IACxC,IAAI,GAAG,CAAC,IAAI,KAAK,KAAK,EAAE,CAAC;QACvB,OAAO,IAAI,CAAC;IACd,CAAC;IAED,MAAM,CAAC,GAAoC,eAAe,CAAC,gCAAgC,CAAC,CAAC;IAC7F,CAAC,CAAC,IAAI,GAAG,GAAG,CAAC,IAAI,IAAI,WAAW,CAAC;IAEjC,IAAI,OAAO,GAAG,CAAC,GAAG,KAAK,QAAQ,EAAE,CAAC;QAChC,IAAI,CAAC;YACH,CAAC,CAAC,KAAK,GAAG,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,WAAW,EAAE,GAAG,EAAE,CAAC,CAAC;QAC3D,CAAC;QAAC,MAAM,CAAC;YACP,mBAAmB;QACrB,CAAC;IACH,CAAC;IAED,IAAI,OAAO,GAAG,CAAC,kBAAkB,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,CAAC;QAC1F,CAAC,CAAC,kBAAkB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC,CAAC;IACzE,CAAC;IACD,IAAI,OAAO,GAAG,CAAC,aAAa,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,aAAa,CAAC,EAAE,CAAC;QAChF,CAAC,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,aAAa,CAAC,CAAC,CAAC;IAChE,CAAC;IACD,IAAI,OAAO,GAAG,CAAC,cAAc,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,cAAc,CAAC,EAAE,CAAC;QAClF,CAAC,CAAC,cAAc,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,CAAC,cAAc,CAAC,CAAC,CAAC;IAClE,CAAC;IACD,IAAI,OAAO,GAAG,CAAC,oBAAoB,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,oBAAoB,CAAC,EAAE,CAAC;QAC9F,CAAC,CAAC,oBAAoB,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC;IAC7E,CAAC;IACD,IAAI,GAAG,CAAC,KAAK,EAAE,CAAC;QACd,CAAC,CAAC,KAAK,GAAG,GAAG,CAAC,KAAK,CAAC;IACtB,CAAC;IACD,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;QACjB,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC,QAAQ,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACxF,CAAC,CAAC,QAAQ,CAAC,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;QACvE,CAAC;QACD,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1F,CAAC,CAAC,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;QACzE,CAAC;QACD,IAAI,OAAO,GAAG,CAAC,QAAQ,CAAC,SAAS,KAAK,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAC1F,CAAC,CAAC,QAAQ,CAAC,SAAS,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC,CAAC;QACzE,CAAC;IACH,CAAC;IACD,IAAI,GAAG,CAAC,SAAS,EAAE,CAAC;QAClB,IAAI,OAAO,GAAG,CAAC,SAAS,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;YAC/C,CAAC,CAAC,SAAS,CAAC,OAAO,GAAG,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC;QAC9C,CAAC;QACD,IAAI,OAAO,GAAG,CAAC,SAAS,CAAC,WAAW,KAAK,QAAQ,IAAI,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,EAAE,CAAC;YACtF,CAAC,CAAC,SAAS,CAAC,WAAW,GAAG,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC;QAC7D,CAAC;IACH,CAAC;IAED,OAAO,CAAC,CAAC;AACX,CAAC"}

package/dist/src/extensions/core/dirty-repo-guard.js

@@ -0,0 +1,42 @@
+/**
+ * Dirty Repo Guard Extension
+ *
+ * Prevents session changes when there are uncommitted git changes.
+ * Useful to ensure work is committed before switching context.
+ */
+async function checkDirtyRepo(pi, ctx, action) {
+    // Check for uncommitted changes
+    const { stdout, code } = await pi.exec("git", ["status", "--porcelain"]);
+    if (code !== 0) {
+        // Not a git repo, allow the action
+        return;
+    }
+    const hasChanges = stdout.trim().length > 0;
+    if (!hasChanges) {
+        return;
+    }
+    if (!ctx.hasUI) {
+        // In non-interactive mode, block by default
+        return { cancel: true };
+    }
+    // Count changed files
+    const changedFiles = stdout.trim().split("\n").filter(Boolean).length;
+    const choice = await ctx.ui.select(`You have ${changedFiles} uncommitted file(s). ${action} anyway?`, [
+        "Yes, proceed anyway",
+        "No, let me commit first",
+    ]);
+    if (choice !== "Yes, proceed anyway") {
+        ctx.ui.notify("Commit your changes first", "warning");
+        return { cancel: true };
+    }
+}
+export default function (pi) {
+    pi.on("session_before_switch", async (event, ctx) => {
+        const action = event.reason === "new" ? "new session" : "switch session";
+        return checkDirtyRepo(pi, ctx, action);
+    });
+    pi.on("session_before_fork", async (_event, ctx) => {
+        return checkDirtyRepo(pi, ctx, "fork");
+    });
+}
+//# sourceMappingURL=dirty-repo-guard.js.map

package/dist/src/extensions/core/dirty-repo-guard.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"dirty-repo-guard.js","sourceRoot":"","sources":["../../../../src/extensions/core/dirty-repo-guard.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAIH,KAAK,UAAU,cAAc,CAC5B,EAAgB,EAChB,GAAqB,EACrB,MAAc;IAEd,gCAAgC;IAChC,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,KAAK,EAAE,CAAC,QAAQ,EAAE,aAAa,CAAC,CAAC,CAAC;IAEzE,IAAI,IAAI,KAAK,CAAC,EAAE,CAAC;QAChB,mCAAmC;QACnC,OAAO;IACR,CAAC;IAED,MAAM,UAAU,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC;IAC5C,IAAI,CAAC,UAAU,EAAE,CAAC;QACjB,OAAO;IACR,CAAC;IAED,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QAChB,4CAA4C;QAC5C,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC;IAED,sBAAsB;IACtB,MAAM,YAAY,GAAG,MAAM,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,MAAM,CAAC;IAEtE,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,YAAY,YAAY,yBAAyB,MAAM,UAAU,EAAE;QACrG,qBAAqB;QACrB,yBAAyB;KACzB,CAAC,CAAC;IAEH,IAAI,MAAM,KAAK,qBAAqB,EAAE,CAAC;QACtC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,2BAA2B,EAAE,SAAS,CAAC,CAAC;QACtD,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC;IACzB,CAAC;AACF,CAAC;AAED,MAAM,CAAC,OAAO,WAAW,EAAgB;IACxC,EAAE,CAAC,EAAE,CAAC,uBAAuB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE;QACnD,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,KAAK,KAAK,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,gBAAgB,CAAC;QACzE,OAAO,cAAc,CAAC,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,EAAE,CAAC,qBAAqB,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE;QAClD,OAAO,cAAc,CAAC,EAAE,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;IACxC,CAAC,CAAC,CAAC;AACJ,CAAC"}