From e882bd1a5ba39c0f3e95029e2773898a3b528d2d Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <shaohu.zhang@fit2cloud.com>
Date: Wed, 3 Jul 2024 15:00:15 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E3=80=90=E5=BA=94?=
 =?UTF-8?q?=E7=94=A8=E7=BC=96=E6=8E=92=E3=80=91=E5=AE=A2=E6=88=B7=E7=AB=AF?=
 =?UTF-8?q?=E6=8F=90=E9=97=AE=E8=BE=BE=E5=88=B0=E9=99=90=E5=88=B6=E6=97=B6?=
 =?UTF-8?q?=EF=BC=8C=E9=9A=90=E8=97=8F=E7=9F=A5=E8=AF=86=E6=9D=A5=E6=BA=90?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/api/type/application.ts      | 7 +++++++
 ui/src/components/ai-chat/index.vue | 8 +++++---
 2 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/ui/src/api/type/application.ts b/ui/src/api/type/application.ts
index 4dc9047f858..4e4c4564eba 100644
--- a/ui/src/api/type/application.ts
+++ b/ui/src/api/type/application.ts
@@ -29,6 +29,7 @@ interface chatType {
   is_stop?: boolean
   record_id: string
   vote_status: string
+  status?: number
 }
 
 export class ChatRecordManage {
@@ -105,6 +106,12 @@ export class ChatManagement {
       chatRecord.append(content)
     }
   }
+  static updateStatus(chatRecordId: string, code: number) {
+    const chatRecord = this.chatMessageContainer[chatRecordId]
+    if (chatRecord) {
+      chatRecord.chat.status = code
+    }
+  }
   /**
    * 持续从缓存区 写出数据
    * @param chatRecordId 对话记录id
diff --git a/ui/src/components/ai-chat/index.vue b/ui/src/components/ai-chat/index.vue
index 9be3820fdd9..d015defc1bf 100644
--- a/ui/src/components/ai-chat/index.vue
+++ b/ui/src/components/ai-chat/index.vue
@@ -104,7 +104,7 @@
                   >
                 </div>
               </div>
-              <div v-if="item.write_ed && props.appId" class="flex-between">
+              <div v-if="item.write_ed && props.appId && 500 != item.status" class="flex-between">
                 <OperationButton
                   :data="item"
                   :applicationId="appId"
@@ -280,7 +280,7 @@ watch(
 function showSource(row: any) {
   if (props.log) {
     return true
-  } else if (row.write_ed) {
+  } else if (row.write_ed && 500 !== row.status) {
     if (id || props.data?.show_source) {
       return true
     }
@@ -453,6 +453,7 @@ const errorWrite = (chat: any, message?: string) => {
   ChatManagement.addChatRecord(chat, 50, loading)
   ChatManagement.write(chat.id)
   ChatManagement.append(chat.id, message || '抱歉，当前正在维护，无法提供服务，请稍后再试！')
+  ChatManagement.updateStatus(chat.id, 500)
   ChatManagement.close(chat.id)
 }
 function chatMessage(chat?: any, problem?: string, re_chat?: boolean) {
@@ -466,7 +467,8 @@ function chatMessage(chat?: any, problem?: string, re_chat?: boolean) {
       write_ed: false,
       is_stop: false,
       record_id: '',
-      vote_status: '-1'
+      vote_status: '-1',
+      status: undefined
     })
     chatList.value.push(chat)
     ChatManagement.addChatRecord(chat, 50, loading)