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

package/dist/{chunk-AHQOJZT2.cjs → chunk-YYKN3NDZ.cjs}

@@ -342,7 +342,7 @@ condition,loopType:"dountil",serializedCondition:{id:`${step.id}-condition`,fn:c
    */async createRunAsync(options){if(this.stepFlow.length===0){throw new Error("Execution flow of workflow is not defined. Add steps to the workflow via .then(), .branch(), etc.");}if(!this.executionGraph.steps){throw new Error("Uncommitted step flow changes detected. Call .commit() to register the steps.");}const runIdToUse=options?.runId||this.#mastra?.generateId()||crypto2.randomUUID();const run=this.#runs.get(runIdToUse)??new Run({workflowId:this.id,stateSchema:this.stateSchema,inputSchema:this.inputSchema,runId:runIdToUse,resourceId:options?.resourceId,executionEngine:this.executionEngine,executionGraph:this.executionGraph,mastra:this.#mastra,retryConfig:this.retryConfig,serializedStepGraph:this.serializedStepGraph,disableScorers:options?.disableScorers,cleanup:()=>this.#runs.delete(runIdToUse),tracingPolicy:this.#options?.tracingPolicy,workflowSteps:this.steps,validateInputs:this.#options?.validateInputs,workflowEngineType:this.engineType});this.#runs.set(runIdToUse,run);const shouldPersistSnapshot=this.#options.shouldPersistSnapshot({workflowStatus:run.workflowRunStatus,stepResults:{}});const workflowSnapshotInStorage=await this.getWorkflowRunExecutionResult(runIdToUse,false);if(!workflowSnapshotInStorage&&shouldPersistSnapshot){await this.mastra?.getStorage()?.persistWorkflowSnapshot({workflowName:this.id,runId:runIdToUse,resourceId:options?.resourceId,snapshot:{runId:runIdToUse,status:"pending",value:{},context:this.#nestedWorkflowInput?{input:this.#nestedWorkflowInput}:{},activePaths:[],activeStepsPath:{},serializedStepGraph:this.serializedStepGraph,suspendedPaths:{},resumeLabels:{},waitingPaths:{},result:void 0,error:void 0,// @ts-ignore
 timestamp:Date.now()}});}return run;}async getScorers({runtimeContext=new chunkGPWMM745_cjs.RuntimeContext()}={}){const steps=this.steps;if(!steps||Object.keys(steps).length===0){return {};}const scorers={};for(const step of Object.values(steps)){if(step.scorers){let scorersToUse=step.scorers;if(typeof scorersToUse==="function"){scorersToUse=await scorersToUse({runtimeContext});}for(const[id,scorer]of Object.entries(scorersToUse)){scorers[id]=scorer;}}}return scorers;}// This method should only be called internally for nested workflow execution, as well as from mastra server handlers
 // To run a workflow use `.createRunAsync` and then `.start` or `.resume`
-async execute({runId,inputData,resumeData,state,setState,suspend,restart,resume,timeTravel,[chunkABJOUEVA_cjs.EMITTER_SYMBOL]:emitter,mastra,runtimeContext,abort,abortSignal,runCount,tracingContext,writer,validateInputs}){this.__registerMastra(mastra);if(validateInputs){this.#options={...(this.#options||{}),validateInputs};}this.executionEngine.options={...(this.executionEngine.options||{}),validateInputs:validateInputs??false};const isResume=!!(resume?.steps&&resume.steps.length>0)||!!resume?.label||!!(resume?.steps&&resume.steps.length===0&&(!runCount||runCount===0));if(!restart&&!isResume){this.#nestedWorkflowInput=inputData;}const isTimeTravel=!!(timeTravel&&timeTravel.steps.length>0);const run=isResume?await this.createRunAsync({runId:resume.runId}):await this.createRunAsync({runId});const nestedAbortCb=()=>{abort();};run.abortController.signal.addEventListener("abort",nestedAbortCb);abortSignal.addEventListener("abort",async()=>{run.abortController.signal.removeEventListener("abort",nestedAbortCb);await run.cancel();});const unwatchV2=run.watch(event=>{emitter.emit("nested-watch-v2",{event,workflowId:this.id});},"watch-v2");const unwatch=run.watch(event=>{emitter.emit("nested-watch",{event,workflowId:this.id,runId:run.runId,isResume:!!resume?.steps?.length});},"watch");if(runCount&&runCount>0&&isResume&&runtimeContext){runtimeContext.set("__mastraWorflowInputData",inputData);}let res;if(isTimeTravel){res=await run.timeTravel({inputData:timeTravel?.inputData,resumeData:timeTravel?.resumeData,initialState:state,step:timeTravel?.steps,context:timeTravel?.nestedStepResults?.[this.id]??{},nestedStepsContext:timeTravel?.nestedStepResults,runtimeContext,tracingContext,writableStream:writer,outputOptions:{includeState:true,includeResumeLabels:true}});}else if(restart){res=await run.restart({runtimeContext,tracingContext,writableStream:writer});}else if(isResume){res=await run.resume({resumeData,step:resume.steps?.length>0?resume.steps:void 0,runtimeContext,tracingContext,outputOptions:{includeState:true,includeResumeLabels:true},label:resume.label});}else {res=await run.start({inputData,runtimeContext,tracingContext,writableStream:writer,initialState:state,outputOptions:{includeState:true,includeResumeLabels:true}});}unwatch();unwatchV2();const suspendedSteps=Object.entries(res.steps).filter(([_stepName,stepResult])=>{const stepRes=stepResult;return stepRes?.status==="suspended";});if(res.state){setState(res.state);}if(suspendedSteps?.length){for(const[stepName,stepResult]of suspendedSteps){const suspendPath=[stepName,...(stepResult?.suspendPayload?.__workflow_meta?.path??[])];await suspend({...stepResult?.suspendPayload,__workflow_meta:{runId:run.runId,path:suspendPath}},{resumeLabel:Object.keys(res.resumeLabels??{})});}}if(res.status==="failed"){throw res.error;}return res.status==="success"?res.result:void 0;}async getWorkflowRuns(args){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow runs. Mastra storage is not initialized");return {runs:[],total:0};}return storage.getWorkflowRuns({workflowName:this.id,...(args??{})});}async getActiveWorkflowRuns(){const runningRuns=await this.getWorkflowRuns({status:"running"});const waitingRuns=await this.getWorkflowRuns({status:"waiting"});return {runs:[...runningRuns.runs,...waitingRuns.runs],total:runningRuns.total+waitingRuns.total};}async restartAllActiveWorkflowRuns(){if(this.engineType!=="default"){this.logger.debug(`Cannot restart active workflow runs for ${this.engineType} engine`);return;}const activeRuns=await this.getActiveWorkflowRuns();if(activeRuns.runs.length>0){this.logger.debug(`Restarting ${activeRuns.runs.length} active workflow run${activeRuns.runs.length>1?"s":""}`);}for(const runSnapshot of activeRuns.runs){try{const run=await this.createRunAsync({runId:runSnapshot.runId});await run.restart();this.logger.debug(`Restarted ${this.id} workflow run ${runSnapshot.runId}`);}catch(error){this.logger.error(`Failed to restart ${this.id} workflow run ${runSnapshot.runId}: ${error}`);}}}async getWorkflowRunById(runId){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow runs from storage. Mastra storage is not initialized");return this.#runs.get(runId)?{...this.#runs.get(runId),workflowName:this.id}:null;}const run=await storage.getWorkflowRunById({runId,workflowName:this.id});return run??(this.#runs.get(runId)?{...this.#runs.get(runId),workflowName:this.id}:null);}async getWorkflowRunSteps({runId,workflowId}){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow run steps. Mastra storage is not initialized");return {};}const run=await storage.getWorkflowRunById({runId,workflowName:workflowId});let snapshot=run?.snapshot;if(!snapshot){return {};}if(typeof snapshot==="string"){try{snapshot=JSON.parse(snapshot);}catch(e){this.logger.debug("Cannot get workflow run execution result. Snapshot is not a valid JSON string",e);return {};}}const{serializedStepGraph,context}=snapshot;const{input,...steps}=context;let finalSteps={};for(const step of Object.keys(steps)){const stepGraph=serializedStepGraph.find(stepGraph2=>stepGraph2?.step?.id===step);finalSteps[step]=steps[step];if(stepGraph&&stepGraph?.step?.component==="WORKFLOW"){const nestedSteps=await this.getWorkflowRunSteps({runId,workflowId:step});if(nestedSteps){const updatedNestedSteps=Object.entries(nestedSteps).reduce((acc,[key,value])=>{acc[`${step}.${key}`]=value;return acc;},{});finalSteps={...finalSteps,...updatedNestedSteps};}}}return finalSteps;}async getWorkflowRunExecutionResult(runId,withNestedWorkflows=true){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow run execution result. Mastra storage is not initialized");return null;}const run=await storage.getWorkflowRunById({runId,workflowName:this.id});let snapshot=run?.snapshot;if(!snapshot){return null;}if(typeof snapshot==="string"){try{snapshot=JSON.parse(snapshot);}catch(e){this.logger.debug("Cannot get workflow run execution result. Snapshot is not a valid JSON string",e);return null;}}const fullSteps=withNestedWorkflows?await this.getWorkflowRunSteps({runId,workflowId:this.id}):snapshot.context;return {status:snapshot.status,result:snapshot.result,error:snapshot.error,payload:snapshot.context?.input,steps:fullSteps,activeStepsPath:snapshot.activeStepsPath,serializedStepGraph:snapshot.serializedStepGraph};}};var Run=class{#abortController;emitter;/**
+async execute({runId,inputData,resumeData,state,setState,suspend,restart,resume,timeTravel,[chunkABJOUEVA_cjs.EMITTER_SYMBOL]:emitter,mastra,runtimeContext,abort,abortSignal,runCount,tracingContext,writer,validateInputs}){this.__registerMastra(mastra);if(validateInputs){this.#options={...(this.#options||{}),validateInputs};}this.executionEngine.options={...(this.executionEngine.options||{}),validateInputs:validateInputs??false};const isResume=!!(resume?.steps&&resume.steps.length>0)||!!resume?.label||!!(resume?.steps&&resume.steps.length===0&&(!runCount||runCount===0));if(!restart&&!isResume){this.#nestedWorkflowInput=inputData;}const isTimeTravel=!!(timeTravel&&timeTravel.steps.length>0);const run=isResume?await this.createRunAsync({runId:resume.runId}):await this.createRunAsync({runId});const nestedAbortCb=()=>{abort();};run.abortController.signal.addEventListener("abort",nestedAbortCb);abortSignal.addEventListener("abort",async()=>{run.abortController.signal.removeEventListener("abort",nestedAbortCb);await run.cancel();});const unwatchV2=run.watch(event=>{emitter.emit("nested-watch-v2",{event,workflowId:this.id});},"watch-v2");const unwatch=run.watch(event=>{emitter.emit("nested-watch",{event,workflowId:this.id,runId:run.runId,isResume:!!resume?.steps?.length});},"watch");if(runCount&&runCount>0&&isResume&&runtimeContext){runtimeContext.set("__mastraWorflowInputData",inputData);}let res;if(isTimeTravel){res=await run.timeTravel({inputData:timeTravel?.inputData,resumeData:timeTravel?.resumeData,initialState:state,step:timeTravel?.steps,context:timeTravel?.nestedStepResults?.[this.id]??{},nestedStepsContext:timeTravel?.nestedStepResults,runtimeContext,tracingContext,writableStream:writer,outputOptions:{includeState:true,includeResumeLabels:true}});}else if(restart){res=await run.restart({runtimeContext,tracingContext,writableStream:writer});}else if(isResume){res=await run.resume({resumeData,step:resume.steps?.length>0?resume.steps:void 0,runtimeContext,tracingContext,writableStream:writer,outputOptions:{includeState:true,includeResumeLabels:true},label:resume.label});}else {res=await run.start({inputData,runtimeContext,tracingContext,writableStream:writer,initialState:state,outputOptions:{includeState:true,includeResumeLabels:true}});}unwatch();unwatchV2();const suspendedSteps=Object.entries(res.steps).filter(([_stepName,stepResult])=>{const stepRes=stepResult;return stepRes?.status==="suspended";});if(res.state){setState(res.state);}if(suspendedSteps?.length){for(const[stepName,stepResult]of suspendedSteps){const suspendPath=[stepName,...(stepResult?.suspendPayload?.__workflow_meta?.path??[])];await suspend({...stepResult?.suspendPayload,__workflow_meta:{runId:run.runId,path:suspendPath}},{resumeLabel:Object.keys(res.resumeLabels??{})});}}if(res.status==="failed"){throw res.error;}return res.status==="success"?res.result:void 0;}async getWorkflowRuns(args){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow runs. Mastra storage is not initialized");return {runs:[],total:0};}return storage.getWorkflowRuns({workflowName:this.id,...(args??{})});}async getActiveWorkflowRuns(){const runningRuns=await this.getWorkflowRuns({status:"running"});const waitingRuns=await this.getWorkflowRuns({status:"waiting"});return {runs:[...runningRuns.runs,...waitingRuns.runs],total:runningRuns.total+waitingRuns.total};}async restartAllActiveWorkflowRuns(){if(this.engineType!=="default"){this.logger.debug(`Cannot restart active workflow runs for ${this.engineType} engine`);return;}const activeRuns=await this.getActiveWorkflowRuns();if(activeRuns.runs.length>0){this.logger.debug(`Restarting ${activeRuns.runs.length} active workflow run${activeRuns.runs.length>1?"s":""}`);}for(const runSnapshot of activeRuns.runs){try{const run=await this.createRunAsync({runId:runSnapshot.runId});await run.restart();this.logger.debug(`Restarted ${this.id} workflow run ${runSnapshot.runId}`);}catch(error){this.logger.error(`Failed to restart ${this.id} workflow run ${runSnapshot.runId}: ${error}`);}}}async getWorkflowRunById(runId){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow runs from storage. Mastra storage is not initialized");return this.#runs.get(runId)?{...this.#runs.get(runId),workflowName:this.id}:null;}const run=await storage.getWorkflowRunById({runId,workflowName:this.id});return run??(this.#runs.get(runId)?{...this.#runs.get(runId),workflowName:this.id}:null);}async getWorkflowRunSteps({runId,workflowId}){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow run steps. Mastra storage is not initialized");return {};}const run=await storage.getWorkflowRunById({runId,workflowName:workflowId});let snapshot=run?.snapshot;if(!snapshot){return {};}if(typeof snapshot==="string"){try{snapshot=JSON.parse(snapshot);}catch(e){this.logger.debug("Cannot get workflow run execution result. Snapshot is not a valid JSON string",e);return {};}}const{serializedStepGraph,context}=snapshot;const{input,...steps}=context;let finalSteps={};for(const step of Object.keys(steps)){const stepGraph=serializedStepGraph.find(stepGraph2=>stepGraph2?.step?.id===step);finalSteps[step]=steps[step];if(stepGraph&&stepGraph?.step?.component==="WORKFLOW"){const nestedSteps=await this.getWorkflowRunSteps({runId,workflowId:step});if(nestedSteps){const updatedNestedSteps=Object.entries(nestedSteps).reduce((acc,[key,value])=>{acc[`${step}.${key}`]=value;return acc;},{});finalSteps={...finalSteps,...updatedNestedSteps};}}}return finalSteps;}async getWorkflowRunExecutionResult(runId,withNestedWorkflows=true){const storage=this.#mastra?.getStorage();if(!storage){this.logger.debug("Cannot get workflow run execution result. Mastra storage is not initialized");return null;}const run=await storage.getWorkflowRunById({runId,workflowName:this.id});let snapshot=run?.snapshot;if(!snapshot){return null;}if(typeof snapshot==="string"){try{snapshot=JSON.parse(snapshot);}catch(e){this.logger.debug("Cannot get workflow run execution result. Snapshot is not a valid JSON string",e);return null;}}const fullSteps=withNestedWorkflows?await this.getWorkflowRunSteps({runId,workflowId:this.id}):snapshot.context;return {status:snapshot.status,result:snapshot.result,error:snapshot.error,payload:snapshot.context?.input,steps:fullSteps,activeStepsPath:snapshot.activeStepsPath,serializedStepGraph:snapshot.serializedStepGraph};}};var Run=class{#abortController;emitter;/**
    * Unique identifier for this workflow
    */workflowId;/**
    * Unique identifier for this run
@@ -1314,5 +1314,5 @@ exports.updateStepInHierarchy = updateStepInHierarchy;
 exports.validateStepInput = validateStepInput;
 exports.validateStepResumeData = validateStepResumeData;
 exports.workflowToStep = workflowToStep;
-//# sourceMappingURL=chunk-AHQOJZT2.cjs.map
-//# sourceMappingURL=chunk-AHQOJZT2.cjs.map
+//# sourceMappingURL=chunk-YYKN3NDZ.cjs.map
+//# sourceMappingURL=chunk-YYKN3NDZ.cjs.map