shariq-pi-extensions 0.2.31 → 0.2.33

package/extensions/smart-compaction/README.md

@@ -25,11 +25,11 @@ When long-running agent sessions reach context thresholds, standard compaction f
 - **Fail-Closed Validation**: Accepts only `stopReason === "stop"` and rejects tool calls, empty output, or summaries missing required section headers.
 - **Lockfile & Bundle Diff Exclusion**: Automatically excludes `package-lock.json`, `Cargo.lock`, `yarn.lock`, `pnpm-lock.yaml`, and minified assets from raw diffs, recording their status under `<modified-lockfiles-and-assets>` to preserve 100% of diff token headroom for source code.
 - **Background Daemon & Terminal Awareness**: Automatically detects running background terminals/processes and injects their status into `<active-background-processes>` so the successor agent never launches duplicate services.
-- **Retry Ladder**: If an attempt encounters output limits or transient reasoning timeouts:
+- **Retry Ladder & Extended Prefill Timeout**: Compaction requests receive an extended 10-minute timeout (600,000ms) to accommodate large 1M context prefills. If an attempt encounters output limits or transient reasoning timeouts:
   1. Primary configured model with requested reasoning.
   2. Primary model with reasoning off (unblocks reasoning/token caps).
   3. Session model with reasoning off.
-  4. Graceful fallback to Pi's default compactor if all stages fail.
+  4. Strict Fail-Closed Protection: If all stages fail, compaction is cancelled to preserve 100% of the conversation transcript rather than silently degrading to Pi's generic compactor.
 - **Two-Ended Head & Tail Truncation**: Preserves both the beginning (context) and end (stack traces, compiler errors, exit codes, test summaries) of tool results and command logs.
 - **100% Full-Fidelity Data Preservation**: Preserves all user-provided data, credentials, environment variables, tool inputs, and code verbatim across compactions without stripping or redaction.
 - **Deterministic 10+ Cycle Stability**: Persists machine-readable touch, dirty-file, bounded-patch, and cycle ledgers in `CompactionEntry.details`; hierarchical delta merging keeps immutable constraints while condensing obsolete history.

package/extensions/smart-compaction/engine.ts

@@ -15,6 +15,9 @@ import {
   serializeConversationForCompaction,
 } from "./prompt.ts";
 
+// Extended 10-minute timeout for large context prefills (up to 1M+ tokens on Gemini/Claude)
+export const COMPACTION_TIMEOUT_MS = 10 * 60 * 1000;
+
 export interface DirtyFileState {
   path: string;
   status: string;
@@ -289,7 +292,8 @@ const REQUIRED_SECTION_PATTERNS = [
 
 export function validateSummaryOutput(response: AssistantMessage): string {
   if (response.stopReason !== "stop") {
-    throw new Error(`Compaction model did not complete successfully (stopReason="${response.stopReason}").`);
+    const errorDetails = response.errorMessage ? `: ${response.errorMessage}` : "";
+    throw new Error(`Compaction model did not complete successfully (stopReason="${response.stopReason}"${errorDetails}).`);
   }
 
   // Reject accidental tool calls
@@ -483,9 +487,21 @@ export async function runSmartCompaction(
     activeIsInherited = plan.isInherited;
 
     const tokenCeiling = computeCompactionTokenCeiling(plan.model, config, reserveTokens);
+
+    const timeoutController = new AbortController();
+    const timeoutTimer = setTimeout(() => {
+      timeoutController.abort(
+        new Error(`Compaction stage "${plan.stageLabel}" timed out after 10 minutes.`),
+      );
+    }, COMPACTION_TIMEOUT_MS);
+
+    const stageSignal = signal
+      ? AbortSignal.any([signal, timeoutController.signal])
+      : timeoutController.signal;
+
     const completeOptions: Record<string, unknown> = {
       maxTokens: tokenCeiling,
-      signal,
+      signal: stageSignal,
       cacheRetention: "none",
       sessionId: uuidv7(),
     };
@@ -496,7 +512,7 @@ export async function runSmartCompaction(
 
     try {
       const response = await ctx.modelRegistry.complete(plan.model, context, completeOptions as any);
-      signal?.throwIfAborted();
+      stageSignal.throwIfAborted();
       if (response.usage) {
         accumulatedUsage = combineCompactionUsage(accumulatedUsage, response.usage);
       }
@@ -509,6 +525,8 @@ export async function runSmartCompaction(
         throw err instanceof Error ? err : new Error(String(err));
       }
       lastError = err instanceof Error ? err : new Error(String(err));
+    } finally {
+      clearTimeout(timeoutTimer);
     }
   }
 

package/extensions/smart-compaction/index.ts

@@ -57,8 +57,10 @@ export function createSmartCompactionExtension(options: SmartCompactionExtension
           throw error;
         }
         const message = error instanceof Error ? error.message : String(error);
-        ctx.ui?.notify(`Smart Compaction failed: ${message}. Falling back to default compactor.`, "warning");
-        return undefined;
+        ctx.ui?.notify(`Smart Compaction failed: ${message}. Compaction cancelled to protect context.`, "error");
+        // Fail-Closed: Return cancel: true so Pi aborts compaction and preserves all conversation context
+        // rather than silently falling back to Pi's generic compactor.
+        return { cancel: true };
       } finally {
         updateStatus();
       }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "shariq-pi-extensions",
-  "version": "0.2.31",
+  "version": "0.2.33",
   "description": "Cross-platform extension suite for the Pi coding agent.",
   "license": "MIT",
   "author": "Shariq Riaz",