@mastra/core 0.15.3-alpha.3 → 0.15.3-alpha.4

package/dist/{chunk-DJKIK6ZB.cjs → chunk-KUNWELBC.cjs}

@@ -3,14 +3,14 @@
 var chunkFBBP67MQ_cjs = require('./chunk-FBBP67MQ.cjs');
 var chunkNFXTYMWZ_cjs = require('./chunk-NFXTYMWZ.cjs');
 var chunkZOU4K5MI_cjs = require('./chunk-ZOU4K5MI.cjs');
-var chunkQHEB6ZLO_cjs = require('./chunk-QHEB6ZLO.cjs');
-var chunkTWNFR6MQ_cjs = require('./chunk-TWNFR6MQ.cjs');
+var chunkF2CAC2R2_cjs = require('./chunk-F2CAC2R2.cjs');
+var chunk5CJDO3UO_cjs = require('./chunk-5CJDO3UO.cjs');
 var chunkTSNDVBUU_cjs = require('./chunk-TSNDVBUU.cjs');
-var chunk24TFSB6Z_cjs = require('./chunk-24TFSB6Z.cjs');
+var chunkLVGGMWSE_cjs = require('./chunk-LVGGMWSE.cjs');
 var chunkGPWMM745_cjs = require('./chunk-GPWMM745.cjs');
 var chunkS6MAHT7F_cjs = require('./chunk-S6MAHT7F.cjs');
-var chunkZAJTIZZF_cjs = require('./chunk-ZAJTIZZF.cjs');
-var chunk4DKPMUAC_cjs = require('./chunk-4DKPMUAC.cjs');
+var chunkABRPHTOG_cjs = require('./chunk-ABRPHTOG.cjs');
+var chunkI7OAONIW_cjs = require('./chunk-I7OAONIW.cjs');
 var chunkC73WLCY3_cjs = require('./chunk-C73WLCY3.cjs');
 var chunkKXCUCBEI_cjs = require('./chunk-KXCUCBEI.cjs');
 var chunkV5WKCX3G_cjs = require('./chunk-V5WKCX3G.cjs');
@@ -536,7 +536,7 @@ var ProcessorRunner = class {
     this.logger = logger;
     this.agentName = agentName;
   }
-  async runOutputProcessors(messageList, telemetry) {
+  async runOutputProcessors(messageList, tracingContext, telemetry) {
     const responseMessages = messageList.clear.response.v2();
     let processableMessages = [...responseMessages];
     const ctx = {
@@ -556,13 +556,15 @@ var ProcessorRunner = class {
       if (!telemetry) {
         processableMessages = await processMethod({
           messages: processableMessages,
-          abort: ctx.abort
+          abort: ctx.abort,
+          tracingContext
         });
       } else {
         await telemetry.traceMethod(async () => {
           processableMessages = await processMethod({
             messages: processableMessages,
-            abort: ctx.abort
+            abort: ctx.abort,
+            tracingContext
           });
           return processableMessages;
         }, {
@@ -583,7 +585,7 @@ var ProcessorRunner = class {
   /**
    * Process a stream part through all output processors with state management
    */
-  async processPart(part, processorStates) {
+  async processPart(part, processorStates, tracingContext) {
     if (!this.outputProcessors.length) {
       return {
         part,
@@ -607,7 +609,8 @@ var ProcessorRunner = class {
               state: state.customState,
               abort: reason => {
                 throw new TripWire(reason || `Stream part blocked by ${processor.name}`);
-              }
+              },
+              tracingContext
             });
             processedPart = result;
           }
@@ -634,7 +637,7 @@ var ProcessorRunner = class {
       };
     }
   }
-  async runOutputProcessorsForStream(streamResult) {
+  async runOutputProcessorsForStream(streamResult, tracingContext) {
     return new ReadableStream({
       start: async controller => {
         const reader = streamResult.fullStream.getReader();
@@ -653,7 +656,7 @@ var ProcessorRunner = class {
               part: processedPart,
               blocked,
               reason
-            } = await this.processPart(value, processorStates);
+            } = await this.processPart(value, processorStates, tracingContext);
             if (blocked) {
               void this.logger.debug(`[Agent:${this.agentName}] - Stream part blocked by output processor`, {
                 reason,
@@ -675,7 +678,7 @@ var ProcessorRunner = class {
       }
     });
   }
-  async runInputProcessors(messageList, telemetry) {
+  async runInputProcessors(messageList, tracingContext, telemetry) {
     const userMessages = messageList.clear.input.v2();
     let processableMessages = [...userMessages];
     const ctx = {
@@ -695,13 +698,15 @@ var ProcessorRunner = class {
       if (!telemetry) {
         processableMessages = await processMethod({
           messages: processableMessages,
-          abort: ctx.abort
+          abort: ctx.abort,
+          tracingContext
         });
       } else {
         await telemetry.traceMethod(async () => {
           processableMessages = await processMethod({
             messages: processableMessages,
-            abort: ctx.abort
+            abort: ctx.abort,
+            tracingContext
           });
           return processableMessages;
         }, {
@@ -1168,7 +1173,7 @@ function convertMastraChunkToAISDKv5({
       if (mode === "generate") {
         return {
           type: "file",
-          file: new chunkTWNFR6MQ_cjs.DefaultGeneratedFile({
+          file: new chunk5CJDO3UO_cjs.DefaultGeneratedFile({
             data: chunk.payload.data,
             mediaType: chunk.payload.mimeType
           })
@@ -1176,7 +1181,7 @@ function convertMastraChunkToAISDKv5({
       }
       return {
         type: "file",
-        file: new chunkTWNFR6MQ_cjs.DefaultGeneratedFileWithType({
+        file: new chunk5CJDO3UO_cjs.DefaultGeneratedFileWithType({
           data: chunk.payload.data,
           mediaType: chunk.payload.mimeType
         })
@@ -2220,7 +2225,7 @@ var MastraModelOutput = class extends chunkKXCUCBEI_cjs.MastraBase {
             try {
               if (self.processorRunner) {
                 await self.processorRunner.runOutputProcessors(self.messageList);
-                const outputText = self.messageList.get.response.aiV4.core().map(m => chunkTWNFR6MQ_cjs.MessageList.coreContentToString(m.content)).join("\n");
+                const outputText = self.messageList.get.response.aiV4.core().map(m => chunk5CJDO3UO_cjs.MessageList.coreContentToString(m.content)).join("\n");
                 const messages = self.messageList.get.response.v2();
                 const messagesWithStructuredData = messages.filter(msg => msg.content.metadata && msg.content.metadata.structuredOutput);
                 if (messagesWithStructuredData[0] && messagesWithStructuredData[0].content.metadata?.structuredOutput) {
@@ -3764,12 +3769,23 @@ function workflowLoopStream({
   modelSettings,
   _internal,
   modelStreamSpan,
+  llmAISpan,
   ...rest
 }) {
   return new web.ReadableStream({
     start: async controller => {
       const writer = new WritableStream({
         write: chunk => {
+          if (llmAISpan && chunk.type === "text-delta") {
+            llmAISpan.createEventSpan({
+              type: "llm_chunk" /* LLM_CHUNK */,
+              name: `llm chunk: ${chunk.type}`,
+              output: chunk.payload.text,
+              attributes: {
+                chunkType: chunk.type
+              }
+            });
+          }
           controller.enqueue(chunk);
         }
       });
@@ -3870,6 +3886,9 @@ function workflowLoopStream({
             user: rest.messageList.get.input.aiV5.model(),
             nonUser: []
           }
+        },
+        tracingContext: {
+          currentSpan: llmAISpan
         }
       });
       if (executionResult.status !== "success") {
@@ -3912,6 +3931,7 @@ function loop({
   _internal,
   mode = "stream",
   outputProcessors,
+  llmAISpan,
   ...rest
 }) {
   let loggerToUse = logger || new chunkV5WKCX3G_cjs.ConsoleLogger({
@@ -3969,6 +3989,7 @@ function loop({
     telemetry_settings,
     modelSettings,
     outputProcessors,
+    llmAISpan,
     ...rest
   };
   const streamFn = workflowLoopStream(workflowLoopProps);
@@ -4081,7 +4102,8 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
     output,
     options,
     outputProcessors,
-    providerOptions
+    providerOptions,
+    tracingContext
     // ...rest
   }) {
     let stopWhenToUse;
@@ -4101,8 +4123,22 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
     if (output) {
       output = this._applySchemaCompat(output);
     }
+    const llmAISpan = tracingContext?.currentSpan?.createChildSpan({
+      name: `llm stream: '${model.modelId}'`,
+      type: "llm_generation" /* LLM_GENERATION */,
+      input: messages,
+      attributes: {
+        model: model.modelId,
+        provider: model.provider,
+        streaming: true
+      },
+      metadata: {
+        threadId,
+        resourceId
+      }
+    });
     try {
-      const messageList = new chunkTWNFR6MQ_cjs.MessageList({
+      const messageList = new chunk5CJDO3UO_cjs.MessageList({
         threadId,
         resourceId
       });
@@ -4121,6 +4157,7 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
         },
         output,
         outputProcessors,
+        llmAISpan,
         options: {
           ...options,
           onStepFinish: async props => {
@@ -4161,7 +4198,7 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
               this.logger.warn("Rate limit approaching, waiting 10 seconds", {
                 runId
               });
-              await chunk24TFSB6Z_cjs.delay(10 * 1e3);
+              await chunkLVGGMWSE_cjs.delay(10 * 1e3);
             }
           },
           onFinish: async props => {
@@ -4203,7 +4240,11 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
           }
         }
       };
-      return loop(loopOptions);
+      const result = loop(loopOptions);
+      llmAISpan?.end({
+        output: result
+      });
+      return result;
     } catch (e) {
       const mastraError = new chunkC73WLCY3_cjs.MastraError({
         id: "LLM_STREAM_TEXT_AI_SDK_EXECUTION_FAILED",
@@ -4217,6 +4258,9 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
           resourceId: resourceId ?? "unknown"
         }
       }, e);
+      llmAISpan?.error({
+        error: mastraError
+      });
       throw mastraError;
     }
   }
@@ -4380,6 +4424,7 @@ function runScorer({
   input,
   output,
   runtimeContext,
+  tracingContext,
   entity,
   structuredOutput,
   source,
@@ -4412,6 +4457,7 @@ function runScorer({
     input,
     output,
     runtimeContext: Object.fromEntries(runtimeContext.entries()),
+    tracingContext,
     runId,
     source,
     entity,
@@ -5047,7 +5093,7 @@ var Machine = class extends EventEmitter__default.default {
         const logger = this.logger;
         let mastraProxy = void 0;
         if (this.#mastra) {
-          mastraProxy = chunk24TFSB6Z_cjs.createMastraProxy({
+          mastraProxy = chunkLVGGMWSE_cjs.createMastraProxy({
             mastra: this.#mastra,
             logger
           });
@@ -7625,7 +7671,8 @@ var ModerationProcessor = class _ModerationProcessor {
     try {
       const {
         messages,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (messages.length === 0) {
         return messages;
@@ -7637,7 +7684,7 @@ var ModerationProcessor = class _ModerationProcessor {
           passedMessages.push(message);
           continue;
         }
-        const moderationResult = await this.moderateContent(textContent);
+        const moderationResult = await this.moderateContent(textContent, false, tracingContext);
         if (this.isModerationFlagged(moderationResult)) {
           this.handleFlaggedContent(moderationResult, this.strategy, abort);
           if (this.strategy === "filter") {
@@ -7662,13 +7709,14 @@ var ModerationProcessor = class _ModerationProcessor {
       const {
         part,
         streamParts,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (part.type !== "text-delta") {
         return part;
       }
       const contentToModerate = this.buildContextFromChunks(streamParts);
-      const moderationResult = await this.moderateContent(contentToModerate, true);
+      const moderationResult = await this.moderateContent(contentToModerate, true, tracingContext);
       if (this.isModerationFlagged(moderationResult)) {
         this.handleFlaggedContent(moderationResult, this.strategy, abort);
         if (this.strategy === "filter") {
@@ -7687,7 +7735,7 @@ var ModerationProcessor = class _ModerationProcessor {
   /**
    * Moderate content using the internal agent
    */
-  async moderateContent(content, isStream = false) {
+  async moderateContent(content, isStream = false, tracingContext) {
     const prompt = this.createModerationPrompt(content, isStream);
     try {
       const model = await this.moderationAgent.getModel();
@@ -7704,12 +7752,14 @@ var ModerationProcessor = class _ModerationProcessor {
           output: schema,
           modelSettings: {
             temperature: 0
-          }
+          },
+          tracingContext
         });
       } else {
         response = await this.moderationAgent.generate(prompt, {
           output: schema,
-          temperature: 0
+          temperature: 0,
+          tracingContext
         });
       }
       const result = response.object;
@@ -7862,7 +7912,8 @@ var PromptInjectionDetector = class _PromptInjectionDetector {
     try {
       const {
         messages,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (messages.length === 0) {
         return messages;
@@ -7874,7 +7925,7 @@ var PromptInjectionDetector = class _PromptInjectionDetector {
           processedMessages.push(message);
           continue;
         }
-        const detectionResult = await this.detectPromptInjection(textContent);
+        const detectionResult = await this.detectPromptInjection(textContent, tracingContext);
         if (this.isInjectionFlagged(detectionResult)) {
           const processedMessage = this.handleDetectedInjection(message, detectionResult, this.strategy, abort);
           if (this.strategy === "filter") {
@@ -7899,7 +7950,7 @@ var PromptInjectionDetector = class _PromptInjectionDetector {
   /**
    * Detect prompt injection using the internal agent
    */
-  async detectPromptInjection(content) {
+  async detectPromptInjection(content, tracingContext) {
     const prompt = this.createDetectionPrompt(content);
     try {
       const model = await this.detectionAgent.getModel();
@@ -7917,12 +7968,14 @@ var PromptInjectionDetector = class _PromptInjectionDetector {
           output: schema,
           modelSettings: {
             temperature: 0
-          }
+          },
+          tracingContext
         });
       } else {
         response = await this.detectionAgent.generate(prompt, {
           output: schema,
-          temperature: 0
+          temperature: 0,
+          tracingContext
         });
       }
       const result = response.object;
@@ -8093,7 +8146,8 @@ var PIIDetector = class _PIIDetector {
     try {
       const {
         messages,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (messages.length === 0) {
         return messages;
@@ -8105,7 +8159,7 @@ var PIIDetector = class _PIIDetector {
           processedMessages.push(message);
           continue;
         }
-        const detectionResult = await this.detectPII(textContent);
+        const detectionResult = await this.detectPII(textContent, tracingContext);
         if (this.isPIIFlagged(detectionResult)) {
           const processedMessage = this.handleDetectedPII(message, detectionResult, this.strategy, abort);
           if (this.strategy === "filter") {
@@ -8132,7 +8186,7 @@ var PIIDetector = class _PIIDetector {
   /**
    * Detect PII using the internal agent
    */
-  async detectPII(content) {
+  async detectPII(content, tracingContext) {
     const prompt = this.createDetectionPrompt(content);
     const schema = z__default.default.object({
       categories: z__default.default.object(this.detectionTypes.reduce((props, type) => {
@@ -8157,12 +8211,14 @@ var PIIDetector = class _PIIDetector {
           output: schema,
           modelSettings: {
             temperature: 0
-          }
+          },
+          tracingContext
         });
       } else {
         response = await this.detectionAgent.generate(prompt, {
           output: schema,
-          temperature: 0
+          temperature: 0,
+          tracingContext
         });
       }
       const result = response.object;
@@ -8356,7 +8412,8 @@ IMPORTANT: IF NO PII IS DETECTED, RETURN AN EMPTY OBJECT, DO NOT INCLUDE ANYTHIN
   async processOutputStream(args) {
     const {
       part,
-      abort
+      abort,
+      tracingContext
     } = args;
     try {
       if (part.type !== "text-delta") {
@@ -8366,7 +8423,7 @@ IMPORTANT: IF NO PII IS DETECTED, RETURN AN EMPTY OBJECT, DO NOT INCLUDE ANYTHIN
       if (!textContent.trim()) {
         return part;
       }
-      const detectionResult = await this.detectPII(textContent);
+      const detectionResult = await this.detectPII(textContent, tracingContext);
       if (this.isPIIFlagged(detectionResult)) {
         switch (this.strategy) {
           case "block":
@@ -9205,13 +9262,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
       this.logger.error(mastraError.toString());
       throw mastraError;
     }
-    return chunk24TFSB6Z_cjs.ensureToolProperties(this.#tools);
+    return chunkLVGGMWSE_cjs.ensureToolProperties(this.#tools);
   }
   getTools({
     runtimeContext = new chunkGPWMM745_cjs.RuntimeContext()
   } = {}) {
     if (typeof this.#tools !== "function") {
-      return chunk24TFSB6Z_cjs.ensureToolProperties(this.#tools);
+      return chunkLVGGMWSE_cjs.ensureToolProperties(this.#tools);
     }
     const result = this.#tools({
       runtimeContext,
@@ -9232,7 +9289,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         this.logger.error(mastraError.toString());
         throw mastraError;
       }
-      return chunk24TFSB6Z_cjs.ensureToolProperties(tools);
+      return chunkLVGGMWSE_cjs.ensureToolProperties(tools);
     });
   }
   get llm() {
@@ -9276,7 +9333,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           mastra: this.#mastra
         });
       } else {
-        llm = new chunkQHEB6ZLO_cjs.MastraLLMV1({
+        llm = new chunkF2CAC2R2_cjs.MastraLLMV1({
           model: resolvedModel,
           mastra: this.#mastra
         });
@@ -9384,6 +9441,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
   async generateTitleFromUserMessage({
     message,
     runtimeContext = new chunkGPWMM745_cjs.RuntimeContext(),
+    tracingContext,
     model,
     instructions
   }) {
@@ -9391,7 +9449,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
       runtimeContext,
       model
     });
-    const normMessage = new chunkTWNFR6MQ_cjs.MessageList().add(message, "user").get.all.ui().at(-1);
+    const normMessage = new chunk5CJDO3UO_cjs.MessageList().add(message, "user").get.all.ui().at(-1);
     if (!normMessage) {
       throw new Error(`Could not generate title from input ${JSON.stringify(message)}`);
     }
@@ -9416,6 +9474,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     if (llm.getModel().specificationVersion === "v2") {
       const result = llm.stream({
         runtimeContext,
+        tracingContext,
         messages: [{
           role: "system",
           content: systemInstructions
@@ -9428,6 +9487,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     } else {
       const result = await llm.__text({
         runtimeContext,
+        tracingContext,
         messages: [{
           role: "system",
           content: systemInstructions
@@ -9445,14 +9505,15 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     const userMessages = messages.filter(message => message.role === "user");
     return userMessages.at(-1);
   }
-  async genTitle(userMessage, runtimeContext, model, instructions) {
+  async genTitle(userMessage, runtimeContext, tracingContext, model, instructions) {
     try {
       if (userMessage) {
-        const normMessage = new chunkTWNFR6MQ_cjs.MessageList().add(userMessage, "user").get.all.ui().at(-1);
+        const normMessage = new chunk5CJDO3UO_cjs.MessageList().add(userMessage, "user").get.all.ui().at(-1);
         if (normMessage) {
           return await this.generateTitleFromUserMessage({
             message: normMessage,
             runtimeContext,
+            tracingContext,
             model,
             instructions
           });
@@ -9473,7 +9534,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     runId,
     userMessages,
     systemMessage,
-    messageList = new chunkTWNFR6MQ_cjs.MessageList({
+    messageList = new chunk5CJDO3UO_cjs.MessageList({
       threadId,
       resourceId
     }),
@@ -9528,7 +9589,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         systemMessage: systemMessages,
         memorySystemMessage: memorySystemMessage || void 0
       });
-      const returnList = new chunkTWNFR6MQ_cjs.MessageList().addSystem(systemMessages).add(processedMemoryMessages, "memory").add(newMessages, "user");
+      const returnList = new chunk5CJDO3UO_cjs.MessageList().addSystem(systemMessages).add(processedMemoryMessages, "memory").add(newMessages, "user");
       return {
         threadId: thread.id,
         messages: returnList.get.all.prompt()
@@ -9544,8 +9605,8 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     resourceId,
     threadId,
     runtimeContext,
-    mastraProxy,
-    agentAISpan
+    tracingContext,
+    mastraProxy
   }) {
     let convertedMemoryTools = {};
     const memory = await this.getMemory({
@@ -9568,12 +9629,12 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           memory,
           agentName: this.name,
           runtimeContext,
+          tracingContext,
           model: typeof this.model === "function" ? await this.getModel({
             runtimeContext
-          }) : this.model,
-          agentAISpan
+          }) : this.model
         };
-        const convertedToCoreTool = chunk24TFSB6Z_cjs.makeCoreTool(toolObj, options);
+        const convertedToCoreTool = chunkLVGGMWSE_cjs.makeCoreTool(toolObj, options);
         convertedMemoryTools[toolName] = convertedToCoreTool;
       }
     }
@@ -9581,6 +9642,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
   }
   async __runInputProcessors({
     runtimeContext,
+    tracingContext,
     messageList,
     inputProcessorOverrides
   }) {
@@ -9591,13 +9653,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         runtimeContext,
         inputProcessorOverrides
       });
-      const tracedRunInputProcessors = messageList2 => {
+      const tracedRunInputProcessors = (messageList2, tracingContext2) => {
         const telemetry = this.#mastra?.getTelemetry();
         if (!telemetry) {
-          return runner.runInputProcessors(messageList2, void 0);
+          return runner.runInputProcessors(messageList2, tracingContext2, void 0);
         }
         return telemetry.traceMethod(async data => {
-          return runner.runInputProcessors(data.messageList, telemetry);
+          return runner.runInputProcessors(data.messageList, tracingContext2, telemetry);
         }, {
           spanName: `agent.${this.name}.inputProcessors`,
           attributes: {
@@ -9610,7 +9672,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         });
       };
       try {
-        messageList = await tracedRunInputProcessors(messageList);
+        messageList = await tracedRunInputProcessors(messageList, tracingContext);
       } catch (error) {
         if (error instanceof TripWire) {
           tripwireTriggered = true;
@@ -9633,6 +9695,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
   }
   async __runOutputProcessors({
     runtimeContext,
+    tracingContext,
     messageList,
     outputProcessorOverrides
   }) {
@@ -9643,13 +9706,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         runtimeContext,
         outputProcessorOverrides
       });
-      const tracedRunOutputProcessors = messageList2 => {
+      const tracedRunOutputProcessors = (messageList2, tracingContext2) => {
         const telemetry = this.#mastra?.getTelemetry();
         if (!telemetry) {
-          return runner.runOutputProcessors(messageList2, void 0);
+          return runner.runOutputProcessors(messageList2, tracingContext2, void 0);
         }
         return telemetry.traceMethod(async data => {
-          return runner.runOutputProcessors(data.messageList, telemetry);
+          return runner.runOutputProcessors(data.messageList, tracingContext2, telemetry);
         }, {
           spanName: `agent.${this.name}.outputProcessors`,
           attributes: {
@@ -9662,7 +9725,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         });
       };
       try {
-        messageList = await tracedRunOutputProcessors(messageList);
+        messageList = await tracedRunOutputProcessors(messageList, tracingContext);
       } catch (e) {
         if (e instanceof TripWire) {
           tripwireTriggered = true;
@@ -9701,13 +9764,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     }).then(r => r.messagesV2);
   }
   async getAssignedTools({
-    runtimeContext,
     runId,
     resourceId,
     threadId,
+    runtimeContext,
+    tracingContext,
     mastraProxy,
-    writableStream,
-    agentAISpan
+    writableStream
   }) {
     let toolsForRequest = {};
     this.logger.debug(`[Agents:${this.name}] - Assembling assigned tools`, {
@@ -9736,13 +9799,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         memory,
         agentName: this.name,
         runtimeContext,
+        tracingContext,
         model: typeof this.model === "function" ? await this.getModel({
           runtimeContext
         }) : this.model,
-        writableStream,
-        agentAISpan
+        writableStream
       };
-      return [k, chunk24TFSB6Z_cjs.makeCoreTool(tool, options)];
+      return [k, chunkLVGGMWSE_cjs.makeCoreTool(tool, options)];
     }));
     const assignedToolEntriesConverted = Object.fromEntries(assignedCoreToolEntries.filter(entry => Boolean(entry)));
     toolsForRequest = {
@@ -9756,8 +9819,8 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     resourceId,
     toolsets,
     runtimeContext,
-    mastraProxy,
-    agentAISpan
+    tracingContext,
+    mastraProxy
   }) {
     let toolsForRequest = {};
     const memory = await this.getMemory({
@@ -9781,12 +9844,12 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
             memory,
             agentName: this.name,
             runtimeContext,
+            tracingContext,
             model: typeof this.model === "function" ? await this.getModel({
               runtimeContext
-            }) : this.model,
-            agentAISpan
+            }) : this.model
           };
-          const convertedToCoreTool = chunk24TFSB6Z_cjs.makeCoreTool(toolObj, options, "toolset");
+          const convertedToCoreTool = chunkLVGGMWSE_cjs.makeCoreTool(toolObj, options, "toolset");
           toolsForRequest[toolName] = convertedToCoreTool;
         }
       }
@@ -9798,9 +9861,9 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     threadId,
     resourceId,
     runtimeContext,
+    tracingContext,
     mastraProxy,
-    clientTools,
-    agentAISpan
+    clientTools
   }) {
     let toolsForRequest = {};
     const memory = await this.getMemory({
@@ -9826,12 +9889,12 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           memory,
           agentName: this.name,
           runtimeContext,
+          tracingContext,
           model: typeof this.model === "function" ? await this.getModel({
             runtimeContext
-          }) : this.model,
-          agentAISpan
+          }) : this.model
         };
-        const convertedToCoreTool = chunk24TFSB6Z_cjs.makeCoreTool(rest, options, "client-tool");
+        const convertedToCoreTool = chunkLVGGMWSE_cjs.makeCoreTool(rest, options, "client-tool");
         toolsForRequest[toolName] = convertedToCoreTool;
       }
     }
@@ -9842,7 +9905,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     threadId,
     resourceId,
     runtimeContext,
-    agentAISpan
+    tracingContext
   }) {
     let convertedWorkflowTools = {};
     const workflows = await this.getWorkflows({
@@ -9859,7 +9922,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           // manually wrap workflow tools with ai tracing, so that we can pass the
           // current tool span onto the workflow to maintain continuity of the trace
           execute: async args => {
-            const toolAISpan = agentAISpan?.createChildSpan({
+            const toolAISpan = tracingContext.currentSpan?.createChildSpan({
               type: "tool_call" /* TOOL_CALL */,
               name: `tool: '${workflowName}'`,
               input: args,
@@ -9881,7 +9944,9 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
               const result = await run.start({
                 inputData: args,
                 runtimeContext,
-                currentSpan: toolAISpan
+                tracingContext: {
+                  currentSpan: toolAISpan
+                }
               });
               toolAISpan?.end({
                 output: result
@@ -9921,13 +9986,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     resourceId,
     runId,
     runtimeContext,
-    writableStream,
-    agentAISpan
+    tracingContext,
+    writableStream
   }) {
     let mastraProxy = void 0;
     const logger = this.logger;
     if (this.#mastra) {
-      mastraProxy = chunk24TFSB6Z_cjs.createMastraProxy({
+      mastraProxy = chunkLVGGMWSE_cjs.createMastraProxy({
         mastra: this.#mastra,
         logger
       });
@@ -9937,42 +10002,42 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
       resourceId,
       threadId,
       runtimeContext,
+      tracingContext,
       mastraProxy,
-      writableStream,
-      agentAISpan
+      writableStream
     });
     const memoryTools = await this.getMemoryTools({
       runId,
       resourceId,
       threadId,
       runtimeContext,
-      mastraProxy,
-      agentAISpan
+      tracingContext,
+      mastraProxy
     });
     const toolsetTools = await this.getToolsets({
       runId,
       resourceId,
       threadId,
       runtimeContext,
+      tracingContext,
       mastraProxy,
-      toolsets,
-      agentAISpan
+      toolsets
     });
     const clientSideTools = await this.getClientTools({
       runId,
       resourceId,
       threadId,
       runtimeContext,
+      tracingContext,
       mastraProxy,
-      clientTools,
-      agentAISpan
+      clientTools
     });
     const workflowTools = await this.getWorkflowTools({
       runId,
       resourceId,
       threadId,
       runtimeContext,
-      agentAISpan
+      tracingContext
     });
     return this.formatTools({
       ...assignedTools,
@@ -10057,7 +10122,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     runtimeContext,
     saveQueueManager,
     writableStream,
-    currentSpan
+    tracingContext
   }) {
     return {
       before: async () => {
@@ -10066,8 +10131,10 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
             runId
           });
         }
-        const spanArgs = {
+        const agentAISpan = chunkI7OAONIW_cjs.getOrCreateSpan({
+          type: "agent_run" /* AGENT_RUN */,
           name: `agent run: '${this.id}'`,
+          input: messages,
           attributes: {
             agentId: this.id,
             instructions,
@@ -10077,28 +10144,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
             runId,
             resourceId,
             threadId: thread ? thread.id : void 0
-          }
+          },
+          tracingContext,
+          runtimeContext
+        });
+        const innerTracingContext = {
+          currentSpan: agentAISpan
         };
-        let agentAISpan;
-        if (currentSpan) {
-          agentAISpan = currentSpan.createChildSpan({
-            type: "agent_run" /* AGENT_RUN */,
-            ...spanArgs
-          });
-        } else {
-          const aiTracing = chunk4DKPMUAC_cjs.getSelectedAITracing({
-            runtimeContext
-          });
-          if (aiTracing) {
-            agentAISpan = aiTracing.startSpan({
-              type: "agent_run" /* AGENT_RUN */,
-              ...spanArgs,
-              startOptions: {
-                runtimeContext
-              }
-            });
-          }
-        }
         const memory = await this.getMemory({
           runtimeContext
         });
@@ -10122,10 +10174,10 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           resourceId,
           runId,
           runtimeContext,
-          writableStream,
-          agentAISpan
+          tracingContext: innerTracingContext,
+          writableStream
         });
-        const messageList = new chunkTWNFR6MQ_cjs.MessageList({
+        const messageList = new chunk5CJDO3UO_cjs.MessageList({
           threadId,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10142,6 +10194,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
             tripwireReason: tripwireReason2
           } = await this.__runInputProcessors({
             runtimeContext,
+            tracingContext: innerTracingContext,
             messageList
           });
           return {
@@ -10212,7 +10265,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         let [memoryMessages, memorySystemMessage] = await Promise.all([existingThread ? this.getMemoryMessages({
           resourceId,
           threadId: threadObject.id,
-          vectorMessageSearch: new chunkTWNFR6MQ_cjs.MessageList().add(messages, `user`).getLatestUserContent() || "",
+          vectorMessageSearch: new chunk5CJDO3UO_cjs.MessageList().add(messages, `user`).getLatestUserContent() || "",
           memoryConfig,
           runtimeContext
         }) : [], memory.getSystemMessage({
@@ -10235,7 +10288,7 @@ The following messages were remembered from a different conversation:
 <remembered_from_other_conversation>
 ${(() => {
             let result = ``;
-            const messages2 = new chunkTWNFR6MQ_cjs.MessageList().add(resultsFromOtherThreads, "memory").get.all.v1();
+            const messages2 = new chunk5CJDO3UO_cjs.MessageList().add(resultsFromOtherThreads, "memory").get.all.v1();
             let lastYmd = null;
             for (const msg of messages2) {
               const date = msg.createdAt;
@@ -10274,6 +10327,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           tripwireReason
         } = await this.__runInputProcessors({
           runtimeContext,
+          tracingContext: innerTracingContext,
           messageList
         });
         const systemMessage = [...messageList.getSystemMessages(), ...messageList.getSystemMessages("memory")]?.map(m => m.content)?.join(`
@@ -10287,7 +10341,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           systemMessage,
           memorySystemMessage: memorySystemMessage || void 0
         });
-        const processedList = new chunkTWNFR6MQ_cjs.MessageList({
+        const processedList = new chunk5CJDO3UO_cjs.MessageList({
           threadId: threadObject.id,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10338,23 +10392,12 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
             };
           })
         };
-        agentAISpan?.end({
-          output: {
-            text: result?.text,
-            object: result?.object
-          },
-          metadata: {
-            usage: result?.usage,
-            toolResults: result?.toolResults,
-            toolCalls: result?.toolCalls
-          }
-        });
         this.logger.debug(`[Agent:${this.name}] - Post processing LLM response`, {
           runId: runId2,
           result: resToLog,
           threadId
         });
-        const messageListResponses = new chunkTWNFR6MQ_cjs.MessageList({
+        const messageListResponses = new chunk5CJDO3UO_cjs.MessageList({
           threadId,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10410,7 +10453,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
                 instructions: titleInstructions
               } = this.resolveTitleGenerationConfig(config?.threads?.generateTitle);
               if (shouldGenerate && userMessage) {
-                promises.push(this.genTitle(userMessage, runtimeContext, titleModel, titleInstructions).then(title => {
+                promises.push(this.genTitle(userMessage, runtimeContext, {
+                  currentSpan: agentAISpan
+                }, titleModel, titleInstructions).then(title => {
                   if (title) {
                     return memory.createThread({
                       threadId: thread2.id,
@@ -10427,6 +10472,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           } catch (e) {
             await saveQueueManager.flushMessages(messageList, threadId, memoryConfig2);
             if (e instanceof chunkC73WLCY3_cjs.MastraError) {
+              agentAISpan?.error({
+                error: e
+              });
               throw e;
             }
             const mastraError = new chunkC73WLCY3_cjs.MastraError({
@@ -10442,6 +10490,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
             }, e);
             this.logger.trackException(mastraError);
             this.logger.error(mastraError.toString());
+            agentAISpan?.error({
+              error: mastraError
+            });
             throw mastraError;
           }
         } else {
@@ -10466,6 +10517,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           outputText,
           instructions,
           runtimeContext,
+          tracingContext: {
+            currentSpan: agentAISpan
+          },
           structuredOutput,
           overrideScorers,
           threadId,
@@ -10480,6 +10534,17 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           },
           output: messageList.getPersisted.response.ui()
         };
+        agentAISpan?.end({
+          output: {
+            text: result?.text,
+            object: result?.object
+          },
+          metadata: {
+            usage: result?.usage,
+            toolResults: result?.toolResults,
+            toolCalls: result?.toolCalls
+          }
+        });
         return {
           scoringData
         };
@@ -10492,6 +10557,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     outputText,
     instructions,
     runtimeContext,
+    tracingContext,
     structuredOutput,
     overrideScorers,
     threadId,
@@ -10513,9 +10579,15 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         });
       }
     }
-    const scorers = overrideScorers ?? (await this.getScorers({
-      runtimeContext
-    }));
+    let scorers = {};
+    try {
+      scorers = overrideScorers ? this.resolveOverrideScorerReferences(overrideScorers) : await this.getScorers({
+        runtimeContext
+      });
+    } catch (e) {
+      this.logger.warn(`[Agent:${this.name}] - Failed to get scorers: ${e}`);
+      return;
+    }
     const scorerInput = {
       inputMessages: messageList.getPersisted.input.ui(),
       rememberedMessages: messageList.getPersisted.remembered.ui(),
@@ -10526,12 +10598,13 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     if (Object.keys(scorers || {}).length > 0) {
       for (const [id, scorerObject] of Object.entries(scorers)) {
         runScorer({
-          scorerId: id,
+          scorerId: overrideScorers ? scorerObject.scorer.name : id,
           scorerObject,
           runId,
           input: scorerInput,
           output: scorerOutput,
           runtimeContext,
+          tracingContext,
           entity: {
             id: this.id,
             name: this.name
@@ -10545,6 +10618,41 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       }
     }
   }
+  resolveOverrideScorerReferences(overrideScorers) {
+    const result = {};
+    for (const [id, scorerObject] of Object.entries(overrideScorers)) {
+      if (typeof scorerObject.scorer === "string") {
+        try {
+          if (!this.#mastra) {
+            throw new chunkC73WLCY3_cjs.MastraError({
+              id: "AGENT_GENEREATE_SCORER_NOT_FOUND",
+              domain: "AGENT" /* AGENT */,
+              category: "USER" /* USER */,
+              text: `Mastra not found when fetching scorer. Make sure to fetch agent from mastra.getAgent()`
+            });
+          }
+          const scorer = this.#mastra.getScorerByName(scorerObject.scorer);
+          result[id] = {
+            scorer,
+            sampling: scorerObject.sampling
+          };
+        } catch (error) {
+          this.logger.warn(`[Agent:${this.name}] - Failed to get scorer ${scorerObject.scorer}: ${error}`);
+        }
+      } else {
+        result[id] = scorerObject;
+      }
+    }
+    if (Object.keys(result).length === 0) {
+      throw new chunkC73WLCY3_cjs.MastraError({
+        id: "AGENT_GENEREATE_SCORER_NOT_FOUND",
+        domain: "AGENT" /* AGENT */,
+        category: "USER" /* USER */,
+        text: `No scorers found in overrideScorers`
+      });
+    }
+    return result;
+  }
   async prepareLLMOptions(messages, options) {
     const {
       context,
@@ -10557,6 +10665,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       temperature,
       toolChoice = "auto",
       runtimeContext = new chunkGPWMM745_cjs.RuntimeContext(),
+      tracingContext,
       savePerStep,
       writableStream,
       ...args
@@ -10621,7 +10730,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       runtimeContext,
       saveQueueManager,
       writableStream,
-      currentSpan: args.tracingContext?.currentSpan
+      tracingContext
     });
     let messageList;
     let thread;
@@ -10688,7 +10797,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         result,
         outputText,
         structuredOutput = false,
-        agentAISpan
+        agentAISpan,
+        overrideScorers
       }) => {
         const afterResult = await after({
           result,
@@ -10700,7 +10810,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           messageList,
           structuredOutput,
           threadExists,
-          agentAISpan
+          agentAISpan,
+          overrideScorers
         });
         return afterResult;
       }
@@ -10724,6 +10835,22 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     const instructions = options.instructions || (await this.getInstructions({
       runtimeContext
     }));
+    const agentAISpan = chunkI7OAONIW_cjs.getOrCreateSpan({
+      type: "agent_run" /* AGENT_RUN */,
+      name: `agent run: '${this.id}'`,
+      input: options.messages,
+      attributes: {
+        agentId: this.id,
+        instructions
+      },
+      metadata: {
+        runId,
+        resourceId,
+        threadId: threadFromArgs ? threadFromArgs.id : void 0
+      },
+      tracingContext: options.tracingContext,
+      runtimeContext
+    });
     const activeSpan = chunkZOU4K5MI_cjs.Telemetry.getActiveSpan();
     const baggageEntries = {};
     if (threadFromArgs?.id) {
@@ -10784,7 +10911,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           resourceId,
           runId,
           runtimeContext,
-          writableStream: options.writableStream
+          writableStream: options.writableStream,
+          tracingContext: {
+            currentSpan: agentAISpan
+          }
         });
         return {
           convertedTools
@@ -10802,9 +10932,11 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         tripwire: z.z.boolean().optional(),
         tripwireReason: z.z.string().optional()
       }),
-      execute: async () => {
+      execute: async ({
+        tracingContext
+      }) => {
         const thread = threadFromArgs;
-        const messageList = new chunkTWNFR6MQ_cjs.MessageList({
+        const messageList = new chunk5CJDO3UO_cjs.MessageList({
           threadId: thread?.id,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10821,6 +10953,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
             tripwireReason: tripwireReason2
           } = await this.__runInputProcessors({
             runtimeContext,
+            tracingContext,
             messageList
           });
           return {
@@ -10886,7 +11019,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         let [memoryMessages, memorySystemMessage] = await Promise.all([existingThread ? this.getMemoryMessages({
           resourceId,
           threadId: threadObject.id,
-          vectorMessageSearch: new chunkTWNFR6MQ_cjs.MessageList().add(options.messages, `user`).getLatestUserContent() || "",
+          vectorMessageSearch: new chunk5CJDO3UO_cjs.MessageList().add(options.messages, `user`).getLatestUserContent() || "",
           memoryConfig,
           runtimeContext
         }) : [], memory.getSystemMessage({
@@ -10908,8 +11041,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
 The following messages were remembered from a different conversation:
 <remembered_from_other_conversation>
 ${(() => {
-            let result = ``;
-            const messages = new chunkTWNFR6MQ_cjs.MessageList().add(resultsFromOtherThreads, "memory").get.all.v1();
+            let result2 = ``;
+            const messages = new chunk5CJDO3UO_cjs.MessageList().add(resultsFromOtherThreads, "memory").get.all.v1();
             let lastYmd = null;
             for (const msg of messages) {
               const date = msg.createdAt;
@@ -10925,15 +11058,14 @@ ${(() => {
               const ampm = utcHour < 12 ? "AM" : "PM";
               const timeofday = `${hour12}:${utcMinute < 10 ? "0" : ""}${utcMinute} ${ampm}`;
               if (!lastYmd || lastYmd !== ymd) {
-                result += `
+                result2 += `
 the following messages are from ${ymd}
 `;
               }
-              result += `
-Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conversation" : ""} at ${timeofday}: ${JSON.stringify(msg)}`;
+              result2 += `Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conversation" : ""} at ${timeofday}: ${JSON.stringify(msg)}`;
               lastYmd = ymd;
             }
-            return result;
+            return result2;
           })()}
 <end_remembered_from_other_conversation>`;
         }
@@ -10948,6 +11080,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           tripwireReason
         } = await this.__runInputProcessors({
           runtimeContext,
+          tracingContext,
           messageList
         });
         const systemMessage = [...messageList.getSystemMessages(), ...messageList.getSystemMessages("memory")]?.map(m => m.content)?.join(`
@@ -10961,7 +11094,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           systemMessage,
           memorySystemMessage: memorySystemMessage || void 0
         });
-        const processedList = new chunkTWNFR6MQ_cjs.MessageList({
+        const processedList = new chunk5CJDO3UO_cjs.MessageList({
           threadId: threadObject.id,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10986,7 +11119,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       inputSchema: z.z.any(),
       outputSchema: z.z.any(),
       execute: async ({
-        inputData
+        inputData,
+        tracingContext
       }) => {
         this.logger.debug(`Starting agent ${this.name} llm stream call`, {
           runId
@@ -10996,7 +11130,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         }) : this.#outputProcessors : []);
         const streamResult = llm.stream({
           ...inputData,
-          outputProcessors
+          outputProcessors,
+          tracingContext
         });
         if (options.format === "aisdk") {
           return streamResult.aisdk.v5;
@@ -11011,9 +11146,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       steps: [prepareToolsStep, prepareMemory]
     }).parallel([prepareToolsStep, prepareMemory]).map(async ({
       inputData,
-      bail
+      bail,
+      tracingContext
     }) => {
-      const result = {
+      const result2 = {
         ...options,
         messages: inputData["prepare-memory-step"].messageObjects,
         tools: inputData["prepare-tools-step"].convertedTools,
@@ -11055,7 +11191,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           tripwireReason: inputData["prepare-memory-step"].tripwireReason
         })
       };
-      if (result.tripwire) {
+      if (result2.tripwire) {
         const emptyResult = {
           textStream: async function* () {}(),
           fullStream: new globalThis.ReadableStream({
@@ -11076,7 +11212,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           }),
           finishReason: Promise.resolve("other"),
           tripwire: true,
-          tripwireReason: result.tripwireReason,
+          tripwireReason: result2.tripwireReason,
           response: {
             id: crypto2.randomUUID(),
             timestamp: /* @__PURE__ */new Date(),
@@ -11099,23 +11235,27 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
         return bail(emptyResult);
       }
       let effectiveOutputProcessors = options.outputProcessors || (this.#outputProcessors ? typeof this.#outputProcessors === "function" ? await this.#outputProcessors({
-        runtimeContext: result.runtimeContext
+        runtimeContext: result2.runtimeContext
       }) : this.#outputProcessors : []);
       if (options.structuredOutput) {
         const structuredProcessor = new StructuredOutputProcessor(options.structuredOutput);
         effectiveOutputProcessors = effectiveOutputProcessors ? [...effectiveOutputProcessors, structuredProcessor] : [structuredProcessor];
       }
       const loopOptions = {
-        messages: result.messages,
-        runtimeContext: result.runtimeContext,
+        messages: result2.messages,
+        runtimeContext: result2.runtimeContext,
+        tracingContext: {
+          currentSpan: agentAISpan
+        },
         runId,
-        toolChoice: result.toolChoice,
-        tools: result.tools,
-        resourceId: result.resourceId,
-        threadId: result.threadId,
-        structuredOutput: result.structuredOutput,
-        stopWhen: result.stopWhen,
-        maxSteps: result.maxSteps,
+        toolChoice: result2.toolChoice,
+        tools: result2.tools,
+        resourceId: result2.resourceId,
+        threadId: result2.threadId,
+        structuredOutput: result2.structuredOutput,
+        stopWhen: result2.stopWhen,
+        maxSteps: result2.maxSteps,
+        providerOptions: result2.providerOptions,
         options: {
           onFinish: async payload => {
             if (payload.finishReason === "error") {
@@ -11133,16 +11273,18 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
                 result: payload,
                 outputText,
                 instructions,
-                thread: result.thread,
-                threadId: result.threadId,
+                thread: result2.thread,
+                threadId: result2.threadId,
                 resourceId,
                 memoryConfig,
                 runtimeContext,
+                tracingContext,
                 runId,
                 messageList,
                 threadExists: inputData["prepare-memory-step"].threadExists,
                 structuredOutput: !!options.output,
-                saveQueueManager
+                saveQueueManager,
+                overrideScorers: options.scorers
               });
             } catch (e) {
               this.logger.error("Error saving memory on finish", {
@@ -11151,11 +11293,12 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
               });
             }
             await options?.onFinish?.({
-              ...result,
-              runId
+              ...result2,
+              runId,
+              messages: messageList.get.response.aiV5.model()
             });
           },
-          onStepFinish: result.onStepFinish
+          onStepFinish: result2.onStepFinish
         },
         output: options.output,
         outputProcessors: effectiveOutputProcessors,
@@ -11167,7 +11310,15 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       return loopOptions;
     }).then(streamStep).commit();
     const run = await executionWorkflow.createRunAsync();
-    return await run.start({});
+    const result = await run.start({
+      tracingContext: {
+        currentSpan: agentAISpan
+      }
+    });
+    agentAISpan?.end({
+      output: result
+    });
+    return result;
   }
   async #executeOnFinish({
     result,
@@ -11178,11 +11329,13 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     memoryConfig,
     outputText,
     runtimeContext,
+    tracingContext,
     runId,
     messageList,
     threadExists,
     structuredOutput = false,
-    saveQueueManager
+    saveQueueManager,
+    overrideScorers
   }) {
     const resToLog = {
       text: result?.text,
@@ -11257,7 +11410,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
             instructions: titleInstructions
           } = this.resolveTitleGenerationConfig(config?.threads?.generateTitle);
           if (shouldGenerate && userMessage) {
-            promises.push(this.genTitle(userMessage, runtimeContext, titleModel, titleInstructions).then(title => {
+            promises.push(this.genTitle(userMessage, runtimeContext, tracingContext, titleModel, titleInstructions).then(title => {
               if (title) {
                 return memory.createThread({
                   threadId: thread.id,
@@ -11313,7 +11466,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       outputText,
       instructions,
       runtimeContext,
-      structuredOutput
+      tracingContext,
+      structuredOutput,
+      overrideScorers
     });
   }
   async generateVNext(messages, options) {
@@ -11373,6 +11528,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     return result.result;
   }
   async generate(messages, generateOptions = {}) {
+    this.logger.warn("Deprecation NOTICE:\nGenerate method will switch to use generateVNext implementation September 16th. Please use generateLegacy if you don't want to upgrade just yet.");
+    return this.generateLegacy(messages, generateOptions);
+  }
+  async generateLegacy(messages, generateOptions = {}) {
     const defaultGenerateOptions = await this.getDefaultGenerateOptions({
       runtimeContext: generateOptions.runtimeContext
     });
@@ -11440,6 +11599,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       agentAISpan,
       ...llmOptions
     } = beforeResult;
+    const tracingContext = {
+      currentSpan: agentAISpan
+    };
     let finalOutputProcessors = mergedGenerateOptions.outputProcessors;
     if (mergedGenerateOptions.structuredOutput) {
       const structuredProcessor = new StructuredOutputProcessor(mergedGenerateOptions.structuredOutput);
@@ -11448,13 +11610,14 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     if (!output || experimental_output) {
       const result2 = await llmToUse.__text({
         ...llmOptions,
-        agentAISpan,
+        tracingContext,
         experimental_output
       });
       const outputProcessorResult2 = await this.__runOutputProcessors({
         runtimeContext: mergedGenerateOptions.runtimeContext || new chunkGPWMM745_cjs.RuntimeContext(),
+        tracingContext,
         outputProcessorOverrides: finalOutputProcessors,
-        messageList: new chunkTWNFR6MQ_cjs.MessageList({
+        messageList: new chunk5CJDO3UO_cjs.MessageList({
           threadId: llmOptions.threadId || "",
           resourceId: llmOptions.resourceId || ""
         }).add({
@@ -11524,12 +11687,13 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           }
         }
       }
+      const overrideScorers = mergedGenerateOptions.scorers;
       const afterResult2 = await after({
         result: result2,
         outputText: newText2,
         agentAISpan,
-        ...(generateOptions.scorers ? {
-          overrideScorers: generateOptions.scorers
+        ...(overrideScorers ? {
+          overrideScorers
         } : {})
       });
       if (generateOptions.returnScorerData) {
@@ -11539,13 +11703,14 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     }
     const result = await llmToUse.__textObject({
       ...llmOptions,
-      agentAISpan,
+      tracingContext,
       structuredOutput: output
     });
     const outputText = JSON.stringify(result.object);
     const outputProcessorResult = await this.__runOutputProcessors({
       runtimeContext: mergedGenerateOptions.runtimeContext || new chunkGPWMM745_cjs.RuntimeContext(),
-      messageList: new chunkTWNFR6MQ_cjs.MessageList({
+      tracingContext,
+      messageList: new chunk5CJDO3UO_cjs.MessageList({
         threadId: llmOptions.threadId || "",
         resourceId: llmOptions.resourceId || ""
       }).add({
@@ -11613,6 +11778,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     return result;
   }
   async stream(messages, streamOptions = {}) {
+    this.logger.warn("Deprecation NOTICE:\nStream method will switch to use streamVNext implementation September 16th. Please use streamLegacy if you don't want to upgrade just yet.");
+    return this.streamLegacy(messages, streamOptions);
+  }
+  async streamLegacy(messages, streamOptions = {}) {
     const defaultStreamOptions = await this.getDefaultStreamOptions({
       runtimeContext: streamOptions.runtimeContext
     });
@@ -11713,6 +11882,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       agentAISpan,
       ...llmOptions
     } = beforeResult;
+    const overrideScorers = mergedStreamOptions.scorers;
+    const tracingContext = {
+      currentSpan: agentAISpan
+    };
     if (!output || experimental_output) {
       this.logger.debug(`Starting agent ${this.name} llm stream call`, {
         runId
@@ -11720,14 +11893,17 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       const streamResult = llm.__stream({
         ...llmOptions,
         experimental_output,
-        agentAISpan,
+        tracingContext,
         onFinish: async result => {
           try {
             const outputText = result.text;
             await after({
               result,
               outputText,
-              agentAISpan
+              agentAISpan,
+              ...(overrideScorers ? {
+                overrideScorers
+              } : {})
             });
           } catch (e) {
             this.logger.error("Error saving memory on finish", {
@@ -11749,7 +11925,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     });
     return llm.__streamObject({
       ...llmOptions,
-      agentAISpan,
+      tracingContext,
       onFinish: async result => {
         try {
           const outputText = JSON.stringify(result.object);
@@ -11757,7 +11933,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
             result,
             outputText,
             structuredOutput: true,
-            agentAISpan
+            agentAISpan,
+            ...(overrideScorers ? {
+              overrideScorers
+            } : {})
           });
         } catch (e) {
           this.logger.error("Error saving memory on finish", {
@@ -12075,6 +12254,21 @@ var ExecutionEngine = class extends chunkKXCUCBEI_cjs.MastraBase {
 
 // src/workflows/default.ts
 var DefaultExecutionEngine = class extends ExecutionEngine {
+  /**
+   * Preprocesses an error caught during workflow execution.
+   *
+   * - Wraps a non-MastraError exception
+   * - Logs error details
+   */
+  preprocessExecutionError(e, errorDefinition, logPrefix) {
+    const error = e instanceof chunkC73WLCY3_cjs.MastraError ? e : new chunkC73WLCY3_cjs.MastraError(errorDefinition, e);
+    if (!(e instanceof chunkC73WLCY3_cjs.MastraError) && e instanceof Error && e.stack) {
+      error.stack = e.stack;
+    }
+    this.logger?.trackException(error);
+    this.logger?.error(logPrefix + error?.stack);
+    return error;
+  }
   /**
    * The runCounts map is used to keep track of the run count for each step.
    * The step id is used as the key and the run count is the value.
@@ -12171,7 +12365,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
       resume,
       retryConfig,
       runtimeContext,
-      currentSpan,
+      tracingContext,
       disableScorers
     } = params;
     const {
@@ -12180,33 +12374,16 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     } = retryConfig ?? {};
     const steps = graph.steps;
     this.runCounts.clear();
-    const spanArgs = {
+    const workflowAISpan = chunkI7OAONIW_cjs.getOrCreateSpan({
+      type: "workflow_run" /* WORKFLOW_RUN */,
       name: `workflow run: '${workflowId}'`,
       input,
       attributes: {
         workflowId
-      }
-    };
-    let workflowAISpan;
-    if (currentSpan) {
-      workflowAISpan = currentSpan.createChildSpan({
-        type: "workflow_run" /* WORKFLOW_RUN */,
-        ...spanArgs
-      });
-    } else {
-      const aiTracing = chunk4DKPMUAC_cjs.getSelectedAITracing({
-        runtimeContext
-      });
-      if (aiTracing) {
-        workflowAISpan = aiTracing.startSpan({
-          type: "workflow_run" /* WORKFLOW_RUN */,
-          ...spanArgs,
-          startOptions: {
-            runtimeContext
-          }
-        });
-      }
-    }
+      },
+      tracingContext,
+      runtimeContext
+    });
     if (steps.length === 0) {
       const empty_graph_error = new chunkC73WLCY3_cjs.MastraError({
         id: "WORKFLOW_EXECUTE_EMPTY_GRAPH",
@@ -12299,7 +12476,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           return result2;
         }
       } catch (e) {
-        const error = e instanceof chunkC73WLCY3_cjs.MastraError ? e : new chunkC73WLCY3_cjs.MastraError({
+        const error = this.preprocessExecutionError(e, {
           id: "WORKFLOW_ENGINE_STEP_EXECUTION_FAILED",
           domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
           category: "USER" /* USER */,
@@ -12307,9 +12484,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             workflowId,
             runId
           }
-        }, e);
-        this.logger?.trackException(error);
-        this.logger?.error(`Error executing step: ${error?.stack}`);
+        }, "Error executing step: ");
         const result2 = await this.fmtReturnValue(executionSpan, params.emitter, stepResults, lastOutput.result, e);
         await this.persistStepUpdate({
           workflowId,
@@ -12623,11 +12798,15 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     const stepAISpan = tracingContext.currentSpan?.createChildSpan({
       name: `workflow step: '${step.id}'`,
       type: "workflow_step" /* WORKFLOW_STEP */,
-      input: prevOutput,
+
+      //input: prevOutput,
       attributes: {
         stepId: step.id
       }
     });
+    const innerTracingContext = {
+      currentSpan: stepAISpan
+    };
     if (!skipEmits) {
       await emitter.emit("watch", {
         type: "watch",
@@ -12703,16 +12882,12 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         const result = await runStep({
           runId,
           workflowId,
-          mastra: this.mastra ? chunk4DKPMUAC_cjs.wrapMastra(this.mastra, {
-            currentSpan: stepAISpan
-          }) : void 0,
+          mastra: this.mastra ? chunkI7OAONIW_cjs.wrapMastra(this.mastra, innerTracingContext) : void 0,
           runtimeContext,
           inputData: prevOutput,
           runCount: this.getOrGenerateRunCount(step.id),
           resumeData: resume?.steps[0] === step.id ? resume?.resumePayload : void 0,
-          tracingContext: {
-            currentSpan: stepAISpan
-          },
+          tracingContext: innerTracingContext,
           getInitData: () => stepResults?.input,
           getStepResult: step2 => {
             if (!step2?.id) {
@@ -12767,6 +12942,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             workflowId,
             stepId: step.id,
             runtimeContext,
+            tracingContext: innerTracingContext,
             disableScorers
           });
         }
@@ -12791,7 +12967,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         }
         break;
       } catch (e) {
-        const error = e instanceof chunkC73WLCY3_cjs.MastraError ? e : new chunkC73WLCY3_cjs.MastraError({
+        const error = this.preprocessExecutionError(e, {
           id: "WORKFLOW_STEP_INVOKE_FAILED",
           domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
           category: "USER" /* USER */,
@@ -12800,9 +12976,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             runId,
             stepId: step.id
           }
-        }, e);
-        this.logger.trackException(error);
-        this.logger.error(`Error executing step ${step.id}: ` + error?.stack);
+        }, `Error executing step ${step.id}: `);
         stepAISpan?.error({
           error,
           attributes: {
@@ -12889,6 +13063,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     workflowId,
     stepId,
     runtimeContext,
+    tracingContext,
     disableScorers
   }) {
     let scorersToUse = scorers;
@@ -12898,18 +13073,16 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           runtimeContext
         });
       } catch (error) {
-        const mastraError = new chunkC73WLCY3_cjs.MastraError({
+        this.preprocessExecutionError(error, {
           id: "WORKFLOW_FAILED_TO_FETCH_SCORERS",
-          domain: "MASTRA_WORKFLOW",
+          domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
           category: "USER" /* USER */,
           details: {
             runId,
             workflowId,
             stepId
           }
-        }, error);
-        this.logger.trackException(mastraError);
-        this.logger.error(mastraError.toString(), error);
+        }, "Error fetching scorers: ");
       }
     }
     if (!disableScorers && scorersToUse && Object.keys(scorersToUse || {}).length > 0) {
@@ -12921,6 +13094,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           input: [input],
           output,
           runtimeContext,
+          tracingContext,
           entity: {
             id: workflowId,
             stepId
@@ -13102,7 +13276,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
         });
         return result ? index : null;
       } catch (e) {
-        const error = e instanceof chunkC73WLCY3_cjs.MastraError ? e : new chunkC73WLCY3_cjs.MastraError({
+        const error = this.preprocessExecutionError(e, {
           id: "WORKFLOW_CONDITION_EVALUATION_FAILED",
           domain: "MASTRA_WORKFLOW" /* MASTRA_WORKFLOW */,
           category: "USER" /* USER */,
@@ -13110,9 +13284,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             workflowId,
             runId
           }
-        }, e);
-        this.logger.trackException(error);
-        this.logger.error("Error evaluating condition: " + error?.stack);
+        }, "Error evaluating condition: ");
         evalSpan?.error({
           error,
           attributes: {
@@ -13292,7 +13464,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
       const evalSpan = loopSpan?.createChildSpan({
         type: "workflow_conditional_eval" /* WORKFLOW_CONDITIONAL_EVAL */,
         name: `condition: ${entry.loopType}`,
-        input: result.output,
+        input: chunkI7OAONIW_cjs.selectFields(result.output, ["stepResult", "output.text", "output.object", "messages"]),
         attributes: {
           conditionIndex: iteration
         }
@@ -14249,7 +14421,7 @@ function createStep(params) {
       }
     };
   }
-  if (params instanceof chunkZAJTIZZF_cjs.Tool) {
+  if (params instanceof chunkABRPHTOG_cjs.Tool) {
     if (!params.inputSchema || !params.outputSchema) {
       throw new Error("Tool must have input and output schemas defined");
     }
@@ -14790,7 +14962,7 @@ var Workflow = class extends chunkKXCUCBEI_cjs.MastraBase {
       cleanup: () => this.#runs.delete(runIdToUse)
     });
     this.#runs.set(runIdToUse, run);
-    this.mastra?.getLogger().warn("createRun() is deprecated. Use createRunAsync() instead.");
+    this.mastra?.getLogger().warn("createRun() will be removed on September 16th. Use createRunAsync() instead.");
     return run;
   }
   /**
@@ -14875,7 +15047,7 @@ var Workflow = class extends chunkKXCUCBEI_cjs.MastraBase {
     abort,
     abortSignal,
     runCount,
-    currentSpan
+    tracingContext
   }) {
     this.__registerMastra(mastra);
     const isResume = !!(resume?.steps && resume.steps.length > 0);
@@ -14911,11 +15083,11 @@ var Workflow = class extends chunkKXCUCBEI_cjs.MastraBase {
       resumeData,
       step: resume.steps,
       runtimeContext,
-      currentSpan
+      tracingContext
     }) : await run.start({
       inputData,
       runtimeContext,
-      currentSpan
+      tracingContext
     });
     unwatch();
     unwatchV2();
@@ -15081,7 +15253,7 @@ var Run = class {
     inputData,
     runtimeContext,
     writableStream,
-    currentSpan
+    tracingContext
   }) {
     const result = await this.executionEngine.execute({
       workflowId: this.workflowId,
@@ -15108,7 +15280,7 @@ var Run = class {
       runtimeContext: runtimeContext ?? new chunkGPWMM745_cjs.RuntimeContext(),
       abortController: this.abortController,
       writableStream,
-      currentSpan
+      tracingContext
     });
     if (result.status !== "suspended") {
       this.cleanup?.();
@@ -15445,7 +15617,7 @@ var Run = class {
       },
       runtimeContext: runtimeContextToUse,
       abortController: this.abortController,
-      currentSpan: params.currentSpan
+      tracingContext: params.tracingContext
     }).then(result => {
       if (result.status !== "suspended") {
         this.closeStreamAction?.().catch(() => {});
@@ -15550,5 +15722,5 @@ exports.recursivelyCheckForFinalState = recursivelyCheckForFinalState;
 exports.resolveVariables = resolveVariables;
 exports.updateStepInHierarchy = updateStepInHierarchy;
 exports.workflowToStep = workflowToStep;
-//# sourceMappingURL=chunk-DJKIK6ZB.cjs.map
-//# sourceMappingURL=chunk-DJKIK6ZB.cjs.map
+//# sourceMappingURL=chunk-KUNWELBC.cjs.map
+//# sourceMappingURL=chunk-KUNWELBC.cjs.map