fix: answer state

web-apps/ui/src/modules/chat-message/ai-message-item.ts

@@ -80,10 +80,13 @@ export class AIChatMessageItem extends ChatMessageItem {
   }
 
   appendChunk(e: ChatEventChunk) {
+    this.state = AnswerState.RECEIVING;
     this.content = `${this.content}${e.output}`;
   }
 
   handleError(e: ChatEventError) {
+    this.state = AnswerState.FAIL;
+
     // {"error": {"error_msg": "The node type is not supported"}, "type": "error"}
     this.error = e.error;
   }
@@ -95,6 +98,7 @@ export class AIChatMessageItem extends ChatMessageItem {
   }
 
   handleResult(e: ChatEventResult) {
+    this.state = AnswerState.SUCCESS;
     this.content = e.output;
   }
 }

web-apps/ui/src/modules/chat-message/chat-message-model.ts

@@ -248,7 +248,7 @@ export class ChatMessageModel implements Disposable {
           this.handleChatEvent(value, ai);
         }
       }
-      ai.state = AnswerState.SUCCESS;
+      // ai.state = AnswerState.SUCCESS;
       return;
     }
   };

web-packages/magent-au/src/au-chat-message/ai-message-item.ts

@@ -77,6 +77,7 @@ export class AUAgentChatMessageItem extends AIChatMessageItemModel {
   }
 
   override appendChunk(e: ChatEventChunk) {
+    this.state = AnswerState.RECEIVING;
     this.content = `${this.content}${e.output}`;
   }
 

web-packages/magent-chat/src/chat-base/ai-message-item-model.ts

@@ -39,15 +39,17 @@ export class AIChatMessageItemModel extends DefaultChatMessageItemModel {
   }
 
   appendChunk(e: ChatEventChunk) {
+    this.state = AnswerState.RECEIVING;
     this.content = `${this.content}${e.output}`;
   }
 
   handleResult(e: ChatEventResult) {
-    this.content = e.output;
     this.state = AnswerState.SUCCESS;
+    this.content = e.output;
   }
 
   handleError(e: ChatEventError) {
+    this.state = AnswerState.FAIL;
     // {"error": {"error_msg": "The node type is not supported"}, "type": "error"}
     this.error = { message: e.message };
   }