@zenning/ai 5.0.42 → 5.0.43

package/dist/index.js

@@ -3805,8 +3805,11 @@ function processUIMessageStream({
             }
             case "start": {
               if (chunk.messageId != null) {
-                if (state.isFinalized && state.message.parts.length > 0) {
-                  state.messageQueue.push(state.message);
+                const isNewMessage = state.isFinalized && state.message.id !== chunk.messageId;
+                if (isNewMessage) {
+                  if (state.message.parts.length > 0 || state.message.id) {
+                    state.messageQueue.push(structuredClone(state.message));
+                  }
                   state.message = {
                     id: chunk.messageId,
                     metadata: chunk.messageMetadata,
@@ -3817,15 +3820,24 @@ function processUIMessageStream({
                   state.activeReasoningParts = {};
                   state.partialToolCalls = {};
                   state.isFinalized = false;
-                } else {
+                  write();
+                } else if (!state.message.id) {
                   state.message.id = chunk.messageId;
                   await updateMessageMetadata(chunk.messageMetadata);
+                  if (chunk.messageId != null || chunk.messageMetadata != null) {
+                    write();
+                  }
+                } else {
+                  await updateMessageMetadata(chunk.messageMetadata);
+                  if (chunk.messageMetadata != null) {
+                    write();
+                  }
                 }
               } else {
                 await updateMessageMetadata(chunk.messageMetadata);
-              }
-              if (chunk.messageId != null || chunk.messageMetadata != null) {
-                write();
+                if (chunk.messageMetadata != null) {
+                  write();
+                }
               }
               break;
             }
@@ -9790,33 +9802,38 @@ var AbstractChat = class {
               this.setStatus({ status: "streaming" });
               if (activeResponse.state.messageQueue && activeResponse.state.messageQueue.length > 0) {
                 for (const queuedMessage of activeResponse.state.messageQueue) {
-                  const existingMessageIndex2 = this.state.messages.findIndex(
+                  const existingIndex = this.state.messages.findIndex(
                     (msg) => msg.id === queuedMessage.id
                   );
-                  if (existingMessageIndex2 >= 0) {
-                    this.state.replaceMessage(existingMessageIndex2, queuedMessage);
+                  if (existingIndex >= 0) {
+                    this.state.replaceMessage(existingIndex, queuedMessage);
                   } else {
                     this.state.pushMessage(queuedMessage);
                   }
                 }
                 activeResponse.state.messageQueue = [];
               }
-              const replaceLastMessage = activeResponse.state.message.id === ((_a18 = this.lastMessage) == null ? void 0 : _a18.id);
-              const existingMessageIndex = this.state.messages.findIndex(
+              if (!activeResponse.state.message.id) {
+                return;
+              }
+              const currentMessageIndex = this.state.messages.findIndex(
                 (msg) => msg.id === activeResponse.state.message.id
               );
-              if (existingMessageIndex >= 0) {
+              if (currentMessageIndex >= 0) {
                 this.state.replaceMessage(
-                  existingMessageIndex,
-                  activeResponse.state.message
-                );
-              } else if (replaceLastMessage) {
-                this.state.replaceMessage(
-                  this.state.messages.length - 1,
+                  currentMessageIndex,
                   activeResponse.state.message
                 );
               } else {
-                this.state.pushMessage(activeResponse.state.message);
+                const replaceLastMessage = activeResponse.state.message.id === ((_a18 = this.lastMessage) == null ? void 0 : _a18.id);
+                if (replaceLastMessage && this.state.messages.length > 0) {
+                  this.state.replaceMessage(
+                    this.state.messages.length - 1,
+                    activeResponse.state.message
+                  );
+                } else {
+                  this.state.pushMessage(activeResponse.state.message);
+                }
               }
             }
           })
@@ -9851,6 +9868,16 @@ var AbstractChat = class {
         }
         activeResponse.state.messageQueue = [];
       }
+      if (activeResponse.state.isFinalized && activeResponse.state.message.parts.length > 0) {
+        const existingMessageIndex = this.state.messages.findIndex(
+          (msg) => msg.id === activeResponse.state.message.id
+        );
+        if (existingMessageIndex >= 0) {
+          this.state.replaceMessage(existingMessageIndex, activeResponse.state.message);
+        } else {
+          this.state.pushMessage(activeResponse.state.message);
+        }
+      }
       this.setStatus({ status: "ready" });
     } catch (err) {
       if (isAbort || err.name === "AbortError") {