@mastra/core 0.0.0-commonjs-20250227130920

package/dist/chunk-B3M27AMP.js

@@ -0,0 +1,1479 @@
+import { MastraBase } from './chunk-OZ4XVJ6F.js';
+import { setTimeout } from 'node:timers/promises';
+import { context, trace } from '@opentelemetry/api';
+import EventEmitter from 'node:events';
+import { get } from 'radash';
+import sift from 'sift';
+import { createActor, assign, fromPromise, setup } from 'xstate';
+
+// src/workflows/utils.ts
+function isErrorEvent(stateEvent) {
+  return stateEvent.type.startsWith("xstate.error.actor.");
+}
+function isTransitionEvent(stateEvent) {
+  return stateEvent.type.startsWith("xstate.done.actor.");
+}
+function isVariableReference(value) {
+  return typeof value === "object" && "step" in value && "path" in value;
+}
+function getStepResult(result) {
+  if (result?.status === "success") return result.output;
+  return void 0;
+}
+function getSuspendedPaths({
+  value,
+  path,
+  suspendedPaths
+}) {
+  if (typeof value === "string") {
+    if (value === "suspended") {
+      suspendedPaths.add(path);
+    }
+  } else {
+    Object.keys(value).forEach(
+      (key) => getSuspendedPaths({ value: value[key], path: path ? `${path}.${key}` : key, suspendedPaths })
+    );
+  }
+}
+function isFinalState(status) {
+  return ["completed", "failed"].includes(status);
+}
+function recursivelyCheckForFinalState({
+  value,
+  suspendedPaths,
+  path
+}) {
+  if (typeof value === "string") {
+    return isFinalState(value) || suspendedPaths.has(path);
+  }
+  return Object.keys(value).every(
+    (key) => recursivelyCheckForFinalState({ value: value[key], suspendedPaths, path: path ? `${path}.${key}` : key })
+  );
+}
+function getActivePathsAndStatus(value) {
+  const paths = [];
+  const traverse = (current, path = []) => {
+    for (const [key, value2] of Object.entries(current)) {
+      const currentPath = [...path, key];
+      if (typeof value2 === "string") {
+        paths.push({
+          stepPath: currentPath,
+          stepId: key,
+          status: value2
+        });
+      } else if (typeof value2 === "object" && value2 !== null) {
+        traverse(value2, currentPath);
+      }
+    }
+  };
+  traverse(value);
+  return paths;
+}
+function mergeChildValue(startStepId, parent, child) {
+  const traverse = (current) => {
+    const obj = {};
+    for (const [key, value] of Object.entries(current)) {
+      if (key === startStepId) {
+        obj[key] = { ...child };
+      } else if (typeof value === "string") {
+        obj[key] = value;
+      } else if (typeof value === "object" && value !== null) {
+        obj[key] = traverse(value);
+      }
+    }
+    return obj;
+  };
+  return traverse(parent);
+}
+var updateStepInHierarchy = (value, targetStepId) => {
+  const result = {};
+  for (const key of Object.keys(value)) {
+    const currentValue = value[key];
+    if (key === targetStepId) {
+      result[key] = "pending";
+    } else if (typeof currentValue === "object" && currentValue !== null) {
+      result[key] = updateStepInHierarchy(currentValue, targetStepId);
+    } else {
+      result[key] = currentValue;
+    }
+  }
+  return result;
+};
+var Machine = class extends EventEmitter {
+  logger;
+  #mastra;
+  #workflowInstance;
+  #executionSpan;
+  #stepGraph;
+  #machine;
+  #runId;
+  #startStepId;
+  name;
+  #actor = null;
+  #steps = {};
+  #retryConfig;
+  constructor({
+    logger,
+    mastra,
+    workflowInstance,
+    executionSpan,
+    name,
+    runId,
+    steps,
+    stepGraph,
+    retryConfig,
+    startStepId
+  }) {
+    super();
+    this.#mastra = mastra;
+    this.#workflowInstance = workflowInstance;
+    this.#executionSpan = executionSpan;
+    this.logger = logger;
+    this.#runId = runId;
+    this.#startStepId = startStepId;
+    this.name = name;
+    this.#stepGraph = stepGraph;
+    this.#steps = steps;
+    this.#retryConfig = retryConfig;
+    this.initializeMachine();
+  }
+  get startStepId() {
+    return this.#startStepId;
+  }
+  async execute({
+    stepId,
+    input,
+    snapshot
+  } = {}) {
+    if (snapshot) {
+      this.logger.debug(`Workflow snapshot received`, { runId: this.#runId, snapshot });
+    }
+    this.logger.debug(`Machine input prepared`, { runId: this.#runId, input });
+    const actorSnapshot = snapshot ? {
+      ...snapshot,
+      context: input
+    } : void 0;
+    this.logger.debug(`Creating actor with configuration`, {
+      input,
+      actorSnapshot,
+      runId: this.#runId,
+      machineStates: this.#machine.config.states
+    });
+    this.#actor = createActor(this.#machine, {
+      inspect: (inspectionEvent) => {
+        this.logger.debug("XState inspection event", {
+          type: inspectionEvent.type,
+          event: inspectionEvent.event,
+          runId: this.#runId
+        });
+      },
+      input,
+      snapshot: actorSnapshot
+    });
+    this.#actor.start();
+    if (stepId) {
+      this.#actor.send({ type: "RESET_TO_PENDING", stepId });
+    }
+    this.logger.debug("Actor started", { runId: this.#runId });
+    return new Promise((resolve, reject) => {
+      if (!this.#actor) {
+        const e = new Error("Actor not initialized");
+        this.#executionSpan?.recordException(e);
+        this.#executionSpan?.end();
+        reject(e);
+        return;
+      }
+      const suspendedPaths = /* @__PURE__ */ new Set();
+      this.#actor.subscribe(async (state) => {
+        this.emit("state-update", this.#startStepId, state.value, state.context);
+        getSuspendedPaths({
+          value: state.value,
+          path: "",
+          suspendedPaths
+        });
+        const allStatesValue = state.value;
+        const allStatesComplete = recursivelyCheckForFinalState({
+          value: allStatesValue,
+          suspendedPaths,
+          path: ""
+        });
+        this.logger.debug("State completion check", {
+          allStatesComplete,
+          suspendedPaths: Array.from(suspendedPaths),
+          runId: this.#runId
+        });
+        if (!allStatesComplete) return;
+        try {
+          await this.#workflowInstance.persistWorkflowSnapshot();
+          this.#cleanup();
+          this.#executionSpan?.end();
+          resolve({
+            results: state.context.steps
+          });
+        } catch (error) {
+          this.logger.debug("Failed to persist final snapshot", { error });
+          this.#cleanup();
+          this.#executionSpan?.end();
+          resolve({
+            results: state.context.steps
+          });
+        }
+      });
+    });
+  }
+  #cleanup() {
+    if (this.#actor) {
+      this.#actor.stop();
+      this.#actor = null;
+    }
+    this.removeAllListeners();
+  }
+  #makeDelayMap() {
+    const delayMap = {};
+    Object.keys(this.#steps).forEach((stepId) => {
+      delayMap[stepId] = this.#steps[stepId]?.retryConfig?.delay || this.#retryConfig?.delay || 1e3;
+    });
+    return delayMap;
+  }
+  #getDefaultActions() {
+    return {
+      updateStepResult: assign({
+        steps: ({ context, event }) => {
+          if (!isTransitionEvent(event)) return context.steps;
+          const { stepId, result } = event.output;
+          return {
+            ...context.steps,
+            [stepId]: {
+              status: "success",
+              output: result
+            }
+          };
+        }
+      }),
+      setStepError: assign({
+        steps: ({ context, event }, params) => {
+          if (!isErrorEvent(event)) return context.steps;
+          const { stepId } = params;
+          if (!stepId) return context.steps;
+          return {
+            ...context.steps,
+            [stepId]: {
+              status: "failed",
+              error: event.error.message
+            }
+          };
+        }
+      }),
+      notifyStepCompletion: async (_, params) => {
+        const { stepId } = params;
+        this.logger.debug(`Step ${stepId} completed`);
+      },
+      snapshotStep: assign({
+        _snapshot: ({}, params) => {
+          const { stepId } = params;
+          return { stepId };
+        }
+      }),
+      persistSnapshot: async ({ context }) => {
+        if (context._snapshot) {
+          await this.#workflowInstance.persistWorkflowSnapshot();
+        }
+        return;
+      },
+      decrementAttemptCount: assign({
+        attempts: ({ context, event }, params) => {
+          if (!isTransitionEvent(event)) return context.attempts;
+          const { stepId } = params;
+          const attemptCount = context.attempts[stepId];
+          if (attemptCount === void 0) return context.attempts;
+          return { ...context.attempts, [stepId]: attemptCount - 1 };
+        }
+      })
+    };
+  }
+  #getDefaultActors() {
+    return {
+      resolverFunction: fromPromise(async ({ input }) => {
+        const { stepNode, context } = input;
+        const resolvedData = this.#resolveVariables({
+          stepConfig: stepNode.config,
+          context,
+          stepId: stepNode.step.id
+        });
+        this.logger.debug(`Resolved variables for ${stepNode.step.id}`, {
+          resolvedData,
+          runId: this.#runId
+        });
+        const result = await stepNode.config.handler({
+          context: resolvedData,
+          suspend: async () => {
+            await this.#workflowInstance.suspend(stepNode.step.id, this);
+            if (this.#actor) {
+              context.steps[stepNode.step.id] = {
+                status: "suspended"
+              };
+              this.logger.debug(`Sending SUSPENDED event for step ${stepNode.step.id}`);
+              this.#actor?.send({ type: "SUSPENDED", stepId: stepNode.step.id });
+            } else {
+              this.logger.debug(`Actor not available for step ${stepNode.step.id}`);
+            }
+          },
+          runId: this.#runId,
+          mastra: this.#mastra
+        });
+        this.logger.debug(`Step ${stepNode.step.id} result`, {
+          stepId: stepNode.step.id,
+          result,
+          runId: this.#runId
+        });
+        return {
+          stepId: stepNode.step.id,
+          result
+        };
+      }),
+      conditionCheck: fromPromise(async ({ input }) => {
+        const { context, stepNode } = input;
+        const stepConfig = stepNode.config;
+        const attemptCount = context.attempts[stepNode.step.id];
+        this.logger.debug(`Checking conditions for step ${stepNode.step.id}`, {
+          stepId: stepNode.step.id,
+          runId: this.#runId
+        });
+        this.logger.debug(`Attempt count for step ${stepNode.step.id}`, {
+          attemptCount,
+          attempts: context.attempts,
+          runId: this.#runId,
+          stepId: stepNode.step.id
+        });
+        if (!attemptCount || attemptCount < 0) {
+          if (stepConfig?.snapshotOnTimeout) {
+            return { type: "SUSPENDED", stepId: stepNode.step.id };
+          }
+          return { type: "CONDITION_FAILED", error: `Step:${stepNode.step.id} condition check failed` };
+        }
+        if (!stepConfig?.when) {
+          return { type: "CONDITIONS_MET" };
+        }
+        this.logger.debug(`Checking conditions for step ${stepNode.step.id}`, {
+          stepId: stepNode.step.id,
+          runId: this.#runId
+        });
+        if (typeof stepConfig?.when === "function") {
+          const conditionMet = await stepConfig.when({
+            context: {
+              ...context,
+              getStepResult: (stepId) => {
+                if (stepId === "trigger") {
+                  return context.triggerData;
+                }
+                const result = context.steps[stepId];
+                if (result && result.status === "success") {
+                  return result.output;
+                }
+                return void 0;
+              }
+            },
+            mastra: this.#mastra
+          });
+          if (conditionMet) {
+            this.logger.debug(`Condition met for step ${stepNode.step.id}`, {
+              stepId: stepNode.step.id,
+              runId: this.#runId
+            });
+            return { type: "CONDITIONS_MET" };
+          }
+          if (!attemptCount || attemptCount < 0) {
+            return { type: "CONDITION_FAILED", error: `Step:${stepNode.step.id} condition check failed` };
+          }
+          return { type: "WAITING", stepId: stepNode.step.id };
+        } else {
+          const conditionMet = this.#evaluateCondition(stepConfig.when, context);
+          if (!conditionMet) {
+            return {
+              type: "CONDITION_FAILED",
+              error: `Step:${stepNode.step.id} condition check failed`
+            };
+          }
+        }
+        return { type: "CONDITIONS_MET" };
+      }),
+      spawnSubscriberFunction: fromPromise(
+        async ({
+          input
+        }) => {
+          const { parentStepId, context } = input;
+          const result = await this.#workflowInstance.runMachine(parentStepId, context);
+          return Promise.resolve({ steps: result?.results });
+        }
+      )
+    };
+  }
+  #resolveVariables({
+    stepConfig,
+    context,
+    stepId
+  }) {
+    this.logger.debug(`Resolving variables for step ${stepId}`, {
+      stepId,
+      runId: this.#runId
+    });
+    const resolvedData = {
+      ...context,
+      getStepResult: (stepId2) => {
+        if (stepId2 === "trigger") {
+          return context.triggerData;
+        }
+        const result = context.steps[stepId2];
+        if (result && result.status === "success") {
+          return result.output;
+        }
+        return void 0;
+      }
+    };
+    for (const [key, variable] of Object.entries(stepConfig.data)) {
+      const sourceData = variable.step === "trigger" ? context.triggerData : getStepResult(context.steps[variable.step.id]);
+      this.logger.debug(
+        `Got source data for ${key} variable from ${variable.step === "trigger" ? "trigger" : variable.step.id}`,
+        {
+          sourceData,
+          path: variable.path,
+          runId: this.#runId
+        }
+      );
+      if (!sourceData && variable.step !== "trigger") {
+        resolvedData[key] = void 0;
+        continue;
+      }
+      const value = variable.path === "" || variable.path === "." ? sourceData : get(sourceData, variable.path);
+      this.logger.debug(`Resolved variable ${key}`, {
+        value,
+        runId: this.#runId
+      });
+      resolvedData[key] = value;
+    }
+    return resolvedData;
+  }
+  initializeMachine() {
+    const machine = setup({
+      types: {},
+      delays: this.#makeDelayMap(),
+      actions: this.#getDefaultActions(),
+      actors: this.#getDefaultActors()
+    }).createMachine({
+      id: this.name,
+      type: "parallel",
+      context: ({ input }) => ({
+        ...input
+      }),
+      states: this.#buildStateHierarchy(this.#stepGraph)
+    });
+    this.#machine = machine;
+    return machine;
+  }
+  #buildStateHierarchy(stepGraph) {
+    const states = {};
+    stepGraph.initial.forEach((stepNode) => {
+      const nextSteps = [...stepGraph[stepNode.step.id] || []];
+      states[stepNode.step.id] = {
+        ...this.#buildBaseState(stepNode, nextSteps)
+      };
+    });
+    return states;
+  }
+  #buildBaseState(stepNode, nextSteps = []) {
+    const nextStep = nextSteps.shift();
+    return {
+      initial: "pending",
+      on: {
+        RESET_TO_PENDING: {
+          target: ".pending"
+          // Note the dot to target child state
+        }
+      },
+      states: {
+        pending: {
+          entry: () => {
+            this.logger.debug(`Step ${stepNode.step.id} pending`, {
+              stepId: stepNode.step.id,
+              runId: this.#runId
+            });
+          },
+          exit: () => {
+            this.logger.debug(`Step ${stepNode.step.id} finished pending`, {
+              stepId: stepNode.step.id,
+              runId: this.#runId
+            });
+          },
+          invoke: {
+            src: "conditionCheck",
+            input: ({ context }) => {
+              return {
+                context,
+                stepNode
+              };
+            },
+            onDone: [
+              {
+                guard: ({ event }) => {
+                  return event.output.type === "SUSPENDED";
+                },
+                target: "suspended",
+                actions: [
+                  assign({
+                    steps: ({ context, event }) => {
+                      if (event.output.type !== "SUSPENDED") return context.steps;
+                      return {
+                        ...context.steps,
+                        [stepNode.step.id]: {
+                          status: "suspended",
+                          ...context.steps?.[stepNode.step.id] || {}
+                        }
+                      };
+                    },
+                    attempts: ({ context, event }) => {
+                      if (event.output.type !== "SUSPENDED") return context.attempts;
+                      return { ...context.attempts, [stepNode.step.id]: stepNode.step.retryConfig?.attempts || 3 };
+                    }
+                  })
+                ]
+              },
+              {
+                guard: ({ event }) => {
+                  return event.output.type === "WAITING";
+                },
+                target: "waiting",
+                actions: [
+                  { type: "decrementAttemptCount", params: { stepId: stepNode.step.id } },
+                  assign({
+                    steps: ({ context, event }) => {
+                      if (event.output.type !== "WAITING") return context.steps;
+                      return {
+                        ...context.steps,
+                        [stepNode.step.id]: {
+                          status: "waiting"
+                        }
+                      };
+                    }
+                  })
+                ]
+              },
+              {
+                guard: ({ event }) => {
+                  return event.output.type === "CONDITIONS_MET";
+                },
+                target: "executing"
+              },
+              {
+                guard: ({ event }) => {
+                  return event.output.type === "CONDITION_FAILED";
+                },
+                target: "failed",
+                actions: assign({
+                  steps: ({ context, event }) => {
+                    if (event.output.type !== "CONDITION_FAILED") return context.steps;
+                    this.logger.debug(`Workflow condition check failed`, {
+                      error: event.output.error,
+                      stepId: stepNode.step.id
+                    });
+                    return {
+                      ...context.steps,
+                      [stepNode.step.id]: {
+                        status: "failed",
+                        error: event.output.error
+                      }
+                    };
+                  }
+                })
+              }
+            ]
+          }
+        },
+        waiting: {
+          entry: () => {
+            this.logger.debug(`Step ${stepNode.step.id} waiting`, {
+              stepId: stepNode.step.id,
+              timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+              runId: this.#runId
+            });
+          },
+          exit: () => {
+            this.logger.debug(`Step ${stepNode.step.id} finished waiting`, {
+              stepId: stepNode.step.id,
+              timestamp: (/* @__PURE__ */ new Date()).toISOString(),
+              runId: this.#runId
+            });
+          },
+          after: {
+            [stepNode.step.id]: {
+              target: "pending"
+            }
+          }
+        },
+        suspended: {
+          type: "final",
+          entry: [
+            () => {
+              this.logger.debug(`Step ${stepNode.step.id} suspended`, {
+                stepId: stepNode.step.id,
+                runId: this.#runId
+              });
+            },
+            assign({
+              steps: ({ context }) => ({
+                ...context.steps,
+                [stepNode.step.id]: {
+                  ...context?.steps?.[stepNode.step.id] || {},
+                  status: "suspended"
+                }
+              })
+            })
+          ]
+        },
+        executing: {
+          entry: () => {
+            this.logger.debug(`Step ${stepNode.step.id} executing`, {
+              stepId: stepNode.step.id,
+              runId: this.#runId
+            });
+          },
+          on: {
+            SUSPENDED: {
+              target: "suspended",
+              actions: [
+                assign({
+                  steps: ({ context }) => ({
+                    ...context.steps,
+                    [stepNode.step.id]: {
+                      status: "suspended"
+                    }
+                  })
+                })
+              ]
+            }
+          },
+          invoke: {
+            src: "resolverFunction",
+            input: ({ context }) => ({
+              context,
+              stepNode
+            }),
+            onDone: {
+              target: "runningSubscribers",
+              actions: [
+                ({ event }) => this.logger.debug(`Step ${stepNode.step.id} finished executing`, {
+                  stepId: stepNode.step.id,
+                  output: event.output,
+                  runId: this.#runId
+                }),
+                { type: "updateStepResult", params: { stepId: stepNode.step.id } },
+                { type: "spawnSubscribers", params: { stepId: stepNode.step.id } }
+              ]
+            },
+            onError: {
+              target: "failed",
+              actions: [{ type: "setStepError", params: { stepId: stepNode.step.id } }]
+            }
+          }
+        },
+        runningSubscribers: {
+          entry: () => {
+            this.logger.debug(`Step ${stepNode.step.id} running subscribers`, {
+              stepId: stepNode.step.id,
+              runId: this.#runId
+            });
+          },
+          exit: () => {
+            this.logger.debug(`Step ${stepNode.step.id} finished running subscribers`, {
+              stepId: stepNode.step.id,
+              runId: this.#runId
+            });
+          },
+          invoke: {
+            src: "spawnSubscriberFunction",
+            input: ({ context }) => ({
+              parentStepId: stepNode.step.id,
+              context
+            }),
+            onDone: {
+              target: nextStep ? nextStep.step.id : "completed",
+              actions: [
+                assign({
+                  steps: ({ context, event }) => ({
+                    ...context.steps,
+                    ...event.output.steps
+                  })
+                }),
+                () => this.logger.debug(`Subscriber execution completed`, { stepId: stepNode.step.id })
+              ]
+            },
+            onError: {
+              target: nextStep ? nextStep.step.id : "completed",
+              actions: ({ event }) => {
+                this.logger.debug(`Subscriber execution failed`, {
+                  error: event.error,
+                  stepId: stepNode.step.id
+                });
+              }
+            }
+          }
+        },
+        completed: {
+          type: "final",
+          entry: [
+            { type: "notifyStepCompletion", params: { stepId: stepNode.step.id } },
+            { type: "snapshotStep", params: { stepId: stepNode.step.id } },
+            { type: "persistSnapshot" }
+          ]
+        },
+        failed: {
+          type: "final",
+          entry: [
+            { type: "notifyStepCompletion", params: { stepId: stepNode.step.id } },
+            { type: "snapshotStep", params: { stepId: stepNode.step.id } },
+            { type: "persistSnapshot" }
+          ]
+        },
+        // build chain of next steps recursively
+        ...nextStep ? { [nextStep.step.id]: { ...this.#buildBaseState(nextStep, nextSteps) } } : {}
+      }
+    };
+  }
+  #evaluateCondition(condition, context) {
+    let andBranchResult = true;
+    let baseResult = true;
+    let orBranchResult = true;
+    const simpleCondition = Object.entries(condition).find(([key]) => key.includes("."));
+    if (simpleCondition) {
+      const [key, queryValue] = simpleCondition;
+      const [stepId, ...pathParts] = key.split(".");
+      const path = pathParts.join(".");
+      const sourceData = stepId === "trigger" ? context.triggerData : getStepResult(context.steps[stepId]);
+      this.logger.debug(`Got condition data from step ${stepId}`, {
+        stepId,
+        sourceData,
+        runId: this.#runId
+      });
+      if (!sourceData) {
+        return false;
+      }
+      let value = get(sourceData, path);
+      if (stepId !== "trigger" && path === "status" && !value) {
+        value = "success";
+      }
+      if (typeof queryValue === "object" && queryValue !== null) {
+        baseResult = sift(queryValue)(value);
+      } else {
+        baseResult = value === queryValue;
+      }
+    }
+    if ("ref" in condition) {
+      const { ref, query } = condition;
+      const sourceData = ref.step === "trigger" ? context.triggerData : getStepResult(context.steps[ref.step.id]);
+      this.logger.debug(`Got condition data from ${ref.step === "trigger" ? "trigger" : ref.step.id}`, {
+        sourceData,
+        runId: this.#runId
+      });
+      if (!sourceData) {
+        return false;
+      }
+      let value = get(sourceData, ref.path);
+      if (ref.step !== "trigger" && ref.path === "status" && !value) {
+        value = "success";
+      }
+      baseResult = sift(query)(value);
+    }
+    if ("and" in condition) {
+      andBranchResult = condition.and.every((cond) => this.#evaluateCondition(cond, context));
+      this.logger.debug(`Evaluated AND condition`, {
+        andBranchResult,
+        runId: this.#runId
+      });
+    }
+    if ("or" in condition) {
+      orBranchResult = condition.or.some((cond) => this.#evaluateCondition(cond, context));
+      this.logger.debug(`Evaluated OR condition`, {
+        orBranchResult,
+        runId: this.#runId
+      });
+    }
+    const finalResult = baseResult && andBranchResult && orBranchResult;
+    this.logger.debug(`Evaluated condition`, {
+      finalResult,
+      runId: this.#runId
+    });
+    return finalResult;
+  }
+  getSnapshot() {
+    const snapshot = this.#actor?.getSnapshot();
+    return snapshot;
+  }
+};
+
+// src/workflows/workflow-instance.ts
+var WorkflowInstance = class {
+  name;
+  #mastra;
+  #machines = {};
+  logger;
+  #steps = {};
+  #stepGraph;
+  #stepSubscriberGraph = {};
+  #retryConfig;
+  #runId;
+  #state = null;
+  #executionSpan;
+  #onStepTransition = /* @__PURE__ */ new Set();
+  #onFinish;
+  // indexed by stepId
+  #suspendedMachines = {};
+  constructor({
+    name,
+    logger,
+    steps,
+    runId,
+    retryConfig,
+    mastra,
+    stepGraph,
+    stepSubscriberGraph,
+    onStepTransition,
+    onFinish
+  }) {
+    this.name = name;
+    this.logger = logger;
+    this.#steps = steps;
+    this.#stepGraph = stepGraph;
+    this.#stepSubscriberGraph = stepSubscriberGraph;
+    this.#retryConfig = retryConfig;
+    this.#mastra = mastra;
+    this.#runId = runId ?? crypto.randomUUID();
+    this.#onStepTransition = onStepTransition;
+    this.#onFinish = onFinish;
+  }
+  setState(state) {
+    this.#state = state;
+  }
+  get runId() {
+    return this.#runId;
+  }
+  get executionSpan() {
+    return this.#executionSpan;
+  }
+  async start({ triggerData } = {}) {
+    const results = await this.execute({ triggerData });
+    if (this.#onFinish) {
+      this.#onFinish();
+    }
+    return {
+      ...results,
+      runId: this.runId
+    };
+  }
+  async execute({
+    triggerData,
+    snapshot,
+    stepId
+  } = {}) {
+    this.#executionSpan = this.#mastra?.telemetry?.tracer.startSpan(`workflow.${this.name}.execute`, {
+      attributes: { componentName: this.name, runId: this.runId }
+    });
+    let machineInput = {
+      // Maintain the original step results and their output
+      steps: {},
+      triggerData: triggerData || {},
+      attempts: Object.keys(this.#steps).reduce(
+        (acc, stepKey) => {
+          acc[stepKey] = this.#steps[stepKey]?.retryConfig?.attempts || this.#retryConfig?.attempts || 3;
+          return acc;
+        },
+        {}
+      )
+    };
+    let stepGraph = this.#stepGraph;
+    let startStepId = "trigger";
+    if (snapshot) {
+      const runState = snapshot;
+      machineInput = runState.context;
+      if (stepId && runState?.suspendedSteps?.[stepId]) {
+        startStepId = runState.suspendedSteps[stepId];
+        stepGraph = this.#stepSubscriberGraph[startStepId] ?? this.#stepGraph;
+      }
+    }
+    const defaultMachine = new Machine({
+      logger: this.logger,
+      mastra: this.#mastra,
+      workflowInstance: this,
+      name: this.name,
+      runId: this.runId,
+      steps: this.#steps,
+      stepGraph,
+      executionSpan: this.#executionSpan,
+      startStepId
+    });
+    this.#machines[startStepId] = defaultMachine;
+    const stateUpdateHandler = (startStepId2, state, context) => {
+      if (startStepId2 === "trigger") {
+        this.#state = state;
+      } else {
+        this.#state = mergeChildValue(startStepId2, this.#state, state);
+      }
+      const now = Date.now();
+      if (this.#onStepTransition) {
+        this.#onStepTransition.forEach((onTransition) => {
+          void onTransition({
+            runId: this.#runId,
+            value: this.#state,
+            context,
+            activePaths: getActivePathsAndStatus(this.#state),
+            timestamp: now
+          });
+        });
+      }
+    };
+    defaultMachine.on("state-update", stateUpdateHandler);
+    const { results } = await defaultMachine.execute({ snapshot, stepId, input: machineInput });
+    await this.persistWorkflowSnapshot();
+    return { results };
+  }
+  async runMachine(parentStepId, input) {
+    if (!this.#stepSubscriberGraph[parentStepId]) {
+      return;
+    }
+    const stateUpdateHandler = (startStepId, state, context) => {
+      if (startStepId === "trigger") {
+        this.#state = state;
+      } else {
+        this.#state = mergeChildValue(startStepId, this.#state, state);
+      }
+      const now = Date.now();
+      if (this.#onStepTransition) {
+        this.#onStepTransition.forEach((onTransition) => {
+          void onTransition({
+            runId: this.#runId,
+            value: this.#state,
+            context,
+            activePaths: getActivePathsAndStatus(this.#state),
+            timestamp: now
+          });
+        });
+      }
+    };
+    const machine = new Machine({
+      logger: this.logger,
+      mastra: this.#mastra,
+      workflowInstance: this,
+      name: parentStepId === "trigger" ? this.name : `${this.name}-${parentStepId}`,
+      runId: this.runId,
+      steps: this.#steps,
+      stepGraph: this.#stepSubscriberGraph[parentStepId],
+      executionSpan: this.#executionSpan,
+      startStepId: parentStepId
+    });
+    machine.on("state-update", stateUpdateHandler);
+    this.#machines[parentStepId] = machine;
+    return await machine.execute({ input });
+  }
+  async suspend(stepId, machine) {
+    this.#suspendedMachines[stepId] = machine;
+  }
+  /**
+   * Persists the workflow state to the database
+   */
+  async persistWorkflowSnapshot() {
+    const existingSnapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
+      workflowName: this.name,
+      runId: this.#runId
+    });
+    const machineSnapshots = {};
+    for (const [stepId, machine] of Object.entries(this.#machines)) {
+      const machineSnapshot = machine?.getSnapshot();
+      if (machineSnapshot) {
+        machineSnapshots[stepId] = { ...machineSnapshot };
+      }
+    }
+    let snapshot = machineSnapshots["trigger"];
+    delete machineSnapshots["trigger"];
+    const suspendedSteps = Object.entries(this.#suspendedMachines).reduce(
+      (acc, [stepId, machine]) => {
+        acc[stepId] = machine.startStepId;
+        return acc;
+      },
+      {}
+    );
+    if (!snapshot && existingSnapshot) {
+      existingSnapshot.childStates = { ...existingSnapshot.childStates, ...machineSnapshots };
+      existingSnapshot.suspendedSteps = { ...existingSnapshot.suspendedSteps, ...suspendedSteps };
+      await this.#mastra?.storage?.persistWorkflowSnapshot({
+        workflowName: this.name,
+        runId: this.#runId,
+        snapshot: existingSnapshot
+      });
+      return;
+    } else if (snapshot && !existingSnapshot) {
+      snapshot.suspendedSteps = suspendedSteps;
+      snapshot.childStates = { ...machineSnapshots };
+      await this.#mastra?.storage?.persistWorkflowSnapshot({
+        workflowName: this.name,
+        runId: this.#runId,
+        snapshot
+      });
+      return;
+    } else if (!snapshot) {
+      this.logger.debug("Snapshot cannot be persisted. No snapshot received.", { runId: this.#runId });
+      return;
+    }
+    snapshot.suspendedSteps = { ...existingSnapshot.suspendedSteps, ...suspendedSteps };
+    if (!existingSnapshot || snapshot === existingSnapshot) {
+      await this.#mastra?.storage?.persistWorkflowSnapshot({
+        workflowName: this.name,
+        runId: this.#runId,
+        snapshot
+      });
+      return;
+    }
+    if (existingSnapshot?.childStates) {
+      snapshot.childStates = { ...existingSnapshot.childStates, ...machineSnapshots };
+    } else {
+      snapshot.childStates = machineSnapshots;
+    }
+    await this.#mastra?.storage?.persistWorkflowSnapshot({
+      workflowName: this.name,
+      runId: this.#runId,
+      snapshot
+    });
+  }
+  async getState() {
+    const storedSnapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
+      workflowName: this.name,
+      runId: this.runId
+    });
+    const prevSnapshot = storedSnapshot ? {
+      trigger: storedSnapshot,
+      ...Object.entries(storedSnapshot?.childStates ?? {}).reduce(
+        (acc, [stepId, snapshot2]) => ({ ...acc, [stepId]: snapshot2 }),
+        {}
+      )
+    } : {};
+    const currentSnapshot = Object.entries(this.#machines).reduce(
+      (acc, [stepId, machine]) => {
+        const snapshot2 = machine.getSnapshot();
+        if (!snapshot2) {
+          return acc;
+        }
+        return {
+          ...acc,
+          [stepId]: snapshot2
+        };
+      },
+      {}
+    );
+    Object.assign(prevSnapshot, currentSnapshot);
+    const trigger = prevSnapshot.trigger;
+    delete prevSnapshot.trigger;
+    const snapshot = { ...trigger};
+    const m = getActivePathsAndStatus(prevSnapshot.value);
+    return {
+      runId: this.runId,
+      value: snapshot.value,
+      context: snapshot.context,
+      activePaths: m,
+      timestamp: Date.now()
+    };
+  }
+};
+
+// src/workflows/workflow.ts
+var Workflow = class extends MastraBase {
+  name;
+  triggerSchema;
+  #retryConfig;
+  #mastra;
+  #runs = /* @__PURE__ */ new Map();
+  // registers stepIds on `after` calls
+  #afterStepStack = [];
+  #lastStepStack = [];
+  #stepGraph = { initial: [] };
+  #stepSubscriberGraph = {};
+  #steps = {};
+  #onStepTransition = /* @__PURE__ */ new Set();
+  /**
+   * Creates a new Workflow instance
+   * @param name - Identifier for the workflow (not necessarily unique)
+   * @param logger - Optional logger instance
+   */
+  constructor({ name, triggerSchema, retryConfig, mastra }) {
+    super({ component: "WORKFLOW", name });
+    this.name = name;
+    this.#retryConfig = retryConfig;
+    this.triggerSchema = triggerSchema;
+    this.#mastra = mastra;
+    if (mastra?.logger) {
+      this.logger = mastra?.logger;
+    }
+  }
+  step(step, config) {
+    const { variables = {} } = config || {};
+    const requiredData = {};
+    for (const [key, variable] of Object.entries(variables)) {
+      if (variable && isVariableReference(variable)) {
+        requiredData[key] = variable;
+      }
+    }
+    const stepKey = this.#makeStepKey(step);
+    const graphEntry = {
+      step,
+      config: {
+        ...this.#makeStepDef(stepKey),
+        ...config,
+        data: requiredData
+      }
+    };
+    this.#steps[stepKey] = step;
+    const parentStepKey = this.#afterStepStack[this.#afterStepStack.length - 1];
+    const stepGraph = this.#stepSubscriberGraph[parentStepKey || ""];
+    if (parentStepKey && stepGraph) {
+      if (!stepGraph.initial.some((step2) => step2.step.id === stepKey)) {
+        stepGraph.initial.push(graphEntry);
+      }
+      stepGraph[stepKey] = [];
+    } else {
+      if (!this.#stepGraph[stepKey]) this.#stepGraph[stepKey] = [];
+      this.#stepGraph.initial.push(graphEntry);
+    }
+    this.#lastStepStack.push(stepKey);
+    return this;
+  }
+  #makeStepKey(step) {
+    return `${step.id}`;
+  }
+  then(step, config) {
+    const { variables = {} } = config || {};
+    const requiredData = {};
+    for (const [key, variable] of Object.entries(variables)) {
+      if (variable && isVariableReference(variable)) {
+        requiredData[key] = variable;
+      }
+    }
+    const lastStepKey = this.#lastStepStack[this.#lastStepStack.length - 1];
+    const stepKey = this.#makeStepKey(step);
+    const graphEntry = {
+      step,
+      config: {
+        ...this.#makeStepDef(stepKey),
+        ...config,
+        data: requiredData
+      }
+    };
+    this.#steps[stepKey] = step;
+    if (!lastStepKey) return this;
+    const parentStepKey = this.#afterStepStack[this.#afterStepStack.length - 1];
+    const stepGraph = this.#stepSubscriberGraph[parentStepKey || ""];
+    if (parentStepKey && stepGraph && stepGraph[lastStepKey]) {
+      stepGraph[lastStepKey].push(graphEntry);
+    } else {
+      if (!this.#stepGraph[lastStepKey]) this.#stepGraph[lastStepKey] = [];
+      this.#stepGraph[lastStepKey].push(graphEntry);
+    }
+    return this;
+  }
+  after(step) {
+    const stepKey = this.#makeStepKey(step);
+    this.#afterStepStack.push(stepKey);
+    if (!this.#stepSubscriberGraph[stepKey]) {
+      this.#stepSubscriberGraph[stepKey] = { initial: [] };
+    }
+    return this;
+  }
+  /**
+   * Executes the workflow with the given trigger data
+   * @param triggerData - Initial data to start the workflow with
+   * @returns Promise resolving to workflow results or rejecting with error
+   * @throws Error if trigger schema validation fails
+   */
+  createRun() {
+    const run = new WorkflowInstance({
+      logger: this.logger,
+      name: this.name,
+      mastra: this.#mastra,
+      retryConfig: this.#retryConfig,
+      steps: this.#steps,
+      stepGraph: this.#stepGraph,
+      stepSubscriberGraph: this.#stepSubscriberGraph,
+      onStepTransition: this.#onStepTransition,
+      onFinish: () => {
+        this.#runs.delete(run.runId);
+      }
+    });
+    this.#runs.set(run.runId, run);
+    return {
+      start: run.start.bind(run),
+      runId: run.runId
+    };
+  }
+  /**
+   * Rebuilds the machine with the current steps configuration and validates the workflow
+   *
+   * This is the last step of a workflow builder method chain
+   * @throws Error if validation fails
+   *
+   * @returns this instance for method chaining
+   */
+  commit() {
+    return this;
+  }
+  // record all object paths that leads to a suspended state
+  #getSuspendedPaths({
+    value,
+    path,
+    suspendedPaths
+  }) {
+    if (typeof value === "string") {
+      if (value === "suspended") {
+        suspendedPaths.add(path);
+      }
+    } else {
+      Object.keys(value).forEach(
+        (key) => this.#getSuspendedPaths({ value: value[key], path: path ? `${path}.${key}` : key, suspendedPaths })
+      );
+    }
+  }
+  async #loadWorkflowSnapshot(runId) {
+    if (!this.#mastra?.storage) {
+      this.logger.debug("Snapshot cannot be loaded. Mastra engine is not initialized", { runId });
+      return;
+    }
+    const activeRun = this.#runs.get(runId);
+    if (activeRun) {
+      await activeRun.persistWorkflowSnapshot();
+    }
+    return this.#mastra.storage.loadWorkflowSnapshot({ runId, workflowName: this.name });
+  }
+  getExecutionSpan(runId) {
+    return this.#runs.get(runId)?.executionSpan;
+  }
+  #makeStepDef(stepId) {
+    const executeStep = (handler2, spanName, attributes) => {
+      return async (data) => {
+        return await context.with(
+          trace.setSpan(context.active(), this.getExecutionSpan(attributes?.runId ?? data?.runId)),
+          async () => {
+            return this.#mastra.telemetry.traceMethod(handler2, {
+              spanName,
+              attributes
+            })(data);
+          }
+        );
+      };
+    };
+    const handler = async ({ context, ...rest }) => {
+      const targetStep = this.#steps[stepId];
+      if (!targetStep) throw new Error(`Step not found`);
+      const { payload = {}, execute = async () => {
+      } } = targetStep;
+      const mergedData = {
+        ...payload,
+        ...context
+      };
+      const finalAction = this.#mastra?.telemetry ? executeStep(execute, `workflow.${this.name}.action.${stepId}`, {
+        componentName: this.name,
+        runId: rest.runId
+      }) : execute;
+      return finalAction ? await finalAction({ context: mergedData, ...rest }) : {};
+    };
+    const finalHandler = ({ context, ...rest }) => {
+      if (this.getExecutionSpan(rest?.runId)) {
+        return executeStep(handler, `workflow.${this.name}.step.${stepId}`, {
+          componentName: this.name,
+          runId: rest?.runId
+        })({ context, ...rest });
+      }
+      return handler({ context, ...rest });
+    };
+    return {
+      handler: finalHandler,
+      data: {}
+    };
+  }
+  #getActivePathsAndStatus(value) {
+    const paths = [];
+    const traverse = (current, path = []) => {
+      for (const [key, value2] of Object.entries(current)) {
+        const currentPath = [...path, key];
+        if (typeof value2 === "string") {
+          paths.push({
+            stepPath: currentPath,
+            stepId: key,
+            status: value2
+          });
+        } else if (typeof value2 === "object" && value2 !== null) {
+          traverse(value2, currentPath);
+        }
+      }
+    };
+    traverse(value);
+    return paths;
+  }
+  async getState(runId) {
+    const run = this.#runs.get(runId);
+    if (run) {
+      return run.getState();
+    }
+    const storedSnapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
+      runId,
+      workflowName: this.name
+    });
+    if (storedSnapshot) {
+      const parsed = storedSnapshot;
+      const m = this.#getActivePathsAndStatus(parsed.value);
+      return {
+        runId,
+        value: parsed.value,
+        context: parsed.context,
+        activePaths: m,
+        timestamp: Date.now()
+      };
+    }
+    return null;
+  }
+  watch(onTransition) {
+    this.#onStepTransition.add(onTransition);
+    return () => {
+      this.#onStepTransition.delete(onTransition);
+    };
+  }
+  async resume({
+    runId,
+    stepId,
+    context: resumeContext
+  }) {
+    await setTimeout(0);
+    return this._resume({ runId, stepId, context: resumeContext });
+  }
+  async _resume({
+    runId,
+    stepId,
+    context: resumeContext
+  }) {
+    const snapshot = await this.#loadWorkflowSnapshot(runId);
+    if (!snapshot) {
+      throw new Error(`No snapshot found for workflow run ${runId}`);
+    }
+    let parsedSnapshot;
+    try {
+      parsedSnapshot = typeof snapshot === "string" ? JSON.parse(snapshot) : snapshot;
+    } catch (error) {
+      this.logger.debug("Failed to parse workflow snapshot for resume", { error, runId });
+      throw new Error("Failed to parse workflow snapshot");
+    }
+    const origSnapshot = parsedSnapshot;
+    const startStepId = parsedSnapshot.suspendedSteps?.[stepId];
+    if (!startStepId) {
+      return;
+    }
+    parsedSnapshot = startStepId === "trigger" ? parsedSnapshot : { ...parsedSnapshot?.childStates?.[startStepId], ...{ suspendedSteps: parsedSnapshot.suspendedSteps } };
+    if (!parsedSnapshot) {
+      throw new Error(`No snapshot found for step: ${stepId} starting at ${startStepId}`);
+    }
+    if (resumeContext) {
+      parsedSnapshot.context.steps[stepId] = {
+        status: "success",
+        output: {
+          ...parsedSnapshot?.context?.steps?.[stepId]?.output || {},
+          ...resumeContext
+        }
+      };
+    }
+    if (parsedSnapshot.children) {
+      Object.entries(parsedSnapshot.children).forEach(([_childId, child]) => {
+        if (child.snapshot?.input?.stepNode) {
+          const stepDef = this.#makeStepDef(child.snapshot.input.stepNode.step.id);
+          child.snapshot.input.stepNode.config = {
+            ...child.snapshot.input.stepNode.config,
+            ...stepDef
+          };
+          child.snapshot.input.context = parsedSnapshot.context;
+        }
+      });
+    }
+    parsedSnapshot.value = updateStepInHierarchy(parsedSnapshot.value, stepId);
+    if (parsedSnapshot.context?.attempts) {
+      parsedSnapshot.context.attempts[stepId] = this.#steps[stepId]?.retryConfig?.attempts || this.#retryConfig?.attempts || 3;
+    }
+    this.logger.debug("Resuming workflow with updated snapshot", {
+      updatedSnapshot: parsedSnapshot,
+      runId,
+      stepId
+    });
+    const run = this.#runs.get(runId) ?? new WorkflowInstance({
+      logger: this.logger,
+      name: this.name,
+      mastra: this.#mastra,
+      retryConfig: this.#retryConfig,
+      steps: this.#steps,
+      stepGraph: this.#stepGraph,
+      stepSubscriberGraph: this.#stepSubscriberGraph,
+      onStepTransition: this.#onStepTransition,
+      runId,
+      onFinish: () => {
+        this.#runs.delete(run.runId);
+      }
+    });
+    run.setState(origSnapshot?.value);
+    this.#runs.set(run.runId, run);
+    return run?.execute({
+      snapshot: parsedSnapshot,
+      stepId
+    });
+  }
+  __registerPrimitives(p) {
+    if (p.telemetry) {
+      this.__setTelemetry(p.telemetry);
+    }
+    if (p.logger) {
+      this.__setLogger(p.logger);
+    }
+    this.#mastra = p;
+  }
+  get stepGraph() {
+    return this.#stepGraph;
+  }
+  get stepSubscriberGraph() {
+    return this.#stepSubscriberGraph;
+  }
+  get steps() {
+    return this.#steps;
+  }
+};
+
+// src/workflows/step.ts
+var Step = class {
+  id;
+  description;
+  inputSchema;
+  outputSchema;
+  payload;
+  execute;
+  retryConfig;
+  mastra;
+  constructor({
+    id,
+    description,
+    execute,
+    payload,
+    outputSchema,
+    inputSchema,
+    retryConfig
+  }) {
+    this.id = id;
+    this.description = description ?? "";
+    this.inputSchema = inputSchema;
+    this.payload = payload;
+    this.outputSchema = outputSchema;
+    this.execute = execute;
+    this.retryConfig = retryConfig;
+  }
+};
+function createStep(opts) {
+  return new Step(opts);
+}
+
+export { Step, Workflow, createStep, getActivePathsAndStatus, getStepResult, getSuspendedPaths, isErrorEvent, isFinalState, isTransitionEvent, isVariableReference, mergeChildValue, recursivelyCheckForFinalState, updateStepInHierarchy };