fix: 修复【应用编排】客户端提问达到限制时，隐藏知识来源 (#696)

Author: shaohuzhang1

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

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)