@rivetkit/workflow-engine 0.0.0-2-3-0-rc-rollback.f0e199e → 0.0.0-claude-edge-bundle-slim.fb63356

package/dist/tsup/index.cjs

@@ -43,7 +43,7 @@
 
 
 
-var _chunkA33YVYOFcjs = require('./chunk-A33YVYOF.cjs');
+var _chunkGQWOLYBAcjs = require('./chunk-GQWOLYBA.cjs');
 
 
 
@@ -89,5 +89,5 @@ var _chunkA33YVYOFcjs = require('./chunk-A33YVYOF.cjs');
 
 
 
-exports.CancelledError = _chunkA33YVYOFcjs.CancelledError; exports.CriticalError = _chunkA33YVYOFcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkA33YVYOFcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkA33YVYOFcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkA33YVYOFcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkA33YVYOFcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkA33YVYOFcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkA33YVYOFcjs.EntryInProgressError; exports.EvictedError = _chunkA33YVYOFcjs.EvictedError; exports.HistoryDivergedError = _chunkA33YVYOFcjs.HistoryDivergedError; exports.JoinError = _chunkA33YVYOFcjs.JoinError; exports.Loop = _chunkA33YVYOFcjs.Loop; exports.MessageWaitError = _chunkA33YVYOFcjs.MessageWaitError; exports.RaceError = _chunkA33YVYOFcjs.RaceError; exports.RollbackCheckpointError = _chunkA33YVYOFcjs.RollbackCheckpointError; exports.RollbackError = _chunkA33YVYOFcjs.RollbackError; exports.SleepError = _chunkA33YVYOFcjs.SleepError; exports.StepExhaustedError = _chunkA33YVYOFcjs.StepExhaustedError; exports.StepFailedError = _chunkA33YVYOFcjs.StepFailedError; exports.WorkflowContextImpl = _chunkA33YVYOFcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkA33YVYOFcjs.appendLoopIteration; exports.appendName = _chunkA33YVYOFcjs.appendName; exports.createEntry = _chunkA33YVYOFcjs.createEntry; exports.createHistorySnapshot = _chunkA33YVYOFcjs.createHistorySnapshot; exports.createStorage = _chunkA33YVYOFcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkA33YVYOFcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkA33YVYOFcjs.emptyLocation; exports.extractErrorInfo = _chunkA33YVYOFcjs.extractErrorInfo; exports.flush = _chunkA33YVYOFcjs.flush; exports.generateId = _chunkA33YVYOFcjs.generateId; exports.getEntry = _chunkA33YVYOFcjs.getEntry; exports.getOrCreateMetadata = _chunkA33YVYOFcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkA33YVYOFcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkA33YVYOFcjs.isLoopIterationMarker; exports.loadMetadata = _chunkA33YVYOFcjs.loadMetadata; exports.loadStorage = _chunkA33YVYOFcjs.loadStorage; exports.locationToKey = _chunkA33YVYOFcjs.locationToKey; exports.locationsEqual = _chunkA33YVYOFcjs.locationsEqual; exports.parentLocation = _chunkA33YVYOFcjs.parentLocation; exports.registerName = _chunkA33YVYOFcjs.registerName; exports.replayWorkflowFromStep = _chunkA33YVYOFcjs.replayWorkflowFromStep; exports.resolveName = _chunkA33YVYOFcjs.resolveName; exports.runWorkflow = _chunkA33YVYOFcjs.runWorkflow; exports.setEntry = _chunkA33YVYOFcjs.setEntry;
+exports.CancelledError = _chunkGQWOLYBAcjs.CancelledError; exports.CriticalError = _chunkGQWOLYBAcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkGQWOLYBAcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkGQWOLYBAcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkGQWOLYBAcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkGQWOLYBAcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkGQWOLYBAcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkGQWOLYBAcjs.EntryInProgressError; exports.EvictedError = _chunkGQWOLYBAcjs.EvictedError; exports.HistoryDivergedError = _chunkGQWOLYBAcjs.HistoryDivergedError; exports.JoinError = _chunkGQWOLYBAcjs.JoinError; exports.Loop = _chunkGQWOLYBAcjs.Loop; exports.MessageWaitError = _chunkGQWOLYBAcjs.MessageWaitError; exports.RaceError = _chunkGQWOLYBAcjs.RaceError; exports.RollbackCheckpointError = _chunkGQWOLYBAcjs.RollbackCheckpointError; exports.RollbackError = _chunkGQWOLYBAcjs.RollbackError; exports.SleepError = _chunkGQWOLYBAcjs.SleepError; exports.StepExhaustedError = _chunkGQWOLYBAcjs.StepExhaustedError; exports.StepFailedError = _chunkGQWOLYBAcjs.StepFailedError; exports.WorkflowContextImpl = _chunkGQWOLYBAcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkGQWOLYBAcjs.appendLoopIteration; exports.appendName = _chunkGQWOLYBAcjs.appendName; exports.createEntry = _chunkGQWOLYBAcjs.createEntry; exports.createHistorySnapshot = _chunkGQWOLYBAcjs.createHistorySnapshot; exports.createStorage = _chunkGQWOLYBAcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkGQWOLYBAcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkGQWOLYBAcjs.emptyLocation; exports.extractErrorInfo = _chunkGQWOLYBAcjs.extractErrorInfo; exports.flush = _chunkGQWOLYBAcjs.flush; exports.generateId = _chunkGQWOLYBAcjs.generateId; exports.getEntry = _chunkGQWOLYBAcjs.getEntry; exports.getOrCreateMetadata = _chunkGQWOLYBAcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkGQWOLYBAcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkGQWOLYBAcjs.isLoopIterationMarker; exports.loadMetadata = _chunkGQWOLYBAcjs.loadMetadata; exports.loadStorage = _chunkGQWOLYBAcjs.loadStorage; exports.locationToKey = _chunkGQWOLYBAcjs.locationToKey; exports.locationsEqual = _chunkGQWOLYBAcjs.locationsEqual; exports.parentLocation = _chunkGQWOLYBAcjs.parentLocation; exports.registerName = _chunkGQWOLYBAcjs.registerName; exports.replayWorkflowFromStep = _chunkGQWOLYBAcjs.replayWorkflowFromStep; exports.resolveName = _chunkGQWOLYBAcjs.resolveName; exports.runWorkflow = _chunkGQWOLYBAcjs.runWorkflow; exports.setEntry = _chunkGQWOLYBAcjs.setEntry;
 //# sourceMappingURL=index.cjs.map

package/dist/tsup/index.d.cts

@@ -345,6 +345,8 @@ interface StepConfig<T> {
     retryBackoffMax?: number;
     /** Timeout in ms for step execution (default: 30000). Set to 0 to disable. */
     timeout?: number;
+    /** If true, step timeouts retry like any other error instead of failing immediately as critical. Default: false. */
+    retryOnTimeout?: boolean;
 }
 type TryStepCatchKind = "critical" | "timeout" | "exhausted" | "rollback";
 interface TryStepFailure {
@@ -398,7 +400,7 @@ type LoopResult<S, T> = {
  * Stateless loops (state = undefined) may return void/undefined, which is treated as
  * `Loop.continue(undefined)`.
  */
-type LoopIterationResult<S, T> = Promise<LoopResult<S, T> | (S extends undefined ? void : never)>;
+type LoopIterationResult<S, T> = Promise<LoopResult<S, T> | (S extends undefined ? undefined : never)>;
 /**
  * Configuration for a loop.
  */
@@ -697,12 +699,12 @@ declare class WorkflowContextImpl implements WorkflowContextInterface {
      */
     evict(): void;
     /**
-     * Wait for eviction message.
-     *
-     * The event listener uses { once: true } to auto-remove after firing,
-     * preventing memory leaks if this method is called multiple times.
+     * Wait for `ms`, rejecting early with EvictedError if the workflow is
+     * evicted. Both the timer and the abort listener are torn down on either
+     * outcome, so a completed sleep never leaves a dangling listener on the
+     * long-lived run abort signal.
      */
-    waitForEviction(): Promise<never>;
+    private sleepOrEvict;
     step<T>(nameOrConfig: string | StepConfig<T>, run?: () => Promise<T>): Promise<T>;
     tryStep<T>(nameOrConfig: string | TryStepConfig<T>, run?: () => Promise<T>): Promise<TryStepResult<T>>;
     try<T>(nameOrConfig: string | TryBlockConfig<T>, run?: (ctx: WorkflowContextInterface) => Promise<T>): Promise<TryBlockResult<T>>;
@@ -713,7 +715,9 @@ declare class WorkflowContextImpl implements WorkflowContextInterface {
      *
      * Note: This does NOT cancel the underlying operation. JavaScript Promises
      * cannot be cancelled once started. When a timeout occurs:
-     * - The step is marked as failed with StepTimeoutError
+     * - The step is rejected with StepTimeoutError. By default this is treated
+     *   as a critical failure with no retry. Set retryOnTimeout: true on the
+     *   step config to retry timeouts like any other error.
      * - The underlying async operation continues running in the background
      * - Any side effects from the operation may still occur
      *
@@ -779,6 +783,11 @@ declare class WorkflowContextImpl implements WorkflowContextInterface {
     private executeRemoved;
 }
 
+/**
+ * Extract structured error information from an error.
+ */
+declare function extractErrorInfo(error: unknown): WorkflowError;
+
 /**
  * Thrown from steps to prevent retry.
  * Use this when an error is unrecoverable and retrying would be pointless.
@@ -883,11 +892,6 @@ declare class EntryInProgressError extends Error {
     constructor();
 }
 
-/**
- * Extract structured error information from an error.
- */
-declare function extractErrorInfo(error: unknown): WorkflowError;
-
 /**
  * Check if a path segment is a loop iteration marker.
  */

package/dist/tsup/index.d.ts

@@ -345,6 +345,8 @@ interface StepConfig<T> {
     retryBackoffMax?: number;
     /** Timeout in ms for step execution (default: 30000). Set to 0 to disable. */
     timeout?: number;
+    /** If true, step timeouts retry like any other error instead of failing immediately as critical. Default: false. */
+    retryOnTimeout?: boolean;
 }
 type TryStepCatchKind = "critical" | "timeout" | "exhausted" | "rollback";
 interface TryStepFailure {
@@ -398,7 +400,7 @@ type LoopResult<S, T> = {
  * Stateless loops (state = undefined) may return void/undefined, which is treated as
  * `Loop.continue(undefined)`.
  */
-type LoopIterationResult<S, T> = Promise<LoopResult<S, T> | (S extends undefined ? void : never)>;
+type LoopIterationResult<S, T> = Promise<LoopResult<S, T> | (S extends undefined ? undefined : never)>;
 /**
  * Configuration for a loop.
  */
@@ -697,12 +699,12 @@ declare class WorkflowContextImpl implements WorkflowContextInterface {
      */
     evict(): void;
     /**
-     * Wait for eviction message.
-     *
-     * The event listener uses { once: true } to auto-remove after firing,
-     * preventing memory leaks if this method is called multiple times.
+     * Wait for `ms`, rejecting early with EvictedError if the workflow is
+     * evicted. Both the timer and the abort listener are torn down on either
+     * outcome, so a completed sleep never leaves a dangling listener on the
+     * long-lived run abort signal.
      */
-    waitForEviction(): Promise<never>;
+    private sleepOrEvict;
     step<T>(nameOrConfig: string | StepConfig<T>, run?: () => Promise<T>): Promise<T>;
     tryStep<T>(nameOrConfig: string | TryStepConfig<T>, run?: () => Promise<T>): Promise<TryStepResult<T>>;
     try<T>(nameOrConfig: string | TryBlockConfig<T>, run?: (ctx: WorkflowContextInterface) => Promise<T>): Promise<TryBlockResult<T>>;
@@ -713,7 +715,9 @@ declare class WorkflowContextImpl implements WorkflowContextInterface {
      *
      * Note: This does NOT cancel the underlying operation. JavaScript Promises
      * cannot be cancelled once started. When a timeout occurs:
-     * - The step is marked as failed with StepTimeoutError
+     * - The step is rejected with StepTimeoutError. By default this is treated
+     *   as a critical failure with no retry. Set retryOnTimeout: true on the
+     *   step config to retry timeouts like any other error.
      * - The underlying async operation continues running in the background
      * - Any side effects from the operation may still occur
      *
@@ -779,6 +783,11 @@ declare class WorkflowContextImpl implements WorkflowContextInterface {
     private executeRemoved;
 }
 
+/**
+ * Extract structured error information from an error.
+ */
+declare function extractErrorInfo(error: unknown): WorkflowError;
+
 /**
  * Thrown from steps to prevent retry.
  * Use this when an error is unrecoverable and retrying would be pointless.
@@ -883,11 +892,6 @@ declare class EntryInProgressError extends Error {
     constructor();
 }
 
-/**
- * Extract structured error information from an error.
- */
-declare function extractErrorInfo(error: unknown): WorkflowError;
-
 /**
  * Check if a path segment is a loop iteration marker.
  */

package/dist/tsup/index.js

@@ -43,7 +43,7 @@ import {
   resolveName,
   runWorkflow,
   setEntry
-} from "./chunk-EAAPMGGE.js";
+} from "./chunk-KA2T56AJ.js";
 export {
   CancelledError,
   CriticalError,

package/dist/tsup/testing.cjs

@@ -47,7 +47,7 @@
 
 
 
-var _chunkA33YVYOFcjs = require('./chunk-A33YVYOF.cjs');
+var _chunkGQWOLYBAcjs = require('./chunk-GQWOLYBA.cjs');
 
 // src/testing.ts
 var InMemoryWorkflowMessageDriver = class {
@@ -97,7 +97,7 @@ var InMemoryWorkflowMessageDriver = class {
   }
   async waitForMessages(messageNames, abortSignal) {
     if (abortSignal.aborted) {
-      throw new (0, _chunkA33YVYOFcjs.EvictedError)();
+      throw new (0, _chunkGQWOLYBAcjs.EvictedError)();
     }
     const nameSet = messageNames.length > 0 ? new Set(messageNames) : void 0;
     if (this.#messages.some(
@@ -118,7 +118,7 @@ var InMemoryWorkflowMessageDriver = class {
         },
         abortSignal,
         onAbort: () => {
-          waiter.reject(new (0, _chunkA33YVYOFcjs.EvictedError)());
+          waiter.reject(new (0, _chunkGQWOLYBAcjs.EvictedError)());
         }
       };
       abortSignal.addEventListener("abort", waiter.onAbort, {
@@ -158,56 +158,56 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
   __init4() {this.workerPollInterval = 100}
   __init5() {this.messageDriver = this.#inMemoryMessageDriver}
   async get(key) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
-    const entry = this.kv.get(_chunkA33YVYOFcjs.keyToHex.call(void 0, key));
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
+    const entry = this.kv.get(_chunkGQWOLYBAcjs.keyToHex.call(void 0, key));
     return _nullishCoalesce((entry == null ? void 0 : entry.value), () => ( null));
   }
   async set(key, value) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
-    this.kv.set(_chunkA33YVYOFcjs.keyToHex.call(void 0, key), { key, value });
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
+    this.kv.set(_chunkGQWOLYBAcjs.keyToHex.call(void 0, key), { key, value });
   }
   async delete(key) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
-    this.kv.delete(_chunkA33YVYOFcjs.keyToHex.call(void 0, key));
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
+    this.kv.delete(_chunkGQWOLYBAcjs.keyToHex.call(void 0, key));
   }
   async deletePrefix(prefix) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     for (const [hexKey, entry] of this.kv) {
-      if (_chunkA33YVYOFcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
+      if (_chunkGQWOLYBAcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
         this.kv.delete(hexKey);
       }
     }
   }
   async deleteRange(start, end) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     for (const [hexKey, entry] of this.kv) {
-      if (_chunkA33YVYOFcjs.compareKeys.call(void 0, entry.key, start) >= 0 && _chunkA33YVYOFcjs.compareKeys.call(void 0, entry.key, end) < 0) {
+      if (_chunkGQWOLYBAcjs.compareKeys.call(void 0, entry.key, start) >= 0 && _chunkGQWOLYBAcjs.compareKeys.call(void 0, entry.key, end) < 0) {
         this.kv.delete(hexKey);
       }
     }
   }
   async list(prefix) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     const results = [];
     for (const entry of this.kv.values()) {
-      if (_chunkA33YVYOFcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
+      if (_chunkGQWOLYBAcjs.keyStartsWith.call(void 0, entry.key, prefix)) {
         results.push({ key: entry.key, value: entry.value });
       }
     }
-    return results.sort((a, b) => _chunkA33YVYOFcjs.compareKeys.call(void 0, a.key, b.key));
+    return results.sort((a, b) => _chunkGQWOLYBAcjs.compareKeys.call(void 0, a.key, b.key));
   }
   async batch(writes) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     for (const { key, value } of writes) {
-      this.kv.set(_chunkA33YVYOFcjs.keyToHex.call(void 0, key), { key, value });
+      this.kv.set(_chunkGQWOLYBAcjs.keyToHex.call(void 0, key), { key, value });
     }
   }
   async setAlarm(workflowId, wakeAt) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     this.alarms.set(workflowId, wakeAt);
   }
   async clearAlarm(workflowId) {
-    await _chunkA33YVYOFcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     this.alarms.delete(workflowId);
   }
   async waitForMessages(messageNames, abortSignal) {
@@ -218,7 +218,7 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
     }
     while (true) {
       if (abortSignal.aborted) {
-        throw new (0, _chunkA33YVYOFcjs.EvictedError)();
+        throw new (0, _chunkGQWOLYBAcjs.EvictedError)();
       }
       const messages = await this.messageDriver.receiveMessages({
         names: messageNames.length > 0 ? messageNames : void 0,
@@ -228,7 +228,7 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
       if (messages.length > 0) {
         return;
       }
-      await _chunkA33YVYOFcjs.sleep.call(void 0, Math.max(1, this.latency));
+      await _chunkGQWOLYBAcjs.sleep.call(void 0, Math.max(1, this.latency));
     }
   }
   /**
@@ -324,5 +324,5 @@ var InMemoryDriver = (_class = class {constructor() { _class.prototype.__init.ca
 
 
 
-exports.CancelledError = _chunkA33YVYOFcjs.CancelledError; exports.CriticalError = _chunkA33YVYOFcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkA33YVYOFcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkA33YVYOFcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkA33YVYOFcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkA33YVYOFcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkA33YVYOFcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkA33YVYOFcjs.EntryInProgressError; exports.EvictedError = _chunkA33YVYOFcjs.EvictedError; exports.HistoryDivergedError = _chunkA33YVYOFcjs.HistoryDivergedError; exports.InMemoryDriver = InMemoryDriver; exports.JoinError = _chunkA33YVYOFcjs.JoinError; exports.Loop = _chunkA33YVYOFcjs.Loop; exports.MessageWaitError = _chunkA33YVYOFcjs.MessageWaitError; exports.RaceError = _chunkA33YVYOFcjs.RaceError; exports.RollbackCheckpointError = _chunkA33YVYOFcjs.RollbackCheckpointError; exports.RollbackError = _chunkA33YVYOFcjs.RollbackError; exports.SleepError = _chunkA33YVYOFcjs.SleepError; exports.StepExhaustedError = _chunkA33YVYOFcjs.StepExhaustedError; exports.StepFailedError = _chunkA33YVYOFcjs.StepFailedError; exports.WorkflowContextImpl = _chunkA33YVYOFcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkA33YVYOFcjs.appendLoopIteration; exports.appendName = _chunkA33YVYOFcjs.appendName; exports.createEntry = _chunkA33YVYOFcjs.createEntry; exports.createHistorySnapshot = _chunkA33YVYOFcjs.createHistorySnapshot; exports.createStorage = _chunkA33YVYOFcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkA33YVYOFcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkA33YVYOFcjs.emptyLocation; exports.extractErrorInfo = _chunkA33YVYOFcjs.extractErrorInfo; exports.flush = _chunkA33YVYOFcjs.flush; exports.generateId = _chunkA33YVYOFcjs.generateId; exports.getEntry = _chunkA33YVYOFcjs.getEntry; exports.getOrCreateMetadata = _chunkA33YVYOFcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkA33YVYOFcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkA33YVYOFcjs.isLoopIterationMarker; exports.loadMetadata = _chunkA33YVYOFcjs.loadMetadata; exports.loadStorage = _chunkA33YVYOFcjs.loadStorage; exports.locationToKey = _chunkA33YVYOFcjs.locationToKey; exports.locationsEqual = _chunkA33YVYOFcjs.locationsEqual; exports.parentLocation = _chunkA33YVYOFcjs.parentLocation; exports.registerName = _chunkA33YVYOFcjs.registerName; exports.replayWorkflowFromStep = _chunkA33YVYOFcjs.replayWorkflowFromStep; exports.resolveName = _chunkA33YVYOFcjs.resolveName; exports.runWorkflow = _chunkA33YVYOFcjs.runWorkflow; exports.setEntry = _chunkA33YVYOFcjs.setEntry;
+exports.CancelledError = _chunkGQWOLYBAcjs.CancelledError; exports.CriticalError = _chunkGQWOLYBAcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkGQWOLYBAcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkGQWOLYBAcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkGQWOLYBAcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkGQWOLYBAcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkGQWOLYBAcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkGQWOLYBAcjs.EntryInProgressError; exports.EvictedError = _chunkGQWOLYBAcjs.EvictedError; exports.HistoryDivergedError = _chunkGQWOLYBAcjs.HistoryDivergedError; exports.InMemoryDriver = InMemoryDriver; exports.JoinError = _chunkGQWOLYBAcjs.JoinError; exports.Loop = _chunkGQWOLYBAcjs.Loop; exports.MessageWaitError = _chunkGQWOLYBAcjs.MessageWaitError; exports.RaceError = _chunkGQWOLYBAcjs.RaceError; exports.RollbackCheckpointError = _chunkGQWOLYBAcjs.RollbackCheckpointError; exports.RollbackError = _chunkGQWOLYBAcjs.RollbackError; exports.SleepError = _chunkGQWOLYBAcjs.SleepError; exports.StepExhaustedError = _chunkGQWOLYBAcjs.StepExhaustedError; exports.StepFailedError = _chunkGQWOLYBAcjs.StepFailedError; exports.WorkflowContextImpl = _chunkGQWOLYBAcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkGQWOLYBAcjs.appendLoopIteration; exports.appendName = _chunkGQWOLYBAcjs.appendName; exports.createEntry = _chunkGQWOLYBAcjs.createEntry; exports.createHistorySnapshot = _chunkGQWOLYBAcjs.createHistorySnapshot; exports.createStorage = _chunkGQWOLYBAcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkGQWOLYBAcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkGQWOLYBAcjs.emptyLocation; exports.extractErrorInfo = _chunkGQWOLYBAcjs.extractErrorInfo; exports.flush = _chunkGQWOLYBAcjs.flush; exports.generateId = _chunkGQWOLYBAcjs.generateId; exports.getEntry = _chunkGQWOLYBAcjs.getEntry; exports.getOrCreateMetadata = _chunkGQWOLYBAcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkGQWOLYBAcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkGQWOLYBAcjs.isLoopIterationMarker; exports.loadMetadata = _chunkGQWOLYBAcjs.loadMetadata; exports.loadStorage = _chunkGQWOLYBAcjs.loadStorage; exports.locationToKey = _chunkGQWOLYBAcjs.locationToKey; exports.locationsEqual = _chunkGQWOLYBAcjs.locationsEqual; exports.parentLocation = _chunkGQWOLYBAcjs.parentLocation; exports.registerName = _chunkGQWOLYBAcjs.registerName; exports.replayWorkflowFromStep = _chunkGQWOLYBAcjs.replayWorkflowFromStep; exports.resolveName = _chunkGQWOLYBAcjs.resolveName; exports.runWorkflow = _chunkGQWOLYBAcjs.runWorkflow; exports.setEntry = _chunkGQWOLYBAcjs.setEntry;
 //# sourceMappingURL=testing.cjs.map

package/dist/tsup/testing.js

@@ -47,7 +47,7 @@ import {
   runWorkflow,
   setEntry,
   sleep
-} from "./chunk-EAAPMGGE.js";
+} from "./chunk-KA2T56AJ.js";
 
 // src/testing.ts
 var InMemoryWorkflowMessageDriver = class {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/workflow-engine",
-	"version": "0.0.0-2-3-0-rc-rollback.f0e199e",
+	"version": "0.0.0-claude-edge-bundle-slim.fb63356",
 	"description": "Durable workflow engine with reentrant execution",
 	"license": "Apache-2.0",
 	"keywords": [

package/schemas/serde.ts

@@ -18,7 +18,6 @@ import type {
 	EntryMetadata as InternalEntryMetadata,
 	EntryStatus as InternalEntryStatus,
 	Location as InternalLocation,
-	LoopIterationMarker as InternalLoopIterationMarker,
 	PathSegment as InternalPathSegment,
 	SleepState as InternalSleepState,
 	WorkflowState as InternalWorkflowState,
@@ -27,7 +26,6 @@ import {
 	CURRENT_VERSION,
 	ENTRY_METADATA_VERSIONED,
 	ENTRY_VERSIONED,
-	WORKFLOW_METADATA_VERSIONED,
 } from "./versioned.js";
 
 // === Helper: ArrayBuffer to/from utilities ===
@@ -74,7 +72,7 @@ function assertString(
 /**
  * Validate that a value is a number.
  */
-function assertNumber(
+function _assertNumber(
 	value: unknown,
 	context: string,
 ): asserts value is number {

package/src/context.ts

@@ -21,7 +21,7 @@ import {
 	StepExhaustedError,
 	StepFailedError,
 } from "./errors.js";
-import { buildLoopIterationRange, buildEntryMetadataKey } from "./keys.js";
+import { buildEntryMetadataKey, buildLoopIterationRange } from "./keys.js";
 import {
 	appendLoopIteration,
 	appendName,
@@ -36,13 +36,12 @@ import {
 	flush,
 	getOrCreateMetadata,
 	loadMetadata,
-	setEntry,
 	type PendingDeletions,
+	setEntry,
 } from "./storage.js";
 import type {
 	BranchConfig,
 	BranchOutput,
-	BranchStatus,
 	Entry,
 	EntryKindType,
 	EntryMetadata,
@@ -66,13 +65,12 @@ import type {
 	WorkflowError,
 	WorkflowErrorEvent,
 	WorkflowErrorHandler,
+	WorkflowMessageDriver,
 	WorkflowQueue,
 	WorkflowQueueMessage,
 	WorkflowQueueNextBatchOptions,
 	WorkflowQueueNextOptions,
-	WorkflowMessageDriver,
 } from "./types.js";
-import { sleep } from "./utils.js";
 
 /**
  * Default values for step configuration.
@@ -505,8 +503,7 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 	 * Throws HistoryDivergedError if duplicate detected.
 	 */
 	private checkDuplicateName(name: string): void {
-		const fullKey =
-			locationToKey(this.storage, this.currentLocation) + "/" + name;
+		const fullKey = `${locationToKey(this.storage, this.currentLocation)}/${name}`;
 		if (this.usedNamesInExecution.has(fullKey)) {
 			throw new HistoryDivergedError(
 				`Duplicate entry name "${name}" at location "${locationToKey(this.storage, this.currentLocation)}". ` +
@@ -580,7 +577,7 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 			const isUnderPrefix =
 				prefix === ""
 					? true // Root: all keys are children
-					: key.startsWith(prefix + "/") || key === prefix;
+					: key.startsWith(`${prefix}/`) || key === prefix;
 
 			if (isUnderPrefix) {
 				if (!this.visitedKeys.has(key)) {
@@ -603,25 +600,33 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 	}
 
 	/**
-	 * Wait for eviction message.
-	 *
-	 * The event listener uses { once: true } to auto-remove after firing,
-	 * preventing memory leaks if this method is called multiple times.
+	 * Wait for `ms`, rejecting early with EvictedError if the workflow is
+	 * evicted. Both the timer and the abort listener are torn down on either
+	 * outcome, so a completed sleep never leaves a dangling listener on the
+	 * long-lived run abort signal.
 	 */
-	waitForEviction(): Promise<never> {
-		return new Promise((_, reject) => {
-			if (this.abortSignal.aborted) {
-				reject(new EvictedError());
-				return;
+	private async sleepOrEvict(ms: number): Promise<void> {
+		if (this.abortSignal.aborted) {
+			throw new EvictedError();
+		}
+		let timer: ReturnType<typeof setTimeout> | undefined;
+		let onAbort: (() => void) | undefined;
+		try {
+			await new Promise<void>((resolve, reject) => {
+				timer = setTimeout(resolve, ms);
+				onAbort = () => reject(new EvictedError());
+				this.abortSignal.addEventListener("abort", onAbort, {
+					once: true,
+				});
+			});
+		} finally {
+			if (timer !== undefined) {
+				clearTimeout(timer);
 			}
-			this.abortSignal.addEventListener(
-				"abort",
-				() => {
-					reject(new EvictedError());
-				},
-				{ once: true },
-			);
-		});
+			if (onAbort) {
+				this.abortSignal.removeEventListener("abort", onAbort);
+			}
+		}
 	}
 
 	// === Step ===
@@ -943,8 +948,9 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 			}
 			entry.dirty = true;
 
-			// Timeout errors are treated as critical (no retry)
-			if (error instanceof StepTimeoutError) {
+			// Timeout errors are treated as critical by default. Steps opt
+			// into retrying on timeout with retryOnTimeout: true.
+			if (error instanceof StepTimeoutError && !config.retryOnTimeout) {
 				metadata.status = "exhausted";
 				metadata.error = String(error);
 				await this.notifyStepError(config, metadata.attempts, error, {
@@ -1010,7 +1016,10 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 				attachTryStepFailure(
 					new StepExhaustedError(config.name, String(error)),
 					{
-						kind: "exhausted",
+						kind:
+							error instanceof StepTimeoutError
+								? "timeout"
+								: "exhausted",
 						stepName: config.name,
 						attempts: metadata.attempts,
 						error: extractErrorInfo(error),
@@ -1029,7 +1038,9 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 	 *
 	 * Note: This does NOT cancel the underlying operation. JavaScript Promises
 	 * cannot be cancelled once started. When a timeout occurs:
-	 * - The step is marked as failed with StepTimeoutError
+	 * - The step is rejected with StepTimeoutError. By default this is treated
+	 *   as a critical failure with no retry. Set retryOnTimeout: true on the
+	 *   step config to retry timeouts like any other error.
 	 * - The underlying async operation continues running in the background
 	 * - Any side effects from the operation may still occur
 	 *
@@ -1480,7 +1491,7 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 
 		// Short sleep: wait in memory
 		if (remaining < this.driver.workerPollInterval) {
-			await Promise.race([sleep(remaining), this.waitForEviction()]);
+			await this.sleepOrEvict(remaining);
 
 			this.checkEvicted();
 

package/src/index.ts

@@ -13,6 +13,7 @@ export {
 } from "./context.js";
 // Driver
 export type { EngineDriver, KVEntry, KVWrite } from "./driver.js";
+export { extractErrorInfo } from "./error-utils.js";
 // Errors
 export {
 	CancelledError,
@@ -29,7 +30,6 @@ export {
 	StepExhaustedError,
 	StepFailedError,
 } from "./errors.js";
-export { extractErrorInfo } from "./error-utils.js";
 
 // Location utilities
 export {
@@ -82,9 +82,6 @@ export type {
 	PathSegment,
 	RaceEntry,
 	RemovedEntry,
-	WorkflowEntryMetadataSnapshot,
-	WorkflowHistoryEntry,
-	WorkflowHistorySnapshot,
 	RollbackCheckpointEntry,
 	RollbackContextInterface,
 	RunWorkflowOptions,
@@ -102,20 +99,23 @@ export type {
 	TryStepFailure,
 	TryStepResult,
 	WorkflowContextInterface,
+	WorkflowEntryMetadataSnapshot,
 	WorkflowError,
 	WorkflowErrorEvent,
 	WorkflowErrorHandler,
 	WorkflowFunction,
 	WorkflowHandle,
-	WorkflowRollbackErrorEvent,
+	WorkflowHistoryEntry,
+	WorkflowHistorySnapshot,
+	WorkflowMessageDriver,
 	WorkflowQueue,
 	WorkflowQueueMessage,
 	WorkflowQueueNextBatchOptions,
 	WorkflowQueueNextOptions,
-	WorkflowMessageDriver,
 	WorkflowResult,
-	WorkflowRunMode,
+	WorkflowRollbackErrorEvent,
 	WorkflowRunErrorEvent,
+	WorkflowRunMode,
 	WorkflowState,
 	WorkflowStepErrorEvent,
 } from "./types.js";
@@ -185,9 +185,9 @@ import type {
 	RunWorkflowOptions,
 	Storage,
 	WorkflowErrorEvent,
-	WorkflowHistorySnapshot,
 	WorkflowFunction,
 	WorkflowHandle,
+	WorkflowHistorySnapshot,
 	WorkflowMessageDriver,
 	WorkflowResult,
 	WorkflowRunMode,
@@ -532,26 +532,46 @@ async function executeLiveWorkflow<TInput, TOutput>(
 		const hasDeadline = result.sleepUntil !== undefined;
 
 		if (hasMessages && hasDeadline) {
-			// Wait for EITHER a message OR the deadline (for queue.next timeout)
+			// Wait for EITHER a message OR the deadline (for queue.next timeout).
+			// Scope both waiters to a per-iteration controller chained to the run
+			// signal so that whichever loses the race is cancelled immediately.
+			// Otherwise the loser (a message waiter or an armed sleep timer) stays
+			// registered on the long-lived run signal and leaks once per cycle.
+			const iterationAbort = new AbortController();
+			const onRunAbort = () => iterationAbort.abort();
+			if (abortController.signal.aborted) {
+				iterationAbort.abort();
+			} else {
+				abortController.signal.addEventListener("abort", onRunAbort, {
+					once: true,
+				});
+			}
+			const messagePromise = awaitWithEviction(
+				driver.waitForMessages(
+					result.waitingForMessages!,
+					iterationAbort.signal,
+				),
+				iterationAbort.signal,
+			);
+			const sleepPromise = waitForSleep(
+				runtime,
+				result.sleepUntil!,
+				iterationAbort.signal,
+			);
+			// Swallow the loser's rejection once it is aborted below so it does
+			// not surface as an unhandled rejection.
+			messagePromise.catch(() => {});
+			sleepPromise.catch(() => {});
 			try {
-				const messagePromise = awaitWithEviction(
-					driver.waitForMessages(
-						result.waitingForMessages!,
-						abortController.signal,
-					),
-					abortController.signal,
-				);
-				const sleepPromise = waitForSleep(
-					runtime,
-					result.sleepUntil!,
-					abortController.signal,
-				);
 				await Promise.race([messagePromise, sleepPromise]);
 			} catch (error) {
 				if (error instanceof EvictedError) {
 					return lastResult;
 				}
 				throw error;
+			} finally {
+				iterationAbort.abort();
+				abortController.signal.removeEventListener("abort", onRunAbort);
 			}
 			continue;
 		}

package/src/location.ts

@@ -159,7 +159,7 @@ export function getChildEntries(
 		const isChild =
 			parentKey === ""
 				? true
-				: key.startsWith(parentKey + "/") || key === parentKey;
+				: key.startsWith(`${parentKey}/`) || key === parentKey;
 
 		if (isChild) {
 			// Return the actual entry's location, not the parent location

package/src/types.ts

@@ -399,6 +399,8 @@ export interface StepConfig<T> {
 	retryBackoffMax?: number;
 	/** Timeout in ms for step execution (default: 30000). Set to 0 to disable. */
 	timeout?: number;
+	/** If true, step timeouts retry like any other error instead of failing immediately as critical. Default: false. */
+	retryOnTimeout?: boolean;
 }
 
 export type TryStepCatchKind =
@@ -455,7 +457,7 @@ export type LoopResult<S, T> =
  * `Loop.continue(undefined)`.
  */
 export type LoopIterationResult<S, T> = Promise<
-	LoopResult<S, T> | (S extends undefined ? void : never)
+	LoopResult<S, T> | (S extends undefined ? undefined : never)
 >;
 
 /**