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

package/package.json

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

package/src/adapter/cloudflare/agent.ts

@@ -766,8 +766,56 @@ export abstract class AgentRuntimeKernel<
     }
   }
 
+  /**
+   * Run or recover the Submission while an alarm heartbeat keeps the DO resident.
+   *
+   * A Turn makes only outbound model and Tool calls, which do not count as DO
+   * activity. Once every client disconnects, an unheld DO is evicted after
+   * 70-140s, taking the in-memory Turn and its stall watchdog with it while the
+   * row stays `running`. If the DO is still evicted (deploy, runtime restart),
+   * the pending heartbeat alarm wakes it and `onStart` recovers the Submission.
+   * A Turn parked for approval or input returns `released`, which drops the hold.
+   */
+  private executeHarnessSubmission(
+    submission: StoredSubmission,
+    mode: SubmissionExecutionMode,
+  ): Promise<StoredSubmission> {
+    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. */
-  private async executeHarnessSubmission(
+  private async runHarnessSubmission(
     submission: StoredSubmission,
     mode: SubmissionExecutionMode,
   ): Promise<StoredSubmission> {