@mastra/core 0.10.4 → 0.10.6-alpha.0

package/dist/{chunk-UFQERGWG.js → chunk-U3F3AQJP.js}

@@ -1,6 +1,7 @@
 import { EMITTER_SYMBOL } from './chunk-Y7D2JLKS.js';
-import { Agent } from './chunk-SWJKOFCU.js';
+import { Agent } from './chunk-VZDPK54I.js';
 import { Tool } from './chunk-C4LMN2IR.js';
+import { MastraError } from './chunk-LDNCERLU.js';
 import { MastraBase } from './chunk-5IEKR756.js';
 import { RegisteredLogger } from './chunk-5YDTZN2X.js';
 import { RuntimeContext } from './chunk-SGGPJWRQ.js';
@@ -91,7 +92,12 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     const { attempts = 0, delay = 0 } = retryConfig ?? {};
     const steps = graph.steps;
     if (steps.length === 0) {
-      throw new Error("Workflow must have at least one step");
+      throw new MastraError({
+        id: "WORKFLOW_EXECUTE_EMPTY_GRAPH",
+        text: "Workflow must have at least one step",
+        domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
+        category: "USER" /* USER */
+      });
     }
     const executionSpan = this.mastra?.getTelemetry()?.tracer.startSpan(`workflow.${workflowId}.execute`, {
       attributes: { componentName: workflowId, runId }
@@ -125,21 +131,77 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           emitter: params.emitter,
           runtimeContext: params.runtimeContext
         });
-        if (lastOutput.status !== "success") {
-          return this.fmtReturnValue(executionSpan, params.emitter, stepResults, lastOutput);
+        if (lastOutput.result.status !== "success") {
+          const result2 = await this.fmtReturnValue(
+            executionSpan,
+            params.emitter,
+            stepResults,
+            lastOutput.result
+          );
+          await this.persistStepUpdate({
+            workflowId,
+            runId,
+            stepResults: lastOutput.stepResults,
+            serializedStepGraph: params.serializedStepGraph,
+            executionContext: lastOutput.executionContext,
+            workflowStatus: result2.status,
+            result: result2.result,
+            error: result2.error
+          });
+          return result2;
         }
       } catch (e) {
-        this.logger.error("Error executing step: " + (e?.stack ?? e));
-        return this.fmtReturnValue(executionSpan, params.emitter, stepResults, lastOutput, e);
+        const error = e instanceof MastraError ? e : new MastraError(
+          {
+            id: "WORKFLOW_ENGINE_STEP_EXECUTION_FAILED",
+            domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
+            category: "USER" /* USER */,
+            details: { workflowId, runId }
+          },
+          e
+        );
+        this.logger?.trackException(error);
+        this.logger?.error(`Error executing step: ${error?.stack}`);
+        const result2 = await this.fmtReturnValue(
+          executionSpan,
+          params.emitter,
+          stepResults,
+          lastOutput.result,
+          e
+        );
+        await this.persistStepUpdate({
+          workflowId,
+          runId,
+          stepResults: lastOutput.stepResults,
+          serializedStepGraph: params.serializedStepGraph,
+          executionContext: lastOutput.executionContext,
+          workflowStatus: result2.status,
+          result: result2.result,
+          error: result2.error
+        });
+        return result2;
       }
     }
-    return this.fmtReturnValue(executionSpan, params.emitter, stepResults, lastOutput);
+    const result = await this.fmtReturnValue(executionSpan, params.emitter, stepResults, lastOutput.result);
+    await this.persistStepUpdate({
+      workflowId,
+      runId,
+      stepResults: lastOutput.stepResults,
+      serializedStepGraph: params.serializedStepGraph,
+      executionContext: lastOutput.executionContext,
+      workflowStatus: result.status,
+      result: result.result,
+      error: result.error
+    });
+    return result;
   }
   getStepOutput(stepResults, step) {
     if (!step) {
       return stepResults.input;
     } else if (step.type === "step") {
       return stepResults[step.step.id]?.output;
+    } else if (step.type === "sleep" || step.type === "sleepUntil") {
+      return stepResults[step.id]?.output;
     } else if (step.type === "parallel" || step.type === "conditional") {
       return step.steps.reduce(
         (acc, entry) => {
@@ -152,6 +214,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             acc[entry.step.id] = stepResults[entry.step.id]?.output;
           } else if (entry.type === "foreach") {
             acc[entry.step.id] = stepResults[entry.step.id]?.output;
+          } else if (entry.type === "sleep" || entry.type === "sleepUntil") {
+            acc[entry.id] = stepResults[entry.id]?.output;
           }
           return acc;
         },
@@ -163,6 +227,9 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
       return stepResults[step.step.id]?.output;
     }
   }
+  async executeSleep({ duration }) {
+    await new Promise((resolve) => setTimeout(resolve, duration));
+  }
   async executeStep({
     workflowId,
     runId,
@@ -263,7 +330,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             // @ts-ignore
             runId: stepResults[step.id]?.suspendPayload?.__workflow_meta?.runId
           },
-          [EMITTER_SYMBOL]: emitter
+          [EMITTER_SYMBOL]: emitter,
+          engine: {}
         });
         if (suspended) {
           execResults = { status: "suspended", suspendPayload: suspended.payload, suspendedAt: Date.now() };
@@ -272,10 +340,20 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         }
         break;
       } catch (e) {
-        this.logger.error("Error executing step: " + (e?.stack ?? e));
+        const error = e instanceof MastraError ? e : new MastraError(
+          {
+            id: "WORKFLOW_STEP_INVOKE_FAILED",
+            domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
+            category: "USER" /* USER */,
+            details: { workflowId, runId, stepId: step.id }
+          },
+          e
+        );
+        this.logger.trackException(error);
+        this.logger.error("Error executing step: " + error?.stack);
         execResults = {
           status: "failed",
-          error: e instanceof Error ? e?.stack ?? e : typeof e === "string" ? e : new Error("Unknown error: " + e)?.stack ?? new Error("Unknown error: " + e),
+          error: error?.stack,
           endedAt: Date.now()
         };
       }
@@ -366,18 +444,18 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         })
       )
     );
-    const hasFailed = results.find((result) => result.status === "failed");
-    const hasSuspended = results.find((result) => result.status === "suspended");
+    const hasFailed = results.find((result) => result.result.status === "failed");
+    const hasSuspended = results.find((result) => result.result.status === "suspended");
     if (hasFailed) {
-      execResults = { status: "failed", error: hasFailed.error };
+      execResults = { status: "failed", error: hasFailed.result.error };
     } else if (hasSuspended) {
-      execResults = { status: "suspended", payload: hasSuspended.suspendPayload };
+      execResults = { status: "suspended", payload: hasSuspended.result.suspendPayload };
     } else {
       execResults = {
         status: "success",
         output: results.reduce((acc, result, index) => {
-          if (result.status === "success") {
-            acc[entry.steps[index].step.id] = result.output;
+          if (result.result.status === "success") {
+            acc[entry.steps[index].step.id] = result.result.output;
           }
           return acc;
         }, {})
@@ -421,11 +499,22 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             // TODO: this function shouldn't have suspend probably?
             suspend: async (_suspendPayload) => {
             },
-            [EMITTER_SYMBOL]: emitter
+            [EMITTER_SYMBOL]: emitter,
+            engine: {}
           });
           return result ? index : null;
         } catch (e) {
-          this.logger.error("Error evaluating condition: " + (e?.stack ?? e));
+          const error = e instanceof MastraError ? e : new MastraError(
+            {
+              id: "WORKFLOW_CONDITION_EVALUATION_FAILED",
+              domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
+              category: "USER" /* USER */,
+              details: { workflowId, runId }
+            },
+            e
+          );
+          this.logger.trackException(error);
+          this.logger.error("Error evaluating condition: " + error?.stack);
           return null;
         }
       })
@@ -454,18 +543,18 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         })
       )
     );
-    const hasFailed = results.find((result) => result.status === "failed");
-    const hasSuspended = results.find((result) => result.status === "suspended");
+    const hasFailed = results.find((result) => result.result.status === "failed");
+    const hasSuspended = results.find((result) => result.result.status === "suspended");
     if (hasFailed) {
-      execResults = { status: "failed", error: hasFailed.error };
+      execResults = { status: "failed", error: hasFailed.result.error };
     } else if (hasSuspended) {
-      execResults = { status: "suspended", payload: hasSuspended.suspendPayload };
+      execResults = { status: "suspended", payload: hasSuspended.result.suspendPayload };
     } else {
       execResults = {
         status: "success",
         output: results.reduce((acc, result, index) => {
-          if (result.status === "success") {
-            acc[stepsToRun[index].step.id] = result.output;
+          if (result.result.status === "success") {
+            acc[stepsToRun[index].step.id] = result.result.output;
           }
           return acc;
         }, {})
@@ -517,7 +606,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         },
         suspend: async (_suspendPayload) => {
         },
-        [EMITTER_SYMBOL]: emitter
+        [EMITTER_SYMBOL]: emitter,
+        engine: {}
       });
     } while (entry.loopType === "dowhile" ? isTrue : !isTrue);
     return result;
@@ -579,18 +669,24 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     runId,
     stepResults,
     serializedStepGraph,
-    executionContext
+    executionContext,
+    workflowStatus,
+    result,
+    error
   }) {
     await this.mastra?.getStorage()?.persistWorkflowSnapshot({
       workflowName: workflowId,
       runId,
       snapshot: {
         runId,
+        status: workflowStatus,
         value: {},
         context: stepResults,
         activePaths: [],
         serializedStepGraph,
         suspendedPaths: executionContext.suspendedPaths,
+        result,
+        error,
         // @ts-ignore
         timestamp: Date.now()
       }
@@ -697,6 +793,28 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         emitter,
         runtimeContext
       });
+    } else if (entry.type === "sleep") {
+      const startedAt = Date.now();
+      await this.executeSleep({ id: entry.id, duration: entry.duration });
+      const endedAt = Date.now();
+      const stepInfo = {
+        payload: prevOutput,
+        startedAt,
+        endedAt
+      };
+      execResults = { ...stepInfo, status: "success", output: prevOutput };
+      stepResults[entry.id] = { ...stepInfo, status: "success", output: prevOutput };
+    } else if (entry.type === "sleepUntil") {
+      const startedAt = Date.now();
+      await this.executeSleep({ id: entry.id, duration: entry.date.getTime() - Date.now() });
+      const endedAt = Date.now();
+      const stepInfo = {
+        payload: prevOutput,
+        startedAt,
+        endedAt
+      };
+      execResults = { ...stepInfo, status: "success", output: prevOutput };
+      stepResults[entry.id] = { ...stepInfo, status: "success", output: prevOutput };
     }
     if (entry.type === "step" || entry.type === "loop" || entry.type === "foreach") {
       stepResults[entry.step.id] = execResults;
@@ -706,9 +824,10 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
       runId,
       serializedStepGraph,
       stepResults,
-      executionContext
+      executionContext,
+      workflowStatus: "running"
     });
-    return execResults;
+    return { result: execResults, stepResults, executionContext };
   }
 };
 
@@ -915,6 +1034,52 @@ var Workflow = class extends MastraBase {
     this.steps[step.id] = step;
     return this;
   }
+  /**
+   * Adds a sleep step to the workflow
+   * @param duration The duration to sleep for
+   * @returns The workflow instance for chaining
+   */
+  sleep(duration) {
+    const id = `sleep_${randomUUID()}`;
+    this.stepFlow.push({ type: "sleep", id, duration });
+    this.serializedStepFlow.push({
+      type: "sleep",
+      id,
+      duration
+    });
+    this.steps[id] = createStep({
+      id,
+      inputSchema: z.object({}),
+      outputSchema: z.object({}),
+      execute: async () => {
+        return {};
+      }
+    });
+    return this;
+  }
+  /**
+   * Adds a sleep until step to the workflow
+   * @param date The date to sleep until
+   * @returns The workflow instance for chaining
+   */
+  sleepUntil(date) {
+    const id = `sleep_${randomUUID()}`;
+    this.stepFlow.push({ type: "sleepUntil", id, date });
+    this.serializedStepFlow.push({
+      type: "sleepUntil",
+      id,
+      date
+    });
+    this.steps[id] = createStep({
+      id,
+      inputSchema: z.object({}),
+      outputSchema: z.object({}),
+      execute: async () => {
+        return {};
+      }
+    });
+    return this;
+  }
   map(mappingConfig) {
     if (typeof mappingConfig === "function") {
       const mappingStep2 = createStep({
@@ -1030,6 +1195,7 @@ var Workflow = class extends MastraBase {
     this.stepFlow.push({
       type: "conditional",
       steps: steps.map(([_cond, step]) => ({ type: "step", step })),
+      // @ts-ignore
       conditions: steps.map(([cond]) => cond),
       serializedConditions: steps.map(([cond, _step]) => ({ id: `${_step.id}-condition`, fn: cond.toString() }))
     });
@@ -1055,6 +1221,7 @@ var Workflow = class extends MastraBase {
     this.stepFlow.push({
       type: "loop",
       step,
+      // @ts-ignore
       condition,
       loopType: "dowhile",
       serializedCondition: { id: `${step.id}-condition`, fn: condition.toString() }
@@ -1077,6 +1244,7 @@ var Workflow = class extends MastraBase {
     this.stepFlow.push({
       type: "loop",
       step,
+      // @ts-ignore
       condition,
       loopType: "dountil",
       serializedCondition: { id: `${step.id}-condition`, fn: condition.toString() }
@@ -1378,7 +1546,7 @@ var Run = class {
     const steps = (Array.isArray(params.step) ? params.step : [params.step]).map(
       (step) => typeof step === "string" ? step : step?.id
     );
-    const snapshot = await this.#mastra?.storage?.loadWorkflowSnapshot({
+    const snapshot = await this.#mastra?.getStorage()?.loadWorkflowSnapshot({
       workflowName: this.workflowId,
       runId: this.runId
     });

package/dist/{chunk-JYLH5IY3.cjs → chunk-VRDDTY4O.cjs}

@@ -1,12 +1,13 @@
 'use strict';
 
-var chunkZAXPU6F2_cjs = require('./chunk-ZAXPU6F2.cjs');
+var chunkQQ3KHYKE_cjs = require('./chunk-QQ3KHYKE.cjs');
+var chunkOLSQRA2V_cjs = require('./chunk-OLSQRA2V.cjs');
 var chunkP3Q73CAW_cjs = require('./chunk-P3Q73CAW.cjs');
 var chunkRWTSGWWL_cjs = require('./chunk-RWTSGWWL.cjs');
 
 // src/voice/voice.ts
 var _MastraVoice_decorators, _init, _a;
-_MastraVoice_decorators = [chunkZAXPU6F2_cjs.InstrumentClass({
+_MastraVoice_decorators = [chunkQQ3KHYKE_cjs.InstrumentClass({
   prefix: "voice",
   excludeMethods: ["__setTools", "__setLogger", "__setTelemetry", "#log"]
 })];
@@ -154,7 +155,12 @@ var CompositeVoice = class extends exports.MastraVoice {
     } else if (this.speakProvider) {
       return this.speakProvider.speak(input, options);
     }
-    throw new Error("No speak provider or realtime provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_COMPOSITE_NO_SPEAK_PROVIDER",
+      text: "No speak provider or realtime provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   async listen(audioStream, options) {
     if (this.realtimeProvider) {
@@ -162,7 +168,12 @@ var CompositeVoice = class extends exports.MastraVoice {
     } else if (this.listenProvider) {
       return await this.listenProvider.listen(audioStream, options);
     }
-    throw new Error("No listen provider or realtime provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_COMPOSITE_NO_LISTEN_PROVIDER",
+      text: "No listen provider or realtime provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   async getSpeakers() {
     if (this.realtimeProvider) {
@@ -170,7 +181,12 @@ var CompositeVoice = class extends exports.MastraVoice {
     } else if (this.speakProvider) {
       return this.speakProvider.getSpeakers();
     }
-    throw new Error("No speak provider or realtime provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_COMPOSITE_NO_SPEAKERS_PROVIDER",
+      text: "No speak provider or realtime provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   async getListener() {
     if (this.realtimeProvider) {
@@ -178,7 +194,12 @@ var CompositeVoice = class extends exports.MastraVoice {
     } else if (this.listenProvider) {
       return this.listenProvider.getListener();
     }
-    throw new Error("No listener provider or realtime provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_COMPOSITE_NO_LISTENER_PROVIDER",
+      text: "No listener provider or realtime provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   updateConfig(options) {
     if (!this.realtimeProvider) {
@@ -192,7 +213,12 @@ var CompositeVoice = class extends exports.MastraVoice {
    */
   connect(options) {
     if (!this.realtimeProvider) {
-      throw new Error("No realtime provider configured");
+      throw new chunkOLSQRA2V_cjs.MastraError({
+        id: "VOICE_COMPOSITE_NO_REALTIME_PROVIDER_CONNECT",
+        text: "No realtime provider configured",
+        domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+        category: "USER" /* USER */
+      });
     }
     return this.realtimeProvider.connect(options);
   }
@@ -202,7 +228,12 @@ var CompositeVoice = class extends exports.MastraVoice {
    */
   send(audioData) {
     if (!this.realtimeProvider) {
-      throw new Error("No realtime provider configured");
+      throw new chunkOLSQRA2V_cjs.MastraError({
+        id: "VOICE_COMPOSITE_NO_REALTIME_PROVIDER_SEND",
+        text: "No realtime provider configured",
+        domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+        category: "USER" /* USER */
+      });
     }
     return this.realtimeProvider.send(audioData);
   }
@@ -211,7 +242,12 @@ var CompositeVoice = class extends exports.MastraVoice {
    */
   answer(options) {
     if (!this.realtimeProvider) {
-      throw new Error("No realtime provider configured");
+      throw new chunkOLSQRA2V_cjs.MastraError({
+        id: "VOICE_COMPOSITE_NO_REALTIME_PROVIDER_ANSWER",
+        text: "No realtime provider configured",
+        domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+        category: "USER" /* USER */
+      });
     }
     return this.realtimeProvider.answer(options);
   }
@@ -240,7 +276,12 @@ var CompositeVoice = class extends exports.MastraVoice {
    */
   close() {
     if (!this.realtimeProvider) {
-      throw new Error("No realtime provider configured");
+      throw new chunkOLSQRA2V_cjs.MastraError({
+        id: "VOICE_COMPOSITE_NO_REALTIME_PROVIDER_CLOSE",
+        text: "No realtime provider configured",
+        domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+        category: "USER" /* USER */
+      });
     }
     this.realtimeProvider.close();
   }
@@ -251,7 +292,12 @@ var CompositeVoice = class extends exports.MastraVoice {
    */
   on(event, callback) {
     if (!this.realtimeProvider) {
-      throw new Error("No realtime provider configured");
+      throw new chunkOLSQRA2V_cjs.MastraError({
+        id: "VOICE_COMPOSITE_NO_REALTIME_PROVIDER_ON",
+        text: "No realtime provider configured",
+        domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+        category: "USER" /* USER */
+      });
     }
     this.realtimeProvider.on(event, callback);
   }
@@ -262,7 +308,12 @@ var CompositeVoice = class extends exports.MastraVoice {
    */
   off(event, callback) {
     if (!this.realtimeProvider) {
-      throw new Error("No realtime provider configured");
+      throw new chunkOLSQRA2V_cjs.MastraError({
+        id: "VOICE_COMPOSITE_NO_REALTIME_PROVIDER_OFF",
+        text: "No realtime provider configured",
+        domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+        category: "USER" /* USER */
+      });
     }
     this.realtimeProvider.off(event, callback);
   }
@@ -274,16 +325,36 @@ var DefaultVoice = class extends exports.MastraVoice {
     super();
   }
   async speak(_input) {
-    throw new Error("No voice provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_DEFAULT_NO_SPEAK_PROVIDER",
+      text: "No voice provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   async listen(_input) {
-    throw new Error("No voice provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_DEFAULT_NO_LISTEN_PROVIDER",
+      text: "No voice provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   async getSpeakers() {
-    throw new Error("No voice provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_DEFAULT_NO_SPEAKERS_PROVIDER",
+      text: "No voice provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
   async getListener() {
-    throw new Error("No voice provider configured");
+    throw new chunkOLSQRA2V_cjs.MastraError({
+      id: "VOICE_DEFAULT_NO_LISTENER_PROVIDER",
+      text: "No voice provider configured",
+      domain: "MASTRA_VOICE" /* MASTRA_VOICE */,
+      category: "USER" /* USER */
+    });
   }
 };