@mastra/core 0.24.7-alpha.0 → 0.24.7-alpha.2

package/dist/{chunk-H6SCQEQZ.js → chunk-2BY7C6Y7.js}

@@ -1,4 +1,4 @@
-import { validateStepInput, getStepResult, Workflow, Run, ExecutionEngine } from './chunk-BMJ6HNP4.js';
+import { validateStepInput, getStepResult, createTimeTravelExecutionParams, validateStepResumeData, Workflow, Run, ExecutionEngine } from './chunk-Y2BH2LUD.js';
 import { STREAM_FORMAT_SYMBOL, EMITTER_SYMBOL } from './chunk-NLNKQD2T.js';
 import { RuntimeContext } from './chunk-HLRWYUFN.js';
 import { MastraError } from './chunk-PZUZNPFM.js';
@@ -432,6 +432,7 @@ async function processWorkflowForEach({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   resumeData,
   parentWorkflow,
   runtimeContext
@@ -454,6 +455,7 @@ async function processWorkflowForEach({
         executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
         resumeSteps,
         stepResults,
+        timeTravel,
         prevResult: currentResult,
         resumeData,
         activeSteps,
@@ -490,6 +492,7 @@ async function processWorkflowForEach({
           executionPath: [executionPath[0], i],
           resumeSteps,
           stepResults,
+          timeTravel,
           prevResult,
           resumeData,
           activeSteps,
@@ -521,6 +524,7 @@ async function processWorkflowForEach({
       runId,
       executionPath: [executionPath[0], idx],
       resumeSteps,
+      timeTravel,
       stepResults,
       prevResult,
       resumeData,
@@ -536,6 +540,7 @@ async function processWorkflowParallel({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -563,6 +568,7 @@ async function processWorkflowParallel({
           stepResults,
           prevResult,
           resumeData,
+          timeTravel,
           parentWorkflow,
           activeSteps,
           runtimeContext
@@ -578,6 +584,7 @@ async function processWorkflowConditional({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -620,6 +627,7 @@ async function processWorkflowConditional({
             executionPath: executionPath.concat([idx]),
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             resumeData,
             parentWorkflow,
@@ -686,6 +694,7 @@ async function processWorkflowSleep({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -757,6 +766,7 @@ async function processWorkflowSleep({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -776,6 +786,7 @@ async function processWorkflowSleepUntil({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -847,6 +858,7 @@ async function processWorkflowSleepUntil({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -927,6 +939,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     resumeSteps,
     prevResult,
     resumeData,
+    timeTravel,
     executionPath,
     stepResults,
     runtimeContext
@@ -939,13 +952,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: {}
       }
     });
@@ -962,6 +976,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           input: prevResult?.status === "success" ? prevResult.output : void 0
         },
         prevResult,
+        timeTravel,
         runtimeContext,
         resumeData,
         activeSteps: {}
@@ -1012,7 +1027,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",
@@ -1028,7 +1043,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           activeSteps,
           parentWorkflow: parentWorkflow.parentWorkflow,
           parentContext: parentWorkflow,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     }
@@ -1039,7 +1055,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",
@@ -1060,6 +1076,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
               }
             }
           },
+          timeTravel,
           resumeData,
           activeSteps,
           runtimeContext,
@@ -1075,7 +1092,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,
@@ -1095,6 +1122,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           stepResults: parentWorkflow.stepResults,
           prevResult,
+          timeTravel,
           resumeData,
           activeSteps,
           runtimeContext,
@@ -1117,6 +1145,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     stepResults,
     activeSteps,
     resumeSteps,
+    timeTravel,
     prevResult,
     resumeData,
     parentWorkflow,
@@ -1180,6 +1209,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1199,6 +1229,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1219,6 +1250,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1239,6 +1271,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1295,6 +1328,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1387,6 +1421,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 = 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: randomUUID(),
+            stepResults: timeTravelParams.stepResults,
+            prevResult: { status: "success", output: nestedPrevResult?.payload },
+            timeTravel: timeTravelParams,
+            activeSteps,
+            runtimeContext
+          }
+        });
       } else {
         await this.mastra.pubsub.publish("workflows", {
           type: "workflow.start",
@@ -1459,6 +1534,21 @@ var WorkflowEventProcessor = class extends EventProcessor {
     for (const [key, value] of Object.entries(runtimeContext)) {
       rc.set(key, value);
     }
+    const { resumeData: timeTravelResumeData, validationError: timeTravelResumeValidationError } = await 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,
@@ -1469,7 +1559,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
@@ -1523,6 +1613,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
             executionPath,
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             activeSteps,
             runtimeContext,
@@ -1563,6 +1654,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,
@@ -1577,6 +1670,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     runId,
     executionPath,
     resumeSteps,
+    timeTravel,
     prevResult,
     parentWorkflow,
     stepResults,
@@ -1666,6 +1760,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           parentWorkflow,
           stepResults,
+          timeTravel,
           prevResult,
           activeSteps,
           runtimeContext
@@ -1699,7 +1794,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
       await this.mastra.pubsub.publish(`workflow.events.${runId}`, {
@@ -1808,7 +1904,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { status: "success", output: allResults },
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     } else if (step?.type === "foreach") {
@@ -1824,7 +1921,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { ...prevResult, output: prevResult?.payload },
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     } else if (executionPath[0] >= workflow.stepGraph.length - 1) {
@@ -1853,7 +1951,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          runtimeContext
+          runtimeContext,
+          timeTravel
         }
       });
     }
@@ -2128,6 +2227,7 @@ function createWorkflow(params) {
 var EventedWorkflow = class extends Workflow {
   constructor(params) {
     super(params);
+    this.engineType = "evented";
   }
   __registerMastra(mastra) {
     super.__registerMastra(mastra);
@@ -2144,7 +2244,9 @@ var EventedWorkflow = class extends 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?.({
@@ -2163,6 +2265,7 @@ var EventedWorkflow = class extends Workflow {
           context: {},
           activePaths: [],
           serializedStepGraph: this.serializedStepGraph,
+          activeStepsPath: {},
           suspendedPaths: {},
           resumeLabels: {},
           waitingPaths: {},
@@ -2201,15 +2304,16 @@ var EventedRun = class extends 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);
@@ -2246,9 +2350,14 @@ var EventedRun = class extends 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");
     }
@@ -2477,6 +2586,23 @@ var EventedExecutionEngine = class extends 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",
@@ -2535,5 +2661,5 @@ var EventedExecutionEngine = class extends ExecutionEngine {
 };
 
 export { EventedExecutionEngine, EventedRun, EventedWorkflow, StepExecutor, WorkflowEventProcessor, cloneStep, cloneWorkflow, createStep, createWorkflow };
-//# sourceMappingURL=chunk-H6SCQEQZ.js.map
-//# sourceMappingURL=chunk-H6SCQEQZ.js.map
+//# sourceMappingURL=chunk-2BY7C6Y7.js.map
+//# sourceMappingURL=chunk-2BY7C6Y7.js.map