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

package/dist/{chunk-7OSUKFNG.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 chunkVBAWR62U_cjs = require('./chunk-VBAWR62U.cjs');
-var chunkQA2TWVUS_cjs = require('./chunk-QA2TWVUS.cjs');
+var chunkF2CAC2R2_cjs = require('./chunk-F2CAC2R2.cjs');
+var chunk5CJDO3UO_cjs = require('./chunk-5CJDO3UO.cjs');
 var chunkTSNDVBUU_cjs = require('./chunk-TSNDVBUU.cjs');
-var chunk6VROHRAR_cjs = require('./chunk-6VROHRAR.cjs');
+var chunkLVGGMWSE_cjs = require('./chunk-LVGGMWSE.cjs');
 var chunkGPWMM745_cjs = require('./chunk-GPWMM745.cjs');
 var chunkS6MAHT7F_cjs = require('./chunk-S6MAHT7F.cjs');
-var chunkE4XQMNEI_cjs = require('./chunk-E4XQMNEI.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');
@@ -90,230 +90,17 @@ var require_fast_deep_equal = chunkEBVYYC2Q_cjs.__commonJS({
 // src/agent/index.ts
 var import_fast_deep_equal = chunkEBVYYC2Q_cjs.__toESM(require_fast_deep_equal(), 1);
 
-// src/agent/trip-wire.ts
-var TripWire = class extends Error {
-  constructor(reason) {
-    super(reason);
-    Object.setPrototypeOf(this, new.target.prototype);
-  }
-};
-
-// src/processors/runner.ts
-var ProcessorState = class {
-  constructor(processorName) {
-    this.processorName = processorName;
-  }
-  accumulatedText = "";
-  customState = {};
-  streamParts = [];
-  // Internal methods for the runner
-  addPart(part) {
-    if (part.type === "text-delta") {
-      this.accumulatedText += part.payload.text;
-    }
-    this.streamParts.push(part);
-  }
-};
-var ProcessorRunner = class {
-  inputProcessors;
-  outputProcessors;
-  logger;
-  agentName;
-  constructor({
-    inputProcessors,
-    outputProcessors,
-    logger,
-    agentName
-  }) {
-    this.inputProcessors = inputProcessors ?? [];
-    this.outputProcessors = outputProcessors ?? [];
-    this.logger = logger;
-    this.agentName = agentName;
-  }
-  async runOutputProcessors(messageList, telemetry) {
-    const responseMessages = messageList.clear.response.v2();
-    let processableMessages = [...responseMessages];
-    const ctx = {
-      abort: () => {
-        throw new TripWire("Tripwire triggered");
-      }
-    };
-    for (const [index, processor] of this.outputProcessors.entries()) {
-      const abort = reason => {
-        throw new TripWire(reason || `Tripwire triggered by ${processor.name}`);
-      };
-      ctx.abort = abort;
-      const processMethod = processor.processOutputResult?.bind(processor);
-      if (!processMethod) {
-        continue;
-      }
-      if (!telemetry) {
-        processableMessages = await processMethod({
-          messages: processableMessages,
-          abort: ctx.abort
-        });
-      } else {
-        await telemetry.traceMethod(async () => {
-          processableMessages = await processMethod({
-            messages: processableMessages,
-            abort: ctx.abort
-          });
-          return processableMessages;
-        }, {
-          spanName: `agent.outputProcessor.${processor.name}`,
-          attributes: {
-            "processor.name": processor.name,
-            "processor.index": index.toString(),
-            "processor.total": this.outputProcessors.length.toString()
-          }
-        })();
-      }
-    }
-    if (processableMessages.length > 0) {
-      messageList.add(processableMessages, "response");
-    }
-    return messageList;
-  }
-  /**
-   * Process a stream part through all output processors with state management
-   */
-  async processPart(part, processorStates) {
-    if (!this.outputProcessors.length) {
-      return {
-        part,
-        blocked: false
-      };
-    }
-    try {
-      let processedPart = part;
-      for (const processor of this.outputProcessors) {
-        try {
-          if (processor.processOutputStream && processedPart) {
-            let state = processorStates.get(processor.name);
-            if (!state) {
-              state = new ProcessorState(processor.name);
-              processorStates.set(processor.name, state);
-            }
-            state.addPart(processedPart);
-            const result = await processor.processOutputStream({
-              part: processedPart,
-              streamParts: state.streamParts,
-              state: state.customState,
-              abort: reason => {
-                throw new TripWire(reason || `Stream part blocked by ${processor.name}`);
-              }
-            });
-            processedPart = result;
-          }
-        } catch (error) {
-          if (error instanceof TripWire) {
-            return {
-              part: null,
-              blocked: true,
-              reason: error.message
-            };
-          }
-          this.logger.error(`[Agent:${this.agentName}] - Output processor ${processor.name} failed:`, error);
-        }
-      }
-      return {
-        part: processedPart,
-        blocked: false
-      };
-    } catch (error) {
-      this.logger.error(`[Agent:${this.agentName}] - Stream part processing failed:`, error);
-      return {
-        part,
-        blocked: false
-      };
-    }
-  }
-  async runOutputProcessorsForStream(streamResult) {
-    return new ReadableStream({
-      start: async controller => {
-        const reader = streamResult.fullStream.getReader();
-        const processorStates = /* @__PURE__ */new Map();
-        try {
-          while (true) {
-            const {
-              done,
-              value
-            } = await reader.read();
-            if (done) {
-              controller.close();
-              break;
-            }
-            const {
-              part: processedPart,
-              blocked,
-              reason
-            } = await this.processPart(value, processorStates);
-            if (blocked) {
-              void this.logger.debug(`[Agent:${this.agentName}] - Stream part blocked by output processor`, {
-                reason,
-                originalPart: value
-              });
-              controller.enqueue({
-                type: "tripwire",
-                tripwireReason: reason || "Output processor blocked content"
-              });
-              controller.close();
-              break;
-            } else if (processedPart !== null) {
-              controller.enqueue(processedPart);
-            }
-          }
-        } catch (error) {
-          controller.error(error);
-        }
-      }
-    });
-  }
-  async runInputProcessors(messageList, telemetry) {
-    const userMessages = messageList.clear.input.v2();
-    let processableMessages = [...userMessages];
-    const ctx = {
-      abort: () => {
-        throw new TripWire("Tripwire triggered");
-      }
-    };
-    for (const [index, processor] of this.inputProcessors.entries()) {
-      const abort = reason => {
-        throw new TripWire(reason || `Tripwire triggered by ${processor.name}`);
-      };
-      ctx.abort = abort;
-      const processMethod = processor.processInput?.bind(processor);
-      if (!processMethod) {
-        continue;
-      }
-      if (!telemetry) {
-        processableMessages = await processMethod({
-          messages: processableMessages,
-          abort: ctx.abort
-        });
-      } else {
-        await telemetry.traceMethod(async () => {
-          processableMessages = await processMethod({
-            messages: processableMessages,
-            abort: ctx.abort
-          });
-          return processableMessages;
-        }, {
-          spanName: `agent.inputProcessor.${processor.name}`,
-          attributes: {
-            "processor.name": processor.name,
-            "processor.index": index.toString(),
-            "processor.total": this.inputProcessors.length.toString()
-          }
-        })();
-      }
-    }
-    if (processableMessages.length > 0) {
-      messageList.add(processableMessages, "user");
-    }
-    return messageList;
+// src/stream/aisdk/v5/compat/ui-message.ts
+function getResponseUIMessageId({
+  originalMessages,
+  responseMessageId
+}) {
+  if (originalMessages == null) {
+    return void 0;
   }
-};
+  const lastMessage = originalMessages[originalMessages.length - 1];
+  return lastMessage?.role === "assistant" ? lastMessage.id : typeof responseMessageId === "function" ? responseMessageId() : responseMessageId;
+}
 function convertFullStreamChunkToUIMessageStream({
   part,
   messageMetadataValue,
@@ -566,16 +353,6 @@ function convertFullStreamChunkToUIMessageStream({
       }
   }
 }
-function getResponseUIMessageId({
-  originalMessages,
-  responseMessageId
-}) {
-  if (originalMessages == null) {
-    return void 0;
-  }
-  const lastMessage = originalMessages[originalMessages.length - 1];
-  return lastMessage?.role === "assistant" ? lastMessage.id : typeof responseMessageId === "function" ? responseMessageId() : responseMessageId;
-}
 async function safeValidateTypes({
   value,
   schema
@@ -608,6 +385,49 @@ async function safeValidateTypes({
     };
   }
 }
+
+// src/stream/aisdk/v5/compat/delayed-promise.ts
+var DelayedPromise = class {
+  status = {
+    type: "pending"
+  };
+  _promise;
+  _resolve = void 0;
+  _reject = void 0;
+  get promise() {
+    if (this._promise) {
+      return this._promise;
+    }
+    this._promise = new Promise((resolve, reject) => {
+      if (this.status.type === "resolved") {
+        resolve(this.status.value);
+      } else if (this.status.type === "rejected") {
+        reject(this.status.error);
+      }
+      this._resolve = resolve;
+      this._reject = reject;
+    });
+    return this._promise;
+  }
+  resolve(value) {
+    this.status = {
+      type: "resolved",
+      value
+    };
+    if (this._promise) {
+      this._resolve?.(value);
+    }
+  }
+  reject(error) {
+    this.status = {
+      type: "rejected",
+      error
+    };
+    if (this._promise) {
+      this._reject?.(error);
+    }
+  }
+};
 function prepareToolsAndToolChoice({
   tools,
   toolChoice,
@@ -675,45 +495,234 @@ function prepareToolsAndToolChoice({
     }
   };
 }
-var DelayedPromise = class {
-  status = {
-    type: "pending"
-  };
-  _promise;
-  _resolve = void 0;
-  _reject = void 0;
-  get promise() {
-    if (this._promise) {
-      return this._promise;
+
+// src/agent/trip-wire.ts
+var TripWire = class extends Error {
+  constructor(reason) {
+    super(reason);
+    Object.setPrototypeOf(this, new.target.prototype);
+  }
+};
+
+// src/processors/runner.ts
+var ProcessorState = class {
+  constructor(processorName) {
+    this.processorName = processorName;
+  }
+  accumulatedText = "";
+  customState = {};
+  streamParts = [];
+  // Internal methods for the runner
+  addPart(part) {
+    if (part.type === "text-delta") {
+      this.accumulatedText += part.payload.text;
     }
-    this._promise = new Promise((resolve, reject) => {
-      if (this.status.type === "resolved") {
-        resolve(this.status.value);
-      } else if (this.status.type === "rejected") {
-        reject(this.status.error);
+    this.streamParts.push(part);
+  }
+};
+var ProcessorRunner = class {
+  inputProcessors;
+  outputProcessors;
+  logger;
+  agentName;
+  constructor({
+    inputProcessors,
+    outputProcessors,
+    logger,
+    agentName
+  }) {
+    this.inputProcessors = inputProcessors ?? [];
+    this.outputProcessors = outputProcessors ?? [];
+    this.logger = logger;
+    this.agentName = agentName;
+  }
+  async runOutputProcessors(messageList, tracingContext, telemetry) {
+    const responseMessages = messageList.clear.response.v2();
+    let processableMessages = [...responseMessages];
+    const ctx = {
+      abort: () => {
+        throw new TripWire("Tripwire triggered");
+      }
+    };
+    for (const [index, processor] of this.outputProcessors.entries()) {
+      const abort = reason => {
+        throw new TripWire(reason || `Tripwire triggered by ${processor.name}`);
+      };
+      ctx.abort = abort;
+      const processMethod = processor.processOutputResult?.bind(processor);
+      if (!processMethod) {
+        continue;
+      }
+      if (!telemetry) {
+        processableMessages = await processMethod({
+          messages: processableMessages,
+          abort: ctx.abort,
+          tracingContext
+        });
+      } else {
+        await telemetry.traceMethod(async () => {
+          processableMessages = await processMethod({
+            messages: processableMessages,
+            abort: ctx.abort,
+            tracingContext
+          });
+          return processableMessages;
+        }, {
+          spanName: `agent.outputProcessor.${processor.name}`,
+          attributes: {
+            "processor.name": processor.name,
+            "processor.index": index.toString(),
+            "processor.total": this.outputProcessors.length.toString()
+          }
+        })();
+      }
+    }
+    if (processableMessages.length > 0) {
+      messageList.add(processableMessages, "response");
+    }
+    return messageList;
+  }
+  /**
+   * Process a stream part through all output processors with state management
+   */
+  async processPart(part, processorStates, tracingContext) {
+    if (!this.outputProcessors.length) {
+      return {
+        part,
+        blocked: false
+      };
+    }
+    try {
+      let processedPart = part;
+      for (const processor of this.outputProcessors) {
+        try {
+          if (processor.processOutputStream && processedPart) {
+            let state = processorStates.get(processor.name);
+            if (!state) {
+              state = new ProcessorState(processor.name);
+              processorStates.set(processor.name, state);
+            }
+            state.addPart(processedPart);
+            const result = await processor.processOutputStream({
+              part: processedPart,
+              streamParts: state.streamParts,
+              state: state.customState,
+              abort: reason => {
+                throw new TripWire(reason || `Stream part blocked by ${processor.name}`);
+              },
+              tracingContext
+            });
+            processedPart = result;
+          }
+        } catch (error) {
+          if (error instanceof TripWire) {
+            return {
+              part: null,
+              blocked: true,
+              reason: error.message
+            };
+          }
+          this.logger.error(`[Agent:${this.agentName}] - Output processor ${processor.name} failed:`, error);
+        }
+      }
+      return {
+        part: processedPart,
+        blocked: false
+      };
+    } catch (error) {
+      this.logger.error(`[Agent:${this.agentName}] - Stream part processing failed:`, error);
+      return {
+        part,
+        blocked: false
+      };
+    }
+  }
+  async runOutputProcessorsForStream(streamResult, tracingContext) {
+    return new ReadableStream({
+      start: async controller => {
+        const reader = streamResult.fullStream.getReader();
+        const processorStates = /* @__PURE__ */new Map();
+        try {
+          while (true) {
+            const {
+              done,
+              value
+            } = await reader.read();
+            if (done) {
+              controller.close();
+              break;
+            }
+            const {
+              part: processedPart,
+              blocked,
+              reason
+            } = await this.processPart(value, processorStates, tracingContext);
+            if (blocked) {
+              void this.logger.debug(`[Agent:${this.agentName}] - Stream part blocked by output processor`, {
+                reason,
+                originalPart: value
+              });
+              controller.enqueue({
+                type: "tripwire",
+                tripwireReason: reason || "Output processor blocked content"
+              });
+              controller.close();
+              break;
+            } else if (processedPart !== null) {
+              controller.enqueue(processedPart);
+            }
+          }
+        } catch (error) {
+          controller.error(error);
+        }
       }
-      this._resolve = resolve;
-      this._reject = reject;
     });
-    return this._promise;
   }
-  resolve(value) {
-    this.status = {
-      type: "resolved",
-      value
+  async runInputProcessors(messageList, tracingContext, telemetry) {
+    const userMessages = messageList.clear.input.v2();
+    let processableMessages = [...userMessages];
+    const ctx = {
+      abort: () => {
+        throw new TripWire("Tripwire triggered");
+      }
     };
-    if (this._promise) {
-      this._resolve?.(value);
+    for (const [index, processor] of this.inputProcessors.entries()) {
+      const abort = reason => {
+        throw new TripWire(reason || `Tripwire triggered by ${processor.name}`);
+      };
+      ctx.abort = abort;
+      const processMethod = processor.processInput?.bind(processor);
+      if (!processMethod) {
+        continue;
+      }
+      if (!telemetry) {
+        processableMessages = await processMethod({
+          messages: processableMessages,
+          abort: ctx.abort,
+          tracingContext
+        });
+      } else {
+        await telemetry.traceMethod(async () => {
+          processableMessages = await processMethod({
+            messages: processableMessages,
+            abort: ctx.abort,
+            tracingContext
+          });
+          return processableMessages;
+        }, {
+          spanName: `agent.inputProcessor.${processor.name}`,
+          attributes: {
+            "processor.name": processor.name,
+            "processor.index": index.toString(),
+            "processor.total": this.inputProcessors.length.toString()
+          }
+        })();
+      }
     }
-  }
-  reject(error) {
-    this.status = {
-      type: "rejected",
-      error
-    };
-    if (this._promise) {
-      this._reject?.(error);
+    if (processableMessages.length > 0) {
+      messageList.add(processableMessages, "user");
     }
+    return messageList;
   }
 };
 function getTransformedSchema(schema) {
@@ -1164,7 +1173,7 @@ function convertMastraChunkToAISDKv5({
       if (mode === "generate") {
         return {
           type: "file",
-          file: new chunkQA2TWVUS_cjs.DefaultGeneratedFile({
+          file: new chunk5CJDO3UO_cjs.DefaultGeneratedFile({
             data: chunk.payload.data,
             mediaType: chunk.payload.mimeType
           })
@@ -1172,7 +1181,7 @@ function convertMastraChunkToAISDKv5({
       }
       return {
         type: "file",
-        file: new chunkQA2TWVUS_cjs.DefaultGeneratedFileWithType({
+        file: new chunk5CJDO3UO_cjs.DefaultGeneratedFileWithType({
           data: chunk.payload.data,
           mediaType: chunk.payload.mimeType
         })
@@ -2216,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 => chunkQA2TWVUS_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) {
@@ -3311,7 +3320,9 @@ function createLLMExecutionStep({
   toolCallStreaming,
   controller,
   output,
-  headers
+  headers,
+  downloadRetries,
+  downloadConcurrency
 }) {
   return createStep({
     id: "llm-execution",
@@ -3332,11 +3343,16 @@ function createLLMExecutionStep({
       switch (model.specificationVersion) {
         case "v2":
           {
+            const inputMessages = await messageList.get.all.aiV5.llmPrompt({
+              downloadRetries,
+              downloadConcurrency,
+              supportedUrls: model?.supportedUrls
+            });
             modelResult = execute({
               runId,
               model,
               providerOptions,
-              inputMessages: messageList.get.all.aiV5.llmPrompt(),
+              inputMessages,
               tools,
               toolChoice,
               options,
@@ -3753,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);
         }
       });
@@ -3859,6 +3886,9 @@ function workflowLoopStream({
             user: rest.messageList.get.input.aiV5.model(),
             nonUser: []
           }
+        },
+        tracingContext: {
+          currentSpan: llmAISpan
         }
       });
       if (executionResult.status !== "success") {
@@ -3901,6 +3931,7 @@ function loop({
   _internal,
   mode = "stream",
   outputProcessors,
+  llmAISpan,
   ...rest
 }) {
   let loggerToUse = logger || new chunkV5WKCX3G_cjs.ConsoleLogger({
@@ -3958,6 +3989,7 @@ function loop({
     telemetry_settings,
     modelSettings,
     outputProcessors,
+    llmAISpan,
     ...rest
   };
   const streamFn = workflowLoopStream(workflowLoopProps);
@@ -4069,7 +4101,9 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
     resourceId,
     output,
     options,
-    outputProcessors
+    outputProcessors,
+    providerOptions,
+    tracingContext
     // ...rest
   }) {
     let stopWhenToUse;
@@ -4089,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 chunkQA2TWVUS_cjs.MessageList({
+      const messageList = new chunk5CJDO3UO_cjs.MessageList({
         threadId,
         resourceId
       });
@@ -4102,12 +4150,14 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
         stopWhen: stopWhenToUse,
         toolChoice,
         modelSettings,
+        providerOptions,
         telemetry_settings: {
           ...this.experimental_telemetry,
           ...telemetry_settings
         },
         output,
         outputProcessors,
+        llmAISpan,
         options: {
           ...options,
           onStepFinish: async props => {
@@ -4148,7 +4198,7 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
               this.logger.warn("Rate limit approaching, waiting 10 seconds", {
                 runId
               });
-              await chunk6VROHRAR_cjs.delay(10 * 1e3);
+              await chunkLVGGMWSE_cjs.delay(10 * 1e3);
             }
           },
           onFinish: async props => {
@@ -4190,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",
@@ -4204,6 +4258,9 @@ var MastraLLMVNext = class extends chunkKXCUCBEI_cjs.MastraBase {
           resourceId: resourceId ?? "unknown"
         }
       }, e);
+      llmAISpan?.error({
+        error: mastraError
+      });
       throw mastraError;
     }
   }
@@ -4367,6 +4424,7 @@ function runScorer({
   input,
   output,
   runtimeContext,
+  tracingContext,
   entity,
   structuredOutput,
   source,
@@ -4399,6 +4457,7 @@ function runScorer({
     input,
     output,
     runtimeContext: Object.fromEntries(runtimeContext.entries()),
+    tracingContext,
     runId,
     source,
     entity,
@@ -5034,7 +5093,7 @@ var Machine = class extends EventEmitter__default.default {
         const logger = this.logger;
         let mastraProxy = void 0;
         if (this.#mastra) {
-          mastraProxy = chunk6VROHRAR_cjs.createMastraProxy({
+          mastraProxy = chunkLVGGMWSE_cjs.createMastraProxy({
             mastra: this.#mastra,
             logger
           });
@@ -7612,7 +7671,8 @@ var ModerationProcessor = class _ModerationProcessor {
     try {
       const {
         messages,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (messages.length === 0) {
         return messages;
@@ -7624,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") {
@@ -7649,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") {
@@ -7674,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();
@@ -7691,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;
@@ -7849,7 +7912,8 @@ var PromptInjectionDetector = class _PromptInjectionDetector {
     try {
       const {
         messages,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (messages.length === 0) {
         return messages;
@@ -7861,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") {
@@ -7886,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();
@@ -7904,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;
@@ -8080,7 +8146,8 @@ var PIIDetector = class _PIIDetector {
     try {
       const {
         messages,
-        abort
+        abort,
+        tracingContext
       } = args;
       if (messages.length === 0) {
         return messages;
@@ -8092,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") {
@@ -8119,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) => {
@@ -8144,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;
@@ -8343,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") {
@@ -8353,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":
@@ -9192,13 +9262,13 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
       this.logger.error(mastraError.toString());
       throw mastraError;
     }
-    return chunk6VROHRAR_cjs.ensureToolProperties(this.#tools);
+    return chunkLVGGMWSE_cjs.ensureToolProperties(this.#tools);
   }
   getTools({
     runtimeContext = new chunkGPWMM745_cjs.RuntimeContext()
   } = {}) {
     if (typeof this.#tools !== "function") {
-      return chunk6VROHRAR_cjs.ensureToolProperties(this.#tools);
+      return chunkLVGGMWSE_cjs.ensureToolProperties(this.#tools);
     }
     const result = this.#tools({
       runtimeContext,
@@ -9219,7 +9289,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         this.logger.error(mastraError.toString());
         throw mastraError;
       }
-      return chunk6VROHRAR_cjs.ensureToolProperties(tools);
+      return chunkLVGGMWSE_cjs.ensureToolProperties(tools);
     });
   }
   get llm() {
@@ -9263,7 +9333,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           mastra: this.#mastra
         });
       } else {
-        llm = new chunkVBAWR62U_cjs.MastraLLMV1({
+        llm = new chunkF2CAC2R2_cjs.MastraLLMV1({
           model: resolvedModel,
           mastra: this.#mastra
         });
@@ -9371,6 +9441,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
   async generateTitleFromUserMessage({
     message,
     runtimeContext = new chunkGPWMM745_cjs.RuntimeContext(),
+    tracingContext,
     model,
     instructions
   }) {
@@ -9378,7 +9449,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
       runtimeContext,
       model
     });
-    const normMessage = new chunkQA2TWVUS_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)}`);
     }
@@ -9403,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
@@ -9415,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
@@ -9432,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 chunkQA2TWVUS_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
           });
@@ -9460,7 +9534,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     runId,
     userMessages,
     systemMessage,
-    messageList = new chunkQA2TWVUS_cjs.MessageList({
+    messageList = new chunk5CJDO3UO_cjs.MessageList({
       threadId,
       resourceId
     }),
@@ -9515,7 +9589,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
         systemMessage: systemMessages,
         memorySystemMessage: memorySystemMessage || void 0
       });
-      const returnList = new chunkQA2TWVUS_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()
@@ -9531,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({
@@ -9555,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 = chunk6VROHRAR_cjs.makeCoreTool(toolObj, options);
+        const convertedToCoreTool = chunkLVGGMWSE_cjs.makeCoreTool(toolObj, options);
         convertedMemoryTools[toolName] = convertedToCoreTool;
       }
     }
@@ -9568,6 +9642,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
   }
   async __runInputProcessors({
     runtimeContext,
+    tracingContext,
     messageList,
     inputProcessorOverrides
   }) {
@@ -9578,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: {
@@ -9597,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;
@@ -9620,6 +9695,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
   }
   async __runOutputProcessors({
     runtimeContext,
+    tracingContext,
     messageList,
     outputProcessorOverrides
   }) {
@@ -9630,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: {
@@ -9649,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;
@@ -9688,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`, {
@@ -9723,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, chunk6VROHRAR_cjs.makeCoreTool(tool, options)];
+      return [k, chunkLVGGMWSE_cjs.makeCoreTool(tool, options)];
     }));
     const assignedToolEntriesConverted = Object.fromEntries(assignedCoreToolEntries.filter(entry => Boolean(entry)));
     toolsForRequest = {
@@ -9743,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({
@@ -9768,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 = chunk6VROHRAR_cjs.makeCoreTool(toolObj, options, "toolset");
+          const convertedToCoreTool = chunkLVGGMWSE_cjs.makeCoreTool(toolObj, options, "toolset");
           toolsForRequest[toolName] = convertedToCoreTool;
         }
       }
@@ -9785,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({
@@ -9813,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 = chunk6VROHRAR_cjs.makeCoreTool(rest, options, "client-tool");
+        const convertedToCoreTool = chunkLVGGMWSE_cjs.makeCoreTool(rest, options, "client-tool");
         toolsForRequest[toolName] = convertedToCoreTool;
       }
     }
@@ -9829,7 +9905,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     threadId,
     resourceId,
     runtimeContext,
-    agentAISpan
+    tracingContext
   }) {
     let convertedWorkflowTools = {};
     const workflows = await this.getWorkflows({
@@ -9846,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,
@@ -9868,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
@@ -9908,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 = chunk6VROHRAR_cjs.createMastraProxy({
+      mastraProxy = chunkLVGGMWSE_cjs.createMastraProxy({
         mastra: this.#mastra,
         logger
       });
@@ -9924,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,
@@ -10044,7 +10122,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
     runtimeContext,
     saveQueueManager,
     writableStream,
-    currentSpan
+    tracingContext
   }) {
     return {
       before: async () => {
@@ -10053,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,
@@ -10064,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
         });
@@ -10109,10 +10174,10 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
           resourceId,
           runId,
           runtimeContext,
-          writableStream,
-          agentAISpan
+          tracingContext: innerTracingContext,
+          writableStream
         });
-        const messageList = new chunkQA2TWVUS_cjs.MessageList({
+        const messageList = new chunk5CJDO3UO_cjs.MessageList({
           threadId,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10129,6 +10194,7 @@ exports.Agent = class Agent extends (_a = chunkKXCUCBEI_cjs.MastraBase) {
             tripwireReason: tripwireReason2
           } = await this.__runInputProcessors({
             runtimeContext,
+            tracingContext: innerTracingContext,
             messageList
           });
           return {
@@ -10199,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 chunkQA2TWVUS_cjs.MessageList().add(messages, `user`).getLatestUserContent() || "",
+          vectorMessageSearch: new chunk5CJDO3UO_cjs.MessageList().add(messages, `user`).getLatestUserContent() || "",
           memoryConfig,
           runtimeContext
         }) : [], memory.getSystemMessage({
@@ -10222,7 +10288,7 @@ The following messages were remembered from a different conversation:
 <remembered_from_other_conversation>
 ${(() => {
             let result = ``;
-            const messages2 = new chunkQA2TWVUS_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;
@@ -10261,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(`
@@ -10274,7 +10341,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           systemMessage,
           memorySystemMessage: memorySystemMessage || void 0
         });
-        const processedList = new chunkQA2TWVUS_cjs.MessageList({
+        const processedList = new chunk5CJDO3UO_cjs.MessageList({
           threadId: threadObject.id,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10325,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 chunkQA2TWVUS_cjs.MessageList({
+        const messageListResponses = new chunk5CJDO3UO_cjs.MessageList({
           threadId,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10397,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,
@@ -10414,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({
@@ -10429,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 {
@@ -10453,6 +10517,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           outputText,
           instructions,
           runtimeContext,
+          tracingContext: {
+            currentSpan: agentAISpan
+          },
           structuredOutput,
           overrideScorers,
           threadId,
@@ -10467,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
         };
@@ -10479,6 +10557,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     outputText,
     instructions,
     runtimeContext,
+    tracingContext,
     structuredOutput,
     overrideScorers,
     threadId,
@@ -10500,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(),
@@ -10513,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
@@ -10532,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,
@@ -10544,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
@@ -10608,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;
@@ -10675,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,
@@ -10687,7 +10810,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           messageList,
           structuredOutput,
           threadExists,
-          agentAISpan
+          agentAISpan,
+          overrideScorers
         });
         return afterResult;
       }
@@ -10711,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) {
@@ -10771,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
@@ -10789,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 chunkQA2TWVUS_cjs.MessageList({
+        const messageList = new chunk5CJDO3UO_cjs.MessageList({
           threadId: thread?.id,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10808,6 +10953,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
             tripwireReason: tripwireReason2
           } = await this.__runInputProcessors({
             runtimeContext,
+            tracingContext,
             messageList
           });
           return {
@@ -10873,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 chunkQA2TWVUS_cjs.MessageList().add(options.messages, `user`).getLatestUserContent() || "",
+          vectorMessageSearch: new chunk5CJDO3UO_cjs.MessageList().add(options.messages, `user`).getLatestUserContent() || "",
           memoryConfig,
           runtimeContext
         }) : [], memory.getSystemMessage({
@@ -10895,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 chunkQA2TWVUS_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;
@@ -10912,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>`;
         }
@@ -10935,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(`
@@ -10948,7 +11094,7 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           systemMessage,
           memorySystemMessage: memorySystemMessage || void 0
         });
-        const processedList = new chunkQA2TWVUS_cjs.MessageList({
+        const processedList = new chunk5CJDO3UO_cjs.MessageList({
           threadId: threadObject.id,
           resourceId,
           generateMessageId: this.#mastra?.generateId?.bind(this.#mastra),
@@ -10973,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
@@ -10983,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;
@@ -10998,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,
@@ -11042,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({
@@ -11063,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(),
@@ -11086,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") {
@@ -11120,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", {
@@ -11138,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,
@@ -11154,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,
@@ -11165,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,
@@ -11244,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,
@@ -11300,7 +11466,9 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
       outputText,
       instructions,
       runtimeContext,
-      structuredOutput
+      tracingContext,
+      structuredOutput,
+      overrideScorers
     });
   }
   async generateVNext(messages, options) {
@@ -11360,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
     });
@@ -11427,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);
@@ -11435,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 chunkQA2TWVUS_cjs.MessageList({
+        messageList: new chunk5CJDO3UO_cjs.MessageList({
           threadId: llmOptions.threadId || "",
           resourceId: llmOptions.resourceId || ""
         }).add({
@@ -11511,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) {
@@ -11526,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 chunkQA2TWVUS_cjs.MessageList({
+      tracingContext,
+      messageList: new chunk5CJDO3UO_cjs.MessageList({
         threadId: llmOptions.threadId || "",
         resourceId: llmOptions.resourceId || ""
       }).add({
@@ -11600,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
     });
@@ -11700,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
@@ -11707,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", {
@@ -11736,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);
@@ -11744,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", {
@@ -12062,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.
@@ -12158,7 +12365,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
       resume,
       retryConfig,
       runtimeContext,
-      currentSpan,
+      tracingContext,
       disableScorers
     } = params;
     const {
@@ -12167,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",
@@ -12286,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 */,
@@ -12294,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,
@@ -12610,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",
@@ -12690,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) {
@@ -12754,6 +12942,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             workflowId,
             stepId: step.id,
             runtimeContext,
+            tracingContext: innerTracingContext,
             disableScorers
           });
         }
@@ -12778,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 */,
@@ -12787,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: {
@@ -12876,6 +13063,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     workflowId,
     stepId,
     runtimeContext,
+    tracingContext,
     disableScorers
   }) {
     let scorersToUse = scorers;
@@ -12885,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) {
@@ -12908,6 +13094,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           input: [input],
           output,
           runtimeContext,
+          tracingContext,
           entity: {
             id: workflowId,
             stepId
@@ -13089,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 */,
@@ -13097,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: {
@@ -13279,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
         }
@@ -14236,7 +14421,7 @@ function createStep(params) {
       }
     };
   }
-  if (params instanceof chunkE4XQMNEI_cjs.Tool) {
+  if (params instanceof chunkABRPHTOG_cjs.Tool) {
     if (!params.inputSchema || !params.outputSchema) {
       throw new Error("Tool must have input and output schemas defined");
     }
@@ -14777,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;
   }
   /**
@@ -14862,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);
@@ -14898,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();
@@ -15068,7 +15253,7 @@ var Run = class {
     inputData,
     runtimeContext,
     writableStream,
-    currentSpan
+    tracingContext
   }) {
     const result = await this.executionEngine.execute({
       workflowId: this.workflowId,
@@ -15095,7 +15280,7 @@ var Run = class {
       runtimeContext: runtimeContext ?? new chunkGPWMM745_cjs.RuntimeContext(),
       abortController: this.abortController,
       writableStream,
-      currentSpan
+      tracingContext
     });
     if (result.status !== "suspended") {
       this.cleanup?.();
@@ -15432,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(() => {});
@@ -15537,5 +15722,5 @@ exports.recursivelyCheckForFinalState = recursivelyCheckForFinalState;
 exports.resolveVariables = resolveVariables;
 exports.updateStepInHierarchy = updateStepInHierarchy;
 exports.workflowToStep = workflowToStep;
-//# sourceMappingURL=chunk-7OSUKFNG.cjs.map
-//# sourceMappingURL=chunk-7OSUKFNG.cjs.map
+//# sourceMappingURL=chunk-KUNWELBC.cjs.map
+//# sourceMappingURL=chunk-KUNWELBC.cjs.map