ai 5.0.0-beta.12 → 5.0.0-beta.14

package/dist/index.mjs

@@ -3206,13 +3206,31 @@ function handleUIMessageStreamFinish({
   onError,
   stream
 }) {
+  let lastMessage = originalMessages == null ? void 0 : originalMessages[originalMessages.length - 1];
+  if ((lastMessage == null ? void 0 : lastMessage.role) !== "assistant") {
+    lastMessage = void 0;
+  } else {
+    messageId = lastMessage.id;
+  }
+  const idInjectedStream = stream.pipeThrough(
+    new TransformStream({
+      transform(chunk, controller) {
+        if (chunk.type === "start") {
+          const startChunk = chunk;
+          if (startChunk.messageId == null && messageId != null) {
+            startChunk.messageId = messageId;
+          }
+        }
+        controller.enqueue(chunk);
+      }
+    })
+  );
   if (onFinish == null) {
-    return stream;
+    return idInjectedStream;
   }
-  const lastMessage = originalMessages == null ? void 0 : originalMessages[originalMessages.length - 1];
   const state = createStreamingUIMessageState({
     lastMessage: lastMessage ? structuredClone(lastMessage) : void 0,
-    messageId
+    messageId: messageId != null ? messageId : ""
     // will be overridden by the stream
   });
   const runUpdateMessageJob = async (job) => {
@@ -3220,19 +3238,7 @@ function handleUIMessageStreamFinish({
     } });
   };
   return processUIMessageStream({
-    stream: stream.pipeThrough(
-      new TransformStream({
-        transform(chunk, controller) {
-          if (chunk.type === "start") {
-            const startChunk = chunk;
-            if (startChunk.messageId == null) {
-              startChunk.messageId = messageId;
-            }
-          }
-          controller.enqueue(chunk);
-        }
-      })
-    ),
+    stream: idInjectedStream,
     runUpdateMessageJob,
     onError
   }).pipeThrough(
@@ -4596,6 +4602,7 @@ var DefaultStreamTextResult = class {
   }
   toUIMessageStream({
     originalMessages,
+    generateMessageId,
     onFinish,
     messageMetadata,
     sendReasoning = true,
@@ -4788,7 +4795,7 @@ var DefaultStreamTextResult = class {
     );
     return handleUIMessageStreamFinish({
       stream: baseStream,
-      messageId: responseMessageId != null ? responseMessageId : this.generateId(),
+      messageId: responseMessageId != null ? responseMessageId : generateMessageId == null ? void 0 : generateMessageId(),
       originalMessages,
       onFinish,
       onError
@@ -4796,6 +4803,7 @@ var DefaultStreamTextResult = class {
   }
   pipeUIMessageStreamToResponse(response, {
     originalMessages,
+    generateMessageId,
     onFinish,
     messageMetadata,
     sendReasoning,
@@ -4809,6 +4817,7 @@ var DefaultStreamTextResult = class {
       response,
       stream: this.toUIMessageStream({
         originalMessages,
+        generateMessageId,
         onFinish,
         messageMetadata,
         sendReasoning,
@@ -4829,6 +4838,7 @@ var DefaultStreamTextResult = class {
   }
   toUIMessageStreamResponse({
     originalMessages,
+    generateMessageId,
     onFinish,
     messageMetadata,
     sendReasoning,
@@ -4841,6 +4851,7 @@ var DefaultStreamTextResult = class {
     return createUIMessageStreamResponse({
       stream: this.toUIMessageStream({
         originalMessages,
+        generateMessageId,
         onFinish,
         messageMetadata,
         sendReasoning,