ai 6.0.295 → 6.0.296

package/dist/index.mjs

@@ -1224,7 +1224,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.295" : "0.0.0-test";
+var VERSION = true ? "6.0.296" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5950,7 +5950,7 @@ function processUIMessageStream({
     new TransformStream({
       async transform(chunk, controller) {
         await runUpdateMessageJob(async ({ state, write }) => {
-          var _a23, _b, _c, _d;
+          var _a23, _b, _c, _d, _e, _f, _g, _h;
           function getCurrentStepParts() {
             const parts = state.message.parts;
             let currentStepStartIndex = parts.length - 1;
@@ -6384,7 +6384,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: (_e = chunk.toolMetadata) != null ? _e : toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -6400,7 +6400,7 @@ function processUIMessageStream({
                     preliminary: chunk.preliminary,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: (_f = chunk.toolMetadata) != null ? _f : toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -6421,7 +6421,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: (_g = chunk.toolMetadata) != null ? _g : toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -6437,7 +6437,7 @@ function processUIMessageStream({
                     providerExecuted: chunk.providerExecuted,
                     providerMetadata: chunk.providerMetadata,
                     title: toolInvocation.title,
-                    toolMetadata: toolInvocation.toolMetadata
+                    toolMetadata: (_h = chunk.toolMetadata) != null ? _h : toolInvocation.toolMetadata
                   },
                   toolInvocation
                 );
@@ -10272,8 +10272,9 @@ async function createAgentUIStream({
   ...uiMessageStreamOptions
 }) {
   var _a23;
+  const inputMessages = uiMessages != null ? uiMessages : uiMessageStreamOptions.originalMessages;
   const validatedMessages = await validateUIMessagesForAgent({
-    messages: uiMessages,
+    messages: inputMessages,
     tools: agent.tools
   });
   const modelMessages = await convertToModelMessages(validatedMessages, {
@@ -12704,21 +12705,32 @@ function smoothStream({
           controller.enqueue(chunk);
           return;
         }
-        if ((chunk.type !== type || chunk.id !== id) && (buffer.length > 0 || providerMetadata != null)) {
+        if (chunk.text.length === 0 && chunk.providerMetadata != null) {
+          flushBuffer(controller);
+          controller.enqueue(chunk);
+          return;
+        }
+        if (buffer.length > 0 && (chunk.type !== type || chunk.id !== id || providerMetadata != null || chunk.providerMetadata != null)) {
           flushBuffer(controller);
         }
         buffer += chunk.text;
         id = chunk.id;
         type = chunk.type;
-        if (chunk.providerMetadata != null) {
-          providerMetadata = chunk.providerMetadata;
-        }
+        providerMetadata = chunk.providerMetadata;
         let match;
         while ((match = detectChunk(buffer)) != null) {
-          controller.enqueue({ type, text: match, id });
+          controller.enqueue({
+            type,
+            text: match,
+            id,
+            ...providerMetadata != null ? { providerMetadata } : {}
+          });
           buffer = buffer.slice(match.length);
           await delay(isDocumentHidden() ? null : delayInMs);
         }
+        if (buffer.length === 0) {
+          providerMetadata = void 0;
+        }
       }
     });
   };