@cydm/pie 2.1.1 → 2.2.0

package/dist/builtin/extensions/ask-user/index.js

@@ -1,7 +1,7 @@
 import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
 import {
   createAskUserCapability
-} from "../../../chunks/chunk-LU7RKALH.js";
+} from "../../../chunks/chunk-HH2Q2SNX.js";
 import "../../../chunks/chunk-QJH6YN2W.js";
 import "../../../chunks/chunk-IM43UYVV.js";
 import "../../../chunks/chunk-XXGYUYXI.js";

package/dist/builtin/extensions/computer-use/index.js

@@ -6901,17 +6901,28 @@ async function releaseOwnedBrowser(child, port, profile) {
     if (port) await requestBrowserClose(port).catch(() => void 0);
     if (!await waitForExit(child, 700)) {
       if (process.platform === "win32" && child.pid) {
-        await new Promise((resolve, reject) => execFile("taskkill.exe", ["/PID", String(child.pid), "/T", "/F"], {
-          windowsHide: true,
-          timeout: 2e3
-        }, (error) => {
-          if (error && child.exitCode === null) reject(error);
-          else resolve();
-        }));
+        const pid = child.pid;
+        await new Promise((resolve) => {
+          execFile("taskkill.exe", ["/PID", String(pid), "/T", "/F"], {
+            windowsHide: true,
+            timeout: 2e3
+          }, () => resolve());
+        });
+        if (!await waitForExit(child, 1500)) {
+          try {
+            process.kill(pid);
+          } catch {
+          }
+          if (!await waitForExit(child, 2e3)) {
+            throw new Error(`Owned browser process ${pid} survived taskkill escalation.`);
+          }
+        }
       } else child.kill("SIGTERM");
-      if (!await waitForExit(child, 2e3)) {
-        if (process.platform !== "win32") child.kill("SIGKILL");
-        if (!await waitForExit(child, 2e3)) throw new Error("Owned browser did not exit.");
+      if (process.platform !== "win32") {
+        if (!await waitForExit(child, 2e3)) {
+          child.kill("SIGKILL");
+          if (!await waitForExit(child, 2e3)) throw new Error("Owned browser did not exit.");
+        }
       }
     }
   }
@@ -7206,16 +7217,24 @@ function shutdownComputerUseSession() {
 }
 async function releaseComputerUseSession() {
   const drain = resourceScheduler.close();
-  await disconnectCdp();
-  await currentPlatformBackend.shutdown?.();
-  await drain;
-  await releaseOwnedBrowserState(runtimeState);
+  const failures = [];
+  const attempt = (step, work) => work().catch((error) => {
+    failures.push(error instanceof Error ? new Error(`${step}: ${error.message}`) : error);
+  });
+  await attempt("disconnect CDP", () => disconnectCdp());
+  await attempt("platform shutdown", async () => {
+    await currentPlatformBackend.shutdown?.();
+  });
+  await attempt("drain scheduler", () => drain);
+  await attempt("release owned browser", () => releaseOwnedBrowserState(runtimeState));
   if (runtimeState.managedBrowserCdpPort && process.env.PIE_COMPUTER_USE_CDP_PORT === runtimeState.managedBrowserCdpPort) {
     if (runtimeState.previousCdpPort === void 0) delete process.env.PIE_COMPUTER_USE_CDP_PORT;
     else process.env.PIE_COMPUTER_USE_CDP_PORT = runtimeState.previousCdpPort;
   }
-  runtimeState.managedBrowserCdpPort = void 0;
   runtimeState.previousCdpPort = void 0;
+  if (runtimeState.managedBrowser === void 0) {
+    runtimeState.managedBrowserCdpPort = void 0;
+  }
   savedStates.clear();
   clearStoredOutputs();
   runtimeState.windowRefs.clear();
@@ -7227,6 +7246,8 @@ async function releaseComputerUseSession() {
   runtimeState.helperDiagnostics = void 0;
   runtimeState.lastPermissionCheckAt = 0;
   resourceScheduler = new ResourceScheduler();
+  if (failures.length === 1) throw failures[0];
+  if (failures.length > 1) throw new AggregateError(failures, "computer-use session release had partial failures (state was reset)");
 }
 function currentRuntimeMode() {
   return isHeadlessMode() ? "stealth" : "default";

package/dist/builtin/extensions/mcp-adapter/index.js

@@ -24491,6 +24491,10 @@ var McpServerManager = class {
   definitions;
   cachePath;
   log;
+  /** Set by dispose(); after it, no new connects and no late attach. */
+  disposed = false;
+  /** Prefixed names claimed by more than one server in the last loadCatalog(); callTool fail-closes on them. */
+  conflictedNames = /* @__PURE__ */ new Set();
   constructor(definitions, cachePath, log) {
     this.definitions = new Map(definitions.map((definition) => [sanitizeName(definition.name), definition]));
     if (this.definitions.size !== definitions.length) {
@@ -24549,6 +24553,7 @@ var McpServerManager = class {
   }
   /** Connect lazily. Concurrent callers share one in-flight connect. */
   async connect(server, signal) {
+    if (this.disposed) throw new Error("MCP server manager is disposed");
     const key = sanitizeName(server);
     const definition = this.definitions.get(key);
     if (!definition) throw new Error(`unknown MCP server "${server}"`);
@@ -24561,12 +24566,17 @@ var McpServerManager = class {
     runtime.connecting = (async () => {
       this.log(`mcp: connecting to "${key}" (${definition.command ?? definition.url})`);
       const { client, transport } = this.startTransport(definition, runtime);
+      runtime.transport = transport;
       client.onclose = () => {
         this.log(`mcp: server "${key}" connection closed`);
         runtime.client = void 0;
       };
       try {
         await withTimeout(client.connect(transport), connectTimeoutFor(definition), `connect "${key}"`, signal);
+        if (this.disposed) {
+          await transport.close().catch(() => void 0);
+          throw new Error(`MCP manager disposed while connecting to "${key}"`);
+        }
         runtime.client = client;
         await this.refreshTools(runtime, connectTimeoutFor(definition));
         return runtime;
@@ -24576,6 +24586,7 @@ var McpServerManager = class {
       }
     })().finally(() => {
       runtime.connecting = void 0;
+      runtime.transport = void 0;
     });
     try {
       return await runtime.connecting;
@@ -24623,7 +24634,14 @@ var McpServerManager = class {
         if (runtime.tools) snapshot[key] = runtime.tools;
       }
       fs2.mkdirSync(path2.dirname(this.cachePath), { recursive: true });
-      fs2.writeFileSync(this.cachePath, JSON.stringify({ updatedAt: Date.now(), tools: snapshot }, null, 2), "utf8");
+      const tempPath = `${this.cachePath}.${process.pid}.tmp`;
+      fs2.writeFileSync(tempPath, JSON.stringify({ updatedAt: Date.now(), tools: snapshot }, null, 2), "utf8");
+      try {
+        fs2.renameSync(tempPath, this.cachePath);
+      } catch (error2) {
+        fs2.unlinkSync(tempPath);
+        throw error2;
+      }
     } catch (error2) {
       this.log(`mcp: metadata cache write failed: ${error2 instanceof Error ? error2.message : String(error2)}`);
     }
@@ -24635,20 +24653,32 @@ var McpServerManager = class {
   /** Metadata catalog: live runtimes plus the last persisted snapshot. */
   loadCatalog() {
     const catalog = /* @__PURE__ */ new Map();
-    const cached2 = this.readCacheFile();
-    for (const tools of Object.values(cached2)) {
+    const ownerOf = /* @__PURE__ */ new Map();
+    const conflicts = /* @__PURE__ */ new Set();
+    const claim2 = (tool, live) => {
+      const owner = ownerOf.get(tool.prefixedName);
+      if (owner !== void 0 && owner !== tool.server) {
+        conflicts.add(tool.prefixedName);
+        return;
+      }
+      if (owner !== void 0 && !live) return;
+      catalog.set(tool.prefixedName, tool);
+      ownerOf.set(tool.prefixedName, tool.server);
+    };
+    for (const tools of Object.values(this.readCacheFile())) {
       for (const tool of tools) {
         if (!this.definitions.has(tool.server)) continue;
-        if (catalog.has(tool.prefixedName)) {
-          this.log(`mcp: metadata cache collision on ${tool.prefixedName} (server "${tool.server}", first seen from another server) \u2014 keeping the first entry`);
-          continue;
-        }
-        catalog.set(tool.prefixedName, tool);
+        claim2(tool, false);
       }
     }
     for (const runtime of this.runtimes.values()) {
-      for (const tool of runtime.tools ?? []) catalog.set(tool.prefixedName, tool);
+      for (const tool of runtime.tools ?? []) claim2(tool, true);
     }
+    for (const name of conflicts) {
+      catalog.delete(name);
+      this.log(`mcp: tool name "${name}" is claimed by more than one server \u2014 disabled until the servers are renamed (fail-closed)`);
+    }
+    this.conflictedNames = conflicts;
     return [...catalog.values()].sort((a, b) => a.prefixedName.localeCompare(b.prefixedName));
   }
   cacheMtime;
@@ -24676,6 +24706,12 @@ var McpServerManager = class {
   }
   async callTool(prefixedName, args, signal) {
     const catalog = this.loadCatalog();
+    if (this.conflictedNames.has(prefixedName)) {
+      return {
+        delivery: "not_sent",
+        error: `MCP tool name "${prefixedName}" is claimed by more than one configured server. Rename one of the servers so every tool name is unambiguous; the call was NOT sent.`
+      };
+    }
     const descriptor = catalog.find((tool) => tool.prefixedName === prefixedName);
     let server = descriptor?.server;
     let toolName = descriptor?.tool;
@@ -24746,12 +24782,17 @@ ${tail}` : "")
   }
   /** Close every client and reap every child process. */
   async dispose() {
+    this.disposed = true;
     for (const runtime of this.runtimes.values()) {
       try {
         await runtime.client?.close();
       } catch {
       }
+      if (!runtime.client && runtime.transport) {
+        await runtime.transport.close().catch(() => void 0);
+      }
       runtime.client = void 0;
+      runtime.transport = void 0;
     }
     this.runtimes.clear();
   }
@@ -24820,10 +24861,14 @@ var DEFAULT_OUTPUT_MAX_BYTES = 8192;
 var SEARCH_RESULT_LIMIT = 20;
 function truncate(text, maxBytes) {
   if (Buffer.byteLength(text, "utf8") <= maxBytes) return text;
-  const cut = Buffer.from(text, "utf8").subarray(0, maxBytes).toString("utf8");
-  return `${cut}
+  const notice = `
 
 \u2026[truncated: result exceeded ${maxBytes} bytes. Narrow the call arguments or read the resource directly.]`;
+  const noticeBytes = Buffer.byteLength(notice, "utf8");
+  const bodyBudget = Math.max(1, maxBytes - noticeBytes);
+  let cut = Buffer.from(text, "utf8").subarray(0, bodyBudget).toString("utf8");
+  if (cut.endsWith("\uFFFD")) cut = cut.slice(0, -1);
+  return `${cut}${notice}`;
 }
 function summarizeParams(schema) {
   if (!schema || typeof schema !== "object") return "";
@@ -25241,8 +25286,8 @@ function mcpAdapter(ctx) {
       commandCtx.ui.notify(lines.join("\n"), "info");
     }
   });
-  ctx.on("session:changed", () => {
-    void reloadManager().catch(() => void 0);
+  ctx.on("session:changed", async () => {
+    await reloadManager();
   });
 }
 export {

package/dist/builtin/extensions/plan-mode/index.js

@@ -7,7 +7,7 @@ import {
   isPlanModeSafeCommand,
   markCompletedPlanSteps,
   restoreExecutionState
-} from "../../../chunks/chunk-LU7RKALH.js";
+} from "../../../chunks/chunk-HH2Q2SNX.js";
 import "../../../chunks/chunk-QJH6YN2W.js";
 import "../../../chunks/chunk-IM43UYVV.js";
 import {

package/dist/builtin/extensions/subagent/index.js

@@ -1,12 +1,12 @@
 import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
 import {
   createCliHostCapabilities
-} from "../../../chunks/chunk-TJ2F4FL5.js";
+} from "../../../chunks/chunk-G3SE3I2L.js";
 import "../../../chunks/chunk-5353DEEC.js";
 import {
   createSharedFileSystemTools,
   createSubagentCapability
-} from "../../../chunks/chunk-LU7RKALH.js";
+} from "../../../chunks/chunk-HH2Q2SNX.js";
 import "../../../chunks/chunk-QJH6YN2W.js";
 import "../../../chunks/chunk-IM43UYVV.js";
 import "../../../chunks/chunk-XXGYUYXI.js";

package/dist/builtin/extensions/todo/index.js

@@ -8,7 +8,7 @@ import {
   executeManageTodoList,
   executionStateToTodos,
   restoreExecutionState
-} from "../../../chunks/chunk-LU7RKALH.js";
+} from "../../../chunks/chunk-HH2Q2SNX.js";
 import "../../../chunks/chunk-QJH6YN2W.js";
 import "../../../chunks/chunk-IM43UYVV.js";
 import "../../../chunks/chunk-XXGYUYXI.js";

package/dist/builtin/skills/tui-use/tui-use-cli.cjs

@@ -9999,6 +9999,16 @@ function validateRequest(value) {
       string("command", true, 32768);
       string("cwd");
       string("label", false, 1024);
+      if (r.env !== void 0) {
+        if (!r.env || typeof r.env !== "object" || Array.isArray(r.env)) throw new Error("Invalid env: expected an object of string to string.");
+        const entries = Object.entries(r.env);
+        if (entries.length > 512) throw new Error("Invalid env: at most 512 variables.");
+        for (const [k, v] of entries) {
+          if (!k.length || k.length > 256 || k.includes("\0") || typeof v !== "string" || v.length > 65536 || v.includes("\0")) {
+            throw new Error(`Invalid env entry ${JSON.stringify(k)}: expected string key and value without NUL bytes.`);
+          }
+        }
+      }
       if (r.argv !== void 0 && (!Array.isArray(r.argv) || r.argv.length < 1 || r.argv.length > 256 || r.argv.some((a) => typeof a !== "string" || a.includes("\0") || a.length > 32768) || !r.argv[0])) throw new Error("Invalid argv: expected a non-empty array of executable and arguments.");
     // Fall through to the shared PTY size limits.
     case "resize":
@@ -10294,6 +10304,13 @@ var SUPPORTED_KEYS = Object.keys(KEY_MAP);
 var import_node_fs2 = require("node:fs");
 var import_node_path3 = __toESM(require("node:path"));
 var program2 = new Command();
+function callerEnv() {
+  const env = {};
+  for (const [key, value] of Object.entries(process.env)) {
+    if (typeof value === "string") env[key] = value;
+  }
+  return env;
+}
 program2.name("tui-use").description("TUI automation for AI agents \u2014 like BrowserUse, but for the terminal").option("--session <id>", "Target this session explicitly (safe for concurrent tasks)").version(version);
 program2.command("start [args...]").description("Start an interactive program in a PTY session").option("--cwd <dir>", "Working directory for the command").option("--label <name>", "Human-readable label for this session").option("--cols <n>", "Terminal width (default: 120)", "120").option("--rows <n>", "Terminal height (default: 30)", "30").option("--argv-stdin", "Read a JSON array of executable and arguments from UTF-8 stdin; bypass the shell").action(async (args, opts) => {
   let argv;
@@ -10311,6 +10328,7 @@ program2.command("start [args...]").description("Start an interactive program in
     type: "start",
     command,
     cwd: import_node_path3.default.resolve(opts.cwd ?? process.cwd()),
+    env: callerEnv(),
     label: opts.label,
     cols: Number(opts.cols),
     rows: Number(opts.rows),

package/dist/builtin/skills/tui-use/tui-use-daemon.cjs

@@ -7267,7 +7267,10 @@ var Session = class {
         rows,
         useConptyDll: process.platform === "win32",
         cwd: options.cwd ?? process.cwd(),
-        env: process.env
+        // The caller's env, not the daemon starter's: a persistent daemon
+        // outlives many projects, and each session must see the environment
+        // of the shell that started it (PATH setups, PIE_* overrides, ...).
+        env: options.env ?? process.env
       });
     } catch (error) {
       this.terminal.dispose();
@@ -7606,6 +7609,16 @@ function validateRequest(value) {
       string("command", true, 32768);
       string("cwd");
       string("label", false, 1024);
+      if (r.env !== void 0) {
+        if (!r.env || typeof r.env !== "object" || Array.isArray(r.env)) throw new Error("Invalid env: expected an object of string to string.");
+        const entries = Object.entries(r.env);
+        if (entries.length > 512) throw new Error("Invalid env: at most 512 variables.");
+        for (const [k, v] of entries) {
+          if (!k.length || k.length > 256 || k.includes("\0") || typeof v !== "string" || v.length > 65536 || v.includes("\0")) {
+            throw new Error(`Invalid env entry ${JSON.stringify(k)}: expected string key and value without NUL bytes.`);
+          }
+        }
+      }
       if (r.argv !== void 0 && (!Array.isArray(r.argv) || r.argv.length < 1 || r.argv.length > 256 || r.argv.some((a) => typeof a !== "string" || a.includes("\0") || a.length > 32768) || !r.argv[0])) throw new Error("Invalid argv: expected a non-empty array of executable and arguments.");
     // Fall through to the shared PTY size limits.
     case "resize":

package/dist/chunks/{chunk-TJ2F4FL5.js → chunk-G3SE3I2L.js}

@@ -12,7 +12,7 @@ import {
   createSharedWebSearchTool,
   interpretShellExit,
   requestInteraction
-} from "./chunk-LU7RKALH.js";
+} from "./chunk-HH2Q2SNX.js";
 import {
   Type
 } from "./chunk-XXGYUYXI.js";

package/dist/chunks/{chunk-LU7RKALH.js → chunk-HH2Q2SNX.js}

@@ -4856,7 +4856,7 @@ var AgentSessionRetryController = class {
       }
       return;
     }
-    const maxRetries = this.retry.maxRetries ?? 3;
+    const maxRetries = this.retry.maxRetriesForFailure?.(failure) ?? this.retry.maxRetries ?? 3;
     this.retryAttempt += 1;
     if (this.retryAttempt > maxRetries) {
       const terminalFailure = asTerminalFailure(failure);
@@ -4869,7 +4869,8 @@ var AgentSessionRetryController = class {
       return;
     }
     const attempt = this.retryAttempt;
-    const delayMs = (this.retry.baseDelayMs ?? 2e3) * 2 ** (attempt - 1);
+    const rawDelayMs = (this.retry.baseDelayMs ?? 2e3) * 2 ** (attempt - 1);
+    const delayMs = this.retry.maxDelayMs !== void 0 ? Math.min(rawDelayMs, this.retry.maxDelayMs) : rawDelayMs;
     this.ensurePending();
     this.setRecoveryState("retry_pending");
     this.emit({ type: "turn_recovery_pending", recovery: "retry", failure, attempt, maxRetries, delayMs });

package/dist/cli.js

@@ -7,7 +7,7 @@ import {
   createCliWebResearchHealthStore,
   formatSize,
   truncateHeadTail
-} from "./chunks/chunk-TJ2F4FL5.js";
+} from "./chunks/chunk-G3SE3I2L.js";
 import {
   VERSION,
   getAgentDir,
@@ -78,7 +78,7 @@ import {
   setExtensionEnabledInSettings,
   shouldPreserveExecutionStateForUserText,
   supersedeExecutionState
-} from "./chunks/chunk-LU7RKALH.js";
+} from "./chunks/chunk-HH2Q2SNX.js";
 import "./chunks/chunk-QJH6YN2W.js";
 import {
   Deref,
@@ -77119,7 +77119,7 @@ function stripStaleUsageBeforeCompaction(messages, latestCompactionTimestamp) {
 }
 
 // src/runtime/retry-policy.ts
-var TRANSIENT_NETWORK_ERROR_PATTERN = /sse stream aborted|sse stream closed before end|sse stream ended before response completed|(?:read\s+)?ECONNRESET|ETIMEDOUT|EAI_AGAIN|ENOTFOUND|ECONNREFUSED|socket hang up|request timeout|client network socket disconnected before secure TLS connection was established|connection.?error|connection.?refused|other side closed|fetch failed|upstream.?connect|reset before headers|terminated/i;
+var TRANSIENT_NETWORK_ERROR_PATTERN = /sse stream aborted|sse stream closed before end|sse stream ended before response completed|(?:read\s+)?ECONNRESET|ETIMEDOUT|EAI_AGAIN|ENOTFOUND|ECONNREFUSED|(?:read\s+|write\s+)?EPIPE|socket hang up|request timeout|client network socket disconnected before secure TLS connection was established|certificate (?:has )?expired|unable to verify (?:the first certificate|certificate)|self[- ]signed cert|connection.?error|connection.?refused|other side closed|fetch failed|upstream.?connect|reset before headers|terminated/i;
 var PROVIDER_TRANSIENT_ERROR_PATTERN = /overloaded|rate.?limit|too many requests|\b429\b|\b500\b|\b502\b|\b503\b|\b504\b|service.?unavailable|server error|internal error|retry delay/i;
 var USER_ABORT_PATTERN = /request was aborted|local abort signal|user abort|user_interrupt|interrupted by user|esc pressed|ctrl\+c/i;
 var AUTH_CONFIG_PATTERN = /missing api key|invalid api key|api key.*invalid|unauthorized|\b401\b|\b403\b|forbidden|authentication failed|no configured model|no models? are configured/i;
@@ -77127,6 +77127,8 @@ var INVALID_FILE_REFERENCE_PATTERN = /not the owner of this file|owner of this f
 var INVALID_REQUEST_PATTERN = /bad request|\b400\b|invalid request|invalid schema|schema validation|invalid parameter|unsupported parameter|malformed/i;
 var CLI_RETRY_MAX_RETRIES = 3;
 var CLI_RETRY_BASE_DELAY_MS = 2e3;
+var CLI_TRANSIENT_NETWORK_MAX_RETRIES = 10;
+var CLI_RETRY_MAX_DELAY_MS = 6e4;
 function isCliContextOverflowError(errorMessage4) {
   return /context length exceeded|context_length_exceeded|maximum context length|maximum context size|too many tokens|input.*too long|prompt.*too long|context window|token limit exceeded|exceeds.{0,20}context/i.test(
     errorMessage4
@@ -77203,6 +77205,31 @@ function getCliRetryBaseDelayMs() {
   const parsed = Number(override);
   return Number.isFinite(parsed) && parsed >= 0 ? parsed : CLI_RETRY_BASE_DELAY_MS;
 }
+function getCliTransientNetworkMaxRetries() {
+  const override = process.env.PIE_RETRY_TRANSIENT_NETWORK_MAX_RETRIES;
+  if (!override) {
+    return CLI_TRANSIENT_NETWORK_MAX_RETRIES;
+  }
+  const parsed = Number(override);
+  return Number.isFinite(parsed) && parsed >= 0 ? Math.floor(parsed) : CLI_TRANSIENT_NETWORK_MAX_RETRIES;
+}
+function getCliRetryMaxDelayMs() {
+  const override = process.env.PIE_RETRY_MAX_DELAY_MS;
+  if (!override) {
+    return CLI_RETRY_MAX_DELAY_MS;
+  }
+  const parsed = Number(override);
+  return Number.isFinite(parsed) && parsed >= 0 ? parsed : CLI_RETRY_MAX_DELAY_MS;
+}
+function createCliRetryPolicy(isRetryableErrorOverride) {
+  return {
+    isRetryableError: isRetryableErrorOverride ?? isCliRetryableError,
+    maxRetries: CLI_RETRY_MAX_RETRIES,
+    baseDelayMs: getCliRetryBaseDelayMs(),
+    maxDelayMs: getCliRetryMaxDelayMs(),
+    maxRetriesForFailure: (failure) => failure.kind === "transient_network" ? getCliTransientNetworkMaxRetries() : CLI_RETRY_MAX_RETRIES
+  };
+}
 
 // src/runtime/interactive-context-ui.ts
 function calculateContextUsage(host) {
@@ -77921,11 +77948,9 @@ function createInteractiveAgentSessionController(host) {
     },
     getAutoContinueMessage: () => host.getTodoAutoContinueMessage(),
     classifyFailure: (context) => classifyCliFailure(context),
-    retry: {
-      isRetryableError: (errorMessage4) => !host._isContextOverflowError(errorMessage4) && isCliRetryableError(errorMessage4),
-      maxRetries: CLI_RETRY_MAX_RETRIES,
-      baseDelayMs: getCliRetryBaseDelayMs()
-    },
+    retry: createCliRetryPolicy(
+      (errorMessage4) => !host._isContextOverflowError(errorMessage4) && isCliRetryableError(errorMessage4)
+    ),
     compaction: {
       compact: async (context) => {
         const beforeCompact = await host.extensionRunner?.emitSessionBeforeCompact({
@@ -80152,6 +80177,24 @@ async function composeCliRuntime(params) {
   };
 }
 
+// src/app/print-mode-retry-status.ts
+function createPrintModeRetryStatusReporter() {
+  let exhaustedRetryBudget;
+  return (event) => {
+    if (event.type === "turn_recovery_pending" && event.recovery === "retry") {
+      console.error(`Transient request error: ${event.failure.message}. Retrying in ${(event.delayMs ?? 0) / 1e3}s (${event.attempt}/${event.maxRetries}).`);
+    } else if (event.type === "turn_retry_started") {
+      console.error(`Retrying request (${event.attempt}/${event.maxRetries})...`);
+    } else if (event.type === "turn_retry_succeeded") {
+      console.error(`Retry succeeded after ${event.attempts} attempt${event.attempts === 1 ? "" : "s"}.`);
+    } else if (event.type === "retry_exhausted") {
+      exhaustedRetryBudget = event.maxRetries;
+    } else if (event.type === "turn_failed_terminal" && (event.failure.kind === "transient_network" || event.failure.kind === "provider_transient")) {
+      console.error(`Request failed after ${exhaustedRetryBudget ?? CLI_RETRY_MAX_RETRIES} retries: ${event.failure.message}`);
+    }
+  };
+}
+
 // src/app/print-mode.ts
 function createNoOpExtensionUIContext() {
   return {
@@ -80329,6 +80372,7 @@ async function runPrintMode(params) {
     streamFn: params.streamFn
   });
   const sessionTrace = new SessionTraceWriter(runtime.paths.sessionsDir, runtime.sessionManager);
+  const reportRetryStatus = createPrintModeRetryStatusReporter();
   const agentSessionController = new AgentSessionController({
     agent,
     sessionManager: runtime.sessionManager,
@@ -80349,11 +80393,7 @@ async function runPrintMode(params) {
         messages: result.messages?.length ? [...context.messages, ...result.messages] : context.messages
       };
     },
-    retry: {
-      isRetryableError: isCliRetryableError,
-      maxRetries: CLI_RETRY_MAX_RETRIES,
-      baseDelayMs: getCliRetryBaseDelayMs()
-    },
+    retry: createCliRetryPolicy(),
     getAutoCompactDecision: ({ message, messages }) => createCliAutoCompactDecision({
       message,
       messages,
@@ -80388,20 +80428,14 @@ async function runPrintMode(params) {
     classifyFailure: (context) => classifyCliFailure(context),
     onEvent: (event) => {
       sessionTrace.noteRuntimeEvent(event.type, event);
-      if (event.type === "turn_recovery_pending" && event.recovery === "retry") {
-        console.error(`Transient request error: ${event.failure.message}. Retrying in ${(event.delayMs ?? 0) / 1e3}s (${event.attempt}/${event.maxRetries}).`);
-      } else if (event.type === "turn_recovery_pending" && event.recovery === "compaction") {
+      if (event.type === "turn_recovery_pending" && event.recovery === "compaction") {
         console.error(event.failure.kind === "context_overflow" ? "Context overflow detected, compacting..." : "Auto-compacting context before retry...");
       } else if (event.type === "auto_compaction_start") {
         console.error(event.reason === "overflow" ? "Context overflow detected, compacting..." : "Auto-compacting context...");
       } else if (event.type === "auto_compaction_end" && !event.errorMessage) {
         runtime.fileObservationState.clear();
-      } else if (event.type === "turn_retry_started") {
-        console.error(`Retrying request (${event.attempt}/${event.maxRetries})...`);
-      } else if (event.type === "turn_retry_succeeded") {
-        console.error(`Retry succeeded after ${event.attempts} attempt${event.attempts === 1 ? "" : "s"}.`);
-      } else if (event.type === "turn_failed_terminal" && (event.failure.kind === "transient_network" || event.failure.kind === "provider_transient")) {
-        console.error(`Request failed after ${CLI_RETRY_MAX_RETRIES} retries: ${event.failure.message}`);
+      } else {
+        reportRetryStatus(event);
       }
     }
   });
@@ -81518,11 +81552,7 @@ async function runJsonEventsMode(params) {
           messages: result?.messages?.length ? [...context.messages, ...result.messages] : context.messages
         };
       },
-      retry: {
-        isRetryableError: isCliRetryableError,
-        maxRetries: CLI_RETRY_MAX_RETRIES,
-        baseDelayMs: getCliRetryBaseDelayMs()
-      },
+      retry: createCliRetryPolicy(),
       getAutoCompactDecision: ({ message, messages }) => createCliAutoCompactDecision({
         message,
         messages,

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@cydm/pie",
-	"version": "2.1.1",
+	"version": "2.2.0",
 	"description": "Pie AI Agent CLI",
 	"type": "module",
 	"bin": {