@rivetkit/workflow-engine 2.3.18 → 2.3.20

package/dist/tsup/{chunk-TVRZA5O5.js → chunk-EGOPMLG2.js}

@@ -163,6 +163,7 @@ var KEY_PREFIX = {
   // Workflow metadata: [3, field]
   ENTRY_METADATA: 4
   // Entry metadata: [4, entryId]
+  // 5 is reserved for RivetKit's workflow trace context.
 };
 var WORKFLOW_FIELD = {
   STATE: 1,
@@ -1871,7 +1872,7 @@ function selectControlFlowError(current, candidate) {
   return controlFlowErrorPriority(candidate) < controlFlowErrorPriority(current) ? candidate : current;
 }
 var WorkflowContextImpl = class _WorkflowContextImpl {
-  constructor(workflowId, storage, driver, messageDriver, location = emptyLocation(), abortController, mode = "forward", rollbackActions, rollbackCheckpointSet = false, historyNotifier, onError, logger, visitedKeys) {
+  constructor(workflowId, storage, driver, messageDriver, location = emptyLocation(), abortController, mode = "forward", rollbackActions, rollbackCheckpointSet = false, historyNotifier, onError, logger, visitedKeys, runSpan) {
     this.workflowId = workflowId;
     this.storage = storage;
     this.driver = driver;
@@ -1885,11 +1886,13 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
     this.onError = onError;
     this.logger = logger;
     this.visitedKeys = visitedKeys ?? /* @__PURE__ */ new Set();
+    this.runSpan = runSpan;
   }
   entryInProgress = false;
   abortController;
   currentLocation;
   visitedKeys;
+  runSpan;
   mode;
   rollbackActions;
   rollbackCheckpointSet;
@@ -1942,7 +1945,8 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
       this.historyNotifier,
       this.onError,
       this.logger,
-      this.visitedKeys
+      this.visitedKeys,
+      this.runSpan
     );
   }
   /**
@@ -2209,6 +2213,7 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
     }
   }
   async executeStep(config2) {
+    var _a;
     this.ensureRollbackCheckpoint(config2);
     if (this.mode === "rollback") {
       return await this.executeStepRollback(config2);
@@ -2300,9 +2305,13 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
     metadata.lastAttemptAt = Date.now();
     metadata.dirty = true;
     const timeout = config2.timeout ?? DEFAULT_STEP_TIMEOUT;
+    const stepSpan = (_a = this.runSpan) == null ? void 0 : _a.startStep(
+      config2.name,
+      metadata.attempts
+    );
     try {
       const output = await this.executeWithTimeout(
-        config2.run(),
+        stepSpan ? stepSpan.run(() => config2.run()) : config2.run(),
         timeout,
         config2.name
       );
@@ -2321,6 +2330,7 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
         });
         await this.flushStorage();
       }
+      stepSpan == null ? void 0 : stepSpan.finish("ok");
       this.log("debug", {
         msg: "step completed",
         step: config2.name,
@@ -2333,6 +2343,7 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
       }
       entry.dirty = true;
       if (error instanceof StepTimeoutError && !config2.retryOnTimeout) {
+        stepSpan == null ? void 0 : stepSpan.finish("failed", error);
         metadata.status = "exhausted";
         metadata.error = String(error);
         await this.notifyStepError(config2, metadata.attempts, error, {
@@ -2348,6 +2359,7 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
         );
       }
       if (error instanceof CriticalError || error instanceof RollbackError) {
+        stepSpan == null ? void 0 : stepSpan.finish("failed", error);
         metadata.status = "exhausted";
         metadata.error = String(error);
         await this.notifyStepError(config2, metadata.attempts, error, {
@@ -2363,6 +2375,7 @@ var WorkflowContextImpl = class _WorkflowContextImpl {
         );
       }
       const willRetry = metadata.attempts <= maxRetries;
+      stepSpan == null ? void 0 : stepSpan.finish(willRetry ? "retry" : "failed", error);
       metadata.status = willRetry ? "failed" : "exhausted";
       metadata.error = String(error);
       if (willRetry) {
@@ -3682,7 +3695,7 @@ async function awaitWithEviction(promise, abortSignal) {
     cleanup();
   }
 }
-async function executeRollback(workflowId, workflowFn, input, driver, messageDriver, abortController, storage, historyNotifier, onError, logger) {
+async function executeRollback(workflowId, workflowFn, input, driver, messageDriver, abortController, storage, historyNotifier, onError, logger, runSpan) {
   const rollbackActions = [];
   const ctx = new WorkflowContextImpl(
     workflowId,
@@ -3696,7 +3709,9 @@ async function executeRollback(workflowId, workflowFn, input, driver, messageDri
     false,
     historyNotifier,
     onError,
-    logger
+    logger,
+    void 0,
+    runSpan
   );
   try {
     await workflowFn(ctx, input);
@@ -3832,16 +3847,20 @@ async function waitForSleep(runtime, deadline, abortSignal) {
 async function executeLiveWorkflow(workflowId, workflowFn, input, driver, messageDriver, abortController, runtime, onHistoryUpdated, onError, logger) {
   let lastResult;
   while (true) {
-    const result = await executeWorkflow(
-      workflowId,
-      workflowFn,
-      input,
+    const result = await traceRun(
       driver,
-      messageDriver,
-      abortController,
-      onHistoryUpdated,
-      onError,
-      logger
+      (runSpan) => executeWorkflow(
+        workflowId,
+        workflowFn,
+        input,
+        driver,
+        messageDriver,
+        abortController,
+        onHistoryUpdated,
+        onError,
+        logger,
+        runSpan
+      )
     );
     lastResult = result;
     if (result.state !== "sleeping") {
@@ -3940,16 +3959,20 @@ function runWorkflow(workflowId, workflowFn, input, driver, options = {}) {
     options.onHistoryUpdated,
     options.onError,
     logger
-  ) : executeWorkflow(
-    workflowId,
-    workflowFn,
-    input,
+  ) : traceRun(
     driver,
-    messageDriver,
-    abortController,
-    options.onHistoryUpdated,
-    options.onError,
-    logger
+    (runSpan) => executeWorkflow(
+      workflowId,
+      workflowFn,
+      input,
+      driver,
+      messageDriver,
+      abortController,
+      options.onHistoryUpdated,
+      options.onError,
+      logger,
+      runSpan
+    )
   );
   return {
     workflowId,
@@ -4125,7 +4148,42 @@ function findReplayBoundaryEntry(entries, target) {
   }
   return boundary;
 }
-async function executeWorkflow(workflowId, workflowFn, input, driver, messageDriver, abortController, onHistoryUpdated, onError, logger) {
+function runOutcomeFromState(state) {
+  switch (state) {
+    case "completed":
+      return "completed";
+    case "sleeping":
+      return "sleeping";
+    case "failed":
+      return "failed";
+    case "cancelled":
+      return "cancelled";
+    case "pending":
+    case "running":
+    case "rolling_back":
+      return "evicted";
+  }
+}
+async function traceRun(driver, run) {
+  if (!driver.telemetry) {
+    return await run();
+  }
+  const span = await driver.telemetry.startSpan();
+  let outcome = "failed";
+  try {
+    const result = await span.run(() => run(span));
+    outcome = runOutcomeFromState(result.state);
+    return result;
+  } catch (error) {
+    if (error instanceof EvictedError) {
+      outcome = "cancelled";
+    }
+    throw error;
+  } finally {
+    await span.finish(outcome);
+  }
+}
+async function executeWorkflow(workflowId, workflowFn, input, driver, messageDriver, abortController, onHistoryUpdated, onError, logger, runSpan) {
   var _a;
   const storage = await loadStorage(driver);
   const historyNotifier = onHistoryUpdated ? () => onHistoryUpdated(createHistorySnapshot(storage)) : void 0;
@@ -4168,7 +4226,8 @@ async function executeWorkflow(workflowId, workflowFn, input, driver, messageDri
         storage,
         historyNotifier,
         onError,
-        logger
+        logger,
+        runSpan
       );
     } catch (error) {
       if (error instanceof EvictedError) {
@@ -4196,7 +4255,9 @@ async function executeWorkflow(workflowId, workflowFn, input, driver, messageDri
     false,
     historyNotifier,
     onError,
-    logger
+    logger,
+    void 0,
+    runSpan
   );
   storage.state = "running";
   try {
@@ -4263,7 +4324,8 @@ async function executeWorkflow(workflowId, workflowFn, input, driver, messageDri
         storage,
         historyNotifier,
         onError,
-        logger
+        logger,
+        runSpan
       );
     } catch (rollbackError) {
       if (rollbackError instanceof EvictedError) {
@@ -4338,4 +4400,4 @@ export {
   runWorkflow,
   replayWorkflowFromStep
 };
-//# sourceMappingURL=chunk-TVRZA5O5.js.map
+//# sourceMappingURL=chunk-EGOPMLG2.js.map