@mastra/ai-sdk 1.1.2 → 1.1.3-alpha.0

package/dist/index.js

@@ -6416,13 +6416,21 @@ async function handleChatStream({
       }
     }
   }
-  const mergedOptions = {
-    ...defaultOptions2,
-    ...rest,
+  const { structuredOutput: restStructuredOutput, ...restOptions } = rest;
+  const { structuredOutput: defaultStructuredOutput, ...defaultOptionsRest } = defaultOptions2 ?? {};
+  const structuredOutput = restStructuredOutput ?? defaultStructuredOutput;
+  const mergedProviderOptions = {
+    ...defaultOptions2?.providerOptions,
+    ...restOptions.providerOptions
+  };
+  const baseOptions = {
+    ...defaultOptionsRest,
+    ...restOptions,
     ...runId && { runId },
-    requestContext: requestContext || defaultOptions2?.requestContext
+    requestContext: requestContext || defaultOptions2?.requestContext,
+    ...Object.keys(mergedProviderOptions).length > 0 && { providerOptions: mergedProviderOptions }
   };
-  const result = resumeData ? await agentObj.resumeStream(resumeData, mergedOptions) : await agentObj.stream(messagesToSend, mergedOptions);
+  const result = resumeData ? structuredOutput ? await agentObj.resumeStream(resumeData, { ...baseOptions, structuredOutput }) : await agentObj.resumeStream(resumeData, baseOptions) : structuredOutput ? await agentObj.stream(messagesToSend, { ...baseOptions, structuredOutput }) : await agentObj.stream(messagesToSend, baseOptions);
   return createUIMessageStream({
     originalMessages: messages,
     execute: async ({ writer }) => {
@@ -7143,6 +7151,13 @@ function createProcessorMiddleware(options) {
               messages: messageList.get.all.db(),
               messageList,
               state: {},
+              result: {
+                text: "",
+                usage: { inputTokens: 0, outputTokens: 0, totalTokens: 0 },
+                finishReason: "unknown",
+                steps: []
+              },
+              retryCount: 0,
               requestContext,
               abort: (reason) => {
                 throw new TripWire(reason || "Aborted by processor");
@@ -7277,10 +7292,23 @@ function createProcessorMiddleware(options) {
               if (!processor.processOutputResult) continue;
               try {
                 const procState = processorStates.get(processor.id);
+                const finishChunk = (procState?.streamParts ?? []).find((p) => p.type === "finish");
+                const outputResult = {
+                  text: (procState?.streamParts ?? []).filter((p) => p.type === "text-delta").map((p) => p.payload?.text ?? "").join(""),
+                  usage: finishChunk?.payload?.output?.usage ?? {
+                    inputTokens: 0,
+                    outputTokens: 0,
+                    totalTokens: 0
+                  },
+                  finishReason: finishChunk?.payload?.stepResult?.reason ?? "unknown",
+                  steps: []
+                };
                 await processor.processOutputResult({
                   messages: messageList.get.all.db(),
                   messageList,
                   state: procState?.customState ?? {},
+                  result: outputResult,
+                  retryCount: 0,
                   requestContext,
                   abort: (reason) => {
                     throw new TripWire(reason || "Aborted by processor");