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

package/dist/{chunk-H6CZGPZD.js → chunk-XRJQ5ZA7.js}

@@ -1,10 +1,10 @@
-import { validateStepInput, createDeprecationProxy, getStepResult, runCountDeprecationMessage, Workflow, Run, ExecutionEngine } from './chunk-JUBFO5J3.js';
+import { validateStepInput, createDeprecationProxy, getStepResult, validateStepSuspendData, runCountDeprecationMessage, createTimeTravelExecutionParams, validateStepResumeData, Workflow, Run, ExecutionEngine } from './chunk-6CYYFT4O.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 { RequestContext } from './chunk-GRBGQ2GE.js';
-import { MastraBase } from './chunk-S6URFGCZ.js';
-import { RegisteredLogger } from './chunk-KJ2SW6VA.js';
+import { Tool } from './chunk-THZTRBFS.js';
+import { RequestContext } from './chunk-6TBWJV35.js';
+import { MastraBase } from './chunk-LSHPJWM5.js';
+import { RegisteredLogger } from './chunk-NRUZYMHE.js';
 import EventEmitter from 'events';
 import { randomUUID } from 'crypto';
 import z from 'zod';
@@ -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],
@@ -39,6 +39,11 @@ var StepExecutor = class extends MastraBase {
       stepInfo.resumePayload = params.resumeData;
       stepInfo.resumedAt = Date.now();
     }
+    let suspendDataToUse = params.stepResults[step.id]?.status === "suspended" ? params.stepResults[step.id]?.suspendPayload : void 0;
+    if (suspendDataToUse && "__workflow_meta" in suspendDataToUse) {
+      const { __workflow_meta, ...userSuspendData } = suspendDataToUse;
+      suspendDataToUse = userSuspendData;
+    }
     try {
       if (validationError) {
         throw validationError;
@@ -52,15 +57,24 @@ var StepExecutor = class extends MastraBase {
             requestContext,
             inputData,
             state: params.state,
-            setState: (state) => {
+            setState: async (state) => {
               params.state = state;
             },
             retryCount,
             resumeData: params.resumeData,
+            suspendData: suspendDataToUse,
             getInitData: () => stepResults?.input,
             getStepResult: getStepResult.bind(this, stepResults),
             suspend: async (suspendPayload) => {
-              suspended = { payload: { ...suspendPayload, __workflow_meta: { runId, path: [step.id] } } };
+              const { suspendData, validationError: validationError2 } = await validateStepSuspendData({
+                suspendData: suspendPayload,
+                step,
+                validateInputs: params.validateInputs ?? true
+              });
+              if (validationError2) {
+                throw validationError2;
+              }
+              suspended = { payload: { ...suspendData, __workflow_meta: { runId, path: [step.id] } } };
             },
             bail: (result) => {
               bailed = { payload: result };
@@ -92,7 +106,8 @@ var StepExecutor = class extends MastraBase {
         finalResult = {
           ...stepInfo,
           status: "suspended",
-          suspendedAt: endedAt
+          suspendedAt: endedAt,
+          ...stepResult ? { suspendOutput: stepResult } : {}
         };
         if (suspended.payload) {
           finalResult.suspendPayload = suspended.payload;
@@ -186,15 +201,10 @@ var StepExecutor = class extends MastraBase {
           requestContext,
           inputData,
           state,
-          setState: (_state) => {
-          },
           retryCount,
           resumeData,
           getInitData: () => stepResults?.input,
           getStepResult: getStepResult.bind(this, stepResults),
-          suspend: async (_suspendPayload) => {
-            throw new Error("Not implemented");
-          },
           bail: (_result) => {
             throw new Error("Not implemented");
           },
@@ -242,7 +252,7 @@ var StepExecutor = class extends MastraBase {
             inputData: params.input,
             // TODO: implement state
             state: {},
-            setState: (_state) => {
+            setState: async (_state) => {
             },
             retryCount,
             resumeData: params.resumeData,
@@ -301,7 +311,7 @@ var StepExecutor = class extends MastraBase {
             inputData: params.input,
             // TODO: implement state
             state: {},
-            setState: (_state) => {
+            setState: async (_state) => {
             },
             retryCount,
             resumeData: params.resumeData,
@@ -472,6 +482,7 @@ async function processWorkflowForEach({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   resumeData,
   parentWorkflow,
   requestContext
@@ -494,6 +505,7 @@ async function processWorkflowForEach({
         executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
         resumeSteps,
         stepResults,
+        timeTravel,
         prevResult: currentResult,
         resumeData,
         activeSteps,
@@ -530,6 +542,7 @@ async function processWorkflowForEach({
           executionPath: [executionPath[0], i],
           resumeSteps,
           stepResults,
+          timeTravel,
           prevResult,
           resumeData,
           activeSteps,
@@ -561,6 +574,7 @@ async function processWorkflowForEach({
       runId,
       executionPath: [executionPath[0], idx],
       resumeSteps,
+      timeTravel,
       stepResults,
       prevResult,
       resumeData,
@@ -576,6 +590,7 @@ async function processWorkflowParallel({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -603,6 +618,7 @@ async function processWorkflowParallel({
           stepResults,
           prevResult,
           resumeData,
+          timeTravel,
           parentWorkflow,
           activeSteps,
           requestContext
@@ -618,6 +634,7 @@ async function processWorkflowConditional({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -660,6 +677,7 @@ async function processWorkflowConditional({
             executionPath: executionPath.concat([idx]),
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             resumeData,
             parentWorkflow,
@@ -726,6 +744,7 @@ async function processWorkflowSleep({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -797,6 +816,7 @@ async function processWorkflowSleep({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -816,6 +836,7 @@ async function processWorkflowSleepUntil({
   stepResults,
   activeSteps,
   resumeSteps,
+  timeTravel,
   prevResult,
   resumeData,
   parentWorkflow,
@@ -887,6 +908,7 @@ async function processWorkflowSleepUntil({
           runId,
           executionPath: executionPath.slice(0, -1).concat([executionPath[executionPath.length - 1] + 1]),
           resumeSteps,
+          timeTravel,
           stepResults,
           prevResult,
           resumeData,
@@ -967,6 +989,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     resumeSteps,
     prevResult,
     resumeData,
+    timeTravel,
     executionPath,
     stepResults,
     requestContext
@@ -979,13 +1002,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 +1026,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           input: prevResult?.status === "success" ? prevResult.output : void 0
         },
         prevResult,
+        timeTravel,
         requestContext,
         resumeData,
         activeSteps: {}
@@ -1035,7 +1060,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 +1076,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           activeSteps,
           parentWorkflow: parentWorkflow.parentWorkflow,
           parentContext: parentWorkflow,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     }
@@ -1062,7 +1088,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 +1109,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
               }
             }
           },
+          timeTravel,
           resumeData,
           activeSteps,
           requestContext,
@@ -1098,7 +1125,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 +1155,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           stepResults: parentWorkflow.stepResults,
           prevResult,
+          timeTravel,
           resumeData,
           activeSteps,
           requestContext,
@@ -1140,6 +1178,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     stepResults,
     activeSteps,
     resumeSteps,
+    timeTravel,
     prevResult,
     resumeData,
     parentWorkflow,
@@ -1203,6 +1242,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1222,6 +1262,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1242,6 +1283,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1262,6 +1304,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1282,6 +1325,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           activeSteps,
           resumeSteps,
+          timeTravel,
           prevResult,
           resumeData,
           parentWorkflow,
@@ -1374,6 +1418,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 +1514,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 +1539,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 +1593,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
             executionPath,
             resumeSteps,
             stepResults,
+            timeTravel,
             prevResult,
             activeSteps,
             requestContext,
@@ -1533,6 +1634,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 +1650,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
     runId,
     executionPath,
     resumeSteps,
+    timeTravel,
     prevResult,
     parentWorkflow,
     stepResults,
@@ -1636,6 +1740,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           resumeSteps,
           parentWorkflow,
           stepResults,
+          timeTravel,
           prevResult,
           activeSteps,
           requestContext
@@ -1669,7 +1774,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
       await this.mastra.pubsub.publish(`workflow.events.v2.${runId}`, {
@@ -1746,7 +1852,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult: { status: "success", output: allResults },
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     } else if (step?.type === "foreach") {
@@ -1762,7 +1869,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 +1899,8 @@ var WorkflowEventProcessor = class extends EventProcessor {
           stepResults,
           prevResult,
           activeSteps,
-          requestContext
+          requestContext,
+          timeTravel
         }
       });
     }
@@ -1927,7 +2036,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 +2049,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
   };
 }
@@ -1952,7 +2067,7 @@ function isTool(params) {
 function createStep(params) {
   if (isAgent(params)) {
     return {
-      id: params.name,
+      id: params.id,
       description: params.getDescription(),
       // @ts-ignore
       inputSchema: z.object({
@@ -2070,7 +2185,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 +2198,7 @@ function createWorkflow(params) {
 var EventedWorkflow = class extends Workflow {
   constructor(params) {
     super(params);
+    this.engineType = "evented";
   }
   __registerMastra(mastra) {
     super.__registerMastra(mastra);
@@ -2099,7 +2215,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 +2236,7 @@ var EventedWorkflow = class extends Workflow {
           context: {},
           activePaths: [],
           serializedStepGraph: this.serializedStepGraph,
+          activeStepsPath: {},
           suspendedPaths: {},
           resumeLabels: {},
           waitingPaths: {},
@@ -2156,15 +2275,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);
@@ -2194,7 +2314,6 @@ var EventedRun = class extends Run {
       requestContext,
       abortController: this.abortController
     });
-    console.dir({ startResult: result }, { depth: null });
     if (result.status !== "suspended") {
       this.cleanup?.();
     }
@@ -2202,9 +2321,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 +2526,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 +2601,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-XRJQ5ZA7.js.map
+//# sourceMappingURL=chunk-XRJQ5ZA7.js.map