@mastra/core 0.9.4 → 0.10.0-alpha.1

package/dist/{base-H7siSAfu.d.cts → base-aPYtPBT2.d.cts}

@@ -1,6 +1,6 @@
 import * as _opentelemetry_api from '@opentelemetry/api';
 import { Tracer, Span, BaggageEntry, Context } from '@opentelemetry/api';
-import { R as RegisteredLogger, L as Logger } from './index-QV27jkxQ.cjs';
+import { R as RegisteredLogger, I as IMastraLogger } from './logger-EhZkzZOr.cjs';
 import { SpanExporter } from '@opentelemetry/sdk-trace-base';
 
 /** Sampling strategy configuration for OpenTelemetry */
@@ -115,7 +115,7 @@ declare class Telemetry {
 
 declare class MastraBase {
     component: RegisteredLogger;
-    protected logger: Logger;
+    protected logger: IMastraLogger;
     name?: string;
     telemetry?: Telemetry;
     constructor({ component, name }: {
@@ -126,7 +126,7 @@ declare class MastraBase {
      * Set the logger for the agent
      * @param logger
      */
-    __setLogger(logger: Logger): void;
+    __setLogger(logger: IMastraLogger): void;
     /**
      * Set the telemetry for the
      * @param telemetry

package/dist/{base-4Z34GQL8.d.ts → base-tc5kgDTD.d.ts}

@@ -1,6 +1,6 @@
 import * as _opentelemetry_api from '@opentelemetry/api';
 import { Tracer, Span, BaggageEntry, Context } from '@opentelemetry/api';
-import { R as RegisteredLogger, L as Logger } from './index-QV27jkxQ.js';
+import { R as RegisteredLogger, I as IMastraLogger } from './logger-EhZkzZOr.js';
 import { SpanExporter } from '@opentelemetry/sdk-trace-base';
 
 /** Sampling strategy configuration for OpenTelemetry */
@@ -115,7 +115,7 @@ declare class Telemetry {
 
 declare class MastraBase {
     component: RegisteredLogger;
-    protected logger: Logger;
+    protected logger: IMastraLogger;
     name?: string;
     telemetry?: Telemetry;
     constructor({ component, name }: {
@@ -126,7 +126,7 @@ declare class MastraBase {
      * Set the logger for the agent
      * @param logger
      */
-    __setLogger(logger: Logger): void;
+    __setLogger(logger: IMastraLogger): void;
     /**
      * Set the telemetry for the
      * @param telemetry

package/dist/base.cjs

@@ -1,10 +1,10 @@
 'use strict';
 
-var chunkEG74VBSP_cjs = require('./chunk-EG74VBSP.cjs');
+var chunkYEULQPUY_cjs = require('./chunk-YEULQPUY.cjs');
 
 
 
 Object.defineProperty(exports, "MastraBase", {
   enumerable: true,
-  get: function () { return chunkEG74VBSP_cjs.MastraBase; }
+  get: function () { return chunkYEULQPUY_cjs.MastraBase; }
 });

package/dist/base.d.cts

@@ -1,6 +1,5 @@
 import '@opentelemetry/api';
-import './index-QV27jkxQ.cjs';
-export { M as MastraBase } from './base-H7siSAfu.cjs';
+import './logger-EhZkzZOr.cjs';
+export { M as MastraBase } from './base-aPYtPBT2.cjs';
 import 'stream';
-import 'pino';
 import '@opentelemetry/sdk-trace-base';

package/dist/base.d.ts

@@ -1,6 +1,5 @@
 import '@opentelemetry/api';
-import './index-QV27jkxQ.js';
-export { M as MastraBase } from './base-4Z34GQL8.js';
+import './logger-EhZkzZOr.js';
+export { M as MastraBase } from './base-tc5kgDTD.js';
 import 'stream';
-import 'pino';
 import '@opentelemetry/sdk-trace-base';

package/dist/base.js

@@ -1 +1 @@
-export { MastraBase } from './chunk-235X76GC.js';
+export { MastraBase } from './chunk-FI7R232B.js';

package/dist/bundler/index.cjs

@@ -1,10 +1,10 @@
 'use strict';
 
-var chunkRGE4UF6A_cjs = require('../chunk-RGE4UF6A.cjs');
+var chunkTMPFLJLG_cjs = require('../chunk-TMPFLJLG.cjs');
 
 
 
 Object.defineProperty(exports, "MastraBundler", {
   enumerable: true,
-  get: function () { return chunkRGE4UF6A_cjs.MastraBundler; }
+  get: function () { return chunkTMPFLJLG_cjs.MastraBundler; }
 });

package/dist/bundler/index.d.cts

@@ -1,8 +1,7 @@
-import { M as MastraBase } from '../base-H7siSAfu.cjs';
+import { M as MastraBase } from '../base-aPYtPBT2.cjs';
 import '@opentelemetry/api';
-import '../index-QV27jkxQ.cjs';
+import '../logger-EhZkzZOr.cjs';
 import 'stream';
-import 'pino';
 import '@opentelemetry/sdk-trace-base';
 
 interface IBundler {

package/dist/bundler/index.d.ts

@@ -1,8 +1,7 @@
-import { M as MastraBase } from '../base-4Z34GQL8.js';
+import { M as MastraBase } from '../base-tc5kgDTD.js';
 import '@opentelemetry/api';
-import '../index-QV27jkxQ.js';
+import '../logger-EhZkzZOr.js';
 import 'stream';
-import 'pino';
 import '@opentelemetry/sdk-trace-base';
 
 interface IBundler {

package/dist/bundler/index.js

@@ -1 +1 @@
-export { MastraBundler } from '../chunk-NLBJOFNG.js';
+export { MastraBundler } from '../chunk-JW5TMK2L.js';

package/dist/{chunk-W4BSLENO.js → chunk-2NTUAUGH.js}

@@ -1,4 +1,4 @@
-import { MastraBundler } from './chunk-NLBJOFNG.js';
+import { MastraBundler } from './chunk-JW5TMK2L.js';
 
 // src/deployer/index.ts
 var MastraDeployer = class extends MastraBundler {

package/dist/{chunk-S34QVMKE.js → chunk-2PW6UJMW.js}

@@ -1,10 +1,10 @@
+import { MastraBase } from './chunk-FI7R232B.js';
 import { RuntimeContext } from './chunk-SGGPJWRQ.js';
-import { MastraBase } from './chunk-235X76GC.js';
 import { createHash } from 'crypto';
 import { convertToCoreMessages, jsonSchema, zodSchema } from 'ai';
 import jsonSchemaToZod from 'json-schema-to-zod';
 import { z } from 'zod';
-import { jsonSchemaObjectToZodRawShape } from 'zod-from-json-schema';
+import { convertJsonSchemaToZod } from 'zod-from-json-schema';
 import { zodToJsonSchema } from 'zod-to-json-schema';
 
 // src/tools/tool.ts
@@ -37,6 +37,16 @@ var ALL_NUMBER_CHECKS = [
 ];
 var ALL_ARRAY_CHECKS = ["min", "max", "length"];
 var UNSUPPORTED_ZOD_TYPES = ["ZodIntersection", "ZodNever", "ZodNull", "ZodTuple", "ZodUndefined"];
+var SUPPORTED_ZOD_TYPES = [
+  "ZodObject",
+  "ZodArray",
+  "ZodUnion",
+  "ZodString",
+  "ZodNumber",
+  "ZodDate",
+  "ZodAny",
+  "ZodDefault"
+];
 var ToolCompatibility = class extends MastraBase {
   model;
   constructor(model) {
@@ -270,6 +280,13 @@ var ToolCompatibility = class extends MastraBase {
     }
     return result;
   }
+  defaultZodOptionalHandler(value, handleTypes = SUPPORTED_ZOD_TYPES) {
+    if (handleTypes.includes(value._def.innerType._def.typeName)) {
+      return this.processZodType(value._def.innerType).optional();
+    } else {
+      return value;
+    }
+  }
   process(tool) {
     if (isVercelTool(tool)) {
       return {
@@ -299,6 +316,11 @@ var AnthropicToolCompat = class extends ToolCompatibility {
   }
   processZodType(value) {
     switch (value._def.typeName) {
+      case "ZodOptional":
+        const handleTypes = ["ZodObject", "ZodArray", "ZodUnion", "ZodNever", "ZodUndefined"];
+        if (this.getModel().modelId.includes("claude-3.5-haiku")) handleTypes.push("ZodString");
+        if (this.getModel().modelId.includes("claude-3.7")) handleTypes.push("ZodTuple");
+        return this.defaultZodOptionalHandler(value, handleTypes);
       case "ZodObject": {
         return this.defaultZodObjectHandler(value);
       }
@@ -340,6 +362,8 @@ var DeepSeekToolCompat = class extends ToolCompatibility {
   }
   processZodType(value) {
     switch (value._def.typeName) {
+      case "ZodOptional":
+        return this.defaultZodOptionalHandler(value, ["ZodObject", "ZodArray", "ZodUnion", "ZodString", "ZodNumber"]);
       case "ZodObject": {
         return this.defaultZodObjectHandler(value);
       }
@@ -371,6 +395,15 @@ var GoogleToolCompat = class extends ToolCompatibility {
   }
   processZodType(value) {
     switch (value._def.typeName) {
+      case "ZodOptional":
+        return this.defaultZodOptionalHandler(value, [
+          "ZodObject",
+          "ZodArray",
+          "ZodUnion",
+          "ZodString",
+          "ZodNumber",
+          ...UNSUPPORTED_ZOD_TYPES
+        ]);
       case "ZodObject": {
         return this.defaultZodObjectHandler(value);
       }
@@ -407,6 +440,8 @@ var MetaToolCompat = class extends ToolCompatibility {
   }
   processZodType(value) {
     switch (value._def.typeName) {
+      case "ZodOptional":
+        return this.defaultZodOptionalHandler(value, ["ZodObject", "ZodArray", "ZodUnion", "ZodString", "ZodNumber"]);
       case "ZodObject": {
         return this.defaultZodObjectHandler(value);
       }
@@ -444,6 +479,16 @@ var OpenAIToolCompat = class extends ToolCompatibility {
   }
   processZodType(value) {
     switch (value._def.typeName) {
+      case "ZodOptional":
+        return this.defaultZodOptionalHandler(value, [
+          "ZodObject",
+          "ZodArray",
+          "ZodUnion",
+          "ZodString",
+          "ZodNever",
+          "ZodUndefined",
+          "ZodTuple"
+        ]);
       case "ZodObject": {
         return this.defaultZodObjectHandler(value);
       }
@@ -466,8 +511,6 @@ var OpenAIToolCompat = class extends ToolCompatibility {
     }
   }
 };
-
-// src/tools/tool-compatibility/provider-compats/openai-reasoning.ts
 var OpenAIReasoningToolCompat = class extends ToolCompatibility {
   constructor(model) {
     super(model);
@@ -487,7 +530,8 @@ var OpenAIReasoningToolCompat = class extends ToolCompatibility {
   processZodType(value) {
     switch (value._def.typeName) {
       case "ZodOptional":
-        return value.unwrap().nullable();
+        const innerZodType = this.processZodType(value._def.innerType);
+        return innerZodType.nullable();
       case "ZodObject": {
         return this.defaultZodObjectHandler(value);
       }
@@ -521,6 +565,12 @@ var OpenAIReasoningToolCompat = class extends ToolCompatibility {
       case "ZodDate": {
         return this.defaultZodDateHandler(value);
       }
+      case "ZodAny": {
+        return z.string().describe(
+          (value.description ?? "") + `
+Argument was an "any" type, but you (the LLM) do not support "any", so it was cast to a "string" type`
+        );
+      }
       default:
         return this.defaultUnsupportedZodTypeHandler(value);
     }
@@ -549,8 +599,7 @@ function convertVercelToolParameters(tool) {
   } else {
     const jsonSchemaToConvert = "jsonSchema" in schema ? schema.jsonSchema : schema;
     try {
-      const rawShape = jsonSchemaObjectToZodRawShape(jsonSchemaToConvert);
-      return z.object(rawShape);
+      return convertJsonSchemaToZod(jsonSchemaToConvert);
     } catch (e) {
       const errorMessage = `[CoreToolBuilder] Failed to convert Vercel tool JSON schema parameters to Zod. Original schema: ${JSON.stringify(jsonSchemaToConvert)}`;
       console.error(errorMessage, e);
@@ -565,8 +614,7 @@ function convertInputSchema(tool) {
     return schema;
   } else {
     try {
-      const rawShape = jsonSchemaObjectToZodRawShape(schema);
-      return z.object(rawShape);
+      return convertJsonSchemaToZod(schema);
     } catch (e) {
       const errorMessage = `[CoreToolBuilder] Failed to convert tool input JSON schema to Zod. Original schema: ${JSON.stringify(schema)}`;
       console.error(errorMessage, e);
@@ -945,5 +993,24 @@ ${JSON.stringify(message, null, 2)}
     );
   }).flat();
 }
+var SQL_IDENTIFIER_PATTERN = /^[a-zA-Z_][a-zA-Z0-9_]*$/;
+function parseSqlIdentifier(name, kind = "identifier") {
+  if (!SQL_IDENTIFIER_PATTERN.test(name) || name.length > 63) {
+    throw new Error(
+      `Invalid ${kind}: ${name}. Must start with a letter or underscore, contain only letters, numbers, or underscores, and be at most 63 characters long.`
+    );
+  }
+  return name;
+}
+function parseFieldKey(key) {
+  if (!key) throw new Error("Field key cannot be empty");
+  const segments = key.split(".");
+  for (const segment of segments) {
+    if (!SQL_IDENTIFIER_PATTERN.test(segment) || segment.length > 63) {
+      throw new Error(`Invalid field key segment: ${segment} in ${key}`);
+    }
+  }
+  return key;
+}
 
-export { Tool, checkEvalStorageFields, createMastraProxy, createTool, deepMerge, delay, ensureAllMessagesAreCoreMessages, ensureToolProperties, isVercelTool, isZodType, makeCoreTool, maskStreamTags, resolveSerializedZodOutput };
+export { Tool, checkEvalStorageFields, createMastraProxy, createTool, deepMerge, delay, ensureAllMessagesAreCoreMessages, ensureToolProperties, isCoreMessage, isUiMessage, isVercelTool, isZodType, makeCoreTool, maskStreamTags, parseFieldKey, parseSqlIdentifier, resolveSerializedZodOutput };

package/dist/{chunk-WE36CRQV.cjs → chunk-2U7ZZUFO.cjs}

@@ -1,6 +1,6 @@
 'use strict';
 
-var chunk66GQROSI_cjs = require('./chunk-66GQROSI.cjs');
+var chunkAOGUTJUO_cjs = require('./chunk-AOGUTJUO.cjs');
 var cohereAi = require('cohere-ai');
 
 var CohereRelevanceScorer = class {
@@ -38,7 +38,7 @@ Relevance score (0-1):`;
 var MastraAgentRelevanceScorer = class {
   agent;
   constructor(name, model) {
-    this.agent = new chunk66GQROSI_cjs.Agent({
+    this.agent = new chunkAOGUTJUO_cjs.Agent({
       name: `Relevance Scorer ${name}`,
       instructions: `You are a specialized agent for evaluating the relevance of text to queries.
 Your task is to rate how well a text passage answers a given query.

package/dist/{workflows/vNext/index.js → chunk-3SQ3G2EJ.js}

@@ -1,14 +1,14 @@
-import { Agent } from '../../chunk-FW5RS6WV.js';
-import { Tool } from '../../chunk-S34QVMKE.js';
-import { RuntimeContext } from '../../chunk-SGGPJWRQ.js';
-import { MastraBase } from '../../chunk-235X76GC.js';
-import { RegisteredLogger } from '../../chunk-CMPPXW4U.js';
+import { Agent } from './chunk-E7CLY6SK.js';
+import { Tool } from './chunk-2PW6UJMW.js';
+import { MastraBase } from './chunk-FI7R232B.js';
+import { RegisteredLogger } from './chunk-VHLL4AZK.js';
+import { RuntimeContext } from './chunk-SGGPJWRQ.js';
 import { randomUUID } from 'crypto';
 import EventEmitter from 'events';
 import { z } from 'zod';
 import { context, trace } from '@opentelemetry/api';
 
-// src/workflows/vNext/execution-engine.ts
+// src/workflows/execution-engine.ts
 var ExecutionEngine = class extends MastraBase {
   mastra;
   constructor({ mastra }) {
@@ -20,7 +20,7 @@ var ExecutionEngine = class extends MastraBase {
   }
 };
 
-// src/workflows/vNext/default.ts
+// src/workflows/default.ts
 var DefaultExecutionEngine = class extends ExecutionEngine {
   async fmtReturnValue(executionSpan, emitter, stepResults, lastOutput, error) {
     const base = {
@@ -641,7 +641,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
   }
 };
 
-// src/workflows/vNext/workflow.ts
+// src/workflows/workflow.ts
 function createStep(params) {
   if (params instanceof Agent) {
     return {
@@ -705,10 +705,10 @@ function cloneStep(step, opts) {
   };
 }
 function createWorkflow(params) {
-  return new NewWorkflow(params);
+  return new Workflow(params);
 }
 function cloneWorkflow(workflow, opts) {
-  const wf = new NewWorkflow({
+  const wf = new Workflow({
     id: opts.id,
     inputSchema: workflow.inputSchema,
     outputSchema: workflow.outputSchema,
@@ -719,7 +719,7 @@ function cloneWorkflow(workflow, opts) {
   wf.commit();
   return wf;
 }
-var NewWorkflow = class extends MastraBase {
+var Workflow = class extends MastraBase {
   id;
   description;
   inputSchema;
@@ -1011,7 +1011,9 @@ var NewWorkflow = class extends MastraBase {
    */
   createRun(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.");
+      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.");
@@ -1035,14 +1037,15 @@ var NewWorkflow = class extends MastraBase {
     suspend,
     resume,
     emitter,
-    mastra
+    mastra,
+    runtimeContext
   }) {
     this.__registerMastra(mastra);
     const run = resume?.steps?.length ? this.createRun({ runId: resume.runId }) : this.createRun();
     const unwatch = run.watch((event) => {
       emitter.emit("nested-watch", { event, workflowId: this.id, runId: run.runId, isResume: !!resume?.steps?.length });
     });
-    const res = resume?.steps?.length ? await run.resume({ resumeData, step: resume.steps }) : await run.start({ inputData });
+    const res = resume?.steps?.length ? await run.resume({ resumeData, step: resume.steps, runtimeContext }) : await run.start({ inputData, runtimeContext });
     unwatch();
     const suspendedSteps = Object.entries(res.steps).filter(([_stepName, stepResult]) => {
       const stepRes = stepResult;
@@ -1155,7 +1158,7 @@ var Run = class {
         const { type, payload, eventTimestamp } = event;
         const prefixedSteps = Object.fromEntries(
           Object.entries(payload?.workflowState?.steps ?? {}).map(([stepId, step]) => [
-            `${this.workflowId}.${stepId}`,
+            `${workflowId}.${stepId}`,
             step
           ])
         );
@@ -1250,4 +1253,4 @@ function deepMerge(a, b) {
   return result;
 }
 
-export { DefaultExecutionEngine, ExecutionEngine, NewWorkflow, Run, cloneStep, cloneWorkflow, createStep, createWorkflow };
+export { DefaultExecutionEngine, ExecutionEngine, Run, Workflow, cloneStep, cloneWorkflow, createStep, createWorkflow };

package/dist/chunk-5HULBQ2W.cjs

@@ -0,0 +1,38 @@
+'use strict';
+
+// src/storage/storageWithInit.ts
+var isAugmentedSymbol = Symbol("isAugmented");
+function augmentWithInit(storage) {
+  let hasInitialized = null;
+  const ensureInit = async () => {
+    if (!hasInitialized) {
+      hasInitialized = storage.init();
+    }
+    await hasInitialized;
+  };
+  if (storage[isAugmentedSymbol]) {
+    return storage;
+  }
+  const proxy = new Proxy(storage, {
+    get(target, prop) {
+      const value = target[prop];
+      if (typeof value === "function" && prop !== "init") {
+        return async (...args) => {
+          await ensureInit();
+          return Reflect.apply(value, target, args);
+        };
+      }
+      return Reflect.get(target, prop);
+    }
+  });
+  Object.defineProperty(proxy, isAugmentedSymbol, {
+    value: true,
+    enumerable: false,
+    // Won't show up in Object.keys() or for...in loops
+    configurable: true
+    // Allows the property to be deleted or modified later if needed
+  });
+  return proxy;
+}
+
+exports.augmentWithInit = augmentWithInit;

package/dist/{chunk-XRGDVLM4.cjs → chunk-5JRD3NDP.cjs}

@@ -1,19 +1,19 @@
 'use strict';
 
-var chunkEJITEVAB_cjs = require('./chunk-EJITEVAB.cjs');
-var chunkEG74VBSP_cjs = require('./chunk-EG74VBSP.cjs');
-var chunkO44MSFMZ_cjs = require('./chunk-O44MSFMZ.cjs');
+var chunkU3L3NEOM_cjs = require('./chunk-U3L3NEOM.cjs');
+var chunkYEULQPUY_cjs = require('./chunk-YEULQPUY.cjs');
+var chunkSWW4EBUZ_cjs = require('./chunk-SWW4EBUZ.cjs');
 var ai = require('ai');
 var zod = require('zod');
 
 // src/llm/model/base.ts
-var MastraLLMBase = class extends chunkEG74VBSP_cjs.MastraBase {
+var MastraLLMBase = class extends chunkYEULQPUY_cjs.MastraBase {
   // @ts-ignore
   #mastra;
   #model;
   constructor({ name, model }) {
     super({
-      component: chunkO44MSFMZ_cjs.RegisteredLogger.LLM,
+      component: chunkSWW4EBUZ_cjs.RegisteredLogger.LLM,
       name
     });
     this.#model = model;
@@ -151,7 +151,7 @@ var MastraLLM = class extends MastraLLMBase {
       toolChoice,
       maxSteps,
       onStepFinish: async (props) => {
-        void onStepFinish?.(props);
+        await onStepFinish?.(props);
         this.logger.debug("[LLM] - Step Change:", {
           text: props?.text,
           toolCalls: props?.toolCalls,
@@ -162,7 +162,7 @@ var MastraLLM = class extends MastraLLMBase {
         });
         if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
           this.logger.warn("Rate limit approaching, waiting 10 seconds", { runId });
-          await chunkEJITEVAB_cjs.delay(10 * 1e3);
+          await chunkU3L3NEOM_cjs.delay(10 * 1e3);
         }
       },
       ...rest
@@ -220,7 +220,7 @@ var MastraLLM = class extends MastraLLMBase {
       maxSteps,
       toolChoice,
       onStepFinish: async (props) => {
-        void onStepFinish?.(props);
+        await onStepFinish?.(props);
         this.logger.debug("[LLM] - Step Change:", {
           text: props?.text,
           toolCalls: props?.toolCalls,
@@ -231,7 +231,7 @@ var MastraLLM = class extends MastraLLMBase {
         });
         if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
           this.logger.warn("Rate limit approaching, waiting 10 seconds", { runId });
-          await chunkEJITEVAB_cjs.delay(10 * 1e3);
+          await chunkU3L3NEOM_cjs.delay(10 * 1e3);
         }
       },
       ...rest
@@ -293,7 +293,7 @@ var MastraLLM = class extends MastraLLMBase {
       maxSteps,
       toolChoice,
       onStepFinish: async (props) => {
-        void onStepFinish?.(props);
+        await onStepFinish?.(props);
         this.logger.debug("[LLM] - Stream Step Change:", {
           text: props?.text,
           toolCalls: props?.toolCalls,
@@ -304,11 +304,11 @@ var MastraLLM = class extends MastraLLMBase {
         });
         if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
           this.logger.warn("Rate limit approaching, waiting 10 seconds", { runId });
-          await chunkEJITEVAB_cjs.delay(10 * 1e3);
+          await chunkU3L3NEOM_cjs.delay(10 * 1e3);
         }
       },
       onFinish: async (props) => {
-        void onFinish?.(props);
+        await onFinish?.(props);
         this.logger.debug("[LLM] - Stream Finished:", {
           text: props?.text,
           toolCalls: props?.toolCalls,
@@ -382,7 +382,7 @@ var MastraLLM = class extends MastraLLMBase {
       maxSteps,
       toolChoice,
       onStepFinish: async (props) => {
-        void onStepFinish?.(props);
+        await onStepFinish?.(props);
         this.logger.debug("[LLM] - Stream Step Change:", {
           text: props?.text,
           toolCalls: props?.toolCalls,
@@ -395,11 +395,11 @@ var MastraLLM = class extends MastraLLMBase {
         });
         if (props?.response?.headers?.["x-ratelimit-remaining-tokens"] && parseInt(props?.response?.headers?.["x-ratelimit-remaining-tokens"], 10) < 2e3) {
           this.logger.warn("Rate limit approaching, waiting 10 seconds", { runId });
-          await chunkEJITEVAB_cjs.delay(10 * 1e3);
+          await chunkU3L3NEOM_cjs.delay(10 * 1e3);
         }
       },
       onFinish: async (props) => {
-        void onFinish?.(props);
+        await onFinish?.(props);
         this.logger.debug("[LLM] - Stream Finished:", {
           text: props?.text,
           toolCalls: props?.toolCalls,