zidane 7.1.5 → 7.1.8

package/dist/restate.js

@@ -1,6 +1,6 @@
 import { L as buildAliasMaps } from "./openai-compat-Dy56weK8.js";
 import { i as coerceToolResultContent, l as toolOutputByteLength, u as toolResultToText } from "./types-CWgENVLY.js";
-import { At as reserveChildLabel, a as stagedToolState, ar as hydrateReadStateFromSessionAsync, i as isToolStateOps } from "./agent-BlGt76qk.js";
+import { At as reserveChildLabel, a as stagedToolState, ar as hydrateReadStateFromSessionAsync, i as isToolStateOps } from "./agent-CJbmpTmA.js";
 import { r as formatTokenUsage } from "./stats-DAKBEKjc.js";
 import { createHash } from "node:crypto";
 //#region src/restate/awakeable.ts
@@ -22,15 +22,28 @@ import { createHash } from "node:crypto";
 */
 function awakeableWithTimeout(ctx, options = {}) {
 	const awakeable = ctx.awakeable();
-	const timeoutMs = options.timeoutMs;
-	if (typeof timeoutMs !== "number" || !Number.isFinite(timeoutMs) || timeoutMs <= 0) return awakeable;
-	const timer = ctx.sleep(timeoutMs);
-	const raced = options.promises ? options.promises.race([awakeable.promise, timer]) : Promise.race([awakeable.promise, timer]);
 	return {
 		id: awakeable.id,
-		promise: raced.then((value) => value === void 0 ? null : value)
+		promise: raceWithTimeout(ctx, awakeable.promise, options)
 	};
 }
+/**
+* Race an already-minted awakeable promise against a durable
+* `ctx.sleep(timeoutMs)`. Split out from {@link awakeableWithTimeout} so a
+* caller that must run journaled steps between minting and parking (see
+* `restateWaitBackground`) can arm the race at park time: a
+* `RestatePromise.race` starts awaiting the SDK's single-waiter progress
+* channel as soon as it is built, so building it early and then awaiting a
+* `ctx.run` is two concurrent waiters — `awaitNext already pending`.
+*
+* Without a positive `timeoutMs` the promise is returned untouched.
+*/
+function raceWithTimeout(ctx, promise, options = {}) {
+	const timeoutMs = options.timeoutMs;
+	if (typeof timeoutMs !== "number" || !Number.isFinite(timeoutMs) || timeoutMs <= 0) return promise;
+	const timer = ctx.sleep(timeoutMs);
+	return (options.promises ? options.promises.race([promise, timer]) : Promise.race([promise, timer])).then((value) => value === void 0 ? null : value);
+}
 //#endregion
 //#region src/restate/awakeable-tool.ts
 /**
@@ -1491,7 +1504,7 @@ function entryToExit(entry) {
 }
 /**
 * Park on `promise` until it settles, the caller aborts, or (already handled
-* by `awakeableWithTimeout`) the durable deadline elapses. Abort and rejection
+* by `raceWithTimeout`) the durable deadline elapses. Abort and rejection
 * both resolve `null` — the `waitBackground` seam's contract is "exit info, or
 * null when the task didn't cleanly exit before we stopped waiting".
 */
@@ -1532,10 +1545,7 @@ function restateWaitBackground(options) {
 			}, runOpts);
 			return exit;
 		};
-		const awakeable = awakeableWithTimeout(ctx, {
-			...timeoutMs !== void 0 ? { timeoutMs } : {},
-			...options.promises ? { promises: options.promises } : {}
-		});
+		const awakeable = ctx.awakeable();
 		const marker = await ctx.run(`${entry}-register`, async () => {
 			options.replayTracker?.markLive();
 			await register(awakeable.id, taskId, handle);
@@ -1551,10 +1561,13 @@ function restateWaitBackground(options) {
 		}, runOpts)).find((e) => e.taskId === taskId);
 		if (!found) return finalize(null, awakeable.id);
 		if (found.status !== "running") return finalize(entryToExit(found), awakeable.id);
-		return finalize(await parkUntilAbort(awakeable.promise, signal), awakeable.id);
+		return finalize(await parkUntilAbort(raceWithTimeout(ctx, awakeable.promise, {
+			...timeoutMs !== void 0 ? { timeoutMs } : {},
+			...options.promises ? { promises: options.promises } : {}
+		}), signal), awakeable.id);
 	};
 }
 //#endregion
-export { DEFAULT_TOOL_STATE_KEY, awakeableWithTimeout, createReplayTracker, deterministicUuid, replayAwareAbort, restateAwakeableTool, restateBehavior, restateClock, restateMcpToolWrap, restateProvider, restateSessionStore, restateSpawnTool, restateTool, restateToolBatchExecutor, restateToolStateStore, restateWaitBackground, shouldRethrowRestateControlError, wrapAgentTools };
+export { DEFAULT_TOOL_STATE_KEY, awakeableWithTimeout, createReplayTracker, deterministicUuid, raceWithTimeout, replayAwareAbort, restateAwakeableTool, restateBehavior, restateClock, restateMcpToolWrap, restateProvider, restateSessionStore, restateSpawnTool, restateTool, restateToolBatchExecutor, restateToolStateStore, restateWaitBackground, shouldRethrowRestateControlError, wrapAgentTools };
 
 //# sourceMappingURL=restate.js.map