@mastra/core 1.5.0 → 1.6.0-alpha.0

package/dist/{chunk-YV2YGQRY.js → chunk-K6U5VMYO.js}

@@ -1,7 +1,7 @@
-import { validateStepInput, createDeprecationProxy, getStepResult, validateStepSuspendData, runCountDeprecationMessage, TripWire, createTimeTravelExecutionParams, validateStepResumeData, Workflow, isSupportedLanguageModel, ProcessorStepOutputSchema, ProcessorStepSchema, Run, WorkflowRunOutput, ExecutionEngine, cleanStepResult, hydrateSerializedStepErrors, Agent, ProcessorRunner } from './chunk-MZIMV2BB.js';
+import { validateStepInput, createDeprecationProxy, getStepResult, validateStepSuspendData, runCountDeprecationMessage, TripWire, createTimeTravelExecutionParams, validateStepResumeData, Workflow, isSupportedLanguageModel, ProcessorStepOutputSchema, ProcessorStepSchema, Run, WorkflowRunOutput, ExecutionEngine, cleanStepResult, hydrateSerializedStepErrors, Agent, ProcessorRunner } from './chunk-5QSHNUVD.js';
 import { STREAM_FORMAT_SYMBOL, PUBSUB_SYMBOL } from './chunk-2QXNHEDL.js';
-import { createPendingMarker } from './chunk-YIN5F7VO.js';
-import { MessageList } from './chunk-OHLVZVIK.js';
+import { createPendingMarker } from './chunk-EEU5NHHU.js';
+import { MessageList } from './chunk-IC5OUWKJ.js';
 import { ToolStream } from './chunk-DD2VNRQM.js';
 import { Tool } from './chunk-BQHWJLXU.js';
 import { RequestContext } from './chunk-CCLV5CAA.js';
@@ -75,6 +75,7 @@ var StepExecutor = class extends MastraBase {
       const { __workflow_meta, ...userSuspendData } = suspendDataToUse;
       suspendDataToUse = userSuspendData;
     }
+    let stateUpdate;
     try {
       if (validationError) {
         throw validationError;
@@ -91,7 +92,7 @@ var StepExecutor = class extends MastraBase {
             inputData,
             state: params.state,
             setState: async (newState) => {
-              Object.assign(params.state, newState);
+              stateUpdate = { ...stateUpdate ?? params.state, ...newState };
             },
             retryCount,
             resumeData: params.resumeData,
@@ -162,6 +163,7 @@ var StepExecutor = class extends MastraBase {
       const isNestedWorkflowStep = step.component === "WORKFLOW";
       const nestedWflowStepPaused = isNestedWorkflowStep && perStep;
       const endedAt = Date.now();
+      const finalState = stateUpdate ?? params.state;
       let finalResult;
       if (suspended) {
         finalResult = {
@@ -169,7 +171,7 @@ var StepExecutor = class extends MastraBase {
           status: "suspended",
           suspendedAt: endedAt,
           ...stepOutput ? { suspendOutput: stepOutput } : {},
-          __state: params.state
+          __state: finalState
         };
         if (suspended.payload) {
           finalResult.suspendPayload = suspended.payload;
@@ -181,13 +183,13 @@ var StepExecutor = class extends MastraBase {
           status: "bailed",
           endedAt,
           output: bailed.payload,
-          __state: params.state
+          __state: finalState
         };
       } else if (nestedWflowStepPaused) {
         finalResult = {
           ...stepInfo,
           status: "paused",
-          __state: params.state
+          __state: finalState
         };
       } else {
         finalResult = {
@@ -195,7 +197,7 @@ var StepExecutor = class extends MastraBase {
           status: "success",
           endedAt,
           output: stepOutput,
-          __state: params.state
+          __state: finalState
         };
       }
       return finalResult;
@@ -1933,7 +1935,91 @@ var WorkflowEventProcessor = class extends EventProcessor {
     activeSteps[step.step.id] = true;
     const workflowsStore = await this.mastra?.getStorage()?.getStore("workflows");
     if (step.step instanceof EventedWorkflow || step.step.component === "WORKFLOW") {
-      if (resumeSteps?.length > 1) {
+      if (resumeSteps?.length === 1 && resumeSteps[0] === step.step.id) {
+        const stepData = stepResults[step.step.id];
+        const nestedRunId = stepData?.suspendPayload?.__workflow_meta?.runId;
+        if (!nestedRunId) {
+          return this.errorWorkflow(
+            {
+              workflowId,
+              runId,
+              executionPath,
+              stepResults,
+              activeSteps,
+              resumeSteps,
+              prevResult,
+              resumeData,
+              parentWorkflow,
+              requestContext
+            },
+            new MastraError({
+              id: "MASTRA_WORKFLOW",
+              text: `Nested workflow run id not found for auto-detection: ${JSON.stringify(stepResults)}`,
+              domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
+              category: "SYSTEM" /* SYSTEM */
+            })
+          );
+        }
+        const snapshot = await workflowsStore?.loadWorkflowSnapshot({
+          workflowName: step.step.id,
+          runId: nestedRunId
+        });
+        const suspendedStepId = Object.keys(snapshot?.suspendedPaths ?? {})?.[0];
+        if (!suspendedStepId) {
+          return this.errorWorkflow(
+            {
+              workflowId,
+              runId,
+              executionPath,
+              stepResults,
+              activeSteps,
+              resumeSteps,
+              prevResult,
+              resumeData,
+              parentWorkflow,
+              requestContext
+            },
+            new MastraError({
+              id: "MASTRA_WORKFLOW",
+              text: `No suspended step found in nested workflow: ${step.step.id}`,
+              domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
+              category: "SYSTEM" /* SYSTEM */
+            })
+          );
+        }
+        const nestedExecutionPath = snapshot?.suspendedPaths?.[suspendedStepId];
+        const nestedStepResults = snapshot?.context;
+        await this.mastra.pubsub.publish("workflows", {
+          type: "workflow.resume",
+          runId,
+          data: {
+            workflowId: step.step.id,
+            parentWorkflow: {
+              stepId: step.step.id,
+              workflowId,
+              runId,
+              executionPath,
+              resumeSteps,
+              stepResults,
+              input: prevResult,
+              parentWorkflow
+            },
+            executionPath: nestedExecutionPath,
+            runId: nestedRunId,
+            resumeSteps: [suspendedStepId],
+            // Resume the auto-detected inner step
+            stepResults: nestedStepResults,
+            prevResult,
+            resumeData,
+            activeSteps,
+            requestContext,
+            perStep,
+            initialState: currentState,
+            state: currentState,
+            outputOptions
+          }
+        });
+      } else if (resumeSteps?.length > 1) {
         const stepData = stepResults[step.step.id];
         const nestedRunId = stepData?.suspendPayload?.__workflow_meta?.runId;
         if (!nestedRunId) {
@@ -2233,6 +2319,7 @@ var WorkflowEventProcessor = class extends EventProcessor {
           //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: {
             ...stepResults,
+            [step.step.id]: stepResult,
             __state: updatedState
           },
           prevResult: stepResult,
@@ -4401,9 +4488,10 @@ var EventedExecutionEngine = class extends ExecutionEngine {
     }
     let result;
     if (resultData.prevResult.status === "suspended") {
-      const suspendedSteps = Object.entries(resultData.stepResults).map(([_stepId, stepResult]) => {
+      const suspendedSteps = Object.entries(resultData.stepResults).map(([stepId, stepResult]) => {
         if (stepResult.status === "suspended") {
-          return stepResult.suspendPayload?.__workflow_meta?.path ?? [];
+          const existingPath = stepResult.suspendPayload?.__workflow_meta?.path ?? [];
+          return [stepId, ...existingPath];
         }
         return null;
       }).filter(Boolean);
@@ -4446,5 +4534,5 @@ var EventedExecutionEngine = class extends ExecutionEngine {
 };
 
 export { EventedExecutionEngine, EventedRun, EventedWorkflow, StepExecutor, WorkflowEventProcessor, cloneStep, cloneWorkflow, createStep, createWorkflow };
-//# sourceMappingURL=chunk-YV2YGQRY.js.map
-//# sourceMappingURL=chunk-YV2YGQRY.js.map
+//# sourceMappingURL=chunk-K6U5VMYO.js.map
+//# sourceMappingURL=chunk-K6U5VMYO.js.map