@rivetkit/workflow-engine 2.3.0-rc.9 → 2.3.1

package/dist/tsup/index.cjs

@@ -43,7 +43,7 @@
 
 
 
-var _chunkSFJQFMCWcjs = require('./chunk-SFJQFMCW.cjs');
+var _chunkGQWOLYBAcjs = require('./chunk-GQWOLYBA.cjs');
 
 
 
@@ -89,5 +89,5 @@ var _chunkSFJQFMCWcjs = require('./chunk-SFJQFMCW.cjs');
 
 
 
-exports.CancelledError = _chunkSFJQFMCWcjs.CancelledError; exports.CriticalError = _chunkSFJQFMCWcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkSFJQFMCWcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkSFJQFMCWcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkSFJQFMCWcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkSFJQFMCWcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkSFJQFMCWcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkSFJQFMCWcjs.EntryInProgressError; exports.EvictedError = _chunkSFJQFMCWcjs.EvictedError; exports.HistoryDivergedError = _chunkSFJQFMCWcjs.HistoryDivergedError; exports.JoinError = _chunkSFJQFMCWcjs.JoinError; exports.Loop = _chunkSFJQFMCWcjs.Loop; exports.MessageWaitError = _chunkSFJQFMCWcjs.MessageWaitError; exports.RaceError = _chunkSFJQFMCWcjs.RaceError; exports.RollbackCheckpointError = _chunkSFJQFMCWcjs.RollbackCheckpointError; exports.RollbackError = _chunkSFJQFMCWcjs.RollbackError; exports.SleepError = _chunkSFJQFMCWcjs.SleepError; exports.StepExhaustedError = _chunkSFJQFMCWcjs.StepExhaustedError; exports.StepFailedError = _chunkSFJQFMCWcjs.StepFailedError; exports.WorkflowContextImpl = _chunkSFJQFMCWcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkSFJQFMCWcjs.appendLoopIteration; exports.appendName = _chunkSFJQFMCWcjs.appendName; exports.createEntry = _chunkSFJQFMCWcjs.createEntry; exports.createHistorySnapshot = _chunkSFJQFMCWcjs.createHistorySnapshot; exports.createStorage = _chunkSFJQFMCWcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkSFJQFMCWcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkSFJQFMCWcjs.emptyLocation; exports.extractErrorInfo = _chunkSFJQFMCWcjs.extractErrorInfo; exports.flush = _chunkSFJQFMCWcjs.flush; exports.generateId = _chunkSFJQFMCWcjs.generateId; exports.getEntry = _chunkSFJQFMCWcjs.getEntry; exports.getOrCreateMetadata = _chunkSFJQFMCWcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkSFJQFMCWcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkSFJQFMCWcjs.isLoopIterationMarker; exports.loadMetadata = _chunkSFJQFMCWcjs.loadMetadata; exports.loadStorage = _chunkSFJQFMCWcjs.loadStorage; exports.locationToKey = _chunkSFJQFMCWcjs.locationToKey; exports.locationsEqual = _chunkSFJQFMCWcjs.locationsEqual; exports.parentLocation = _chunkSFJQFMCWcjs.parentLocation; exports.registerName = _chunkSFJQFMCWcjs.registerName; exports.replayWorkflowFromStep = _chunkSFJQFMCWcjs.replayWorkflowFromStep; exports.resolveName = _chunkSFJQFMCWcjs.resolveName; exports.runWorkflow = _chunkSFJQFMCWcjs.runWorkflow; exports.setEntry = _chunkSFJQFMCWcjs.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-CIDHCIH7.js";
+} from "./chunk-KA2T56AJ.js";
 export {
   CancelledError,
   CriticalError,

package/dist/tsup/testing.cjs

@@ -47,7 +47,7 @@
 
 
 
-var _chunkSFJQFMCWcjs = require('./chunk-SFJQFMCW.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, _chunkSFJQFMCWcjs.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, _chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
-    const entry = this.kv.get(_chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
-    this.kv.set(_chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
-    this.kv.delete(_chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     for (const [hexKey, entry] of this.kv) {
-      if (_chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     for (const [hexKey, entry] of this.kv) {
-      if (_chunkSFJQFMCWcjs.compareKeys.call(void 0, entry.key, start) >= 0 && _chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     const results = [];
     for (const entry of this.kv.values()) {
-      if (_chunkSFJQFMCWcjs.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) => _chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     for (const { key, value } of writes) {
-      this.kv.set(_chunkSFJQFMCWcjs.keyToHex.call(void 0, key), { key, value });
+      this.kv.set(_chunkGQWOLYBAcjs.keyToHex.call(void 0, key), { key, value });
     }
   }
   async setAlarm(workflowId, wakeAt) {
-    await _chunkSFJQFMCWcjs.sleep.call(void 0, this.latency);
+    await _chunkGQWOLYBAcjs.sleep.call(void 0, this.latency);
     this.alarms.set(workflowId, wakeAt);
   }
   async clearAlarm(workflowId) {
-    await _chunkSFJQFMCWcjs.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, _chunkSFJQFMCWcjs.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 _chunkSFJQFMCWcjs.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 = _chunkSFJQFMCWcjs.CancelledError; exports.CriticalError = _chunkSFJQFMCWcjs.CriticalError; exports.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL = _chunkSFJQFMCWcjs.DEFAULT_LOOP_HISTORY_PRUNE_INTERVAL; exports.DEFAULT_MAX_RETRIES = _chunkSFJQFMCWcjs.DEFAULT_MAX_RETRIES; exports.DEFAULT_RETRY_BACKOFF_BASE = _chunkSFJQFMCWcjs.DEFAULT_RETRY_BACKOFF_BASE; exports.DEFAULT_RETRY_BACKOFF_MAX = _chunkSFJQFMCWcjs.DEFAULT_RETRY_BACKOFF_MAX; exports.DEFAULT_STEP_TIMEOUT = _chunkSFJQFMCWcjs.DEFAULT_STEP_TIMEOUT; exports.EntryInProgressError = _chunkSFJQFMCWcjs.EntryInProgressError; exports.EvictedError = _chunkSFJQFMCWcjs.EvictedError; exports.HistoryDivergedError = _chunkSFJQFMCWcjs.HistoryDivergedError; exports.InMemoryDriver = InMemoryDriver; exports.JoinError = _chunkSFJQFMCWcjs.JoinError; exports.Loop = _chunkSFJQFMCWcjs.Loop; exports.MessageWaitError = _chunkSFJQFMCWcjs.MessageWaitError; exports.RaceError = _chunkSFJQFMCWcjs.RaceError; exports.RollbackCheckpointError = _chunkSFJQFMCWcjs.RollbackCheckpointError; exports.RollbackError = _chunkSFJQFMCWcjs.RollbackError; exports.SleepError = _chunkSFJQFMCWcjs.SleepError; exports.StepExhaustedError = _chunkSFJQFMCWcjs.StepExhaustedError; exports.StepFailedError = _chunkSFJQFMCWcjs.StepFailedError; exports.WorkflowContextImpl = _chunkSFJQFMCWcjs.WorkflowContextImpl; exports.appendLoopIteration = _chunkSFJQFMCWcjs.appendLoopIteration; exports.appendName = _chunkSFJQFMCWcjs.appendName; exports.createEntry = _chunkSFJQFMCWcjs.createEntry; exports.createHistorySnapshot = _chunkSFJQFMCWcjs.createHistorySnapshot; exports.createStorage = _chunkSFJQFMCWcjs.createStorage; exports.deleteEntriesWithPrefix = _chunkSFJQFMCWcjs.deleteEntriesWithPrefix; exports.emptyLocation = _chunkSFJQFMCWcjs.emptyLocation; exports.extractErrorInfo = _chunkSFJQFMCWcjs.extractErrorInfo; exports.flush = _chunkSFJQFMCWcjs.flush; exports.generateId = _chunkSFJQFMCWcjs.generateId; exports.getEntry = _chunkSFJQFMCWcjs.getEntry; exports.getOrCreateMetadata = _chunkSFJQFMCWcjs.getOrCreateMetadata; exports.isLocationPrefix = _chunkSFJQFMCWcjs.isLocationPrefix; exports.isLoopIterationMarker = _chunkSFJQFMCWcjs.isLoopIterationMarker; exports.loadMetadata = _chunkSFJQFMCWcjs.loadMetadata; exports.loadStorage = _chunkSFJQFMCWcjs.loadStorage; exports.locationToKey = _chunkSFJQFMCWcjs.locationToKey; exports.locationsEqual = _chunkSFJQFMCWcjs.locationsEqual; exports.parentLocation = _chunkSFJQFMCWcjs.parentLocation; exports.registerName = _chunkSFJQFMCWcjs.registerName; exports.replayWorkflowFromStep = _chunkSFJQFMCWcjs.replayWorkflowFromStep; exports.resolveName = _chunkSFJQFMCWcjs.resolveName; exports.runWorkflow = _chunkSFJQFMCWcjs.runWorkflow; exports.setEntry = _chunkSFJQFMCWcjs.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-CIDHCIH7.js";
+} from "./chunk-KA2T56AJ.js";
 
 // src/testing.ts
 var InMemoryWorkflowMessageDriver = class {

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@rivetkit/workflow-engine",
-	"version": "2.3.0-rc.9",
+	"version": "2.3.1",
 	"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 ===
@@ -825,10 +830,7 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 			const maxRetries = config.maxRetries ?? DEFAULT_MAX_RETRIES;
 
 			if (metadata.attempts > maxRetries) {
-				// Prefer step history error, but fall back to metadata since
-				// driver implementations may persist metadata without the history
-				// entry error (e.g. partial writes/crashes between attempts).
-				const lastError = stepData.error ?? metadata.error;
+				const lastError = metadata.error;
 				const exhaustedError = new StepExhaustedError(
 					config.name,
 					lastError,
@@ -941,15 +943,16 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 			});
 			return output;
 		} catch (error) {
-			// Timeout errors are treated as critical (no retry)
-			if (error instanceof StepTimeoutError) {
-				if (entry.kind.type === "step") {
-					entry.kind.data.error = String(error);
-				}
-				entry.dirty = true;
+			if (entry.kind.type === "step") {
+				entry.kind.data.error = String(error);
+			}
+			entry.dirty = true;
+
+			// 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.flushStorage();
 				await this.notifyStepError(config, metadata.attempts, error, {
 					willRetry: false,
 				});
@@ -967,13 +970,8 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 				error instanceof CriticalError ||
 				error instanceof RollbackError
 			) {
-				if (entry.kind.type === "step") {
-					entry.kind.data.error = String(error);
-				}
-				entry.dirty = true;
 				metadata.status = "exhausted";
 				metadata.error = String(error);
-				await this.flushStorage();
 				await this.notifyStepError(config, metadata.attempts, error, {
 					willRetry: false,
 				});
@@ -990,15 +988,10 @@ export class WorkflowContextImpl implements WorkflowContextInterface {
 				);
 			}
 
-			if (entry.kind.type === "step") {
-				entry.kind.data.error = String(error);
-			}
-			entry.dirty = true;
 			const willRetry = metadata.attempts <= maxRetries;
 			metadata.status = willRetry ? "failed" : "exhausted";
 			metadata.error = String(error);
 
-			await this.flushStorage();
 			if (willRetry) {
 				const retryDelay = calculateBackoff(
 					metadata.attempts,
@@ -1023,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),
@@ -1042,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
 	 *
@@ -1493,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();