@springbrand/agent-runtime 0.3.0-alpha.28 → 0.3.0-alpha.29

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@springbrand/agent-runtime",
-  "version": "0.3.0-alpha.28",
+  "version": "0.3.0-alpha.29",
   "type": "module",
   "files": [
     "src",

package/src/adapter/cloudflare/agent.ts

@@ -780,7 +780,38 @@ export abstract class AgentRuntimeKernel<
     submission: StoredSubmission,
     mode: SubmissionExecutionMode,
   ): Promise<StoredSubmission> {
-    return this.keepAliveWhile(() => this.runHarnessSubmission(submission, mode));
+    return this.holdResidentWhile(() => this.runHarnessSubmission(submission, mode));
+  }
+
+  /**
+   * `keepAliveWhile` whose release finishes before `run`'s result settles.
+   *
+   * The SDK disposer hands the alarm re-sync to `ctx.waitUntil`. The Turn's last
+   * await can be resumed by another Durable Object in the same isolate, and that
+   * registration then fails as cross-DO I/O (`RefcountedFulfiller`) after the Turn
+   * has already finished. Awaiting the re-sync inline needs no waitUntil. A facet
+   * borrows its root's heartbeat over RPC, so it keeps the SDK path.
+   * Relies on agents 0.28 internals; the peer range pins that minor.
+   */
+  private async holdResidentWhile<T>(run: () => Promise<T>): Promise<T> {
+    const host = this as unknown as {
+      _isFacet: boolean;
+      _keepAliveRefs: number;
+      _syncHostJobs(): Promise<void>;
+    };
+    if (host._isFacet) return this.keepAliveWhile(run);
+    host._keepAliveRefs += 1;
+    if (host._keepAliveRefs === 1) await host._syncHostJobs();
+    try {
+      return await run();
+    } finally {
+      host._keepAliveRefs = Math.max(0, host._keepAliveRefs - 1);
+      if (host._keepAliveRefs === 0) {
+        await host._syncHostJobs().catch((error: unknown) => {
+          console.error("[keep-alive] heartbeat release failed", error);
+        });
+      }
+    }
   }
 
   /** Run or recover the Submission through Pi's single durable AgentHarness. */