pi-invisible-continue 0.3.8 → 0.3.10

package/README.md

@@ -35,6 +35,7 @@ The continuation deliberately goes through `AgentSession`:
 - A concurrent prompt is queued through the native follow-up path rather than racing it.
 - Auto-retry, compaction, abort, and `agent_settled` behavior remain intact.
 - The marker is hidden from the TUI and filtered from every LLM request.
+- When a continuation starts after pi's auto-retry is exhausted, the trailing failed assistant attempts (`stopReason: "error"`/`"aborted"`, no tool calls) are also dropped from the request — the LLM never sees the failure either.
 
 ## How It Works
 

package/continue.ts

@@ -4,6 +4,7 @@ import {
   INVISIBLE_CONTINUE_CUSTOM_TYPE,
   getLastAssistantMessageText,
   isInvisibleContinueMarker,
+  stripTrailingIncompleteAssistants,
 } from "./src/index.js";
 
 /**
@@ -13,8 +14,10 @@ import {
  */
 export default function invisibleContinueExtension(pi: ExtensionAPI) {
   pi.on("context", (event) => {
+    const hasMarker = event.messages.some((message) => isInvisibleContinueMarker(message));
     const messages = event.messages.filter((message) => !isInvisibleContinueMarker(message));
-    if (messages.length !== event.messages.length) return { messages };
+    const next = hasMarker ? stripTrailingIncompleteAssistants(messages) : messages;
+    if (next.length !== event.messages.length) return { messages: next };
   });
 
   pi.registerCommand("continue", {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pi-invisible-continue",
-  "version": "0.3.8",
+  "version": "0.3.10",
   "description": "Invisible session continuation for pi — resume the agentic loop without sending ANY prompt the LLM can see",
   "type": "module",
   "author": "Tom X Nguyen",
@@ -30,8 +30,15 @@
     "src/",
     "README.md"
   ],
+  "scripts": {
+    "test": "vitest run",
+    "test:watch": "vitest",
+    "test:coverage": "vitest run --coverage",
+    "typecheck": "tsc --noEmit",
+    "lint:dead": "knip --no-gitignore"
+  },
   "devDependencies": {
-    "@earendil-works/pi-coding-agent": "^0.84.3",
+    "@earendil-works/pi-coding-agent": "^0.85.0",
     "@types/node": "25.9.1",
     "@vitest/coverage-v8": "4.1.7",
     "knip": "6.14.1",
@@ -47,12 +54,5 @@
     "brace-expansion": "5.0.6",
     "protobufjs": "8.4.0",
     "ws": "8.20.1"
-  },
-  "scripts": {
-    "test": "vitest run",
-    "test:watch": "vitest",
-    "test:coverage": "vitest run --coverage",
-    "typecheck": "tsc --noEmit",
-    "lint:dead": "knip --no-gitignore"
   }
-}
+}

package/src/index.ts

@@ -14,6 +14,54 @@ export function isInvisibleContinueMarker(message: unknown): boolean {
   );
 }
 
+const INCOMPLETE_ASSISTANT_STOP_REASONS = new Set(["error", "aborted"]);
+
+export function isIncompleteAssistantMessage(message: unknown): boolean {
+  if (!message || typeof message !== "object") return false;
+  const candidate = message as { role?: unknown; stopReason?: unknown };
+  return (
+    candidate.role === "assistant" &&
+    typeof candidate.stopReason === "string" &&
+    INCOMPLETE_ASSISTANT_STOP_REASONS.has(candidate.stopReason)
+  );
+}
+
+export function hasToolCallBlocks(message: unknown): boolean {
+  if (!message || typeof message !== "object") return false;
+  const content = (message as { content?: unknown }).content;
+  return (
+    Array.isArray(content) &&
+    content.some(
+      (block) =>
+        !!block &&
+        typeof block === "object" &&
+        (block as { type?: unknown }).type === "toolCall",
+    )
+  );
+}
+
+/**
+ * Pop trailing incomplete assistant attempts (retry-exhausted errors, aborts)
+ * so a continuation does not re-serve the failed run to the LLM.
+ *
+ * Guards: stops at the first message that is not an incomplete assistant,
+ * never pops an assistant carrying toolCall blocks (their tool results must
+ * stay paired), and never empties the context entirely.
+ */
+export function stripTrailingIncompleteAssistants<T>(
+  messages: ReadonlyArray<T>,
+): T[] {
+  const stripped = [...messages];
+  while (
+    stripped.length > 1 &&
+    isIncompleteAssistantMessage(stripped[stripped.length - 1]) &&
+    !hasToolCallBlocks(stripped[stripped.length - 1])
+  ) {
+    stripped.pop();
+  }
+  return stripped;
+}
+
 export function getLastAssistantMessageText(
   entries: ReadonlyArray<{ type: string; message?: { role?: string; content?: unknown } }>,
 ): string | undefined {