@mastra/inngest 0.0.0-remove-unused-model-providers-api-20251030210744 → 0.0.0-remove-ai-peer-dep-from-evals-20260105220639

package/dist/index.js

@@ -1,1688 +1,1594 @@
+import { Agent } from '@mastra/core/agent';
+import { Tool } from '@mastra/core/tools';
+import { DefaultExecutionEngine, createTimeTravelExecutionParams, Run, hydrateSerializedStepErrors, Workflow } from '@mastra/core/workflows';
+import { PUBSUB_SYMBOL, STREAM_FORMAT_SYMBOL } from '@mastra/core/workflows/_constants';
+import { z } from 'zod';
 import { randomUUID } from 'crypto';
-import { ReadableStream } from 'stream/web';
-import { subscribe } from '@inngest/realtime';
-import { wrapMastra, AISpanType } from '@mastra/core/ai-tracing';
 import { RequestContext } from '@mastra/core/di';
+import { NonRetriableError } from 'inngest';
+import { getErrorFromUnknown } from '@mastra/core/error';
+import { subscribe } from '@inngest/realtime';
+import { PubSub } from '@mastra/core/events';
+import { ReadableStream } from 'stream/web';
 import { ChunkFrom, WorkflowRunOutput } from '@mastra/core/stream';
-import { ToolStream, Tool } from '@mastra/core/tools';
-import { Run, Workflow, DefaultExecutionEngine, createDeprecationProxy, getStepResult, runCountDeprecationMessage, validateStepInput } from '@mastra/core/workflows';
-import { EMITTER_SYMBOL, STREAM_FORMAT_SYMBOL } from '@mastra/core/workflows/_constants';
-import { NonRetriableError, RetryAfterError } from 'inngest';
 import { serve as serve$1 } from 'inngest/hono';
-import { z } from '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();
-        }
-        return [];
-      })
-    )
-  );
-  return serve$1({
-    ...registerOptions,
-    client: inngest,
-    functions: [...workflowFunctions, ...userFunctions]
-  });
-}
-var InngestRun = class extends Run {
-  inngest;
-  serializedStepGraph;
-  #mastra;
-  constructor(params, inngest) {
-    super(params);
-    this.inngest = inngest;
-    this.serializedStepGraph = params.serializedStepGraph;
-    this.#mastra = params.mastra;
+var InngestExecutionEngine = class extends DefaultExecutionEngine {
+  inngestStep;
+  inngestAttempts;
+  constructor(mastra, inngestStep, inngestAttempts = 0, options) {
+    super({ mastra, options });
+    this.inngestStep = inngestStep;
+    this.inngestAttempts = inngestAttempts;
   }
-  async getRuns(eventId) {
-    const response = await fetch(`${this.inngest.apiBaseUrl ?? "https://api.inngest.com"}/v1/events/${eventId}/runs`, {
-      headers: {
-        Authorization: `Bearer ${process.env.INNGEST_SIGNING_KEY}`
-      }
+  // =============================================================================
+  // Hook Overrides
+  // =============================================================================
+  /**
+   * Format errors while preserving Error instances and their custom properties.
+   * Uses getErrorFromUnknown to ensure all error properties are preserved.
+   */
+  formatResultError(error, lastOutput) {
+    const outputError = lastOutput?.error;
+    const errorSource = error || outputError;
+    const errorInstance = getErrorFromUnknown(errorSource, {
+      serializeStack: true,
+      // Include stack in JSON for better debugging in Inngest
+      fallbackMessage: "Unknown workflow error"
     });
-    const json = await response.json();
-    return json.data;
+    return errorInstance.toJSON();
   }
-  async getRunOutput(eventId) {
-    let runs = await this.getRuns(eventId);
-    while (runs?.[0]?.status !== "Completed" || runs?.[0]?.event_id !== eventId) {
-      await new Promise((resolve) => setTimeout(resolve, 1e3));
-      runs = await this.getRuns(eventId);
-      if (runs?.[0]?.status === "Failed") {
-        const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
-          workflowName: this.workflowId,
-          runId: this.runId
-        });
-        return {
-          output: { result: { steps: snapshot?.context, status: "failed", error: runs?.[0]?.output?.message } }
-        };
-      }
-      if (runs?.[0]?.status === "Cancelled") {
-        const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
-          workflowName: this.workflowId,
-          runId: this.runId
-        });
-        return { output: { result: { steps: snapshot?.context, status: "canceled" } } };
-      }
-    }
-    return runs?.[0];
+  /**
+   * Detect InngestWorkflow instances for special nested workflow handling
+   */
+  isNestedWorkflowStep(step) {
+    return step instanceof InngestWorkflow;
   }
-  async sendEvent(event, data) {
-    await this.inngest.send({
-      name: `user-event-${event}`,
-      data
-    });
+  /**
+   * 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;
   }
-  async cancel() {
-    await this.inngest.send({
-      name: `cancel.workflow.${this.workflowId}`,
-      data: {
-        runId: this.runId
+  /**
+   * 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) {
+    for (let i = 0; i < params.retries + 1; i++) {
+      if (i > 0 && params.delay) {
+        await new Promise((resolve) => setTimeout(resolve, params.delay));
       }
-    });
-    const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
-      workflowName: this.workflowId,
-      runId: this.runId
-    });
-    if (snapshot) {
-      await this.#mastra?.storage?.persistWorkflowSnapshot({
-        workflowName: this.workflowId,
-        runId: this.runId,
-        resourceId: this.resourceId,
-        snapshot: {
-          ...snapshot,
-          status: "canceled"
+      try {
+        const result = await this.wrapDurableOperation(stepId, runStep);
+        return { ok: true, result };
+      } catch (e) {
+        if (i === params.retries) {
+          const cause = e?.cause;
+          if (cause?.status === "failed") {
+            params.stepSpan?.error({
+              error: e,
+              attributes: { status: "failed" }
+            });
+            if (cause.error && !(cause.error instanceof Error)) {
+              cause.error = getErrorFromUnknown(cause.error, { serializeStack: false });
+            }
+            return { ok: false, error: cause };
+          }
+          const errorInstance = getErrorFromUnknown(e, {
+            serializeStack: false,
+            fallbackMessage: "Unknown step execution error"
+          });
+          params.stepSpan?.error({
+            error: errorInstance,
+            attributes: { status: "failed" }
+          });
+          return {
+            ok: false,
+            error: {
+              status: "failed",
+              error: errorInstance,
+              endedAt: Date.now()
+            }
+          };
         }
-      });
+      }
     }
+    return { ok: false, error: { status: "failed", error: new Error("Unknown error"), endedAt: Date.now() } };
   }
-  async start(params) {
-    return this._start(params);
+  /**
+   * Use Inngest's sleep primitive for durability
+   */
+  async executeSleepDuration(duration, sleepId, workflowId) {
+    await this.inngestStep.sleep(`workflow.${workflowId}.sleep.${sleepId}`, duration < 0 ? 0 : duration);
   }
-  async _start({
-    inputData,
-    initialState,
-    outputOptions,
-    tracingOptions,
-    format
-  }) {
-    await this.#mastra.getStorage()?.persistWorkflowSnapshot({
-      workflowName: this.workflowId,
-      runId: this.runId,
-      resourceId: this.resourceId,
-      snapshot: {
-        runId: this.runId,
-        serializedStepGraph: this.serializedStepGraph,
-        value: {},
-        context: {},
-        activePaths: [],
-        suspendedPaths: {},
-        resumeLabels: {},
-        waitingPaths: {},
-        timestamp: Date.now(),
-        status: "running"
-      }
-    });
-    const inputDataToUse = await this._validateInput(inputData);
-    const initialStateToUse = await this._validateInitialState(initialState ?? {});
-    const eventOutput = await this.inngest.send({
-      name: `workflow.${this.workflowId}`,
-      data: {
-        inputData: inputDataToUse,
-        initialState: initialStateToUse,
-        runId: this.runId,
-        resourceId: this.resourceId,
-        outputOptions,
-        tracingOptions,
-        format
-      }
-    });
-    const eventId = eventOutput.ids[0];
-    if (!eventId) {
-      throw new Error("Event ID is not set");
-    }
-    const runOutput = await this.getRunOutput(eventId);
-    const result = runOutput?.output?.result;
-    if (result.status === "failed") {
-      result.error = new Error(result.error);
-    }
-    if (result.status !== "suspended") {
-      this.cleanup?.();
-    }
-    return result;
+  /**
+   * Use Inngest's sleepUntil primitive for durability
+   */
+  async executeSleepUntilDate(date, sleepUntilId, workflowId) {
+    await this.inngestStep.sleepUntil(`workflow.${workflowId}.sleepUntil.${sleepUntilId}`, date);
   }
-  async resume(params) {
-    const p = this._resume(params).then((result) => {
-      if (result.status !== "suspended") {
-        this.closeStreamAction?.().catch(() => {
-        });
+  /**
+   * 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) {
+    return this.inngestStep.run(operationId, async () => {
+      try {
+        return await operationFn();
+      } catch (e) {
+        throw e;
       }
-      return result;
     });
-    this.executionResults = p;
-    return p;
   }
-  async _resume(params) {
-    const steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
-      (step) => typeof step === "string" ? step : step?.id
-    );
-    const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
-      workflowName: this.workflowId,
-      runId: this.runId
-    });
-    const suspendedStep = this.workflowSteps[steps?.[0] ?? ""];
-    const resumeDataToUse = await this._validateResumeData(params.resumeData, suspendedStep);
-    const eventOutput = await this.inngest.send({
-      name: `workflow.${this.workflowId}`,
-      data: {
-        inputData: resumeDataToUse,
-        initialState: snapshot?.value ?? {},
-        runId: this.runId,
-        workflowId: this.workflowId,
-        stepResults: snapshot?.context,
-        resume: {
-          steps,
-          stepResults: snapshot?.context,
-          resumePayload: resumeDataToUse,
-          // @ts-ignore
-          resumePath: snapshot?.suspendedPaths?.[steps?.[0]]
-        }
-      }
-    });
-    const eventId = eventOutput.ids[0];
-    if (!eventId) {
-      throw new Error("Event ID is not set");
-    }
-    const runOutput = await this.getRunOutput(eventId);
-    const result = runOutput?.output?.result;
-    if (result.status === "failed") {
-      result.error = new Error(result.error);
-    }
-    return result;
+  /**
+   * Provide Inngest step primitive in engine context
+   */
+  getEngineContext() {
+    return { step: this.inngestStep };
   }
-  watch(cb, type = "watch") {
-    let active = true;
-    const streamPromise = subscribe(
-      {
-        channel: `workflow:${this.workflowId}:${this.runId}`,
-        topics: [type],
-        app: this.inngest
-      },
-      (message) => {
-        if (active) {
-          cb(message.data);
-        }
-      }
-    );
-    return () => {
-      active = false;
-      streamPromise.then(async (stream) => {
-        return stream.cancel();
-      }).catch((err) => {
-        console.error(err);
-      });
-    };
+  /**
+   * For Inngest, lifecycle callbacks are invoked in the workflow's finalize step
+   * (wrapped in step.run for durability), not in execute(). Override to skip.
+   */
+  async invokeLifecycleCallbacks(_result) {
   }
-  streamLegacy({ inputData, requestContext } = {}) {
-    const { readable, writable } = new TransformStream();
-    const writer = writable.getWriter();
-    const unwatch = this.watch(async (event) => {
-      try {
-        await writer.write({
-          // @ts-ignore
-          type: "start",
-          // @ts-ignore
-          payload: { runId: this.runId }
+  /**
+   * Actually invoke the lifecycle callbacks. Called from workflow.ts finalize step.
+   */
+  async invokeLifecycleCallbacksInternal(result) {
+    return super.invokeLifecycleCallbacks(result);
+  }
+  /**
+   * 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,
+      pubsub,
+      startedAt,
+      perStep
+    } = 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 ?? randomUUID();
+        const workflowsStore = await this.mastra?.getStorage()?.getStore("workflows");
+        const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+          workflowName: step.id,
+          runId
         });
-        const e = {
-          ...event,
-          type: event.type.replace("workflow-", "")
+        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 },
+            perStep
+          }
+        });
+        result = invokeResp.result;
+        runId = invokeResp.runId;
+        executionContext.state = invokeResp.result.state;
+      } else if (isTimeTravel) {
+        const workflowsStoreForTimeTravel = await this.mastra?.getStorage()?.getStore("workflows");
+        const snapshot = await workflowsStoreForTimeTravel?.loadWorkflowSnapshot({
+          workflowName: step.id,
+          runId: executionContext.runId
+        }) ?? { context: {} };
+        const timeTravelParams = 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 },
+            perStep
+          }
+        });
+        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 },
+            perStep
+          }
+        });
+        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 || randomUUID();
+      } else {
+        runId = randomUUID();
+        result = {
+          status: "failed",
+          error: e instanceof Error ? e : new Error(String(e)),
+          steps: {},
+          input: inputData
         };
-        if (e.type === "step-output") {
-          e.type = e.payload.output.type;
-          e.payload = e.payload.output.payload;
-        }
-        await writer.write(e);
-      } catch {
       }
-    }, "watch-v2");
-    this.closeStreamAction = async () => {
-      await writer.write({
-        type: "finish",
-        // @ts-ignore
-        payload: { runId: this.runId }
-      });
-      unwatch();
-      try {
-        await writer.close();
-      } catch (err) {
-        console.error("Error closing stream:", err);
-      } finally {
-        writer.releaseLock();
-      }
-    };
-    this.executionResults = this._start({ inputData, requestContext, format: "legacy" }).then((result) => {
-      if (result.status !== "suspended") {
-        this.closeStreamAction?.().catch(() => {
-        });
-      }
-      return result;
-    });
-    return {
-      stream: readable,
-      getWorkflowState: () => this.executionResults
-    };
-  }
-  stream({
-    inputData,
-    requestContext,
-    tracingOptions,
-    closeOnSuspend = true,
-    initialState,
-    outputOptions
-  } = {}) {
-    if (this.closeStreamAction && this.streamOutput) {
-      return this.streamOutput;
     }
-    this.closeStreamAction = async () => {
-    };
-    const self = this;
-    const stream = new ReadableStream({
-      async start(controller) {
-        const unwatch = self.watch(async ({ type, from = ChunkFrom.WORKFLOW, payload }) => {
-          controller.enqueue({
-            type,
-            runId: self.runId,
-            from,
-            payload: {
-              stepName: payload?.id,
-              ...payload
+    const res = await this.inngestStep.run(
+      `workflow.${executionContext.workflowId}.step.${step.id}.nestedwf-results`,
+      async () => {
+        if (result.status === "failed") {
+          await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+            type: "watch",
+            runId: executionContext.runId,
+            data: {
+              type: "workflow-step-result",
+              payload: {
+                id: step.id,
+                status: "failed",
+                error: result?.error,
+                payload: prevOutput
+              }
             }
           });
-        }, "watch-v2");
-        self.closeStreamAction = async () => {
-          unwatch();
-          try {
-            await controller.close();
-          } catch (err) {
-            console.error("Error closing stream:", err);
-          }
-        };
-        const executionResultsPromise = self._start({
-          inputData,
-          requestContext,
-          // tracingContext, // We are not able to pass a reference to a span here, what to do?
-          initialState,
-          tracingOptions,
-          outputOptions,
-          format: "vnext"
-        });
-        let executionResults;
-        try {
-          executionResults = await executionResultsPromise;
-          if (closeOnSuspend) {
-            self.closeStreamAction?.().catch(() => {
-            });
-          } else if (executionResults.status !== "suspended") {
-            self.closeStreamAction?.().catch(() => {
+          return { executionContext, result: { status: "failed", error: result?.error, endedAt: Date.now() } };
+        } 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 pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+              type: "watch",
+              runId: executionContext.runId,
+              data: {
+                type: "workflow-step-suspended",
+                payload: {
+                  id: step.id,
+                  status: "suspended"
+                }
+              }
             });
+            return {
+              executionContext,
+              result: {
+                status: "suspended",
+                suspendedAt: Date.now(),
+                payload: stepResult.payload,
+                suspendPayload: {
+                  ...stepResult?.suspendPayload,
+                  __workflow_meta: { runId, path: suspendPath }
+                }
+              }
+            };
           }
-          if (self.streamOutput) {
-            self.streamOutput.updateResults(
-              executionResults
-            );
-          }
-        } catch (err) {
-          self.streamOutput?.rejectResults(err);
-          self.closeStreamAction?.().catch(() => {
+          return {
+            executionContext,
+            result: {
+              status: "suspended",
+              suspendedAt: Date.now(),
+              payload: {}
+            }
+          };
+        } else if (result.status === "tripwire") {
+          await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+            type: "watch",
+            runId: executionContext.runId,
+            data: {
+              type: "workflow-step-result",
+              payload: {
+                id: step.id,
+                status: "tripwire",
+                error: result?.tripwire?.reason,
+                payload: prevOutput
+              }
+            }
+          });
+          return {
+            executionContext,
+            result: {
+              status: "tripwire",
+              tripwire: result?.tripwire,
+              endedAt: Date.now()
+            }
+          };
+        } else if (perStep || result.status === "paused") {
+          await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+            type: "watch",
+            runId: executionContext.runId,
+            data: {
+              type: "workflow-step-result",
+              payload: {
+                id: step.id,
+                status: "paused"
+              }
+            }
           });
+          await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+            type: "watch",
+            runId: executionContext.runId,
+            data: {
+              type: "workflow-step-finish",
+              payload: {
+                id: step.id,
+                metadata: {}
+              }
+            }
+          });
+          return { executionContext, result: { status: "paused" } };
         }
+        await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+          type: "watch",
+          runId: executionContext.runId,
+          data: {
+            type: "workflow-step-result",
+            payload: {
+              id: step.id,
+              status: "success",
+              output: result?.result
+            }
+          }
+        });
+        await pubsub.publish(`workflow.events.v2.${executionContext.runId}`, {
+          type: "watch",
+          runId: executionContext.runId,
+          data: {
+            type: "workflow-step-finish",
+            payload: {
+              id: step.id,
+              metadata: {}
+            }
+          }
+        });
+        return { executionContext, result: { status: "success", output: result?.result, endedAt: Date.now() } };
       }
-    });
-    this.streamOutput = new WorkflowRunOutput({
-      runId: this.runId,
-      workflowId: this.workflowId,
-      stream
-    });
-    return this.streamOutput;
-  }
-  streamVNext(args = {}) {
-    return this.stream(args);
+    );
+    Object.assign(executionContext, res.executionContext);
+    return {
+      ...res.result,
+      startedAt,
+      payload: inputData,
+      resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
+      resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
+    };
   }
 };
-var InngestWorkflow = class _InngestWorkflow extends Workflow {
-  #mastra;
+var InngestPubSub = class extends PubSub {
   inngest;
-  function;
-  flowControlConfig;
-  constructor(params, inngest) {
-    const { concurrency, rateLimit, throttle, debounce, priority, ...workflowParams } = params;
-    super(workflowParams);
-    const flowControlEntries = Object.entries({ concurrency, rateLimit, throttle, debounce, priority }).filter(
-      ([_, value]) => value !== void 0
-    );
-    this.flowControlConfig = flowControlEntries.length > 0 ? Object.fromEntries(flowControlEntries) : void 0;
-    this.#mastra = params.mastra;
+  workflowId;
+  publishFn;
+  subscriptions = /* @__PURE__ */ new Map();
+  constructor(inngest, workflowId, publishFn) {
+    super();
     this.inngest = inngest;
+    this.workflowId = workflowId;
+    this.publishFn = publishFn;
   }
-  async listWorkflowRuns(args) {
-    const storage = this.#mastra?.getStorage();
-    if (!storage) {
-      this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
-      return { runs: [], total: 0 };
+  /**
+   * Publish an event to Inngest's realtime system.
+   *
+   * Topic format: "workflow.events.v2.{runId}"
+   * Maps to Inngest channel: "workflow:{workflowId}:{runId}"
+   */
+  async publish(topic, event) {
+    if (!this.publishFn) {
+      return;
     }
-    return storage.listWorkflowRuns({ workflowName: this.id, ...args ?? {} });
-  }
-  async getWorkflowRunById(runId) {
-    const storage = this.#mastra?.getStorage();
-    if (!storage) {
-      this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
-      return this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null;
+    const match = topic.match(/^workflow\.events\.v2\.(.+)$/);
+    if (!match) {
+      return;
     }
-    const run = await storage.getWorkflowRunById({ runId, workflowName: this.id });
-    return run ?? (this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null);
-  }
-  __registerMastra(mastra) {
-    this.#mastra = mastra;
-    this.executionEngine.__registerMastra(mastra);
-    const updateNested = (step) => {
-      if ((step.type === "step" || step.type === "loop" || step.type === "foreach") && step.step instanceof _InngestWorkflow) {
-        step.step.__registerMastra(mastra);
-      } else if (step.type === "parallel" || step.type === "conditional") {
-        for (const subStep of step.steps) {
-          updateNested(subStep);
-        }
-      }
-    };
-    if (this.executionGraph.steps.length) {
-      for (const step of this.executionGraph.steps) {
-        updateNested(step);
-      }
+    const runId = match[1];
+    try {
+      await this.publishFn({
+        channel: `workflow:${this.workflowId}:${runId}`,
+        topic: "watch",
+        data: event.data
+      });
+    } catch (err) {
+      console.error("InngestPubSub publish error:", err?.message ?? err);
     }
   }
   /**
-   * @deprecated Use createRunAsync() instead.
-   * @throws {Error} Always throws an error directing users to use createRunAsync()
+   * Subscribe to events from Inngest's realtime system.
+   *
+   * Topic format: "workflow.events.v2.{runId}"
+   * Maps to Inngest channel: "workflow:{workflowId}:{runId}"
    */
-  createRun(_options) {
-    throw new Error(
-      "createRun() has been deprecated. Please use createRunAsync() instead.\n\nMigration guide:\n  Before: const run = workflow.createRun();\n  After:  const run = await workflow.createRunAsync();\n\nNote: createRunAsync() is an async method, so make sure your calling function is async."
-    );
-  }
-  async createRunAsync(options) {
-    const runIdToUse = options?.runId || randomUUID();
-    const run = this.runs.get(runIdToUse) ?? new InngestRun(
+  async subscribe(topic, cb) {
+    const match = topic.match(/^workflow\.events\.v2\.(.+)$/);
+    if (!match || !match[1]) {
+      return;
+    }
+    const runId = match[1];
+    if (this.subscriptions.has(topic)) {
+      this.subscriptions.get(topic).callbacks.add(cb);
+      return;
+    }
+    const callbacks = /* @__PURE__ */ new Set([cb]);
+    const channel = `workflow:${this.workflowId}:${runId}`;
+    const streamPromise = subscribe(
       {
-        workflowId: this.id,
-        runId: runIdToUse,
-        resourceId: options?.resourceId,
-        executionEngine: this.executionEngine,
-        executionGraph: this.executionGraph,
-        serializedStepGraph: this.serializedStepGraph,
-        mastra: this.#mastra,
-        retryConfig: this.retryConfig,
-        cleanup: () => this.runs.delete(runIdToUse),
-        workflowSteps: this.steps
+        channel,
+        topics: ["watch"],
+        app: this.inngest
       },
-      this.inngest
+      (message) => {
+        const event = {
+          id: crypto.randomUUID(),
+          type: "watch",
+          runId,
+          data: message.data,
+          createdAt: /* @__PURE__ */ new Date()
+        };
+        for (const callback of callbacks) {
+          callback(event);
+        }
+      }
     );
-    this.runs.set(runIdToUse, run);
-    const shouldPersistSnapshot = this.options.shouldPersistSnapshot({
-      workflowStatus: run.workflowRunStatus,
-      stepResults: {}
+    this.subscriptions.set(topic, {
+      unsubscribe: () => {
+        streamPromise.then((stream) => stream.cancel()).catch((err) => {
+          console.error("InngestPubSub unsubscribe error:", err);
+        });
+      },
+      callbacks
     });
-    const workflowSnapshotInStorage = await this.getWorkflowRunExecutionResult(runIdToUse, false);
-    if (!workflowSnapshotInStorage && shouldPersistSnapshot) {
-      await this.mastra?.getStorage()?.persistWorkflowSnapshot({
-        workflowName: this.id,
-        runId: runIdToUse,
-        resourceId: options?.resourceId,
-        snapshot: {
-          runId: runIdToUse,
-          status: "pending",
-          value: {},
-          context: {},
-          activePaths: [],
-          waitingPaths: {},
-          serializedStepGraph: this.serializedStepGraph,
-          suspendedPaths: {},
-          resumeLabels: {},
-          result: void 0,
-          error: void 0,
-          // @ts-ignore
-          timestamp: Date.now()
+  }
+  /**
+   * Unsubscribe a callback from a topic.
+   * If no callbacks remain, the underlying Inngest subscription is cancelled.
+   */
+  async unsubscribe(topic, cb) {
+    const sub = this.subscriptions.get(topic);
+    if (!sub) {
+      return;
+    }
+    sub.callbacks.delete(cb);
+    if (sub.callbacks.size === 0) {
+      sub.unsubscribe();
+      this.subscriptions.delete(topic);
+    }
+  }
+  /**
+   * Flush any pending operations. No-op for Inngest.
+   */
+  async flush() {
+  }
+  /**
+   * Clean up all subscriptions during graceful shutdown.
+   */
+  async close() {
+    for (const [, sub] of this.subscriptions) {
+      sub.unsubscribe();
+    }
+    this.subscriptions.clear();
+  }
+};
+var InngestRun = class extends Run {
+  inngest;
+  serializedStepGraph;
+  #mastra;
+  constructor(params, inngest) {
+    super(params);
+    this.inngest = inngest;
+    this.serializedStepGraph = params.serializedStepGraph;
+    this.#mastra = params.mastra;
+  }
+  async getRuns(eventId) {
+    const maxRetries = 3;
+    let lastError = null;
+    for (let attempt = 0; attempt < maxRetries; attempt++) {
+      try {
+        const response = await fetch(
+          `${this.inngest.apiBaseUrl ?? "https://api.inngest.com"}/v1/events/${eventId}/runs`,
+          {
+            headers: {
+              Authorization: `Bearer ${process.env.INNGEST_SIGNING_KEY}`
+            }
+          }
+        );
+        if (response.status === 429) {
+          const retryAfter = parseInt(response.headers.get("retry-after") || "2", 10);
+          await new Promise((resolve) => setTimeout(resolve, retryAfter * 1e3));
+          continue;
+        }
+        if (!response.ok) {
+          throw new Error(`Inngest API error: ${response.status} ${response.statusText}`);
+        }
+        const text = await response.text();
+        if (!text) {
+          await new Promise((resolve) => setTimeout(resolve, 1e3 * (attempt + 1)));
+          continue;
+        }
+        const json = JSON.parse(text);
+        return json.data;
+      } catch (error) {
+        lastError = error;
+        if (attempt < maxRetries - 1) {
+          await new Promise((resolve) => setTimeout(resolve, 1e3 * Math.pow(2, attempt)));
+        }
+      }
+    }
+    throw new NonRetriableError(`Failed to get runs after ${maxRetries} attempts: ${lastError?.message}`);
+  }
+  async getRunOutput(eventId, maxWaitMs = 3e5) {
+    const startTime = Date.now();
+    const storage = this.#mastra?.getStorage();
+    const workflowsStore = await storage?.getStore("workflows");
+    while (Date.now() - startTime < maxWaitMs) {
+      let runs;
+      try {
+        runs = await this.getRuns(eventId);
+      } catch (error) {
+        if (error instanceof NonRetriableError) {
+          throw error;
+        }
+        throw new NonRetriableError(
+          `Failed to poll workflow status: ${error instanceof Error ? error.message : String(error)}`
+        );
+      }
+      if (runs?.[0]?.status === "Completed" && runs?.[0]?.event_id === eventId) {
+        return runs[0];
+      }
+      if (runs?.[0]?.status === "Failed") {
+        const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+          workflowName: this.workflowId,
+          runId: this.runId
+        });
+        if (snapshot?.context) {
+          snapshot.context = hydrateSerializedStepErrors(snapshot.context);
+        }
+        return {
+          output: {
+            result: {
+              steps: snapshot?.context,
+              status: "failed",
+              // Get the original error from NonRetriableError's cause (which contains the workflow result)
+              error: getErrorFromUnknown(runs?.[0]?.output?.cause?.error, { serializeStack: false })
+            }
+          }
+        };
+      }
+      if (runs?.[0]?.status === "Cancelled") {
+        const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+          workflowName: this.workflowId,
+          runId: this.runId
+        });
+        return { output: { result: { steps: snapshot?.context, status: "canceled" } } };
+      }
+      await new Promise((resolve) => setTimeout(resolve, 1e3 + Math.random() * 1e3));
+    }
+    throw new NonRetriableError(`Workflow did not complete within ${maxWaitMs}ms`);
+  }
+  async cancel() {
+    const storage = this.#mastra?.getStorage();
+    await this.inngest.send({
+      name: `cancel.workflow.${this.workflowId}`,
+      data: {
+        runId: this.runId
+      }
+    });
+    const workflowsStore = await storage?.getStore("workflows");
+    const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+      workflowName: this.workflowId,
+      runId: this.runId
+    });
+    if (snapshot) {
+      await workflowsStore?.persistWorkflowSnapshot({
+        workflowName: this.workflowId,
+        runId: this.runId,
+        resourceId: this.resourceId,
+        snapshot: {
+          ...snapshot,
+          status: "canceled",
+          value: snapshot.value
         }
       });
     }
-    return run;
   }
-  getFunction() {
-    if (this.function) {
-      return this.function;
+  async start(params) {
+    return this._start(params);
+  }
+  /**
+   * Starts the workflow execution without waiting for completion (fire-and-forget).
+   * Returns immediately with the runId after sending the event to Inngest.
+   * The workflow executes independently in Inngest.
+   * Use this when you don't need to wait for the result or want to avoid polling failures.
+   */
+  async startAsync(params) {
+    const workflowsStore = await this.#mastra.getStorage()?.getStore("workflows");
+    await workflowsStore?.persistWorkflowSnapshot({
+      workflowName: this.workflowId,
+      runId: this.runId,
+      resourceId: this.resourceId,
+      snapshot: {
+        runId: this.runId,
+        serializedStepGraph: this.serializedStepGraph,
+        status: "running",
+        value: {},
+        context: {},
+        activePaths: [],
+        suspendedPaths: {},
+        activeStepsPath: {},
+        resumeLabels: {},
+        waitingPaths: {},
+        timestamp: Date.now()
+      }
+    });
+    const inputDataToUse = await this._validateInput(params.inputData);
+    const initialStateToUse = await this._validateInitialState(params.initialState ?? {});
+    const eventOutput = await this.inngest.send({
+      name: `workflow.${this.workflowId}`,
+      data: {
+        inputData: inputDataToUse,
+        initialState: initialStateToUse,
+        runId: this.runId,
+        resourceId: this.resourceId,
+        outputOptions: params.outputOptions,
+        tracingOptions: params.tracingOptions,
+        requestContext: params.requestContext ? Object.fromEntries(params.requestContext.entries()) : {},
+        perStep: params.perStep
+      }
+    });
+    const eventId = eventOutput.ids[0];
+    if (!eventId) {
+      throw new Error("Event ID is not set");
     }
-    this.function = this.inngest.createFunction(
+    return { runId: this.runId };
+  }
+  async _start({
+    inputData,
+    initialState,
+    outputOptions,
+    tracingOptions,
+    format,
+    requestContext,
+    perStep
+  }) {
+    const workflowsStore = await this.#mastra.getStorage()?.getStore("workflows");
+    await workflowsStore?.persistWorkflowSnapshot({
+      workflowName: this.workflowId,
+      runId: this.runId,
+      resourceId: this.resourceId,
+      snapshot: {
+        runId: this.runId,
+        serializedStepGraph: this.serializedStepGraph,
+        status: "running",
+        value: {},
+        context: {},
+        activePaths: [],
+        suspendedPaths: {},
+        activeStepsPath: {},
+        resumeLabels: {},
+        waitingPaths: {},
+        timestamp: Date.now()
+      }
+    });
+    const inputDataToUse = await this._validateInput(inputData);
+    const initialStateToUse = await this._validateInitialState(initialState ?? {});
+    const eventOutput = await this.inngest.send({
+      name: `workflow.${this.workflowId}`,
+      data: {
+        inputData: inputDataToUse,
+        initialState: initialStateToUse,
+        runId: this.runId,
+        resourceId: this.resourceId,
+        outputOptions,
+        tracingOptions,
+        format,
+        requestContext: requestContext ? Object.fromEntries(requestContext.entries()) : {},
+        perStep
+      }
+    });
+    const eventId = eventOutput.ids[0];
+    if (!eventId) {
+      throw new Error("Event ID is not set");
+    }
+    const runOutput = await this.getRunOutput(eventId);
+    const result = runOutput?.output?.result;
+    this.hydrateFailedResult(result);
+    if (result.status !== "suspended") {
+      this.cleanup?.();
+    }
+    return result;
+  }
+  async resume(params) {
+    const p = this._resume(params).then((result) => {
+      if (result.status !== "suspended") {
+        this.closeStreamAction?.().catch(() => {
+        });
+      }
+      return result;
+    });
+    this.executionResults = p;
+    return p;
+  }
+  async _resume(params) {
+    const storage = this.#mastra?.getStorage();
+    let steps = [];
+    if (typeof params.step === "string") {
+      steps = params.step.split(".");
+    } else {
+      steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
+        (step) => typeof step === "string" ? step : step?.id
+      );
+    }
+    const workflowsStore = await storage?.getStore("workflows");
+    const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+      workflowName: this.workflowId,
+      runId: this.runId
+    });
+    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: {
+        inputData: resumeDataToUse,
+        initialState: snapshot?.value ?? {},
+        runId: this.runId,
+        workflowId: this.workflowId,
+        stepResults: snapshot?.context,
+        resume: {
+          steps,
+          stepResults: snapshot?.context,
+          resumePayload: resumeDataToUse,
+          resumePath: steps?.[0] ? snapshot?.suspendedPaths?.[steps?.[0]] : void 0
+        },
+        requestContext: mergedRequestContext,
+        perStep: params.perStep
+      }
+    });
+    const eventId = eventOutput.ids[0];
+    if (!eventId) {
+      throw new Error("Event ID is not set");
+    }
+    const runOutput = await this.getRunOutput(eventId);
+    const result = runOutput?.output?.result;
+    this.hydrateFailedResult(result);
+    return result;
+  }
+  async timeTravel(params) {
+    const p = this._timeTravel(params).then((result) => {
+      if (result.status !== "suspended") {
+        this.closeStreamAction?.().catch(() => {
+        });
+      }
+      return result;
+    });
+    this.executionResults = p;
+    return p;
+  }
+  async _timeTravel(params) {
+    if (!params.step || Array.isArray(params.step) && params.step?.length === 0) {
+      throw new Error("Step is required and must be a valid step or array of steps");
+    }
+    let steps = [];
+    if (typeof params.step === "string") {
+      steps = params.step.split(".");
+    } else {
+      steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
+        (step) => typeof step === "string" ? step : step?.id
+      );
+    }
+    if (steps.length === 0) {
+      throw new Error("No steps provided to timeTravel");
+    }
+    const storage = this.#mastra?.getStorage();
+    const workflowsStore = await storage?.getStore("workflows");
+    const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+      workflowName: this.workflowId,
+      runId: this.runId
+    });
+    if (!snapshot) {
+      await workflowsStore?.persistWorkflowSnapshot({
+        workflowName: this.workflowId,
+        runId: this.runId,
+        resourceId: this.resourceId,
+        snapshot: {
+          runId: this.runId,
+          serializedStepGraph: this.serializedStepGraph,
+          status: "pending",
+          value: {},
+          context: {},
+          activePaths: [],
+          suspendedPaths: {},
+          activeStepsPath: {},
+          resumeLabels: {},
+          waitingPaths: {},
+          timestamp: Date.now()
+        }
+      });
+    }
+    if (snapshot?.status === "running") {
+      throw new Error("This workflow run is still running, cannot time travel");
+    }
+    let inputDataToUse = params.inputData;
+    if (inputDataToUse && steps.length === 1) {
+      inputDataToUse = await this._validateTimetravelInputData(params.inputData, this.workflowSteps[steps[0]]);
+    }
+    const timeTravelData = createTimeTravelExecutionParams({
+      steps,
+      inputData: inputDataToUse,
+      resumeData: params.resumeData,
+      context: params.context,
+      nestedStepsContext: params.nestedStepsContext,
+      snapshot: snapshot ?? { context: {} },
+      graph: this.executionGraph,
+      initialState: params.initialState,
+      perStep: params.perStep
+    });
+    const eventOutput = await this.inngest.send({
+      name: `workflow.${this.workflowId}`,
+      data: {
+        initialState: timeTravelData.state,
+        runId: this.runId,
+        workflowId: this.workflowId,
+        stepResults: timeTravelData.stepResults,
+        timeTravel: timeTravelData,
+        tracingOptions: params.tracingOptions,
+        outputOptions: params.outputOptions,
+        requestContext: params.requestContext ? Object.fromEntries(params.requestContext.entries()) : {},
+        perStep: params.perStep
+      }
+    });
+    const eventId = eventOutput.ids[0];
+    if (!eventId) {
+      throw new Error("Event ID is not set");
+    }
+    const runOutput = await this.getRunOutput(eventId);
+    const result = runOutput?.output?.result;
+    this.hydrateFailedResult(result);
+    return result;
+  }
+  watch(cb) {
+    let active = true;
+    const streamPromise = subscribe(
       {
-        id: `workflow.${this.id}`,
-        // @ts-ignore
-        retries: this.retryConfig?.attempts ?? 0,
-        cancelOn: [{ event: `cancel.workflow.${this.id}` }],
-        // Spread flow control configuration
-        ...this.flowControlConfig
+        channel: `workflow:${this.workflowId}:${this.runId}`,
+        topics: ["watch"],
+        app: this.inngest
       },
-      { event: `workflow.${this.id}` },
-      async ({ event, step, attempt, publish }) => {
-        let { inputData, initialState, runId, resourceId, resume, outputOptions, format } = event.data;
-        if (!runId) {
-          runId = await step.run(`workflow.${this.id}.runIdGen`, async () => {
-            return randomUUID();
-          });
+      (message) => {
+        if (active) {
+          cb(message.data);
         }
-        const emitter = {
-          emit: async (event2, data) => {
-            if (!publish) {
-              return;
-            }
-            try {
-              await publish({
-                channel: `workflow:${this.id}:${runId}`,
-                topic: event2,
-                data
-              });
-            } catch (err) {
-              this.logger.error("Error emitting event: " + (err?.stack ?? err?.message ?? err));
+      }
+    );
+    return () => {
+      active = false;
+      streamPromise.then(async (stream) => {
+        return stream.cancel();
+      }).catch((err) => {
+        console.error(err);
+      });
+    };
+  }
+  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 {
+        const e = {
+          ...event,
+          type: event.type.replace("workflow-", "")
+        };
+        if (e.type === "step-output") {
+          e.type = e.payload.output.type;
+          e.payload = e.payload.output.payload;
+        }
+        await writer.write(e);
+      } catch {
+      }
+    });
+    this.closeStreamAction = async () => {
+      await writer.write({
+        type: "finish",
+        // @ts-ignore
+        payload: { runId: this.runId }
+      });
+      unwatch();
+      try {
+        await writer.close();
+      } catch (err) {
+        console.error("Error closing stream:", err);
+      } finally {
+        writer.releaseLock();
+      }
+    };
+    this.executionResults = this._start({ inputData, requestContext, format: "legacy" }).then((result) => {
+      if (result.status !== "suspended") {
+        this.closeStreamAction?.().catch(() => {
+        });
+      }
+      return result;
+    });
+    return {
+      stream: readable,
+      getWorkflowState: () => this.executionResults
+    };
+  }
+  stream({
+    inputData,
+    requestContext,
+    tracingOptions,
+    closeOnSuspend = true,
+    initialState,
+    outputOptions,
+    perStep
+  } = {}) {
+    if (this.closeStreamAction && this.streamOutput) {
+      return this.streamOutput;
+    }
+    this.closeStreamAction = async () => {
+    };
+    const self = this;
+    const stream = new ReadableStream({
+      async start(controller) {
+        const unwatch = self.watch(async ({ type, from = ChunkFrom.WORKFLOW, payload }) => {
+          controller.enqueue({
+            type,
+            runId: self.runId,
+            from,
+            payload: {
+              stepName: payload?.id,
+              ...payload
             }
-          },
-          on: (_event, _callback) => {
-          },
-          off: (_event, _callback) => {
-          },
-          once: (_event, _callback) => {
+          });
+        });
+        self.closeStreamAction = async () => {
+          unwatch();
+          try {
+            await controller.close();
+          } catch (err) {
+            console.error("Error closing stream:", err);
           }
         };
-        const engine = new InngestExecutionEngine(this.#mastra, step, attempt, this.options);
-        const result = await engine.execute({
-          workflowId: this.id,
-          runId,
-          resourceId,
-          graph: this.executionGraph,
-          serializedStepGraph: this.serializedStepGraph,
-          input: inputData,
-          initialState,
-          emitter,
-          retryConfig: this.retryConfig,
-          requestContext: new RequestContext(),
-          // TODO
-          resume,
-          format,
-          abortController: new AbortController(),
-          // currentSpan: undefined, // TODO: Pass actual parent AI span from workflow execution context
+        const executionResultsPromise = self._start({
+          inputData,
+          requestContext,
+          // tracingContext, // We are not able to pass a reference to a span here, what to do?
+          initialState,
+          tracingOptions,
           outputOptions,
-          writableStream: new WritableStream({
-            write(chunk) {
-              void emitter.emit("watch-v2", chunk).catch(() => {
-              });
-            }
-          })
+          format: "vnext",
+          perStep
         });
-        await step.run(`workflow.${this.id}.finalize`, async () => {
-          if (result.status === "failed") {
-            throw new NonRetriableError(`Workflow failed`, {
-              cause: result
+        let executionResults;
+        try {
+          executionResults = await executionResultsPromise;
+          if (closeOnSuspend) {
+            self.closeStreamAction?.().catch(() => {
+            });
+          } else if (executionResults.status !== "suspended") {
+            self.closeStreamAction?.().catch(() => {
             });
           }
-          return result;
-        });
-        return { result, runId };
-      }
-    );
-    return this.function;
-  }
-  getNestedFunctions(steps) {
-    return steps.flatMap((step) => {
-      if (step.type === "step" || step.type === "loop" || step.type === "foreach") {
-        if (step.step instanceof _InngestWorkflow) {
-          return [step.step.getFunction(), ...step.step.getNestedFunctions(step.step.executionGraph.steps)];
+          if (self.streamOutput) {
+            self.streamOutput.updateResults(
+              executionResults
+            );
+          }
+        } catch (err) {
+          self.streamOutput?.rejectResults(err);
+          self.closeStreamAction?.().catch(() => {
+          });
         }
-        return [];
-      } else if (step.type === "parallel" || step.type === "conditional") {
-        return this.getNestedFunctions(step.steps);
       }
-      return [];
     });
+    this.streamOutput = new WorkflowRunOutput({
+      runId: this.runId,
+      workflowId: this.workflowId,
+      stream
+    });
+    return this.streamOutput;
   }
-  getFunctions() {
-    return [this.getFunction(), ...this.getNestedFunctions(this.executionGraph.steps)];
-  }
-};
-function isAgent(params) {
-  return params?.component === "AGENT";
-}
-function isTool(params) {
-  return params instanceof Tool;
-}
-function createStep(params, agentOptions) {
-  if (isAgent(params)) {
-    return {
-      id: params.name,
-      description: params.getDescription(),
-      // @ts-ignore
-      inputSchema: z.object({
-        prompt: z.string()
-        // resourceId: z.string().optional(),
-        // threadId: z.string().optional(),
-      }),
-      // @ts-ignore
-      outputSchema: z.object({
-        text: z.string()
-      }),
-      execute: async ({
-        inputData,
-        [EMITTER_SYMBOL]: emitter,
-        [STREAM_FORMAT_SYMBOL]: streamFormat,
-        requestContext,
-        tracingContext,
-        abortSignal,
-        abort,
-        writer
-      }) => {
-        let streamPromise = {};
-        streamPromise.promise = new Promise((resolve, reject) => {
-          streamPromise.resolve = resolve;
-          streamPromise.reject = reject;
+  timeTravelStream({
+    inputData,
+    resumeData,
+    initialState,
+    step,
+    context,
+    nestedStepsContext,
+    requestContext,
+    tracingOptions,
+    outputOptions,
+    perStep
+  }) {
+    this.closeStreamAction = async () => {
+    };
+    const self = this;
+    const stream = new ReadableStream({
+      async start(controller) {
+        const unwatch = self.watch(async ({ type, from = ChunkFrom.WORKFLOW, payload }) => {
+          controller.enqueue({
+            type,
+            runId: self.runId,
+            from,
+            payload: {
+              stepName: payload?.id,
+              ...payload
+            }
+          });
         });
-        const toolData = {
-          name: params.name,
-          args: inputData
+        self.closeStreamAction = async () => {
+          unwatch();
+          try {
+            controller.close();
+          } catch (err) {
+            console.error("Error closing stream:", err);
+          }
         };
-        let stream;
-        if ((await params.getModel()).specificationVersion === "v1") {
-          const { fullStream } = await params.streamLegacy(inputData.prompt, {
-            ...agentOptions ?? {},
-            // resourceId: inputData.resourceId,
-            // threadId: inputData.threadId,
-            requestContext,
-            tracingContext,
-            onFinish: (result) => {
-              streamPromise.resolve(result.text);
-              void agentOptions?.onFinish?.(result);
-            },
-            abortSignal
-          });
-          stream = fullStream;
-        } else {
-          const modelOutput = await params.stream(inputData.prompt, {
-            ...agentOptions ?? {},
-            requestContext,
-            tracingContext,
-            onFinish: (result) => {
-              streamPromise.resolve(result.text);
-              void agentOptions?.onFinish?.(result);
-            },
-            abortSignal
-          });
-          stream = modelOutput.fullStream;
-        }
-        if (streamFormat === "legacy") {
-          await emitter.emit("watch-v2", {
-            type: "tool-call-streaming-start",
-            ...toolData ?? {}
+        const executionResultsPromise = self._timeTravel({
+          inputData,
+          step,
+          context,
+          nestedStepsContext,
+          resumeData,
+          initialState,
+          requestContext,
+          tracingOptions,
+          outputOptions,
+          perStep
+        });
+        self.executionResults = executionResultsPromise;
+        let executionResults;
+        try {
+          executionResults = await executionResultsPromise;
+          self.closeStreamAction?.().catch(() => {
           });
-          for await (const chunk of stream) {
-            if (chunk.type === "text-delta") {
-              await emitter.emit("watch-v2", {
-                type: "tool-call-delta",
-                ...toolData ?? {},
-                argsTextDelta: chunk.textDelta
-              });
-            }
+          if (self.streamOutput) {
+            self.streamOutput.updateResults(executionResults);
           }
-          await emitter.emit("watch-v2", {
-            type: "tool-call-streaming-finish",
-            ...toolData ?? {}
+        } catch (err) {
+          self.streamOutput?.rejectResults(err);
+          self.closeStreamAction?.().catch(() => {
           });
-        } else {
-          for await (const chunk of stream) {
-            await writer.write(chunk);
-          }
-        }
-        if (abortSignal.aborted) {
-          return abort();
         }
-        return {
-          text: await streamPromise.promise
-        };
-      },
-      component: params.component
-    };
+      }
+    });
+    this.streamOutput = new WorkflowRunOutput({
+      runId: this.runId,
+      workflowId: this.workflowId,
+      stream
+    });
+    return this.streamOutput;
   }
-  if (isTool(params)) {
-    if (!params.inputSchema || !params.outputSchema) {
-      throw new Error("Tool must have input and output schemas defined");
+  /**
+   * Hydrates errors in a failed workflow result back to proper Error instances.
+   * This ensures error.cause chains and custom properties are preserved.
+   */
+  hydrateFailedResult(result) {
+    if (result.status === "failed") {
+      result.error = getErrorFromUnknown(result.error, { serializeStack: false });
+      if (result.steps) {
+        hydrateSerializedStepErrors(result.steps);
+      }
     }
-    return {
-      // TODO: tool probably should have strong id type
-      // @ts-ignore
-      id: params.id,
-      description: params.description,
-      inputSchema: params.inputSchema,
-      outputSchema: params.outputSchema,
-      execute: async ({ inputData, mastra, requestContext, tracingContext, suspend, resumeData }) => {
-        return params.execute({
-          context: inputData,
-          mastra: wrapMastra(mastra, tracingContext),
-          requestContext,
-          tracingContext,
-          suspend,
-          resumeData
-        });
-      },
-      component: "TOOL"
-    };
   }
-  return {
-    id: params.id,
-    description: params.description,
-    inputSchema: params.inputSchema,
-    outputSchema: params.outputSchema,
-    resumeSchema: params.resumeSchema,
-    suspendSchema: params.suspendSchema,
-    execute: params.execute
-  };
-}
-function init(inngest) {
-  return {
-    createWorkflow(params) {
-      return new InngestWorkflow(
-        params,
-        inngest
-      );
-    },
-    createStep,
-    cloneStep(step, opts) {
-      return {
-        id: opts.id,
-        description: step.description,
-        inputSchema: step.inputSchema,
-        outputSchema: step.outputSchema,
-        resumeSchema: step.resumeSchema,
-        suspendSchema: step.suspendSchema,
-        stateSchema: step.stateSchema,
-        execute: step.execute,
-        component: step.component
-      };
-    },
-    cloneWorkflow(workflow, opts) {
-      const wf = new Workflow({
-        id: opts.id,
-        inputSchema: workflow.inputSchema,
-        outputSchema: workflow.outputSchema,
-        steps: workflow.stepDefs,
-        mastra: workflow.mastra
-      });
-      wf.setStepFlow(workflow.stepGraph);
-      wf.commit();
-      return wf;
+};
+
+// src/workflow.ts
+var InngestWorkflow = class _InngestWorkflow extends Workflow {
+  #mastra;
+  inngest;
+  function;
+  cronFunction;
+  flowControlConfig;
+  cronConfig;
+  constructor(params, inngest) {
+    const { concurrency, rateLimit, throttle, debounce, priority, cron, inputData, initialState, ...workflowParams } = params;
+    super(workflowParams);
+    this.engineType = "inngest";
+    const flowControlEntries = Object.entries({ concurrency, rateLimit, throttle, debounce, priority }).filter(
+      ([_, value]) => value !== void 0
+    );
+    this.flowControlConfig = flowControlEntries.length > 0 ? Object.fromEntries(flowControlEntries) : void 0;
+    this.#mastra = params.mastra;
+    this.inngest = inngest;
+    if (cron) {
+      this.cronConfig = { cron, inputData, initialState };
+    }
+  }
+  async listWorkflowRuns(args) {
+    const storage = this.#mastra?.getStorage();
+    if (!storage) {
+      this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
+      return { runs: [], total: 0 };
     }
-  };
-}
-var InngestExecutionEngine = class extends DefaultExecutionEngine {
-  inngestStep;
-  inngestAttempts;
-  constructor(mastra, inngestStep, inngestAttempts = 0, options) {
-    super({ mastra, options });
-    this.inngestStep = inngestStep;
-    this.inngestAttempts = inngestAttempts;
+    const workflowsStore = await storage.getStore("workflows");
+    if (!workflowsStore) {
+      return { runs: [], total: 0 };
+    }
+    return workflowsStore.listWorkflowRuns({ workflowName: this.id, ...args ?? {} });
   }
-  async fmtReturnValue(emitter, stepResults, lastOutput, error) {
-    const base = {
-      status: lastOutput.status,
-      steps: stepResults
-    };
-    if (lastOutput.status === "success") {
-      await emitter.emit("watch", {
-        type: "watch",
-        payload: {
-          workflowState: {
-            status: lastOutput.status,
-            steps: stepResults,
-            result: lastOutput.output
-          }
-        },
-        eventTimestamp: Date.now()
-      });
-      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";
-      await emitter.emit("watch", {
-        type: "watch",
-        payload: {
-          workflowState: {
-            status: lastOutput.status,
-            steps: stepResults,
-            result: null,
-            error: base.error
-          }
-        },
-        eventTimestamp: Date.now()
-      });
-    } else if (lastOutput.status === "suspended") {
-      await emitter.emit("watch", {
-        type: "watch",
-        payload: {
-          workflowState: {
-            status: lastOutput.status,
-            steps: stepResults,
-            result: null,
-            error: null
-          }
-        },
-        eventTimestamp: Date.now()
-      });
-      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;
+  async getWorkflowRunById(runId) {
+    const storage = this.#mastra?.getStorage();
+    if (!storage) {
+      this.logger.debug("Cannot get workflow runs. Mastra engine is not initialized");
+      return this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null;
+    }
+    const workflowsStore = await storage.getStore("workflows");
+    if (!workflowsStore) {
+      return this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null;
     }
-    return base;
+    const run = await workflowsStore.getWorkflowRunById({ runId, workflowName: this.id });
+    return run ?? (this.runs.get(runId) ? { ...this.runs.get(runId), workflowName: this.id } : null);
   }
-  // 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: AISpanType.WORKFLOW_SLEEP,
-      name: `sleep: ${duration ? `${duration}ms` : "dynamic"}`,
-      attributes: {
-        durationMs: duration,
-        sleepType: fn ? "dynamic" : "fixed"
-      },
-      tracingPolicy: this.options?.tracingPolicy
-    });
-    if (fn) {
-      const stepCallId = randomUUID();
-      duration = await this.inngestStep.run(`workflow.${workflowId}.sleep.${entry.id}`, async () => {
-        return await fn(
-          createDeprecationProxy(
-            {
-              runId,
-              workflowId,
-              mastra: this.mastra,
-              requestContext,
-              inputData: prevOutput,
-              state: executionContext.state,
-              setState: (state) => {
-                executionContext.state = state;
-              },
-              runCount: -1,
-              retryCount: -1,
-              tracingContext: {
-                currentSpan: sleepSpan
-              },
-              getInitData: () => stepResults?.input,
-              getStepResult: getStepResult.bind(this, stepResults),
-              // TODO: this function shouldn't have suspend probably?
-              suspend: async (_suspendPayload) => {
-              },
-              bail: () => {
-              },
-              abort: () => {
-                abortController?.abort();
-              },
-              [EMITTER_SYMBOL]: emitter,
-              [STREAM_FORMAT_SYMBOL]: executionContext.format,
-              engine: { step: this.inngestStep },
-              abortSignal: abortController?.signal,
-              writer: new ToolStream(
-                {
-                  prefix: "workflow-step",
-                  callId: stepCallId,
-                  name: "sleep",
-                  runId
-                },
-                writableStream
-              )
-            },
-            {
-              paramName: "runCount",
-              deprecationMessage: runCountDeprecationMessage,
-              logger: this.logger
-            }
-          )
-        );
-      });
-      sleepSpan?.update({
-        attributes: {
-          durationMs: duration
+  __registerMastra(mastra) {
+    super.__registerMastra(mastra);
+    this.#mastra = mastra;
+    this.executionEngine.__registerMastra(mastra);
+    const updateNested = (step) => {
+      if ((step.type === "step" || step.type === "loop" || step.type === "foreach") && step.step instanceof _InngestWorkflow) {
+        step.step.__registerMastra(mastra);
+      } else if (step.type === "parallel" || step.type === "conditional") {
+        for (const subStep of step.steps) {
+          updateNested(subStep);
         }
-      });
-    }
-    try {
-      await this.inngestStep.sleep(entry.id, !duration || duration < 0 ? 0 : duration);
-      sleepSpan?.end();
-    } catch (e) {
-      sleepSpan?.error({ error: e });
-      throw e;
+      }
+    };
+    if (this.executionGraph.steps.length) {
+      for (const step of this.executionGraph.steps) {
+        updateNested(step);
+      }
     }
   }
-  async executeSleepUntil({
-    workflowId,
-    runId,
-    entry,
-    prevOutput,
-    stepResults,
-    emitter,
-    abortController,
-    requestContext,
-    executionContext,
-    writableStream,
-    tracingContext
-  }) {
-    let { date, fn } = entry;
-    const sleepUntilSpan = tracingContext?.currentSpan?.createChildSpan({
-      type: AISpanType.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"
+  async createRun(options) {
+    const runIdToUse = options?.runId || randomUUID();
+    const run = this.runs.get(runIdToUse) ?? new InngestRun(
+      {
+        workflowId: this.id,
+        runId: runIdToUse,
+        resourceId: options?.resourceId,
+        executionEngine: this.executionEngine,
+        executionGraph: this.executionGraph,
+        serializedStepGraph: this.serializedStepGraph,
+        mastra: this.#mastra,
+        retryConfig: this.retryConfig,
+        cleanup: () => this.runs.delete(runIdToUse),
+        workflowSteps: this.steps,
+        workflowEngineType: this.engineType,
+        validateInputs: this.options.validateInputs
       },
-      tracingPolicy: this.options?.tracingPolicy
+      this.inngest
+    );
+    this.runs.set(runIdToUse, run);
+    const shouldPersistSnapshot = this.options.shouldPersistSnapshot({
+      workflowStatus: run.workflowRunStatus,
+      stepResults: {}
     });
-    if (fn) {
-      date = await this.inngestStep.run(`workflow.${workflowId}.sleepUntil.${entry.id}`, async () => {
-        const stepCallId = randomUUID();
-        return await fn(
-          createDeprecationProxy(
-            {
-              runId,
-              workflowId,
-              mastra: this.mastra,
-              requestContext,
-              inputData: prevOutput,
-              state: executionContext.state,
-              setState: (state) => {
-                executionContext.state = state;
-              },
-              runCount: -1,
-              retryCount: -1,
-              tracingContext: {
-                currentSpan: sleepUntilSpan
-              },
-              getInitData: () => stepResults?.input,
-              getStepResult: getStepResult.bind(this, stepResults),
-              // TODO: this function shouldn't have suspend probably?
-              suspend: async (_suspendPayload) => {
-              },
-              bail: () => {
-              },
-              abort: () => {
-                abortController?.abort();
-              },
-              [EMITTER_SYMBOL]: emitter,
-              [STREAM_FORMAT_SYMBOL]: executionContext.format,
-              engine: { step: this.inngestStep },
-              abortSignal: abortController?.signal,
-              writer: new ToolStream(
-                {
-                  prefix: "workflow-step",
-                  callId: stepCallId,
-                  name: "sleep",
-                  runId
-                },
-                writableStream
-              )
-            },
-            {
-              paramName: "runCount",
-              deprecationMessage: 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)
+    const workflowSnapshotInStorage = await this.getWorkflowRunExecutionResult(runIdToUse, {
+      withNestedWorkflows: false
+    });
+    if (!workflowSnapshotInStorage && shouldPersistSnapshot) {
+      const workflowsStore = await this.mastra?.getStorage()?.getStore("workflows");
+      await workflowsStore?.persistWorkflowSnapshot({
+        workflowName: this.id,
+        runId: runIdToUse,
+        resourceId: options?.resourceId,
+        snapshot: {
+          runId: runIdToUse,
+          status: "pending",
+          value: {},
+          context: {},
+          activePaths: [],
+          activeStepsPath: {},
+          waitingPaths: {},
+          serializedStepGraph: this.serializedStepGraph,
+          suspendedPaths: {},
+          resumeLabels: {},
+          result: void 0,
+          error: void 0,
+          timestamp: Date.now()
         }
       });
     }
-    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;
-    }
+    return run;
   }
-  async executeWaitForEvent({ event, timeout }) {
-    const eventData = await this.inngestStep.waitForEvent(`user-event-${event}`, {
-      event: `user-event-${event}`,
-      timeout: timeout ?? 5e3
-    });
-    if (eventData === null) {
-      throw "Timeout waiting for event";
+  //createCronFunction is only called if cronConfig.cron is defined.
+  createCronFunction() {
+    if (this.cronFunction) {
+      return this.cronFunction;
     }
-    return eventData?.data;
-  }
-  async executeStep({
-    step,
-    stepResults,
-    executionContext,
-    resume,
-    prevOutput,
-    emitter,
-    abortController,
-    requestContext,
-    tracingContext,
-    writableStream,
-    disableScorers
-  }) {
-    const stepAISpan = tracingContext?.currentSpan?.createChildSpan({
-      name: `workflow step: '${step.id}'`,
-      type: AISpanType.WORKFLOW_STEP,
-      input: prevOutput,
-      attributes: {
-        stepId: step.id
+    this.cronFunction = this.inngest.createFunction(
+      {
+        id: `workflow.${this.id}.cron`,
+        retries: 0,
+        cancelOn: [{ event: `cancel.workflow.${this.id}` }],
+        ...this.flowControlConfig
       },
-      tracingPolicy: this.options?.tracingPolicy
-    });
-    const { inputData, validationError } = await validateStepInput({
-      prevOutput,
-      step,
-      validateInputs: this.options?.validateInputs ?? false
-    });
-    const startedAt = await this.inngestStep.run(
-      `workflow.${executionContext.workflowId}.run.${executionContext.runId}.step.${step.id}.running_ev`,
+      { cron: this.cronConfig?.cron ?? "" },
       async () => {
-        const startedAt2 = Date.now();
-        await emitter.emit("watch", {
-          type: "watch",
-          payload: {
-            currentStep: {
-              id: step.id,
-              status: "running"
-            },
-            workflowState: {
-              status: "running",
-              steps: {
-                ...stepResults,
-                [step.id]: {
-                  status: "running"
-                }
-              },
-              result: null,
-              error: null
-            }
-          },
-          eventTimestamp: Date.now()
-        });
-        await emitter.emit("watch-v2", {
-          type: "workflow-step-start",
-          payload: {
-            id: step.id,
-            status: "running",
-            payload: inputData,
-            startedAt: startedAt2
-          }
+        const run = await this.createRun();
+        const result = await run.start({
+          inputData: this.cronConfig?.inputData,
+          initialState: this.cronConfig?.initialState
         });
-        return startedAt2;
+        return { result, runId: run.runId };
       }
     );
-    if (step instanceof InngestWorkflow) {
-      const isResume = !!resume?.steps?.length;
-      let result;
-      let runId;
-      try {
-        if (isResume) {
-          runId = stepResults[resume?.steps?.[0]]?.suspendPayload?.__workflow_meta?.runId ?? 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,
-                // @ts-ignore
-                resumePath: snapshot?.suspendedPaths?.[resume.steps?.[1]]
-              },
-              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 }
-            }
+    return this.cronFunction;
+  }
+  getFunction() {
+    if (this.function) {
+      return this.function;
+    }
+    this.function = this.inngest.createFunction(
+      {
+        id: `workflow.${this.id}`,
+        retries: 0,
+        cancelOn: [{ event: `cancel.workflow.${this.id}` }],
+        // Spread flow control configuration
+        ...this.flowControlConfig
+      },
+      { event: `workflow.${this.id}` },
+      async ({ event, step, attempt, publish }) => {
+        let { inputData, initialState, runId, resourceId, resume, outputOptions, format, timeTravel, perStep } = event.data;
+        if (!runId) {
+          runId = await step.run(`workflow.${this.id}.runIdGen`, async () => {
+            return randomUUID();
           });
-          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 || randomUUID();
-        } else {
-          runId = 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: "watch",
-              payload: {
-                currentStep: {
-                  id: step.id,
-                  status: "failed",
-                  error: result?.error
-                },
-                workflowState: {
-                  status: "running",
-                  steps: stepResults,
-                  result: null,
-                  error: null
-                }
-              },
-              eventTimestamp: Date.now()
-            });
-            await emitter.emit("watch-v2", {
-              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", {
+        const pubsub = new InngestPubSub(this.inngest, this.id, publish);
+        const engine = new InngestExecutionEngine(this.#mastra, step, attempt, this.options);
+        const result = await engine.execute({
+          workflowId: this.id,
+          runId,
+          resourceId,
+          graph: this.executionGraph,
+          serializedStepGraph: this.serializedStepGraph,
+          input: inputData,
+          initialState,
+          pubsub,
+          retryConfig: this.retryConfig,
+          requestContext: new RequestContext(Object.entries(event.data.requestContext ?? {})),
+          resume,
+          timeTravel,
+          perStep,
+          format,
+          abortController: new AbortController(),
+          // currentSpan: undefined, // TODO: Pass actual parent Span from workflow execution context
+          outputOptions,
+          outputWriter: async (chunk) => {
+            try {
+              await pubsub.publish(`workflow.events.v2.${runId}`, {
                 type: "watch",
-                payload: {
-                  currentStep: {
-                    id: step.id,
-                    status: "suspended",
-                    payload: stepResult.payload,
-                    suspendPayload: {
-                      ...stepResult?.suspendPayload,
-                      __workflow_meta: { runId, path: suspendPath }
-                    }
-                  },
-                  workflowState: {
-                    status: "running",
-                    steps: stepResults,
-                    result: null,
-                    error: null
-                  }
-                },
-                eventTimestamp: Date.now()
-              });
-              await emitter.emit("watch-v2", {
-                type: "workflow-step-suspended",
-                payload: {
-                  id: step.id,
-                  status: "suspended"
-                }
+                runId,
+                data: chunk
               });
-              return {
-                executionContext,
-                result: {
-                  status: "suspended",
-                  payload: stepResult.payload,
-                  suspendPayload: {
-                    ...stepResult?.suspendPayload,
-                    __workflow_meta: { runId, path: suspendPath }
-                  }
-                }
-              };
+            } catch (err) {
+              this.logger.debug?.("Failed to publish watch event:", err);
             }
-            await emitter.emit("watch", {
-              type: "watch",
-              payload: {
-                currentStep: {
-                  id: step.id,
-                  status: "suspended",
-                  payload: {}
-                },
-                workflowState: {
-                  status: "running",
-                  steps: stepResults,
-                  result: null,
-                  error: null
-                }
-              },
-              eventTimestamp: Date.now()
+          }
+        });
+        await step.run(`workflow.${this.id}.finalize`, async () => {
+          if (result.status !== "paused") {
+            await engine.invokeLifecycleCallbacksInternal(result);
+          }
+          if (result.status === "failed") {
+            throw new NonRetriableError(`Workflow failed`, {
+              cause: result
             });
-            return {
-              executionContext,
-              result: {
-                status: "suspended",
-                payload: {}
-              }
-            };
           }
-          await emitter.emit("watch", {
-            type: "watch",
-            payload: {
-              currentStep: {
-                id: step.id,
-                status: "success",
-                output: result?.result
-              },
-              workflowState: {
-                status: "running",
-                steps: stepResults,
-                result: null,
-                error: null
+          return result;
+        });
+        return { result, runId };
+      }
+    );
+    return this.function;
+  }
+  getNestedFunctions(steps) {
+    return steps.flatMap((step) => {
+      if (step.type === "step" || step.type === "loop" || step.type === "foreach") {
+        if (step.step instanceof _InngestWorkflow) {
+          return [step.step.getFunction(), ...step.step.getNestedFunctions(step.step.executionGraph.steps)];
+        }
+        return [];
+      } else if (step.type === "parallel" || step.type === "conditional") {
+        return this.getNestedFunctions(step.steps);
+      }
+      return [];
+    });
+  }
+  getFunctions() {
+    return [
+      this.getFunction(),
+      ...this.cronConfig?.cron ? [this.createCronFunction()] : [],
+      ...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 serve$1({
+    ...registerOptions,
+    client: inngest,
+    functions: [...workflowFunctions, ...userFunctions]
+  });
+}
+
+// src/types.ts
+var _compatibilityCheck = true;
+
+// src/index.ts
+function createStep(params, agentOrToolOptions) {
+  if (params instanceof InngestWorkflow) {
+    return params;
+  }
+  if (params instanceof Agent) {
+    const options = agentOrToolOptions;
+    const outputSchema = options?.structuredOutput?.schema ?? z.object({ text: z.string() });
+    const { retries, scorers, ...agentOptions } = options ?? {};
+    return {
+      id: params.name,
+      description: params.getDescription(),
+      inputSchema: z.object({
+        prompt: z.string()
+        // resourceId: z.string().optional(),
+        // threadId: z.string().optional(),
+      }),
+      outputSchema,
+      retries,
+      scorers,
+      execute: async ({
+        inputData,
+        runId,
+        [PUBSUB_SYMBOL]: pubsub,
+        [STREAM_FORMAT_SYMBOL]: streamFormat,
+        requestContext,
+        tracingContext,
+        abortSignal,
+        abort,
+        writer
+      }) => {
+        let streamPromise = {};
+        streamPromise.promise = new Promise((resolve, reject) => {
+          streamPromise.resolve = resolve;
+          streamPromise.reject = reject;
+        });
+        let structuredResult = null;
+        const toolData = {
+          name: params.name,
+          args: inputData
+        };
+        let stream;
+        if ((await params.getModel()).specificationVersion === "v1") {
+          const { fullStream } = await params.streamLegacy(inputData.prompt, {
+            ...agentOptions ?? {},
+            // resourceId: inputData.resourceId,
+            // threadId: inputData.threadId,
+            requestContext,
+            tracingContext,
+            onFinish: (result) => {
+              const resultWithObject = result;
+              if (agentOptions?.structuredOutput?.schema && resultWithObject.object) {
+                structuredResult = resultWithObject.object;
               }
+              streamPromise.resolve(result.text);
+              void agentOptions?.onFinish?.(result);
             },
-            eventTimestamp: Date.now()
-          });
-          await emitter.emit("watch-v2", {
-            type: "workflow-step-result",
-            payload: {
-              id: step.id,
-              status: "success",
-              output: result?.result
-            }
-          });
-          await emitter.emit("watch-v2", {
-            type: "workflow-step-finish",
-            payload: {
-              id: step.id,
-              metadata: {}
-            }
+            abortSignal
           });
-          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 = randomUUID();
-    let stepRes;
-    try {
-      stepRes = await this.inngestStep.run(`workflow.${executionContext.workflowId}.step.${step.id}`, async () => {
-        let execResults;
-        let suspended;
-        let bailed;
-        try {
-          if (validationError) {
-            throw validationError;
-          }
-          const result = await step.execute({
-            runId: executionContext.runId,
-            mastra: this.mastra,
+          stream = fullStream;
+        } else {
+          const modelOutput = await params.stream(inputData.prompt, {
+            ...agentOptions ?? {},
             requestContext,
-            writer: new ToolStream(
-              {
-                prefix: "workflow-step",
-                callId: stepCallId,
-                name: step.id,
-                runId: executionContext.runId
-              },
-              writableStream
-            ),
-            state: executionContext?.state ?? {},
-            setState: (state) => {
-              executionContext.state = state;
-            },
-            inputData,
-            resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
-            tracingContext: {
-              currentSpan: stepAISpan
-            },
-            getInitData: () => stepResults?.input,
-            getStepResult: getStepResult.bind(this, stepResults),
-            suspend: async (suspendPayload, suspendOptions) => {
-              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
-                  };
-                }
+            tracingContext,
+            onFinish: (result) => {
+              const resultWithObject = result;
+              if (agentOptions?.structuredOutput?.schema && resultWithObject.object) {
+                structuredResult = resultWithObject.object;
               }
-              suspended = { payload: suspendPayload };
-            },
-            bail: (result2) => {
-              bailed = { payload: result2 };
-            },
-            resume: {
-              steps: resume?.steps?.slice(1) || [],
-              resumePayload: resume?.resumePayload,
-              // @ts-ignore
-              runId: stepResults[step.id]?.suspendPayload?.__workflow_meta?.runId
-            },
-            [EMITTER_SYMBOL]: emitter,
-            [STREAM_FORMAT_SYMBOL]: executionContext.format,
-            engine: {
-              step: this.inngestStep
+              streamPromise.resolve(result.text);
+              void agentOptions?.onFinish?.(result);
             },
-            abortSignal: abortController.signal
-          });
-          const endedAt = Date.now();
-          execResults = {
-            status: "success",
-            output: result,
-            startedAt,
-            endedAt,
-            payload: inputData,
-            resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
-            resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
-          };
-        } catch (e) {
-          const stepFailure = {
-            status: "failed",
-            payload: inputData,
-            error: e instanceof Error ? e.message : String(e),
-            endedAt: Date.now(),
-            startedAt,
-            resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
-            resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
-          };
-          execResults = stepFailure;
-          const fallbackErrorMessage = `Step ${step.id} failed`;
-          stepAISpan?.error({ error: new Error(execResults.error ?? fallbackErrorMessage) });
-          throw new RetryAfterError(execResults.error ?? fallbackErrorMessage, executionContext.retryConfig.delay, {
-            cause: execResults
+            abortSignal
           });
+          stream = modelOutput.fullStream;
         }
-        if (suspended) {
-          execResults = {
-            status: "suspended",
-            suspendPayload: suspended.payload,
-            payload: inputData,
-            suspendedAt: Date.now(),
-            startedAt,
-            resumedAt: resume?.steps[0] === step.id ? startedAt : void 0,
-            resumePayload: resume?.steps[0] === step.id ? resume?.resumePayload : void 0
-          };
-        } else if (bailed) {
-          execResults = {
-            status: "bailed",
-            output: bailed.payload,
-            payload: inputData,
-            endedAt: Date.now(),
-            startedAt
-          };
-        }
-        await emitter.emit("watch", {
-          type: "watch",
-          payload: {
-            currentStep: {
-              id: step.id,
-              ...execResults
-            },
-            workflowState: {
-              status: "running",
-              steps: { ...stepResults, [step.id]: execResults },
-              result: null,
-              error: null
-            }
-          },
-          eventTimestamp: Date.now()
-        });
-        if (execResults.status === "suspended") {
-          await emitter.emit("watch-v2", {
-            type: "workflow-step-suspended",
-            payload: {
-              id: step.id,
-              ...execResults
-            }
-          });
-        } else {
-          await emitter.emit("watch-v2", {
-            type: "workflow-step-result",
-            payload: {
-              id: step.id,
-              ...execResults
-            }
+        if (streamFormat === "legacy") {
+          await pubsub.publish(`workflow.events.v2.${runId}`, {
+            type: "watch",
+            runId,
+            data: { type: "tool-call-streaming-start", ...toolData ?? {} }
           });
-          await emitter.emit("watch-v2", {
-            type: "workflow-step-finish",
-            payload: {
-              id: step.id,
-              metadata: {}
+          for await (const chunk of stream) {
+            if (chunk.type === "text-delta") {
+              await pubsub.publish(`workflow.events.v2.${runId}`, {
+                type: "watch",
+                runId,
+                data: { type: "tool-call-delta", ...toolData ?? {}, argsTextDelta: chunk.textDelta }
+              });
             }
+          }
+          await pubsub.publish(`workflow.events.v2.${runId}`, {
+            type: "watch",
+            runId,
+            data: { type: "tool-call-streaming-finish", ...toolData ?? {} }
           });
+        } else {
+          for await (const chunk of stream) {
+            await writer.write(chunk);
+          }
         }
-        stepAISpan?.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 (abortSignal.aborted) {
+          return abort();
         }
-      };
-    }
-    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: stepAISpan }
-          });
+        if (structuredResult !== null) {
+          return structuredResult;
         }
-      });
-    }
-    Object.assign(executionContext.suspendedPaths, stepRes.executionContext.suspendedPaths);
-    Object.assign(stepResults, stepRes.stepResults);
-    executionContext.state = stepRes.executionContext.state;
-    return stepRes.result;
+        return {
+          text: await streamPromise.promise
+        };
+      },
+      component: params.component
+    };
   }
-  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: {
+  if (params instanceof Tool) {
+    const toolOpts = agentOrToolOptions;
+    if (!params.inputSchema || !params.outputSchema) {
+      throw new Error("Tool must have input and output schemas defined");
+    }
+    return {
+      // TODO: tool probably should have strong id type
+      id: params.id,
+      description: params.description,
+      inputSchema: params.inputSchema,
+      outputSchema: params.outputSchema,
+      suspendSchema: params.suspendSchema,
+      resumeSchema: params.resumeSchema,
+      retries: toolOpts?.retries,
+      scorers: toolOpts?.scorers,
+      execute: async ({
+        inputData,
+        mastra,
+        requestContext,
+        tracingContext,
+        suspend,
+        resumeData,
+        runId,
+        workflowId,
+        state,
+        setState
+      }) => {
+        const toolContext = {
+          mastra,
+          requestContext,
+          tracingContext,
+          workflow: {
             runId,
-            value: executionContext.state,
-            context: stepResults,
-            activePaths: [],
-            suspendedPaths: executionContext.suspendedPaths,
-            resumeLabels: executionContext.resumeLabels,
-            waitingPaths: {},
-            serializedStepGraph,
-            status: workflowStatus,
-            result,
-            error,
-            // @ts-ignore
-            timestamp: Date.now()
-          }
-        });
-      }
-    );
-  }
-  async executeConditional({
-    workflowId,
-    runId,
-    entry,
-    prevOutput,
-    stepResults,
-    resume,
-    executionContext,
-    emitter,
-    abortController,
-    requestContext,
-    writableStream,
-    disableScorers,
-    tracingContext
-  }) {
-    const conditionalSpan = tracingContext?.currentSpan?.createChildSpan({
-      type: AISpanType.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: AISpanType.WORKFLOW_CONDITIONAL_EVAL,
-            name: `condition: '${index}'`,
-            input: prevOutput,
-            attributes: {
-              conditionIndex: index
-            },
-            tracingPolicy: this.options?.tracingPolicy
-          });
-          try {
-            const result = await cond(
-              createDeprecationProxy(
-                {
-                  runId,
-                  workflowId,
-                  mastra: this.mastra,
-                  requestContext,
-                  runCount: -1,
-                  retryCount: -1,
-                  inputData: prevOutput,
-                  state: executionContext.state,
-                  setState: (state) => {
-                    executionContext.state = state;
-                  },
-                  tracingContext: {
-                    currentSpan: evalSpan
-                  },
-                  getInitData: () => stepResults?.input,
-                  getStepResult: getStepResult.bind(this, stepResults),
-                  // TODO: this function shouldn't have suspend probably?
-                  suspend: async (_suspendPayload) => {
-                  },
-                  bail: () => {
-                  },
-                  abort: () => {
-                    abortController.abort();
-                  },
-                  [EMITTER_SYMBOL]: emitter,
-                  [STREAM_FORMAT_SYMBOL]: executionContext.format,
-                  engine: {
-                    step: this.inngestStep
-                  },
-                  abortSignal: abortController.signal,
-                  writer: new ToolStream(
-                    {
-                      prefix: "workflow-step",
-                      callId: randomUUID(),
-                      name: "conditional",
-                      runId
-                    },
-                    writableStream
-                  )
-                },
-                {
-                  paramName: "runCount",
-                  deprecationMessage: 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,
-          executionContext: {
+            resumeData,
+            suspend,
             workflowId,
-            runId,
-            executionPath: [...executionContext.executionPath, index],
-            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 };
-    } else {
-      execResults = {
-        status: "success",
-        output: results.reduce((acc, result, index) => {
-          if (result.status === "success") {
-            acc[stepsToRun[index].step.id] = result.output;
+            state,
+            setState
           }
-          return acc;
-        }, {})
+        };
+        return params.execute(inputData, toolContext);
+      },
+      component: "TOOL"
+    };
+  }
+  return {
+    id: params.id,
+    description: params.description,
+    inputSchema: params.inputSchema,
+    outputSchema: params.outputSchema,
+    resumeSchema: params.resumeSchema,
+    suspendSchema: params.suspendSchema,
+    retries: params.retries,
+    scorers: params.scorers,
+    execute: params.execute
+  };
+}
+function init(inngest) {
+  return {
+    createWorkflow(params) {
+      return new InngestWorkflow(
+        params,
+        inngest
+      );
+    },
+    createStep,
+    cloneStep(step, opts) {
+      return {
+        id: opts.id,
+        description: step.description,
+        inputSchema: step.inputSchema,
+        outputSchema: step.outputSchema,
+        resumeSchema: step.resumeSchema,
+        suspendSchema: step.suspendSchema,
+        stateSchema: step.stateSchema,
+        execute: step.execute,
+        retries: step.retries,
+        scorers: step.scorers,
+        component: step.component
       };
-    }
-    if (execResults.status === "failed") {
-      conditionalSpan?.error({
-        error: new Error(execResults.error)
-      });
-    } else {
-      conditionalSpan?.end({
-        output: execResults.output || execResults
+    },
+    cloneWorkflow(workflow, opts) {
+      const wf = new Workflow({
+        id: opts.id,
+        inputSchema: workflow.inputSchema,
+        outputSchema: workflow.outputSchema,
+        steps: workflow.stepDefs,
+        mastra: workflow.mastra,
+        options: workflow.options
       });
+      wf.setStepFlow(workflow.stepGraph);
+      wf.commit();
+      return wf;
     }
-    return execResults;
-  }
-};
+  };
+}
 
-export { InngestExecutionEngine, InngestRun, InngestWorkflow, createStep, init, serve };
+export { InngestExecutionEngine, InngestPubSub, InngestRun, InngestWorkflow, _compatibilityCheck, createStep, init, serve };
 //# sourceMappingURL=index.js.map
 //# sourceMappingURL=index.js.map