ai 6.0.252 → 6.0.255

package/dist/index.mjs

@@ -1193,7 +1193,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "6.0.252" : "0.0.0-test";
+var VERSION = true ? "6.0.255" : "0.0.0-test";
 
 // src/util/download/download.ts
 var download = async ({
@@ -5949,6 +5949,7 @@ function processUIMessageStream({
             case "reasoning-start": {
               const reasoningPart = {
                 type: "reasoning",
+                id: chunk.id,
                 text: "",
                 providerMetadata: chunk.providerMetadata,
                 state: "streaming"
@@ -6259,7 +6260,7 @@ function processUIMessageStream({
               }
               await updateMessageMetadata(chunk.messageMetadata);
               if (chunk.messageId != null || chunk.messageMetadata != null) {
-                write();
+                write({ updateStatus: false });
               }
               break;
             }
@@ -8959,6 +8960,27 @@ function createUIMessageStream({
 }
 
 // src/ui-message-stream/read-ui-message-stream.ts
+function createUIMessageSnapshot(message) {
+  const textByPartIndex = /* @__PURE__ */ new Map();
+  const messageWithoutText = {
+    ...message,
+    parts: message.parts.map((part, index) => {
+      if (part.type === "text" || part.type === "reasoning") {
+        textByPartIndex.set(index, part.text);
+        return { ...part, text: "" };
+      }
+      return part;
+    })
+  };
+  const snapshot = structuredClone(messageWithoutText);
+  for (const [index, text2] of textByPartIndex) {
+    const part = snapshot.parts[index];
+    if (part.type === "text" || part.type === "reasoning") {
+      part.text = text2;
+    }
+  }
+  return snapshot;
+}
 function readUIMessageStream({
   message,
   stream,
@@ -8991,7 +9013,7 @@ function readUIMessageStream({
         return job({
           state,
           write: () => {
-            controller == null ? void 0 : controller.enqueue(structuredClone(state.message));
+            controller == null ? void 0 : controller.enqueue(createUIMessageSnapshot(state.message));
           }
         });
       },
@@ -9276,6 +9298,7 @@ var uiMessagesSchema = lazySchema2(
             }),
             z8.object({
               type: z8.literal("reasoning"),
+              id: z8.string().optional(),
               text: z8.string(),
               state: z8.enum(["streaming", "done"]).optional(),
               providerMetadata: providerMetadataSchema.optional()
@@ -14063,12 +14086,14 @@ var AbstractChat = class {
           }
           return job({
             state: response.state,
-            write: () => {
+            write: ({ updateStatus = true } = {}) => {
               var _a23;
               if (response.abortController.signal.aborted) {
                 return;
               }
-              this.setStatus({ status: "streaming" });
+              if (updateStatus) {
+                this.setStatus({ status: "streaming" });
+              }
               const replaceLastMessage = response.state.message.id === ((_a23 = this.lastMessage) == null ? void 0 : _a23.id);
               if (replaceLastMessage) {
                 this.state.replaceMessage(