@mastra/inngest 1.0.0-beta.3 → 1.0.0-beta.5

package/dist/index.cjs

@@ -1,43 +1,302 @@
 'use strict';
 
-var crypto = require('crypto');
-var web = require('stream/web');
-var realtime = require('@inngest/realtime');
-var di = require('@mastra/core/di');
-var observability = require('@mastra/core/observability');
-var stream = require('@mastra/core/stream');
 var tools = require('@mastra/core/tools');
 var workflows = require('@mastra/core/workflows');
 var _constants = require('@mastra/core/workflows/_constants');
+var zod = require('zod');
+var crypto = require('crypto');
+var di = require('@mastra/core/di');
 var inngest = require('inngest');
+var web = require('stream/web');
+var realtime = require('@inngest/realtime');
+var stream = require('@mastra/core/stream');
 var hono = require('inngest/hono');
-var zod = require('zod');
 
 // src/index.ts
-function serve({
-  mastra,
-  inngest,
-  functions: userFunctions = [],
-  registerOptions
-}) {
-  const wfs = mastra.listWorkflows();
-  const workflowFunctions = Array.from(
-    new Set(
-      Object.values(wfs).flatMap((wf) => {
-        if (wf instanceof InngestWorkflow) {
-          wf.__registerMastra(mastra);
-          return wf.getFunctions();
+var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
+  inngestStep;
+  inngestAttempts;
+  constructor(mastra, inngestStep, inngestAttempts = 0, options) {
+    super({ mastra, options });
+    this.inngestStep = inngestStep;
+    this.inngestAttempts = inngestAttempts;
+  }
+  // =============================================================================
+  // Hook Overrides
+  // =============================================================================
+  /**
+   * Format errors with stack traces for better debugging in Inngest
+   */
+  formatResultError(error, lastOutput) {
+    if (error instanceof Error) {
+      return error.stack ?? error.message;
+    }
+    const outputError = lastOutput?.error;
+    if (outputError instanceof Error) {
+      return outputError.message;
+    }
+    return outputError ?? error ?? "Unknown error";
+  }
+  /**
+   * Detect InngestWorkflow instances for special nested workflow handling
+   */
+  isNestedWorkflowStep(step) {
+    return step instanceof InngestWorkflow;
+  }
+  /**
+   * Inngest requires requestContext serialization for memoization.
+   * When steps are replayed, the original function doesn't re-execute,
+   * so requestContext modifications must be captured and restored.
+   */
+  requiresDurableContextSerialization() {
+    return true;
+  }
+  /**
+   * Execute a step with retry logic for Inngest.
+   * Retries are handled via step-level retry (RetryAfterError thrown INSIDE step.run()).
+   * After retries exhausted, error propagates here and we return a failed result.
+   */
+  async executeStepWithRetry(stepId, runStep, params) {
+    try {
+      const result = await this.wrapDurableOperation(stepId, runStep, { delay: params.delay });
+      return { ok: true, result };
+    } catch (e) {
+      const cause = e?.cause;
+      if (cause?.status === "failed") {
+        params.stepSpan?.error({
+          error: e,
+          attributes: { status: "failed" }
+        });
+        return { ok: false, error: cause };
+      }
+      const errorMessage = e instanceof Error ? e.message : String(e);
+      params.stepSpan?.error({
+        error: e,
+        attributes: { status: "failed" }
+      });
+      return {
+        ok: false,
+        error: {
+          status: "failed",
+          error: `Error: ${errorMessage}`,
+          endedAt: Date.now()
         }
-        return [];
-      })
-    )
-  );
-  return hono.serve({
-    ...registerOptions,
-    client: inngest,
-    functions: [...workflowFunctions, ...userFunctions]
-  });
-}
+      };
+    }
+  }
+  /**
+   * Use Inngest's sleep primitive for durability
+   */
+  async executeSleepDuration(duration, sleepId, workflowId) {
+    await this.inngestStep.sleep(`workflow.${workflowId}.sleep.${sleepId}`, duration < 0 ? 0 : duration);
+  }
+  /**
+   * Use Inngest's sleepUntil primitive for durability
+   */
+  async executeSleepUntilDate(date, sleepUntilId, workflowId) {
+    await this.inngestStep.sleepUntil(`workflow.${workflowId}.sleepUntil.${sleepUntilId}`, date);
+  }
+  /**
+   * Wrap durable operations in Inngest step.run() for durability.
+   * If retryConfig is provided, throws RetryAfterError INSIDE step.run() to trigger
+   * Inngest's step-level retry mechanism (not function-level retry).
+   */
+  async wrapDurableOperation(operationId, operationFn, retryConfig) {
+    return this.inngestStep.run(operationId, async () => {
+      try {
+        return await operationFn();
+      } catch (e) {
+        if (retryConfig) {
+          const errorMessage = e instanceof Error ? e.message : String(e);
+          throw new inngest.RetryAfterError(errorMessage, retryConfig.delay, {
+            cause: {
+              status: "failed",
+              error: `Error: ${errorMessage}`,
+              endedAt: Date.now()
+            }
+          });
+        }
+        throw e;
+      }
+    });
+  }
+  /**
+   * Provide Inngest step primitive in engine context
+   */
+  getEngineContext() {
+    return { step: this.inngestStep };
+  }
+  /**
+   * Execute nested InngestWorkflow using inngestStep.invoke() for durability.
+   * This MUST be called directly (not inside step.run()) due to Inngest constraints.
+   */
+  async executeWorkflowStep(params) {
+    if (!(params.step instanceof InngestWorkflow)) {
+      return null;
+    }
+    const { step, stepResults, executionContext, resume, timeTravel, prevOutput, inputData, emitter, startedAt } = params;
+    const isResume = !!resume?.steps?.length;
+    let result;
+    let runId;
+    const isTimeTravel = !!(timeTravel && timeTravel.steps?.length > 1 && timeTravel.steps[0] === step.id);
+    try {
+      if (isResume) {
+        runId = stepResults[resume?.steps?.[0] ?? ""]?.suspendPayload?.__workflow_meta?.runId ?? crypto.randomUUID();
+        const snapshot = await this.mastra?.getStorage()?.loadWorkflowSnapshot({
+          workflowName: step.id,
+          runId
+        });
+        const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
+          function: step.getFunction(),
+          data: {
+            inputData,
+            initialState: executionContext.state ?? snapshot?.value ?? {},
+            runId,
+            resume: {
+              runId,
+              steps: resume.steps.slice(1),
+              stepResults: snapshot?.context,
+              resumePayload: resume.resumePayload,
+              resumePath: resume.steps?.[1] ? snapshot?.suspendedPaths?.[resume.steps?.[1]] : void 0
+            },
+            outputOptions: { includeState: true }
+          }
+        });
+        result = invokeResp.result;
+        runId = invokeResp.runId;
+        executionContext.state = invokeResp.result.state;
+      } else if (isTimeTravel) {
+        const snapshot = await this.mastra?.getStorage()?.loadWorkflowSnapshot({
+          workflowName: step.id,
+          runId: executionContext.runId
+        }) ?? { context: {} };
+        const timeTravelParams = workflows.createTimeTravelExecutionParams({
+          steps: timeTravel.steps.slice(1),
+          inputData: timeTravel.inputData,
+          resumeData: timeTravel.resumeData,
+          context: timeTravel.nestedStepResults?.[step.id] ?? {},
+          nestedStepsContext: timeTravel.nestedStepResults ?? {},
+          snapshot,
+          graph: step.buildExecutionGraph()
+        });
+        const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
+          function: step.getFunction(),
+          data: {
+            timeTravel: timeTravelParams,
+            initialState: executionContext.state ?? {},
+            runId: executionContext.runId,
+            outputOptions: { includeState: true }
+          }
+        });
+        result = invokeResp.result;
+        runId = invokeResp.runId;
+        executionContext.state = invokeResp.result.state;
+      } else {
+        const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
+          function: step.getFunction(),
+          data: {
+            inputData,
+            initialState: executionContext.state ?? {},
+            outputOptions: { includeState: true }
+          }
+        });
+        result = invokeResp.result;
+        runId = invokeResp.runId;
+        executionContext.state = invokeResp.result.state;
+      }
+    } catch (e) {
+      const errorCause = e?.cause;
+      if (errorCause && typeof errorCause === "object") {
+        result = errorCause;
+        runId = errorCause.runId || crypto.randomUUID();
+      } else {
+        runId = crypto.randomUUID();
+        result = {
+          status: "failed",
+          error: e instanceof Error ? e : new Error(String(e)),
+          steps: {},
+          input: inputData
+        };
+      }
+    }
+    const res = await this.inngestStep.run(
+      `workflow.${executionContext.workflowId}.step.${step.id}.nestedwf-results`,
+      async () => {
+        if (result.status === "failed") {
+          await emitter.emit("watch", {
+            type: "workflow-step-result",
+            payload: {
+              id: step.id,
+              status: "failed",
+              error: result?.error,
+              payload: prevOutput
+            }
+          });
+          return { executionContext, result: { status: "failed", error: result?.error } };
+        } else if (result.status === "suspended") {
+          const suspendedSteps = Object.entries(result.steps).filter(([_stepName, stepResult]) => {
+            const stepRes = stepResult;
+            return stepRes?.status === "suspended";
+          });
+          for (const [stepName, stepResult] of suspendedSteps) {
+            const suspendPath = [stepName, ...stepResult?.suspendPayload?.__workflow_meta?.path ?? []];
+            executionContext.suspendedPaths[step.id] = executionContext.executionPath;
+            await emitter.emit("watch", {
+              type: "workflow-step-suspended",
+              payload: {
+                id: step.id,
+                status: "suspended"
+              }
+            });
+            return {
+              executionContext,
+              result: {
+                status: "suspended",
+                payload: stepResult.payload,
+                suspendPayload: {
+                  ...stepResult?.suspendPayload,
+                  __workflow_meta: { runId, path: suspendPath }
+                }
+              }
+            };
+          }
+          return {
+            executionContext,
+            result: {
+              status: "suspended",
+              payload: {}
+            }
+          };
+        }
+        await emitter.emit("watch", {
+          type: "workflow-step-result",
+          payload: {
+            id: step.id,
+            status: "success",
+            output: result?.result
+          }
+        });
+        await emitter.emit("watch", {
+          type: "workflow-step-finish",
+          payload: {
+            id: step.id,
+            metadata: {}
+          }
+        });
+        return { executionContext, result: { status: "success", output: result?.result } };
+      }
+    );
+    Object.assign(executionContext, res.executionContext);
+    return {
+      ...res.result,
+      startedAt,
+      endedAt: Date.now(),
+      payload: inputData,
+      resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
+      resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
+    };
+  }
+};
 var InngestRun = class extends workflows.Run {
   inngest;
   serializedStepGraph;
@@ -115,7 +374,8 @@ var InngestRun = class extends workflows.Run {
     initialState,
     outputOptions,
     tracingOptions,
-    format
+    format,
+    requestContext
   }) {
     await this.#mastra.getStorage()?.persistWorkflowSnapshot({
       workflowName: this.workflowId,
@@ -146,7 +406,8 @@ var InngestRun = class extends workflows.Run {
         resourceId: this.resourceId,
         outputOptions,
         tracingOptions,
-        format
+        format,
+        requestContext: requestContext ? Object.fromEntries(requestContext.entries()) : {}
       }
     });
     const eventId = eventOutput.ids[0];
@@ -190,6 +451,9 @@ var InngestRun = class extends workflows.Run {
     });
     const suspendedStep = this.workflowSteps[steps?.[0] ?? ""];
     const resumeDataToUse = await this._validateResumeData(params.resumeData, suspendedStep);
+    const persistedRequestContext = snapshot?.requestContext ?? {};
+    const newRequestContext = params.requestContext ? Object.fromEntries(params.requestContext.entries()) : {};
+    const mergedRequestContext = { ...persistedRequestContext, ...newRequestContext };
     const eventOutput = await this.inngest.send({
       name: `workflow.${this.workflowId}`,
       data: {
@@ -203,7 +467,8 @@ var InngestRun = class extends workflows.Run {
           stepResults: snapshot?.context,
           resumePayload: resumeDataToUse,
           resumePath: steps?.[0] ? snapshot?.suspendedPaths?.[steps?.[0]] : void 0
-        }
+        },
+        requestContext: mergedRequestContext
       }
     });
     const eventId = eventOutput.ids[0];
@@ -294,7 +559,8 @@ var InngestRun = class extends workflows.Run {
         stepResults: timeTravelData.stepResults,
         timeTravel: timeTravelData,
         tracingOptions: params.tracingOptions,
-        outputOptions: params.outputOptions
+        outputOptions: params.outputOptions,
+        requestContext: params.requestContext ? Object.fromEntries(params.requestContext.entries()) : {}
       }
     });
     const eventId = eventOutput.ids[0];
@@ -334,14 +600,14 @@ var InngestRun = class extends workflows.Run {
   streamLegacy({ inputData, requestContext } = {}) {
     const { readable, writable } = new TransformStream();
     const writer = writable.getWriter();
+    void writer.write({
+      // @ts-ignore
+      type: "start",
+      // @ts-ignore
+      payload: { runId: this.runId }
+    });
     const unwatch = this.watch(async (event) => {
       try {
-        await writer.write({
-          // @ts-ignore
-          type: "start",
-          // @ts-ignore
-          payload: { runId: this.runId }
-        });
         const e = {
           ...event,
           type: event.type.replace("workflow-", "")
@@ -487,7 +753,7 @@ var InngestRun = class extends workflows.Run {
         self.closeStreamAction = async () => {
           unwatch();
           try {
-            await controller.close();
+            controller.close();
           } catch (err) {
             console.error("Error closing stream:", err);
           }
@@ -527,6 +793,8 @@ var InngestRun = class extends workflows.Run {
     return this.streamOutput;
   }
 };
+
+// src/workflow.ts
 var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
   #mastra;
   inngest;
@@ -680,20 +948,17 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
           initialState,
           emitter,
           retryConfig: this.retryConfig,
-          requestContext: new di.RequestContext(),
-          // TODO
+          requestContext: new di.RequestContext(Object.entries(event.data.requestContext ?? {})),
           resume,
           timeTravel,
           format,
           abortController: new AbortController(),
           // currentSpan: undefined, // TODO: Pass actual parent Span from workflow execution context
           outputOptions,
-          writableStream: new web.WritableStream({
-            write(chunk) {
-              void emitter.emit("watch", chunk).catch(() => {
-              });
-            }
-          })
+          outputWriter: async (chunk) => {
+            void emitter.emit("watch", chunk).catch(() => {
+            });
+          }
         });
         await step.run(`workflow.${this.id}.finalize`, async () => {
           if (result.status === "failed") {
@@ -725,14 +990,50 @@ var InngestWorkflow = class _InngestWorkflow extends workflows.Workflow {
     return [this.getFunction(), ...this.getNestedFunctions(this.executionGraph.steps)];
   }
 };
+function serve({
+  mastra,
+  inngest,
+  functions: userFunctions = [],
+  registerOptions
+}) {
+  const wfs = mastra.listWorkflows();
+  const workflowFunctions = Array.from(
+    new Set(
+      Object.values(wfs).flatMap((wf) => {
+        if (wf instanceof InngestWorkflow) {
+          wf.__registerMastra(mastra);
+          return wf.getFunctions();
+        }
+        return [];
+      })
+    )
+  );
+  return hono.serve({
+    ...registerOptions,
+    client: inngest,
+    functions: [...workflowFunctions, ...userFunctions]
+  });
+}
+
+// src/types.ts
+var _compatibilityCheck = true;
+
+// src/index.ts
 function isAgent(params) {
   return params?.component === "AGENT";
 }
 function isTool(params) {
   return params instanceof tools.Tool;
 }
+function isInngestWorkflow(params) {
+  return params instanceof InngestWorkflow;
+}
 function createStep(params, agentOptions) {
+  if (isInngestWorkflow(params)) {
+    return params;
+  }
   if (isAgent(params)) {
+    const outputSchema = agentOptions?.structuredOutput?.schema ?? zod.z.object({ text: zod.z.string() });
     return {
       id: params.name,
       description: params.getDescription(),
@@ -741,9 +1042,7 @@ function createStep(params, agentOptions) {
         // resourceId: z.string().optional(),
         // threadId: z.string().optional(),
       }),
-      outputSchema: zod.z.object({
-        text: zod.z.string()
-      }),
+      outputSchema,
       execute: async ({
         inputData,
         [_constants.EMITTER_SYMBOL]: emitter,
@@ -759,6 +1058,7 @@ function createStep(params, agentOptions) {
           streamPromise.resolve = resolve;
           streamPromise.reject = reject;
         });
+        let structuredResult = null;
         const toolData = {
           name: params.name,
           args: inputData
@@ -772,6 +1072,10 @@ function createStep(params, agentOptions) {
             requestContext,
             tracingContext,
             onFinish: (result) => {
+              const resultWithObject = result;
+              if (agentOptions?.structuredOutput?.schema && resultWithObject.object) {
+                structuredResult = resultWithObject.object;
+              }
               streamPromise.resolve(result.text);
               void agentOptions?.onFinish?.(result);
             },
@@ -784,6 +1088,10 @@ function createStep(params, agentOptions) {
             requestContext,
             tracingContext,
             onFinish: (result) => {
+              const resultWithObject = result;
+              if (agentOptions?.structuredOutput?.schema && resultWithObject.object) {
+                structuredResult = resultWithObject.object;
+              }
               streamPromise.resolve(result.text);
               void agentOptions?.onFinish?.(result);
             },
@@ -817,6 +1125,9 @@ function createStep(params, agentOptions) {
         if (abortSignal.aborted) {
           return abort();
         }
+        if (structuredResult !== null) {
+          return structuredResult;
+        }
         return {
           text: await streamPromise.promise
         };
@@ -906,7 +1217,8 @@ function init(inngest) {
         inputSchema: workflow.inputSchema,
         outputSchema: workflow.outputSchema,
         steps: workflow.stepDefs,
-        mastra: workflow.mastra
+        mastra: workflow.mastra,
+        options: workflow.options
       });
       wf.setStepFlow(workflow.stepGraph);
       wf.commit();
@@ -914,856 +1226,11 @@ function init(inngest) {
     }
   };
 }
-var InngestExecutionEngine = class extends workflows.DefaultExecutionEngine {
-  inngestStep;
-  inngestAttempts;
-  constructor(mastra, inngestStep, inngestAttempts = 0, options) {
-    super({ mastra, options });
-    this.inngestStep = inngestStep;
-    this.inngestAttempts = inngestAttempts;
-  }
-  async fmtReturnValue(emitter, stepResults, lastOutput, error) {
-    const base = {
-      status: lastOutput.status,
-      steps: stepResults
-    };
-    if (lastOutput.status === "success") {
-      base.result = lastOutput.output;
-    } else if (lastOutput.status === "failed") {
-      base.error = error instanceof Error ? error?.stack ?? error.message : lastOutput?.error instanceof Error ? lastOutput.error.message : lastOutput.error ?? error ?? "Unknown error";
-    } else if (lastOutput.status === "suspended") {
-      const suspendedStepIds = Object.entries(stepResults).flatMap(([stepId, stepResult]) => {
-        if (stepResult?.status === "suspended") {
-          const nestedPath = stepResult?.suspendPayload?.__workflow_meta?.path;
-          return nestedPath ? [[stepId, ...nestedPath]] : [[stepId]];
-        }
-        return [];
-      });
-      base.suspended = suspendedStepIds;
-    }
-    return base;
-  }
-  // async executeSleep({ id, duration }: { id: string; duration: number }): Promise<void> {
-  //   await this.inngestStep.sleep(id, duration);
-  // }
-  async executeSleep({
-    workflowId,
-    runId,
-    entry,
-    prevOutput,
-    stepResults,
-    emitter,
-    abortController,
-    requestContext,
-    executionContext,
-    writableStream,
-    tracingContext
-  }) {
-    let { duration, fn } = entry;
-    const sleepSpan = tracingContext?.currentSpan?.createChildSpan({
-      type: observability.SpanType.WORKFLOW_SLEEP,
-      name: `sleep: ${duration ? `${duration}ms` : "dynamic"}`,
-      attributes: {
-        durationMs: duration,
-        sleepType: fn ? "dynamic" : "fixed"
-      },
-      tracingPolicy: this.options?.tracingPolicy
-    });
-    if (fn) {
-      const stepCallId = crypto.randomUUID();
-      duration = await this.inngestStep.run(`workflow.${workflowId}.sleep.${entry.id}`, async () => {
-        return await fn(
-          workflows.createDeprecationProxy(
-            {
-              runId,
-              workflowId,
-              mastra: this.mastra,
-              requestContext,
-              inputData: prevOutput,
-              state: executionContext.state,
-              setState: (state) => {
-                executionContext.state = state;
-              },
-              retryCount: -1,
-              tracingContext: {
-                currentSpan: sleepSpan
-              },
-              getInitData: () => stepResults?.input,
-              getStepResult: workflows.getStepResult.bind(this, stepResults),
-              // TODO: this function shouldn't have suspend probably?
-              suspend: async (_suspendPayload) => {
-              },
-              bail: () => {
-              },
-              abort: () => {
-                abortController?.abort();
-              },
-              [_constants.EMITTER_SYMBOL]: emitter,
-              [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
-              engine: { step: this.inngestStep },
-              abortSignal: abortController?.signal,
-              writer: new tools.ToolStream(
-                {
-                  prefix: "workflow-step",
-                  callId: stepCallId,
-                  name: "sleep",
-                  runId
-                },
-                writableStream
-              )
-            },
-            {
-              paramName: "runCount",
-              deprecationMessage: workflows.runCountDeprecationMessage,
-              logger: this.logger
-            }
-          )
-        );
-      });
-      sleepSpan?.update({
-        attributes: {
-          durationMs: duration
-        }
-      });
-    }
-    try {
-      await this.inngestStep.sleep(entry.id, !duration || duration < 0 ? 0 : duration);
-      sleepSpan?.end();
-    } catch (e) {
-      sleepSpan?.error({ error: e });
-      throw e;
-    }
-  }
-  async executeSleepUntil({
-    workflowId,
-    runId,
-    entry,
-    prevOutput,
-    stepResults,
-    emitter,
-    abortController,
-    requestContext,
-    executionContext,
-    writableStream,
-    tracingContext
-  }) {
-    let { date, fn } = entry;
-    const sleepUntilSpan = tracingContext?.currentSpan?.createChildSpan({
-      type: observability.SpanType.WORKFLOW_SLEEP,
-      name: `sleepUntil: ${date ? date.toISOString() : "dynamic"}`,
-      attributes: {
-        untilDate: date,
-        durationMs: date ? Math.max(0, date.getTime() - Date.now()) : void 0,
-        sleepType: fn ? "dynamic" : "fixed"
-      },
-      tracingPolicy: this.options?.tracingPolicy
-    });
-    if (fn) {
-      date = await this.inngestStep.run(`workflow.${workflowId}.sleepUntil.${entry.id}`, async () => {
-        const stepCallId = crypto.randomUUID();
-        return await fn(
-          workflows.createDeprecationProxy(
-            {
-              runId,
-              workflowId,
-              mastra: this.mastra,
-              requestContext,
-              inputData: prevOutput,
-              state: executionContext.state,
-              setState: (state) => {
-                executionContext.state = state;
-              },
-              retryCount: -1,
-              tracingContext: {
-                currentSpan: sleepUntilSpan
-              },
-              getInitData: () => stepResults?.input,
-              getStepResult: workflows.getStepResult.bind(this, stepResults),
-              // TODO: this function shouldn't have suspend probably?
-              suspend: async (_suspendPayload) => {
-              },
-              bail: () => {
-              },
-              abort: () => {
-                abortController?.abort();
-              },
-              [_constants.EMITTER_SYMBOL]: emitter,
-              [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
-              engine: { step: this.inngestStep },
-              abortSignal: abortController?.signal,
-              writer: new tools.ToolStream(
-                {
-                  prefix: "workflow-step",
-                  callId: stepCallId,
-                  name: "sleep",
-                  runId
-                },
-                writableStream
-              )
-            },
-            {
-              paramName: "runCount",
-              deprecationMessage: workflows.runCountDeprecationMessage,
-              logger: this.logger
-            }
-          )
-        );
-      });
-      if (date && !(date instanceof Date)) {
-        date = new Date(date);
-      }
-      const time = !date ? 0 : date.getTime() - Date.now();
-      sleepUntilSpan?.update({
-        attributes: {
-          durationMs: Math.max(0, time)
-        }
-      });
-    }
-    if (!(date instanceof Date)) {
-      sleepUntilSpan?.end();
-      return;
-    }
-    try {
-      await this.inngestStep.sleepUntil(entry.id, date);
-      sleepUntilSpan?.end();
-    } catch (e) {
-      sleepUntilSpan?.error({ error: e });
-      throw e;
-    }
-  }
-  async executeStep({
-    step,
-    stepResults,
-    executionContext,
-    resume,
-    timeTravel,
-    prevOutput,
-    emitter,
-    abortController,
-    requestContext,
-    tracingContext,
-    writableStream,
-    disableScorers
-  }) {
-    const stepSpan = tracingContext?.currentSpan?.createChildSpan({
-      name: `workflow step: '${step.id}'`,
-      type: observability.SpanType.WORKFLOW_STEP,
-      input: prevOutput,
-      attributes: {
-        stepId: step.id
-      },
-      tracingPolicy: this.options?.tracingPolicy
-    });
-    const { inputData, validationError } = await workflows.validateStepInput({
-      prevOutput,
-      step,
-      validateInputs: this.options?.validateInputs ?? true
-    });
-    const startedAt = await this.inngestStep.run(
-      `workflow.${executionContext.workflowId}.run.${executionContext.runId}.step.${step.id}.running_ev`,
-      async () => {
-        const startedAt2 = Date.now();
-        await emitter.emit("watch", {
-          type: "workflow-step-start",
-          payload: {
-            id: step.id,
-            status: "running",
-            payload: inputData,
-            startedAt: startedAt2
-          }
-        });
-        return startedAt2;
-      }
-    );
-    if (step instanceof InngestWorkflow) {
-      const isResume = !!resume?.steps?.length;
-      let result;
-      let runId;
-      const isTimeTravel = !!(timeTravel && timeTravel.steps?.length > 1 && timeTravel.steps[0] === step.id);
-      try {
-        if (isResume) {
-          runId = stepResults[resume?.steps?.[0] ?? ""]?.suspendPayload?.__workflow_meta?.runId ?? crypto.randomUUID();
-          const snapshot = await this.mastra?.getStorage()?.loadWorkflowSnapshot({
-            workflowName: step.id,
-            runId
-          });
-          const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
-            function: step.getFunction(),
-            data: {
-              inputData,
-              initialState: executionContext.state ?? snapshot?.value ?? {},
-              runId,
-              resume: {
-                runId,
-                steps: resume.steps.slice(1),
-                stepResults: snapshot?.context,
-                resumePayload: resume.resumePayload,
-                resumePath: resume.steps?.[1] ? snapshot?.suspendedPaths?.[resume.steps?.[1]] : void 0
-              },
-              outputOptions: { includeState: true }
-            }
-          });
-          result = invokeResp.result;
-          runId = invokeResp.runId;
-          executionContext.state = invokeResp.result.state;
-        } else if (isTimeTravel) {
-          const snapshot = await this.mastra?.getStorage()?.loadWorkflowSnapshot({
-            workflowName: step.id,
-            runId: executionContext.runId
-          }) ?? { context: {} };
-          const timeTravelParams = workflows.createTimeTravelExecutionParams({
-            steps: timeTravel.steps.slice(1),
-            inputData: timeTravel.inputData,
-            resumeData: timeTravel.resumeData,
-            context: timeTravel.nestedStepResults?.[step.id] ?? {},
-            nestedStepsContext: timeTravel.nestedStepResults ?? {},
-            snapshot,
-            graph: step.buildExecutionGraph()
-          });
-          const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
-            function: step.getFunction(),
-            data: {
-              timeTravel: timeTravelParams,
-              initialState: executionContext.state ?? {},
-              runId: executionContext.runId,
-              outputOptions: { includeState: true }
-            }
-          });
-          result = invokeResp.result;
-          runId = invokeResp.runId;
-          executionContext.state = invokeResp.result.state;
-        } else {
-          const invokeResp = await this.inngestStep.invoke(`workflow.${executionContext.workflowId}.step.${step.id}`, {
-            function: step.getFunction(),
-            data: {
-              inputData,
-              initialState: executionContext.state ?? {},
-              outputOptions: { includeState: true }
-            }
-          });
-          result = invokeResp.result;
-          runId = invokeResp.runId;
-          executionContext.state = invokeResp.result.state;
-        }
-      } catch (e) {
-        const errorCause = e?.cause;
-        if (errorCause && typeof errorCause === "object") {
-          result = errorCause;
-          runId = errorCause.runId || crypto.randomUUID();
-        } else {
-          runId = crypto.randomUUID();
-          result = {
-            status: "failed",
-            error: e instanceof Error ? e : new Error(String(e)),
-            steps: {},
-            input: inputData
-          };
-        }
-      }
-      const res = await this.inngestStep.run(
-        `workflow.${executionContext.workflowId}.step.${step.id}.nestedwf-results`,
-        async () => {
-          if (result.status === "failed") {
-            await emitter.emit("watch", {
-              type: "workflow-step-result",
-              payload: {
-                id: step.id,
-                status: "failed",
-                error: result?.error,
-                payload: prevOutput
-              }
-            });
-            return { executionContext, result: { status: "failed", error: result?.error } };
-          } else if (result.status === "suspended") {
-            const suspendedSteps = Object.entries(result.steps).filter(([_stepName, stepResult]) => {
-              const stepRes2 = stepResult;
-              return stepRes2?.status === "suspended";
-            });
-            for (const [stepName, stepResult] of suspendedSteps) {
-              const suspendPath = [stepName, ...stepResult?.suspendPayload?.__workflow_meta?.path ?? []];
-              executionContext.suspendedPaths[step.id] = executionContext.executionPath;
-              await emitter.emit("watch", {
-                type: "workflow-step-suspended",
-                payload: {
-                  id: step.id,
-                  status: "suspended"
-                }
-              });
-              return {
-                executionContext,
-                result: {
-                  status: "suspended",
-                  payload: stepResult.payload,
-                  suspendPayload: {
-                    ...stepResult?.suspendPayload,
-                    __workflow_meta: { runId, path: suspendPath }
-                  }
-                }
-              };
-            }
-            return {
-              executionContext,
-              result: {
-                status: "suspended",
-                payload: {}
-              }
-            };
-          }
-          await emitter.emit("watch", {
-            type: "workflow-step-result",
-            payload: {
-              id: step.id,
-              status: "success",
-              output: result?.result
-            }
-          });
-          await emitter.emit("watch", {
-            type: "workflow-step-finish",
-            payload: {
-              id: step.id,
-              metadata: {}
-            }
-          });
-          return { executionContext, result: { status: "success", output: result?.result } };
-        }
-      );
-      Object.assign(executionContext, res.executionContext);
-      return {
-        ...res.result,
-        startedAt,
-        endedAt: Date.now(),
-        payload: inputData,
-        resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
-        resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
-      };
-    }
-    const stepCallId = crypto.randomUUID();
-    let stepRes;
-    try {
-      stepRes = await this.inngestStep.run(`workflow.${executionContext.workflowId}.step.${step.id}`, async () => {
-        let execResults;
-        let suspended;
-        let bailed;
-        const { resumeData: timeTravelResumeData, validationError: timeTravelResumeValidationError } = await workflows.validateStepResumeData({
-          resumeData: timeTravel?.stepResults[step.id]?.status === "suspended" ? timeTravel?.resumeData : void 0,
-          step
-        });
-        let resumeDataToUse;
-        if (timeTravelResumeData && !timeTravelResumeValidationError) {
-          resumeDataToUse = timeTravelResumeData;
-        } else if (timeTravelResumeData && timeTravelResumeValidationError) {
-          this.logger.warn("Time travel resume data validation failed", {
-            stepId: step.id,
-            error: timeTravelResumeValidationError.message
-          });
-        } else if (resume?.steps[0] === step.id) {
-          resumeDataToUse = resume?.resumePayload;
-        }
-        try {
-          if (validationError) {
-            throw validationError;
-          }
-          const retryCount = this.getOrGenerateRetryCount(step.id);
-          const result = await step.execute({
-            runId: executionContext.runId,
-            workflowId: executionContext.workflowId,
-            mastra: this.mastra,
-            requestContext,
-            retryCount,
-            writer: new tools.ToolStream(
-              {
-                prefix: "workflow-step",
-                callId: stepCallId,
-                name: step.id,
-                runId: executionContext.runId
-              },
-              writableStream
-            ),
-            state: executionContext?.state ?? {},
-            setState: (state) => {
-              executionContext.state = state;
-            },
-            inputData,
-            resumeData: resumeDataToUse,
-            tracingContext: {
-              currentSpan: stepSpan
-            },
-            getInitData: () => stepResults?.input,
-            getStepResult: workflows.getStepResult.bind(this, stepResults),
-            suspend: async (suspendPayload, suspendOptions) => {
-              const { suspendData, validationError: validationError2 } = await workflows.validateStepSuspendData({
-                suspendData: suspendPayload,
-                step
-              });
-              if (validationError2) {
-                throw validationError2;
-              }
-              executionContext.suspendedPaths[step.id] = executionContext.executionPath;
-              if (suspendOptions?.resumeLabel) {
-                const resumeLabel = Array.isArray(suspendOptions.resumeLabel) ? suspendOptions.resumeLabel : [suspendOptions.resumeLabel];
-                for (const label of resumeLabel) {
-                  executionContext.resumeLabels[label] = {
-                    stepId: step.id,
-                    foreachIndex: executionContext.foreachIndex
-                  };
-                }
-              }
-              suspended = { payload: suspendData };
-            },
-            bail: (result2) => {
-              bailed = { payload: result2 };
-            },
-            abort: () => {
-              abortController?.abort();
-            },
-            [_constants.EMITTER_SYMBOL]: emitter,
-            [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
-            engine: {
-              step: this.inngestStep
-            },
-            abortSignal: abortController.signal
-          });
-          const endedAt = Date.now();
-          execResults = {
-            status: "success",
-            output: result,
-            startedAt,
-            endedAt,
-            payload: inputData,
-            resumedAt: resumeDataToUse ? startedAt : void 0,
-            resumePayload: resumeDataToUse
-          };
-        } catch (e) {
-          const stepFailure = {
-            status: "failed",
-            payload: inputData,
-            error: e instanceof Error ? e.message : String(e),
-            endedAt: Date.now(),
-            startedAt,
-            resumedAt: resumeDataToUse ? startedAt : void 0,
-            resumePayload: resumeDataToUse
-          };
-          execResults = stepFailure;
-          const fallbackErrorMessage = `Step ${step.id} failed`;
-          stepSpan?.error({ error: new Error(execResults.error ?? fallbackErrorMessage) });
-          throw new inngest.RetryAfterError(execResults.error ?? fallbackErrorMessage, executionContext.retryConfig.delay, {
-            cause: execResults
-          });
-        }
-        if (suspended) {
-          execResults = {
-            status: "suspended",
-            suspendPayload: suspended.payload,
-            ...execResults.output ? { suspendOutput: execResults.output } : {},
-            payload: inputData,
-            suspendedAt: Date.now(),
-            startedAt,
-            resumedAt: resumeDataToUse ? startedAt : void 0,
-            resumePayload: resumeDataToUse
-          };
-        } else if (bailed) {
-          execResults = {
-            status: "bailed",
-            output: bailed.payload,
-            payload: inputData,
-            endedAt: Date.now(),
-            startedAt
-          };
-        }
-        if (execResults.status === "suspended") {
-          await emitter.emit("watch", {
-            type: "workflow-step-suspended",
-            payload: {
-              id: step.id,
-              ...execResults
-            }
-          });
-        } else {
-          await emitter.emit("watch", {
-            type: "workflow-step-result",
-            payload: {
-              id: step.id,
-              ...execResults
-            }
-          });
-          await emitter.emit("watch", {
-            type: "workflow-step-finish",
-            payload: {
-              id: step.id,
-              metadata: {}
-            }
-          });
-        }
-        stepSpan?.end({ output: execResults });
-        return { result: execResults, executionContext, stepResults };
-      });
-    } catch (e) {
-      const stepFailure = e instanceof Error ? e?.cause : {
-        status: "failed",
-        error: e instanceof Error ? e.message : String(e),
-        payload: inputData,
-        startedAt,
-        endedAt: Date.now()
-      };
-      stepRes = {
-        result: stepFailure,
-        executionContext,
-        stepResults: {
-          ...stepResults,
-          [step.id]: stepFailure
-        }
-      };
-    }
-    if (disableScorers !== false && stepRes.result.status === "success") {
-      await this.inngestStep.run(`workflow.${executionContext.workflowId}.step.${step.id}.score`, async () => {
-        if (step.scorers) {
-          await this.runScorers({
-            scorers: step.scorers,
-            runId: executionContext.runId,
-            input: inputData,
-            output: stepRes.result,
-            workflowId: executionContext.workflowId,
-            stepId: step.id,
-            requestContext,
-            disableScorers,
-            tracingContext: { currentSpan: stepSpan }
-          });
-        }
-      });
-    }
-    Object.assign(executionContext.suspendedPaths, stepRes.executionContext.suspendedPaths);
-    executionContext.state = stepRes.executionContext.state;
-    return stepRes.result;
-  }
-  async persistStepUpdate({
-    workflowId,
-    runId,
-    stepResults,
-    resourceId,
-    executionContext,
-    serializedStepGraph,
-    workflowStatus,
-    result,
-    error
-  }) {
-    await this.inngestStep.run(
-      `workflow.${workflowId}.run.${runId}.path.${JSON.stringify(executionContext.executionPath)}.stepUpdate`,
-      async () => {
-        const shouldPersistSnapshot = this.options.shouldPersistSnapshot({ stepResults, workflowStatus });
-        if (!shouldPersistSnapshot) {
-          return;
-        }
-        await this.mastra?.getStorage()?.persistWorkflowSnapshot({
-          workflowName: workflowId,
-          runId,
-          resourceId,
-          snapshot: {
-            runId,
-            status: workflowStatus,
-            value: executionContext.state,
-            context: stepResults,
-            activePaths: executionContext.executionPath,
-            activeStepsPath: executionContext.activeStepsPath,
-            suspendedPaths: executionContext.suspendedPaths,
-            resumeLabels: executionContext.resumeLabels,
-            waitingPaths: {},
-            serializedStepGraph,
-            result,
-            error,
-            timestamp: Date.now()
-          }
-        });
-      }
-    );
-  }
-  async executeConditional({
-    workflowId,
-    runId,
-    entry,
-    prevOutput,
-    stepResults,
-    timeTravel,
-    resume,
-    executionContext,
-    emitter,
-    abortController,
-    requestContext,
-    writableStream,
-    disableScorers,
-    tracingContext
-  }) {
-    const conditionalSpan = tracingContext?.currentSpan?.createChildSpan({
-      type: observability.SpanType.WORKFLOW_CONDITIONAL,
-      name: `conditional: '${entry.conditions.length} conditions'`,
-      input: prevOutput,
-      attributes: {
-        conditionCount: entry.conditions.length
-      },
-      tracingPolicy: this.options?.tracingPolicy
-    });
-    let execResults;
-    const truthyIndexes = (await Promise.all(
-      entry.conditions.map(
-        (cond, index) => this.inngestStep.run(`workflow.${workflowId}.conditional.${index}`, async () => {
-          const evalSpan = conditionalSpan?.createChildSpan({
-            type: observability.SpanType.WORKFLOW_CONDITIONAL_EVAL,
-            name: `condition: '${index}'`,
-            input: prevOutput,
-            attributes: {
-              conditionIndex: index
-            },
-            tracingPolicy: this.options?.tracingPolicy
-          });
-          try {
-            const result = await cond(
-              workflows.createDeprecationProxy(
-                {
-                  runId,
-                  workflowId,
-                  mastra: this.mastra,
-                  requestContext,
-                  retryCount: -1,
-                  inputData: prevOutput,
-                  state: executionContext.state,
-                  setState: (state) => {
-                    executionContext.state = state;
-                  },
-                  tracingContext: {
-                    currentSpan: evalSpan
-                  },
-                  getInitData: () => stepResults?.input,
-                  getStepResult: workflows.getStepResult.bind(this, stepResults),
-                  // TODO: this function shouldn't have suspend probably?
-                  suspend: async (_suspendPayload) => {
-                  },
-                  bail: () => {
-                  },
-                  abort: () => {
-                    abortController.abort();
-                  },
-                  [_constants.EMITTER_SYMBOL]: emitter,
-                  [_constants.STREAM_FORMAT_SYMBOL]: executionContext.format,
-                  engine: {
-                    step: this.inngestStep
-                  },
-                  abortSignal: abortController.signal,
-                  writer: new tools.ToolStream(
-                    {
-                      prefix: "workflow-step",
-                      callId: crypto.randomUUID(),
-                      name: "conditional",
-                      runId
-                    },
-                    writableStream
-                  )
-                },
-                {
-                  paramName: "runCount",
-                  deprecationMessage: workflows.runCountDeprecationMessage,
-                  logger: this.logger
-                }
-              )
-            );
-            evalSpan?.end({
-              output: result,
-              attributes: {
-                result: !!result
-              }
-            });
-            return result ? index : null;
-          } catch (e) {
-            evalSpan?.error({
-              error: e instanceof Error ? e : new Error(String(e)),
-              attributes: {
-                result: false
-              }
-            });
-            return null;
-          }
-        })
-      )
-    )).filter((index) => index !== null);
-    const stepsToRun = entry.steps.filter((_, index) => truthyIndexes.includes(index));
-    conditionalSpan?.update({
-      attributes: {
-        truthyIndexes,
-        selectedSteps: stepsToRun.map((s) => s.type === "step" ? s.step.id : `control-${s.type}`)
-      }
-    });
-    const results = await Promise.all(
-      stepsToRun.map(async (step, index) => {
-        const currStepResult = stepResults[step.step.id];
-        if (currStepResult && currStepResult.status === "success") {
-          return currStepResult;
-        }
-        const result = await this.executeStep({
-          step: step.step,
-          prevOutput,
-          stepResults,
-          resume,
-          timeTravel,
-          executionContext: {
-            workflowId,
-            runId,
-            executionPath: [...executionContext.executionPath, index],
-            activeStepsPath: executionContext.activeStepsPath,
-            suspendedPaths: executionContext.suspendedPaths,
-            resumeLabels: executionContext.resumeLabels,
-            retryConfig: executionContext.retryConfig,
-            state: executionContext.state
-          },
-          emitter,
-          abortController,
-          requestContext,
-          writableStream,
-          disableScorers,
-          tracingContext: {
-            currentSpan: conditionalSpan
-          }
-        });
-        stepResults[step.step.id] = result;
-        return result;
-      })
-    );
-    const hasFailed = results.find((result) => result.status === "failed");
-    const hasSuspended = results.find((result) => result.status === "suspended");
-    if (hasFailed) {
-      execResults = { status: "failed", error: hasFailed.error };
-    } else if (hasSuspended) {
-      execResults = {
-        status: "suspended",
-        suspendPayload: hasSuspended.suspendPayload,
-        ...hasSuspended.suspendOutput ? { suspendOutput: hasSuspended.suspendOutput } : {}
-      };
-    } else {
-      execResults = {
-        status: "success",
-        output: results.reduce((acc, result, index) => {
-          if (result.status === "success") {
-            if ("step" in stepsToRun[index]) {
-              acc[stepsToRun[index].step.id] = result.output;
-            }
-          }
-          return acc;
-        }, {})
-      };
-    }
-    if (execResults.status === "failed") {
-      conditionalSpan?.error({
-        error: new Error(execResults.error)
-      });
-    } else {
-      conditionalSpan?.end({
-        output: execResults.output || execResults
-      });
-    }
-    return execResults;
-  }
-};
 
 exports.InngestExecutionEngine = InngestExecutionEngine;
 exports.InngestRun = InngestRun;
 exports.InngestWorkflow = InngestWorkflow;
+exports._compatibilityCheck = _compatibilityCheck;
 exports.createStep = createStep;
 exports.init = init;
 exports.serve = serve;