@mastra/core 0.24.6 → 0.24.7-alpha.1

package/dist/{chunk-PIH5FBNQ.cjs → chunk-U57JN7JY.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunkXDMQQZNX_cjs = require('./chunk-XDMQQZNX.cjs');
+var chunkAHQOJZT2_cjs = require('./chunk-AHQOJZT2.cjs');
 var chunkABJOUEVA_cjs = require('./chunk-ABJOUEVA.cjs');
 var chunkGPWMM745_cjs = require('./chunk-GPWMM745.cjs');
 var chunk5NTO7S5I_cjs = require('./chunk-5NTO7S5I.cjs');
@@ -31,7 +31,7 @@ var StepExecutor = class extends chunk6VOPKVYH_cjs.MastraBase {
     let suspended;
     let bailed;
     const startedAt = Date.now();
-    const { inputData, validationError } = await chunkXDMQQZNX_cjs.validateStepInput({
+    const { inputData, validationError } = await chunkAHQOJZT2_cjs.validateStepInput({
       prevOutput: typeof params.foreachIdx === "number" ? params.input?.[params.foreachIdx] : params.input,
       step,
       validateInputs: params.validateInputs ?? false
@@ -63,7 +63,7 @@ var StepExecutor = class extends chunk6VOPKVYH_cjs.MastraBase {
         runCount,
         resumeData: params.resumeData,
         getInitData: () => stepResults?.input,
-        getStepResult: chunkXDMQQZNX_cjs.getStepResult.bind(this, stepResults),
+        getStepResult: chunkAHQOJZT2_cjs.getStepResult.bind(this, stepResults),
         suspend: async (suspendPayload) => {
           suspended = { payload: { ...suspendPayload, __workflow_meta: { runId, path: [step.id] } } };
         },
@@ -183,7 +183,7 @@ var StepExecutor = class extends chunk6VOPKVYH_cjs.MastraBase {
       runCount,
       resumeData,
       getInitData: () => stepResults?.input,
-      getStepResult: chunkXDMQQZNX_cjs.getStepResult.bind(this, stepResults),
+      getStepResult: chunkAHQOJZT2_cjs.getStepResult.bind(this, stepResults),
       suspend: async (_suspendPayload) => {
         throw new Error("Not implemented");
       },
@@ -230,7 +230,7 @@ var StepExecutor = class extends chunk6VOPKVYH_cjs.MastraBase {
         runCount,
         resumeData: params.resumeData,
         getInitData: () => stepResults?.input,
-        getStepResult: chunkXDMQQZNX_cjs.getStepResult.bind(this, stepResults),
+        getStepResult: chunkAHQOJZT2_cjs.getStepResult.bind(this, stepResults),
         suspend: async (_suspendPayload) => {
           throw new Error("Not implemented");
         },
@@ -280,7 +280,7 @@ var StepExecutor = class extends chunk6VOPKVYH_cjs.MastraBase {
         runCount,
         resumeData: params.resumeData,
         getInitData: () => stepResults?.input,
-        getStepResult: chunkXDMQQZNX_cjs.getStepResult.bind(this, stepResults),
+        getStepResult: chunkAHQOJZT2_cjs.getStepResult.bind(this, stepResults),
         suspend: async (_suspendPayload) => {
           throw new Error("Not implemented");
         },
@@ -439,6 +439,7 @@ async function processWorkflowForEach({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   resumeData,
   parentWorkflow,
   runtimeContext
@@ -461,6 +462,7 @@ async function processWorkflowForEach({
         executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
         resumeSteps,
         stepResults,
+        timeTravel,
         prevResult: currentResult,
         resumeData,
         activeSteps,
@@ -497,6 +499,7 @@ async function processWorkflowForEach({
           executionPath: [executionPath[0], i],
           resumeSteps,
           stepResults,
+          timeTravel,
           prevResult,
           resumeData,
           activeSteps,
@@ -528,6 +531,7 @@ async function processWorkflowForEach({
       runId,
       executionPath: [executionPath[0], idx],
       resumeSteps,
+      timeTravel,
       stepResults,
       prevResult,
       resumeData,
@@ -543,6 +547,7 @@ async function processWorkflowParallel({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -570,6 +575,7 @@ async function processWorkflowParallel({
           stepResults,
           prevResult,
           resumeData,
+          timeTravel,
           parentWorkflow,
           activeSteps,
           runtimeContext
@@ -585,6 +591,7 @@ async function processWorkflowConditional({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -627,6 +634,7 @@ async function processWorkflowConditional({
             executionPath: executionPath.concat([idx]),
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             resumeData,
             parentWorkflow,
@@ -693,6 +701,7 @@ async function processWorkflowSleep({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -764,6 +773,7 @@ async function processWorkflowSleep({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -783,6 +793,7 @@ async function processWorkflowSleepUntil({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -854,6 +865,7 @@ async function processWorkflowSleepUntil({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -934,6 +946,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     resumeSteps,
     prevResult,
     resumeData,
+    timeTravel,
     executionPath,
     stepResults,
     runtimeContext
@@ -946,13 +959,14 @@ var WorkflowEventProcessor = class extends EventProcessor {
         suspendedPaths: {},
         resumeLabels: {},
         waitingPaths: {},
+        activeStepsPath: {},
         serializedStepGraph: workflow.serializedStepGraph,
         timestamp: Date.now(),
         runId,
-        status: "running",
         context: stepResults ?? {
           input: prevResult?.status === "success" ? prevResult.output : void 0
         },
+        status: "running",
         value: {}
       }
     });
@@ -969,6 +983,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           input: prevResult?.status === "success" ? prevResult.output : void 0
         },
         prevResult,
+        timeTravel,
         runtimeContext,
         resumeData,
         activeSteps: {}
@@ -1019,7 +1034,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     });
   }
   async processWorkflowEnd(args) {
-    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runtimeContext, runId } = args;
+    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runtimeContext, runId, timeTravel } = args;
     if (parentWorkflow) {
       await this.mastra.pubsub.publish("workflows", {
         type: "workflow.step.end",
@@ -1035,7 +1050,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           activeSteps,
           parentWorkflow: parentWorkflow.parentWorkflow,
           parentContext: parentWorkflow,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     }
@@ -1046,7 +1062,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     });
   }
   async processWorkflowSuspend(args) {
-    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runId, runtimeContext } = args;
+    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runId, runtimeContext, timeTravel } = args;
     if (parentWorkflow) {
       await this.mastra.pubsub.publish("workflows", {
         type: "workflow.step.end",
@@ -1067,6 +1083,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
               }
             }
           },
+          timeTravel,
           resumeData,
           activeSteps,
           runtimeContext,
@@ -1082,7 +1099,17 @@ var WorkflowEventProcessor = class extends EventProcessor {
     });
   }
   async processWorkflowFail(args) {
-    const { workflowId, runId, resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runtimeContext } = args;
+    const {
+      workflowId,
+      runId,
+      resumeSteps,
+      prevResult,
+      resumeData,
+      parentWorkflow,
+      activeSteps,
+      runtimeContext,
+      timeTravel
+    } = args;
     await this.mastra.getStorage()?.updateWorkflowState({
       workflowName: workflowId,
       runId,
@@ -1102,6 +1129,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           stepResults: parentWorkflow.stepResults,
           prevResult,
+          timeTravel,
           resumeData,
           activeSteps,
           runtimeContext,
@@ -1124,6 +1152,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     stepResults,
     activeSteps,
     resumeSteps,
+    timeTravel,
     prevResult,
     resumeData,
     parentWorkflow,
@@ -1187,6 +1216,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1206,6 +1236,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1226,6 +1257,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1246,6 +1278,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1302,6 +1335,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1394,6 +1428,47 @@ var WorkflowEventProcessor = class extends EventProcessor {
             runtimeContext
           }
         });
+      } else if (timeTravel && timeTravel.steps?.length > 1 && timeTravel.steps[0] === step.step.id) {
+        const snapshot = await this.mastra?.getStorage()?.loadWorkflowSnapshot({
+          workflowName: step.step.id,
+          runId
+        }) ?? { context: {} };
+        const timeTravelParams = chunkAHQOJZT2_cjs.createTimeTravelExecutionParams({
+          steps: timeTravel.steps.slice(1),
+          inputData: timeTravel.inputData,
+          resumeData: timeTravel.resumeData,
+          context: timeTravel.nestedStepResults?.[step.step.id] ?? {},
+          nestedStepsContext: timeTravel.nestedStepResults ?? {},
+          snapshot,
+          graph: step.step.buildExecutionGraph()
+        });
+        const nestedPrevStep = getStep(step.step, timeTravelParams.executionPath);
+        const nestedPrevResult = timeTravelParams.stepResults[nestedPrevStep?.id ?? "input"];
+        await this.mastra.pubsub.publish("workflows", {
+          type: "workflow.start",
+          runId,
+          data: {
+            workflowId: step.step.id,
+            parentWorkflow: {
+              stepId: step.step.id,
+              workflowId,
+              runId,
+              executionPath,
+              resumeSteps,
+              stepResults,
+              timeTravel,
+              input: prevResult,
+              parentWorkflow
+            },
+            executionPath: timeTravelParams.executionPath,
+            runId: crypto.randomUUID(),
+            stepResults: timeTravelParams.stepResults,
+            prevResult: { status: "success", output: nestedPrevResult?.payload },
+            timeTravel: timeTravelParams,
+            activeSteps,
+            runtimeContext
+          }
+        });
       } else {
         await this.mastra.pubsub.publish("workflows", {
           type: "workflow.start",
@@ -1466,6 +1541,21 @@ var WorkflowEventProcessor = class extends EventProcessor {
     for (const [key, value] of Object.entries(runtimeContext)) {
       rc.set(key, value);
     }
+    const { resumeData: timeTravelResumeData, validationError: timeTravelResumeValidationError } = await chunkAHQOJZT2_cjs.validateStepResumeData({
+      resumeData: timeTravel?.stepResults[step.step.id]?.status === "suspended" ? timeTravel?.resumeData : void 0,
+      step: step.step
+    });
+    let resumeDataToUse;
+    if (timeTravelResumeData && !timeTravelResumeValidationError) {
+      resumeDataToUse = timeTravelResumeData;
+    } else if (timeTravelResumeData && timeTravelResumeValidationError) {
+      this.mastra.getLogger()?.warn("Time travel resume data validation failed", {
+        stepId: step.step.id,
+        error: timeTravelResumeValidationError.message
+      });
+    } else if (resumeSteps?.length > 0 && resumeSteps?.[0] === step.step.id || step.type === "waitForEvent") {
+      resumeDataToUse = resumeData;
+    }
     const stepResult = await this.stepExecutor.execute({
       workflowId,
       step: step.step,
@@ -1476,7 +1566,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
       emitter: ee,
       runtimeContext: rc,
       input: prevResult?.output,
-      resumeData: step.type === "waitForEvent" || resumeSteps?.length === 1 && resumeSteps?.[0] === step.step.id ? resumeData : void 0,
+      resumeData: resumeDataToUse,
       runCount,
       foreachIdx: step.type === "foreach" ? executionPath[1] : void 0,
       validateInputs: workflow.options.validateInputs
@@ -1530,6 +1620,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
             executionPath,
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             activeSteps,
             runtimeContext,
@@ -1570,6 +1661,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           runId,
           executionPath,
           resumeSteps,
+          timeTravel,
+          //timeTravel is passed in as workflow.step.end ends the step, not the workflow, the timeTravel info is passed to the next step to run.
           stepResults,
           prevResult: stepResult,
           activeSteps,
@@ -1584,6 +1677,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     runId,
     executionPath,
     resumeSteps,
+    timeTravel,
     prevResult,
     parentWorkflow,
     stepResults,
@@ -1673,6 +1767,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           parentWorkflow,
           stepResults,
+          timeTravel,
           prevResult,
           activeSteps,
           runtimeContext
@@ -1706,7 +1801,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
       await this.mastra.pubsub.publish(`workflow.events.${runId}`, {
@@ -1815,7 +1911,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { status: "success", output: allResults },
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     } else if (step?.type === "foreach") {
@@ -1831,7 +1928,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { ...prevResult, output: prevResult?.payload },
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     } else if (executionPath[0] >= workflow.stepGraph.length - 1) {
@@ -1860,7 +1958,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     }
@@ -1991,7 +2090,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
 
 // src/workflows/evented/workflow.ts
 function cloneWorkflow(workflow, opts) {
-  const wf = new chunkXDMQQZNX_cjs.Workflow({
+  const wf = new chunkAHQOJZT2_cjs.Workflow({
     id: opts.id,
     inputSchema: workflow.inputSchema,
     outputSchema: workflow.outputSchema,
@@ -2132,9 +2231,10 @@ function createWorkflow(params) {
     executionEngine
   });
 }
-var EventedWorkflow = class extends chunkXDMQQZNX_cjs.Workflow {
+var EventedWorkflow = class extends chunkAHQOJZT2_cjs.Workflow {
   constructor(params) {
     super(params);
+    this.engineType = "evented";
   }
   __registerMastra(mastra) {
     super.__registerMastra(mastra);
@@ -2151,7 +2251,9 @@ var EventedWorkflow = class extends chunkXDMQQZNX_cjs.Workflow {
       mastra: this.mastra,
       retryConfig: this.retryConfig,
       cleanup: () => this.runs.delete(runIdToUse),
-      workflowSteps: this.steps
+      workflowSteps: this.steps,
+      validateInputs: this.options?.validateInputs,
+      workflowEngineType: this.engineType
     });
     this.runs.set(runIdToUse, run);
     const shouldPersistSnapshot = this.options?.shouldPersistSnapshot?.({
@@ -2170,6 +2272,7 @@ var EventedWorkflow = class extends chunkXDMQQZNX_cjs.Workflow {
           context: {},
           activePaths: [],
           serializedStepGraph: this.serializedStepGraph,
+          activeStepsPath: {},
           suspendedPaths: {},
           resumeLabels: {},
           waitingPaths: {},
@@ -2183,7 +2286,7 @@ var EventedWorkflow = class extends chunkXDMQQZNX_cjs.Workflow {
     return run;
   }
 };
-var EventedRun = class extends chunkXDMQQZNX_cjs.Run {
+var EventedRun = class extends chunkAHQOJZT2_cjs.Run {
   constructor(params) {
     super(params);
     this.serializedStepGraph = params.serializedStepGraph;
@@ -2208,15 +2311,16 @@ var EventedRun = class extends chunkXDMQQZNX_cjs.Run {
       snapshot: {
         runId: this.runId,
         serializedStepGraph: this.serializedStepGraph,
+        status: "running",
         value: {},
         context: {},
         runtimeContext: Object.fromEntries(runtimeContext.entries()),
         activePaths: [],
+        activeStepsPath: {},
         suspendedPaths: {},
         resumeLabels: {},
         waitingPaths: {},
-        timestamp: Date.now(),
-        status: "running"
+        timestamp: Date.now()
       }
     });
     const inputDataToUse = await this._validateInput(inputData);
@@ -2253,9 +2357,14 @@ var EventedRun = class extends chunkXDMQQZNX_cjs.Run {
   }
   // TODO: stream
   async resume(params) {
-    const steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
-      (step) => typeof step === "string" ? step : step?.id
-    );
+    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 resume");
     }
@@ -2441,7 +2550,7 @@ function isExecutableStep(step) {
 }
 
 // src/workflows/evented/execution-engine.ts
-var EventedExecutionEngine = class extends chunkXDMQQZNX_cjs.ExecutionEngine {
+var EventedExecutionEngine = class extends chunkAHQOJZT2_cjs.ExecutionEngine {
   eventProcessor;
   constructor({
     mastra,
@@ -2484,6 +2593,23 @@ var EventedExecutionEngine = class extends chunkXDMQQZNX_cjs.ExecutionEngine {
           format: params.format
         }
       });
+    } else if (params.timeTravel) {
+      const prevStep = getStep(this.mastra.getWorkflow(params.workflowId), params.timeTravel.executionPath);
+      const prevResult = params.timeTravel.stepResults[prevStep?.id ?? "input"];
+      await pubsub.publish("workflows", {
+        type: "workflow.start",
+        runId: params.runId,
+        data: {
+          workflowId: params.workflowId,
+          runId: params.runId,
+          executionPath: params.timeTravel.executionPath,
+          stepResults: params.timeTravel.stepResults,
+          timeTravel: params.timeTravel,
+          prevResult: { status: "success", output: prevResult?.payload },
+          runtimeContext: Object.fromEntries(params.runtimeContext.entries()),
+          format: params.format
+        }
+      });
     } else {
       await pubsub.publish("workflows", {
         type: "workflow.start",
@@ -2550,5 +2676,5 @@ exports.cloneStep = cloneStep;
 exports.cloneWorkflow = cloneWorkflow;
 exports.createStep = createStep;
 exports.createWorkflow = createWorkflow;
-//# sourceMappingURL=chunk-PIH5FBNQ.cjs.map
-//# sourceMappingURL=chunk-PIH5FBNQ.cjs.map
+//# sourceMappingURL=chunk-U57JN7JY.cjs.map
+//# sourceMappingURL=chunk-U57JN7JY.cjs.map