@mastra/core 0.15.3 → 0.16.0-alpha.1

package/dist/{chunk-AVPW677Z.js → chunk-YILDTNQM.js}

@@ -1,6 +1,6 @@
-import { DefaultVoice } from './chunk-4VU6A5XE.js';
-import { EMITTER_SYMBOL } from './chunk-GK5V7YTQ.js';
-import { InstrumentClass, Telemetry } from './chunk-R7K2QO7M.js';
+import { DefaultVoice } from './chunk-QQIBOVFQ.js';
+import { STREAM_FORMAT_SYMBOL, EMITTER_SYMBOL } from './chunk-NLNKQD2T.js';
+import { InstrumentClass, Telemetry } from './chunk-TLJPVRO5.js';
 import { MastraLLMV1 } from './chunk-HGYL2SSF.js';
 import { MessageList, DefaultGeneratedFile, DefaultGeneratedFileWithType } from './chunk-HF7HXOCC.js';
 import { executeHook } from './chunk-TTELJD4F.js';
@@ -2200,7 +2200,7 @@ var MastraModelOutput = class extends MastraBase {
             chunk.payload.output.usage = self.#usageCount;
             try {
               if (self.processorRunner) {
-                await self.processorRunner.runOutputProcessors(self.messageList);
+                self.messageList = await self.processorRunner.runOutputProcessors(self.messageList);
                 const outputText = self.messageList.get.response.aiV4.core().map(m => 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);
@@ -2212,6 +2212,17 @@ var MastraModelOutput = class extends MastraBase {
                 }
                 self.#delayedPromises.text.resolve(outputText);
                 self.#delayedPromises.finishReason.resolve(self.#finishReason);
+                if (chunk.payload.metadata) {
+                  const {
+                    providerMetadata,
+                    request,
+                    ...otherMetadata
+                  } = chunk.payload.metadata;
+                  response = {
+                    ...otherMetadata,
+                    messages: messageList.get.response.aiV5.model()
+                  };
+                }
               } else {
                 self.#delayedPromises.text.resolve(self.#bufferedText.join(""));
                 self.#delayedPromises.finishReason.resolve(self.#finishReason);
@@ -4423,7 +4434,8 @@ function runScorer({
   source,
   entityType,
   threadId,
-  resourceId
+  resourceId,
+  tracingContext
 }) {
   let shouldExecute = false;
   if (!scorerObject?.sampling || scorerObject?.sampling?.type === "none") {
@@ -4456,7 +4468,8 @@ function runScorer({
     structuredOutput,
     entityType,
     threadId,
-    resourceId
+    resourceId,
+    tracingContext
   };
   executeHook("onScorerRun" /* ON_SCORER_RUN */, payload);
 }
@@ -9898,7 +9911,8 @@ var Agent = class extends (_a = MastraBase) {
     resourceId,
     runtimeContext,
     methodType,
-    tracingContext
+    tracingContext,
+    format
   }) {
     const convertedWorkflowTools = {};
     const workflows = await this.getWorkflows({
@@ -9962,8 +9976,8 @@ var Agent = class extends (_a = MastraBase) {
               } else if (methodType === "streamVNext") {
                 const streamResult = run.streamVNext({
                   inputData: context,
-                  runtimeContext
-                  //format,
+                  runtimeContext,
+                  format
                 });
                 if (writer) {
                   await streamResult.pipeTo(writer);
@@ -9973,7 +9987,10 @@ var Agent = class extends (_a = MastraBase) {
               toolAISpan?.end({
                 output: result
               });
-              return result;
+              return {
+                result,
+                runId: run.runId
+              };
             } catch (err) {
               const mastraError = new MastraError({
                 id: "AGENT_WORKFLOW_TOOL_EXECUTION_FAILED",
@@ -10565,7 +10582,10 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           structuredOutput,
           overrideScorers,
           threadId,
-          resourceId
+          resourceId,
+          tracingContext: {
+            currentSpan: agentAISpan
+          }
         });
         const scoringData = {
           input: {
@@ -10602,7 +10622,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
     structuredOutput,
     overrideScorers,
     threadId,
-    resourceId
+    resourceId,
+    tracingContext
   }) {
     const agentName = this.name;
     const userInputMessages = messageList.get.all.ui().filter(m => m.role === "user");
@@ -10653,7 +10674,8 @@ Message ${msg.threadId && msg.threadId !== threadObject.id ? "from previous conv
           entityType: "AGENT",
           structuredOutput: !!structuredOutput,
           threadId,
-          resourceId
+          resourceId,
+          tracingContext
         });
       }
     }
@@ -11538,7 +11560,8 @@ the following messages are from ${ymd}
       instructions,
       runtimeContext,
       structuredOutput,
-      overrideScorers
+      overrideScorers,
+      tracingContext
     });
   }
   async generateVNext(messages, options) {
@@ -12503,7 +12526,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
               attempts,
               delay: delay2
             },
-            executionSpan
+            executionSpan,
+            format: params.format
           },
           tracingContext: {
             currentSpan: workflowAISpan
@@ -12639,6 +12663,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     emitter,
     abortController,
     runtimeContext,
+    executionContext,
     writableStream,
     tracingContext
   }) {
@@ -12684,6 +12709,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           abortController?.abort();
         },
         [EMITTER_SYMBOL]: emitter,
+        [STREAM_FORMAT_SYMBOL]: executionContext.format,
         engine: {},
         abortSignal: abortController?.signal,
         writer: new ToolStream({
@@ -12717,6 +12743,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     emitter,
     abortController,
     runtimeContext,
+    executionContext,
     writableStream,
     tracingContext
   }) {
@@ -12763,6 +12790,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           abortController?.abort();
         },
         [EMITTER_SYMBOL]: emitter,
+        [STREAM_FORMAT_SYMBOL]: executionContext.format,
         engine: {},
         abortSignal: abortController?.signal,
         writer: new ToolStream({
@@ -12994,6 +13022,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             runId: stepResults[step.id]?.suspendPayload?.__workflow_meta?.runId
           } : void 0,
           [EMITTER_SYMBOL]: emitter,
+          [STREAM_FORMAT_SYMBOL]: executionContext.format,
           engine: {},
           abortSignal: abortController?.signal,
           writer: new ToolStream({
@@ -13014,7 +13043,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             workflowId,
             stepId: step.id,
             runtimeContext,
-            disableScorers
+            disableScorers,
+            tracingContext
           });
         }
         if (suspended) {
@@ -13134,7 +13164,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
     workflowId,
     stepId,
     runtimeContext,
-    disableScorers
+    disableScorers,
+    tracingContext
   }) {
     let scorersToUse = scorers;
     if (typeof scorersToUse === "function") {
@@ -13170,7 +13201,8 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           },
           structuredOutput: true,
           source: "LIVE",
-          entityType: "WORKFLOW"
+          entityType: "WORKFLOW",
+          tracingContext
         });
       }
     }
@@ -13328,6 +13360,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
             abortController?.abort();
           },
           [EMITTER_SYMBOL]: emitter,
+          [STREAM_FORMAT_SYMBOL]: executionContext.format,
           engine: {},
           abortSignal: abortController?.signal,
           writer: new ToolStream({
@@ -13562,6 +13595,7 @@ var DefaultExecutionEngine = class extends ExecutionEngine {
           abortController?.abort();
         },
         [EMITTER_SYMBOL]: emitter,
+        [STREAM_FORMAT_SYMBOL]: executionContext.format,
         engine: {},
         abortSignal: abortController?.signal,
         writer: new ToolStream({
@@ -14429,6 +14463,7 @@ function createStep(params) {
       execute: async ({
         inputData,
         [EMITTER_SYMBOL]: emitter,
+        [STREAM_FORMAT_SYMBOL]: streamFormat,
         runtimeContext,
         abortSignal,
         abort
@@ -14444,27 +14479,42 @@ function createStep(params) {
         };
         await emitter.emit("watch-v2", {
           type: "workflow-agent-call-start",
+          from: "WORKFLOW",
           payload: toolData
         });
-        const {
-          fullStream
-        } = await params.stream(inputData.prompt, {
-          // resourceId: inputData.resourceId,
-          // threadId: inputData.threadId,
-          runtimeContext,
-          onFinish: result => {
-            streamPromise.resolve(result.text);
-          },
-          abortSignal
-        });
+        let stream;
+        if (streamFormat === "aisdk") {
+          const {
+            fullStream
+          } = await params.stream(inputData.prompt, {
+            // resourceId: inputData.resourceId,
+            // threadId: inputData.threadId,
+            runtimeContext,
+            onFinish: result => {
+              streamPromise.resolve(result.text);
+            },
+            abortSignal
+          });
+          stream = fullStream;
+        } else {
+          const modelOutput = await params.streamVNext(inputData.prompt, {
+            runtimeContext,
+            onFinish: result => {
+              streamPromise.resolve(result.text);
+            }
+            // abortSignal,
+          });
+          stream = modelOutput.fullStream;
+        }
         if (abortSignal.aborted) {
           return abort();
         }
-        for await (const chunk of fullStream) {
+        for await (const chunk of stream) {
           await emitter.emit("watch-v2", chunk);
         }
         await emitter.emit("watch-v2", {
           type: "workflow-agent-call-finish",
+          from: "WORKFLOW",
           payload: toolData
         });
         return {
@@ -14507,7 +14557,7 @@ function createStep(params) {
     suspendSchema: params.suspendSchema,
     scorers: params.scorers,
     retries: params.retries,
-    execute: params.execute
+    execute: params.execute.bind(params)
   };
 }
 function cloneStep(step, opts) {
@@ -15359,16 +15409,12 @@ var Run = class {
   async sendEvent(event, data) {
     this.emitter.emit(`user-event-${event}`, data);
   }
-  /**
-   * Starts the workflow execution with the provided input
-   * @param input The input data for the workflow
-   * @returns A promise that resolves to the workflow output
-   */
-  async start({
+  async _start({
     inputData,
     runtimeContext,
     writableStream,
-    tracingContext
+    tracingContext,
+    format
   }) {
     const result = await this.executionEngine.execute({
       workflowId: this.workflowId,
@@ -15395,13 +15441,33 @@ var Run = class {
       runtimeContext: runtimeContext ?? new RuntimeContext(),
       abortController: this.abortController,
       writableStream,
-      tracingContext
+      tracingContext,
+      format
     });
     if (result.status !== "suspended") {
       this.cleanup?.();
     }
     return result;
   }
+  /**
+   * Starts the workflow execution with the provided input
+   * @param input The input data for the workflow
+   * @returns A promise that resolves to the workflow output
+   */
+  async start({
+    inputData,
+    runtimeContext,
+    writableStream,
+    tracingContext
+  }) {
+    return this._start({
+      inputData,
+      runtimeContext,
+      writableStream,
+      tracingContext,
+      format: "aisdk"
+    });
+  }
   /**
    * Starts the workflow execution with the provided input as a stream
    * @param input The input data for the workflow
@@ -15471,9 +15537,10 @@ var Run = class {
         runId: this.runId
       }
     });
-    this.executionResults = this.start({
+    this.executionResults = this._start({
       inputData,
-      runtimeContext
+      runtimeContext,
+      format: "aisdk"
     }).then(result => {
       if (result.status !== "suspended") {
         this.closeStreamAction?.().catch(() => {});
@@ -15501,7 +15568,8 @@ var Run = class {
    */
   streamVNext({
     inputData,
-    runtimeContext
+    runtimeContext,
+    format
   } = {}) {
     this.closeStreamAction = async () => {};
     return new MastraWorkflowStream({
@@ -15537,12 +15605,13 @@ var Run = class {
         };
         const unwatch = this.watch(async ({
           type,
+          from = "WORKFLOW" /* WORKFLOW */,
           payload
         }) => {
           buffer.push({
             type,
             runId: this.runId,
-            from: "WORKFLOW" /* WORKFLOW */,
+            from,
             payload: {
               stepName: payload.id,
               ...payload
@@ -15558,10 +15627,11 @@ var Run = class {
             console.error("Error closing stream:", err);
           }
         };
-        const executionResults = this.start({
+        const executionResults = this._start({
           inputData,
           runtimeContext,
-          writableStream: writable
+          writableStream: writable,
+          format
         }).then(result => {
           if (result.status !== "suspended") {
             this.closeStreamAction?.().catch(() => {});
@@ -15799,5 +15869,5 @@ function deepMergeWorkflowState(a, b) {
 }
 
 export { AISDKV5OutputStream, Agent, DefaultExecutionEngine, ExecutionEngine, LanguageDetector, LanguageDetectorInputProcessor, LegacyStep, LegacyWorkflow, MastraModelOutput, ModerationInputProcessor, ModerationProcessor, PIIDetector, PIIDetectorInputProcessor, PromptInjectionDetector, PromptInjectionDetectorInputProcessor, Run, StructuredOutputProcessor, TripWire, UnicodeNormalizer, UnicodeNormalizerInputProcessor, WhenConditionReturnValue, Workflow, agentToStep, cloneStep, cloneWorkflow, createStep, createWorkflow, getActivePathsAndStatus, getResultActivePaths, getStepResult, getSuspendedPaths, isAgent, isConditionalKey, isErrorEvent, isFinalState, isLimboState, isTransitionEvent, isVariableReference, isWorkflow, loop, mapVariable, mergeChildValue, recursivelyCheckForFinalState, resolveVariables, updateStepInHierarchy, workflowToStep };
-//# sourceMappingURL=chunk-AVPW677Z.js.map
-//# sourceMappingURL=chunk-AVPW677Z.js.map
+//# sourceMappingURL=chunk-YILDTNQM.js.map
+//# sourceMappingURL=chunk-YILDTNQM.js.map