@cydm/pie 1.0.14 → 1.0.15

package/dist/chunks/{chunk-LF5Q7BVU.js → chunk-R5LYKDKA.js}

@@ -1,6 +1,8 @@
 import { createRequire as __createRequire } from "node:module"; const require = __createRequire(import.meta.url);
 import {
-  Agent,
+  Agent
+} from "./chunk-VE2HDCNB.js";
+import {
   FileSystemGateway,
   Type,
   completeSimple,
@@ -8,7 +10,7 @@ import {
   getFileSystem,
   getPlatformConfig,
   streamSimple
-} from "./chunk-D7NAXU7F.js";
+} from "./chunk-5DA2D3K2.js";
 import {
   __require
 } from "./chunk-TG2EQLX2.js";
@@ -3279,8 +3281,13 @@ var AgentSessionRetryController = class {
   continueTurn;
   retryAttempt = 0;
   retryTimer;
+  pendingRetry;
+  resolvePendingRetry;
   get hasPendingRetry() {
-    return this.retryTimer !== void 0;
+    return this.retryTimer !== void 0 || this.pendingRetry !== void 0;
+  }
+  waitForRetry() {
+    return this.pendingRetry ?? Promise.resolve();
   }
   async handleTurnEnd(message) {
     if (!this.retry || !message || message.role !== "assistant") {
@@ -3291,10 +3298,15 @@ var AgentSessionRetryController = class {
       if (this.retryAttempt > 0) {
         this.emit({ type: "retry_succeeded", attempts: this.retryAttempt });
         this.retryAttempt = 0;
+        this.resolvePending();
       }
       return;
     }
     if (!this.retry.isRetryableError(errorMessage)) {
+      if (this.retryAttempt > 0) {
+        this.retryAttempt = 0;
+        this.resolvePending();
+      }
       return;
     }
     const maxRetries = this.retry.maxRetries ?? 3;
@@ -3302,10 +3314,12 @@ var AgentSessionRetryController = class {
     if (this.retryAttempt > maxRetries) {
       this.emit({ type: "retry_exhausted", errorMessage, maxRetries });
       this.retryAttempt = 0;
+      this.resolvePending();
       return;
     }
     const attempt = this.retryAttempt;
     const delayMs = (this.retry.baseDelayMs ?? 2e3) * 2 ** (attempt - 1);
+    this.ensurePending();
     this.emit({ type: "retry_scheduled", errorMessage, attempt, maxRetries, delayMs });
     this.cancelTimerOnly();
     const setTimer = this.retry.setTimeout ?? defaultSetTimeout4;
@@ -3313,13 +3327,21 @@ var AgentSessionRetryController = class {
       this.retryTimer = void 0;
       this.emit({ type: "retry_start", attempt, maxRetries });
       this.removeTrailingAssistantError();
-      void this.continueTurn().catch(() => {
+      void this.continueTurn().catch((error) => {
+        this.emit({
+          type: "retry_exhausted",
+          errorMessage: error instanceof Error ? error.message : String(error),
+          maxRetries
+        });
+        this.retryAttempt = 0;
+        this.resolvePending();
       });
     }, delayMs);
   }
   cancel() {
     this.cancelTimerOnly();
     this.retryAttempt = 0;
+    this.resolvePending();
   }
   cancelTimerOnly() {
     if (this.retryTimer === void 0) {
@@ -3329,6 +3351,20 @@ var AgentSessionRetryController = class {
     clearTimer(this.retryTimer);
     this.retryTimer = void 0;
   }
+  ensurePending() {
+    if (this.pendingRetry) {
+      return;
+    }
+    this.pendingRetry = new Promise((resolve2) => {
+      this.resolvePendingRetry = resolve2;
+    });
+  }
+  resolvePending() {
+    const resolve2 = this.resolvePendingRetry;
+    this.pendingRetry = void 0;
+    this.resolvePendingRetry = void 0;
+    resolve2?.();
+  }
 };
 
 // ../../packages/agent-framework/src/session/controller/runtime-guard.ts
@@ -3735,6 +3771,15 @@ var AgentSessionController = class {
   waitForIdle() {
     return this.agent.waitForIdle();
   }
+  async waitForSettled() {
+    while (true) {
+      await this.agent.waitForIdle();
+      if (!this.retryController.hasPendingRetry) {
+        return;
+      }
+      await this.retryController.waitForRetry();
+    }
+  }
   async runAutoCompact(reason, willRetry = false) {
     return runAgentSessionCompaction({
       reason,
@@ -3833,6 +3878,10 @@ var AgentSessionController = class {
     if (this.disposed || this.runtimeGuard.triggered) {
       return;
     }
+    const assistantError = message?.role === "assistant" && message.stopReason === "error" && Boolean(message.errorMessage);
+    if (!assistantError) {
+      await this.retryController.handleTurnEnd(message);
+    }
     const compactDecision = this.getAutoCompactDecision?.({
       agent: this.agent,
       sessionManager: this.sessionManager,
@@ -3844,7 +3893,9 @@ var AgentSessionController = class {
       this.scheduleAutoCompact(compactDecision);
       return;
     }
-    await this.retryController.handleTurnEnd(message);
+    if (assistantError) {
+      await this.retryController.handleTurnEnd(message);
+    }
   }
   async handleAgentEndHook(messages) {
     if (this.disposed || this.runtimeGuard.triggered) {