fix: add msgId

difizen · Nov 27, 2024 · bb8bd8a · bb8bd8a
1 parent 8d17ec5
commit bb8bd8a
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 0 deletions.
diff --git a/.changeset/violet-planes-repeat.md b/.changeset/violet-planes-repeat.md
@@ -0,0 +1,5 @@
+---
+'@difizen/magent-chat': patch
+---
+
+add msgId to fit aistudio scene
diff --git a/web-packages/magent-chat/src/chat-base/ai-message-item-model.ts b/web-packages/magent-chat/src/chat-base/ai-message-item-model.ts
@@ -41,6 +41,9 @@ export class AIChatMessageItemModel extends DefaultChatMessageItemModel {
   appendChunk(e: ChatEventChunk) {
     this.state = AnswerState.RECEIVING;
     this.content = `${this.content}${e.output}`;
+    if (!this.msgId && e.msgId) {
+      this.msgId = e.msgId;
+    }
   }
 
   handleResult(e: ChatEventResult) {

diff --git a/web-packages/magent-chat/src/chat-base/chat-message-item-model.ts b/web-packages/magent-chat/src/chat-base/chat-message-item-model.ts
@@ -23,6 +23,7 @@ export class DefaultChatMessageItemModel {
   };
 
   id?: string;
+  msgId?: string;
   sender: IChatMessageSender;
 
   option: ChatMessageItemOption;

diff --git a/web-packages/magent-chat/src/chat-base/protocol.ts b/web-packages/magent-chat/src/chat-base/protocol.ts
@@ -107,6 +107,7 @@ export interface IChatEvent {
 export interface ChatEventChunk extends IChatEvent {
   output: string;
   type: 'chunk';
+  msgId?: string;
   [key: string]: any;
 }