@mastra/core 1.0.0-beta.1 → 1.0.0-beta.3

package/dist/{chunk-H6CZGPZD.js → chunk-4DWZ4Z6H.js}

@@ -1,7 +1,7 @@
-import { validateStepInput, createDeprecationProxy, getStepResult, runCountDeprecationMessage, Workflow, Run, ExecutionEngine } from './chunk-JUBFO5J3.js';
+import { validateStepInput, createDeprecationProxy, getStepResult, runCountDeprecationMessage, createTimeTravelExecutionParams, validateStepResumeData, Workflow, Run, ExecutionEngine } from './chunk-CKGIPST2.js';
 import { STREAM_FORMAT_SYMBOL, EMITTER_SYMBOL } from './chunk-NLNKQD2T.js';
 import { getErrorFromUnknown, MastraError } from './chunk-JJ5O45LH.js';
-import { Tool } from './chunk-7AHYOMHJ.js';
+import { Tool } from './chunk-KEURQGCQ.js';
 import { RequestContext } from './chunk-GRBGQ2GE.js';
 import { MastraBase } from './chunk-S6URFGCZ.js';
 import { RegisteredLogger } from './chunk-KJ2SW6VA.js';
@@ -27,7 +27,7 @@ var StepExecutor = class extends MastraBase {
     const { inputData, validationError } = await validateStepInput({
       prevOutput: typeof params.foreachIdx === "number" ? params.input?.[params.foreachIdx] : params.input,
       step,
-      validateInputs: params.validateInputs ?? false
+      validateInputs: params.validateInputs ?? true
     });
     let stepInfo = {
       ...stepResults[step.id],
@@ -92,7 +92,8 @@ var StepExecutor = class extends MastraBase {
         finalResult = {
           ...stepInfo,
           status: "suspended",
-          suspendedAt: endedAt
+          suspendedAt: endedAt,
+          ...stepResult ? { suspendOutput: stepResult } : {}
         };
         if (suspended.payload) {
           finalResult.suspendPayload = suspended.payload;
@@ -472,6 +473,7 @@ async function processWorkflowForEach({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   resumeData,
   parentWorkflow,
   requestContext
@@ -494,6 +496,7 @@ async function processWorkflowForEach({
         executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
         resumeSteps,
         stepResults,
+        timeTravel,
         prevResult: currentResult,
         resumeData,
         activeSteps,
@@ -530,6 +533,7 @@ async function processWorkflowForEach({
           executionPath: [executionPath[0], i],
           resumeSteps,
           stepResults,
+          timeTravel,
           prevResult,
           resumeData,
           activeSteps,
@@ -561,6 +565,7 @@ async function processWorkflowForEach({
       runId,
       executionPath: [executionPath[0], idx],
       resumeSteps,
+      timeTravel,
       stepResults,
       prevResult,
       resumeData,
@@ -576,6 +581,7 @@ async function processWorkflowParallel({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -603,6 +609,7 @@ async function processWorkflowParallel({
           stepResults,
           prevResult,
           resumeData,
+          timeTravel,
           parentWorkflow,
           activeSteps,
           requestContext
@@ -618,6 +625,7 @@ async function processWorkflowConditional({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -660,6 +668,7 @@ async function processWorkflowConditional({
             executionPath: executionPath.concat([idx]),
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             resumeData,
             parentWorkflow,
@@ -726,6 +735,7 @@ async function processWorkflowSleep({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -797,6 +807,7 @@ async function processWorkflowSleep({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -816,6 +827,7 @@ async function processWorkflowSleepUntil({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -887,6 +899,7 @@ async function processWorkflowSleepUntil({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -967,6 +980,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     resumeSteps,
     prevResult,
     resumeData,
+    timeTravel,
     executionPath,
     stepResults,
     requestContext
@@ -979,13 +993,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: {}
       }
     });
@@ -1002,6 +1017,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           input: prevResult?.status === "success" ? prevResult.output : void 0
         },
         prevResult,
+        timeTravel,
         requestContext,
         resumeData,
         activeSteps: {}
@@ -1035,7 +1051,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     });
   }
   async processWorkflowEnd(args) {
-    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, requestContext, runId } = args;
+    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, requestContext, runId, timeTravel } = args;
     if (parentWorkflow) {
       await this.mastra.pubsub.publish("workflows", {
         type: "workflow.step.end",
@@ -1051,7 +1067,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           activeSteps,
           parentWorkflow: parentWorkflow.parentWorkflow,
           parentContext: parentWorkflow,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     }
@@ -1062,7 +1079,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     });
   }
   async processWorkflowSuspend(args) {
-    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runId, requestContext } = args;
+    const { resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, runId, requestContext, timeTravel } = args;
     if (parentWorkflow) {
       await this.mastra.pubsub.publish("workflows", {
         type: "workflow.step.end",
@@ -1083,6 +1100,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
               }
             }
           },
+          timeTravel,
           resumeData,
           activeSteps,
           requestContext,
@@ -1098,7 +1116,17 @@ var WorkflowEventProcessor = class extends EventProcessor {
     });
   }
   async processWorkflowFail(args) {
-    const { workflowId, runId, resumeSteps, prevResult, resumeData, parentWorkflow, activeSteps, requestContext } = args;
+    const {
+      workflowId,
+      runId,
+      resumeSteps,
+      prevResult,
+      resumeData,
+      parentWorkflow,
+      activeSteps,
+      requestContext,
+      timeTravel
+    } = args;
     await this.mastra.getStorage()?.updateWorkflowState({
       workflowName: workflowId,
       runId,
@@ -1118,6 +1146,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           stepResults: parentWorkflow.stepResults,
           prevResult,
+          timeTravel,
           resumeData,
           activeSteps,
           requestContext,
@@ -1140,6 +1169,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     stepResults,
     activeSteps,
     resumeSteps,
+    timeTravel,
     prevResult,
     resumeData,
     parentWorkflow,
@@ -1203,6 +1233,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1222,6 +1253,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1242,6 +1274,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1262,6 +1295,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1282,6 +1316,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1374,6 +1409,47 @@ var WorkflowEventProcessor = class extends EventProcessor {
             requestContext
           }
         });
+      } 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,
+            requestContext
+          }
+        });
       } else {
         await this.mastra.pubsub.publish("workflows", {
           type: "workflow.start",
@@ -1429,6 +1505,21 @@ var WorkflowEventProcessor = class extends EventProcessor {
     for (const [key, value] of Object.entries(requestContext)) {
       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) {
+      resumeDataToUse = resumeData;
+    }
     const stepResult = await this.stepExecutor.execute({
       workflowId,
       step: step.step,
@@ -1439,7 +1530,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
       emitter: ee,
       requestContext: rc,
       input: prevResult?.output,
-      resumeData: resumeSteps?.length === 1 && resumeSteps?.[0] === step.step.id ? resumeData : void 0,
+      resumeData: resumeDataToUse,
       retryCount,
       foreachIdx: step.type === "foreach" ? executionPath[1] : void 0,
       validateInputs: workflow.options.validateInputs
@@ -1493,6 +1584,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
             executionPath,
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             activeSteps,
             requestContext,
@@ -1533,6 +1625,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,
@@ -1547,6 +1641,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     runId,
     executionPath,
     resumeSteps,
+    timeTravel,
     prevResult,
     parentWorkflow,
     stepResults,
@@ -1636,6 +1731,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           parentWorkflow,
           stepResults,
+          timeTravel,
           prevResult,
           activeSteps,
           requestContext
@@ -1669,7 +1765,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
       await this.mastra.pubsub.publish(`workflow.events.v2.${runId}`, {
@@ -1746,7 +1843,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { status: "success", output: allResults },
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     } else if (step?.type === "foreach") {
@@ -1762,7 +1860,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { ...prevResult, output: prevResult?.payload },
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     } else if (executionPath[0] >= workflow.stepGraph.length - 1) {
@@ -1791,7 +1890,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     }
@@ -1927,7 +2027,8 @@ function cloneWorkflow(workflow, opts) {
     inputSchema: workflow.inputSchema,
     outputSchema: workflow.outputSchema,
     steps: workflow.stepDefs,
-    mastra: workflow.mastra
+    mastra: workflow.mastra,
+    options: workflow.options
   });
   wf.setStepFlow(workflow.stepGraph);
   wf.commit();
@@ -1939,7 +2040,12 @@ function cloneStep(step, opts) {
     description: step.description,
     inputSchema: step.inputSchema,
     outputSchema: step.outputSchema,
+    suspendSchema: step.suspendSchema,
+    resumeSchema: step.resumeSchema,
+    stateSchema: step.stateSchema,
     execute: step.execute,
+    retries: step.retries,
+    scorers: step.scorers,
     component: step.component
   };
 }
@@ -2070,7 +2176,7 @@ function createWorkflow(params) {
     mastra: params.mastra,
     eventProcessor,
     options: {
-      validateInputs: params.options?.validateInputs ?? false,
+      validateInputs: params.options?.validateInputs ?? true,
       shouldPersistSnapshot: params.options?.shouldPersistSnapshot ?? (() => true),
       tracingPolicy: params.options?.tracingPolicy
     }
@@ -2083,6 +2189,7 @@ function createWorkflow(params) {
 var EventedWorkflow = class extends Workflow {
   constructor(params) {
     super(params);
+    this.engineType = "evented";
   }
   __registerMastra(mastra) {
     super.__registerMastra(mastra);
@@ -2099,7 +2206,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?.({
@@ -2118,6 +2227,7 @@ var EventedWorkflow = class extends Workflow {
           context: {},
           activePaths: [],
           serializedStepGraph: this.serializedStepGraph,
+          activeStepsPath: {},
           suspendedPaths: {},
           resumeLabels: {},
           waitingPaths: {},
@@ -2156,15 +2266,16 @@ var EventedRun = class extends Run {
       snapshot: {
         runId: this.runId,
         serializedStepGraph: this.serializedStepGraph,
+        status: "running",
         value: {},
         context: {},
         requestContext: Object.fromEntries(requestContext.entries()),
         activePaths: [],
+        activeStepsPath: {},
         suspendedPaths: {},
         resumeLabels: {},
         waitingPaths: {},
-        timestamp: Date.now(),
-        status: "running"
+        timestamp: Date.now()
       }
     });
     const inputDataToUse = await this._validateInput(inputData);
@@ -2202,9 +2313,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");
     }
@@ -2402,6 +2518,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 },
+          requestContext: Object.fromEntries(params.requestContext.entries()),
+          format: params.format
+        }
+      });
     } else {
       await pubsub.publish("workflows", {
         type: "workflow.start",
@@ -2460,5 +2593,5 @@ var EventedExecutionEngine = class extends ExecutionEngine {
 };
 
 export { EventedExecutionEngine, EventedRun, EventedWorkflow, StepExecutor, WorkflowEventProcessor, cloneStep, cloneWorkflow, createStep, createWorkflow };
-//# sourceMappingURL=chunk-H6CZGPZD.js.map
-//# sourceMappingURL=chunk-H6CZGPZD.js.map
+//# sourceMappingURL=chunk-4DWZ4Z6H.js.map
+//# sourceMappingURL=chunk-4DWZ4Z6H.js.map