pi-context 2.1.0 → 2.1.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-context",
-  "version": "2.1.0",
+  "version": "2.1.2",
   "description": "Agentic Context Management for Pi",
   "files": [
     "src",
@@ -8,7 +8,8 @@
     "README.md"
   ],
   "scripts": {
-    "typecheck": "tsc --noEmit"
+    "typecheck": "tsc --noEmit",
+    "test": "npm run typecheck && rm -rf dist && tsc --declaration false && node --test test/*.test.mjs"
   },
   "keywords": [
     "pi-agent",

package/src/index.ts

@@ -38,6 +38,33 @@ const formatContextUsage = (usage: ContextUsage | undefined, includeTokens = fal
     return `${percent} (${formatTokens(usage.tokens)}/${formatTokens(usage.contextWindow)})`;
 };
 
+const PassiveCompactionEntryTypes = new Set<SessionEntry["type"]>([
+    "custom",
+    "label",
+    "session_info",
+    "model_change",
+    "thinking_level_change",
+]);
+
+/**
+ * Detect whether session activity after the compact turn should cancel the
+ * requested compaction. Non-contextual session state is ignored; entries that
+ * participate in model context, plus unknown future behavior, fail closed.
+ */
+export const didConversationAdvance = (
+    branch: readonly SessionEntry[],
+    compactTurnLeaf: string | null,
+): boolean => {
+    if (!compactTurnLeaf) return true;
+
+    const compactTurnIndex = branch.findIndex((entry) => entry.id === compactTurnLeaf);
+    if (compactTurnIndex === -1) return true;
+
+    return branch
+        .slice(compactTurnIndex + 1)
+        .some((entry) => !PassiveCompactionEntryTypes.has(entry.type));
+};
+
 const resolveTargetId = (sm: SessionManager, target: string): string => {
     if (target.toLowerCase() === "root") {
         const tree = sm.getTree();
@@ -452,6 +479,7 @@ export default function (pi: ExtensionAPI) {
         const compactParams = CompactParams;
         const commandCtx = CommandCtx;
         CompactParams = null;
+        const compactTurnLeaf = sm.getLeafId();
 
         // `agent_end` is emitted before the core Agent is actually idle. If we
         // call pi.sendMessage({ triggerTurn: true }) inside this handler, pi still
@@ -461,6 +489,24 @@ export default function (pi: ExtensionAPI) {
         setTimeout(async () => {
             try {
                 await commandCtx.waitForIdle();
+
+                const branch = sm.getBranch();
+                if (didConversationAdvance(branch, compactTurnLeaf)) {
+                    commandCtx.ui.notify("context_compact cancelled: conversation advanced before compaction completed.", "warning");
+                    pi.sendMessage({
+                        customType: PiContextCustomMessageType,
+                        content: [
+                            "context_compact cancelled: conversation advanced before the summary branch was created.",
+                            "No compaction was applied; continue from the current path. If still useful, inspect timeline and retry with an updated summary.",
+                        ].join("\n"),
+                        display: false,
+                    }, {
+                        triggerTurn: true,
+                        deliverAs: "followUp",
+                    });
+                    return;
+                }
+
                 const nid = sm.branchWithSummary(compactParams.tid, compactParams.enrichedMessage);
                 compactParams.nid = nid;
                 // branchWithSummary advances the leaf to the summary entry. Reset
@@ -488,7 +534,19 @@ export default function (pi: ExtensionAPI) {
                     deliverAs: "followUp",
                 });
             } catch (err) {
-                commandCtx.ui.notify(`context_compact failed to continue: ${err instanceof Error ? err.message : String(err)}`, "error");
+                const message = err instanceof Error ? err.message : String(err);
+                commandCtx.ui.notify(`context_compact failed: ${message}`, "error");
+                pi.sendMessage({
+                    customType: PiContextCustomMessageType,
+                    content: [
+                        `context_compact failed: ${message}`,
+                        "No compaction was applied; continue from the current path. Retry only with a fresh timeline/summary.",
+                    ].join("\n"),
+                    display: false,
+                }, {
+                    triggerTurn: true,
+                    deliverAs: "followUp",
+                });
             }
         }, 0);
     });