From 9354e2c38ebfcae2b9541ebb785442ddf4af0b43 Mon Sep 17 00:00:00 2001
From: dataeaseShu <tong.shu@fit2cloud.com>
Date: Thu, 23 Oct 2025 10:09:50 +0800
Subject: [PATCH] fix(Custom prompt words): The effective data source has been
 changed from some data sources to all data sources, and the details page has
 not changed.

---
 frontend/src/views/chat/ChatRow.vue           | 10 +++++++-
 frontend/src/views/chat/index.vue             | 23 ++++++++++++++++---
 .../dashboard/editor/ChatChartSelection.vue   |  2 +-
 frontend/src/views/system/prompt/index.vue    |  2 +-
 4 files changed, 31 insertions(+), 6 deletions(-)

diff --git a/frontend/src/views/chat/ChatRow.vue b/frontend/src/views/chat/ChatRow.vue
index ffa4176f..16931d5f 100644
--- a/frontend/src/views/chat/ChatRow.vue
+++ b/frontend/src/views/chat/ChatRow.vue
@@ -9,6 +9,7 @@ withDefaults(
     msg: ChatMessage
     currentChat: ChatInfo
     hideAvatar?: boolean
+    logoAssistant?: string
   }>(),
   {
     hideAvatar: false,
@@ -21,7 +22,14 @@ const appearanceStore = useAppearanceStoreWithOut()
   <div class="chat-row-container">
     <div class="chat-row" :class="{ 'right-to-left': msg.role === 'user' }">
       <div v-if="msg.role === 'assistant'" class="ai-avatar">
-        <el-icon v-if="!hideAvatar">
+        <img
+          v-if="!hideAvatar && appearanceStore.getLogin"
+          :src="logoAssistant ? logoAssistant : appearanceStore.getLogin"
+          alt=""
+          width="28"
+          height="28"
+        />
+        <el-icon v-else-if="!hideAvatar">
           <custom_small v-if="appearanceStore.themeColor === 'custom'"></custom_small>
           <logo_fold v-else />
         </el-icon>
diff --git a/frontend/src/views/chat/index.vue b/frontend/src/views/chat/index.vue
index 0926f2fb..3456c73c 100644
--- a/frontend/src/views/chat/index.vue
+++ b/frontend/src/views/chat/index.vue
@@ -183,7 +183,20 @@
           </div>
         </div>
         <div v-else-if="computedMessages.length == 0 && loading" class="welcome-content-block">
-          <logo />
+          <div style="display: flex; align-items: center; height: 30px">
+            <img
+              height="30"
+              width="30"
+              v-if="logoAssistant || loginBg"
+              :src="logoAssistant ? logoAssistant : loginBg"
+              alt=""
+            />
+            <el-icon size="30" v-else
+              ><custom_small v-if="appearanceStore.themeColor !== 'default'"></custom_small>
+              <LOGO_fold v-else></LOGO_fold
+            ></el-icon>
+            <span style="margin-left: 12px">{{ appearanceStore.name }}</span>
+          </div>
         </div>
         <el-scrollbar
           v-if="computedMessages.length > 0"
@@ -200,7 +213,12 @@
             }"
           >
             <template v-for="(message, _index) in computedMessages" :key="_index">
-              <ChatRow :current-chat="currentChat" :msg="message" :hide-avatar="message.first_chat">
+              <ChatRow
+                :logoAssistant="logoAssistant"
+                :current-chat="currentChat"
+                :msg="message"
+                :hide-avatar="message.first_chat"
+              >
                 <RecommendQuestion
                   v-if="message.role === 'assistant' && message.first_chat"
                   ref="recommendQuestionRef"
@@ -432,7 +450,6 @@ import icon_replace_outlined from '@/assets/svg/icon_replace_outlined.svg'
 import icon_screen_outlined from '@/assets/svg/icon_screen_outlined.svg'
 import icon_start_outlined from '@/assets/svg/icon_start_outlined.svg'
 import logo_fold from '@/assets/svg/logo-custom_small.svg'
-import logo from '@/assets/LOGO.svg'
 import icon_send_filled from '@/assets/svg/icon_send_filled.svg'
 import { useAssistantStore } from '@/stores/assistant'
 import { onClickOutside } from '@vueuse/core'
diff --git a/frontend/src/views/dashboard/editor/ChatChartSelection.vue b/frontend/src/views/dashboard/editor/ChatChartSelection.vue
index f422aa73..893f6191 100644
--- a/frontend/src/views/dashboard/editor/ChatChartSelection.vue
+++ b/frontend/src/views/dashboard/editor/ChatChartSelection.vue
@@ -58,7 +58,7 @@
 </template>
 
 <script setup lang="ts">
-import { computed, onMounted, reactive, ref } from 'vue'
+import { computed, reactive, ref } from 'vue'
 import { useI18n } from 'vue-i18n'
 import { Chat, chatApi, ChatInfo } from '@/api/chat.ts'
 import DashboardChatList from '@/views/dashboard/editor/DashboardChatList.vue'
diff --git a/frontend/src/views/system/prompt/index.vue b/frontend/src/views/system/prompt/index.vue
index 0de18e67..93d6d997 100644
--- a/frontend/src/views/system/prompt/index.vue
+++ b/frontend/src/views/system/prompt/index.vue
@@ -629,7 +629,7 @@ const typeChange = (val: any) => {
       <el-form-item :label="t('ds.title')">
         <div class="content">
           {{
-            pageForm.datasource_names.length
+            pageForm.datasource_names.length && pageForm.specific_ds
               ? pageForm.datasource_names.join()
               : t('training.all_data_sources')
           }}