From c8facdaffb8e1cb0fd53cf07770146c2fb96a940 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Sun, 30 Mar 2025 11:36:03 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=AF=B9=E8=AF=9D=E4=BD=93?=
 =?UTF-8?q?=E9=AA=8C?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/models/__init__.py                  |  2 +-
 web/src/components/ChatComponent.vue    | 12 ++++++++++++
 web/src/components/MessageComponent.vue | 17 ++++++++---------
 3 files changed, 21 insertions(+), 10 deletions(-)

diff --git a/src/models/__init__.py b/src/models/__init__.py
index 9f4c0d8c..b796a52b 100644
--- a/src/models/__init__.py
+++ b/src/models/__init__.py
@@ -41,7 +41,7 @@ def select_model(model_provider=None, model_name=None):
         return OpenModel(model_name)
 
     elif model_provider == "custom":
-        model_info = next((x for x in conf.custom_models if x["custom_id"] == model_name), None)
+        model_info = next((x for x in config.custom_models if x["custom_id"] == model_name), None)
         if model_info is None:
             raise ValueError(f"Model {model_name} not found in custom models")
 
diff --git a/web/src/components/ChatComponent.vue b/web/src/components/ChatComponent.vue
index 4b14477e..3132ecc6 100644
--- a/web/src/components/ChatComponent.vue
+++ b/web/src/components/ChatComponent.vue
@@ -545,6 +545,18 @@ onMounted(() => {
 
   chatContainer.value.addEventListener('scroll', handleUserScroll);
 
+  // 检查现有消息中是否有内容为空的情况
+  if (conv.value.messages && conv.value.messages.length > 0) {
+    conv.value.messages.forEach(msg => {
+      if (msg.role === 'received' && (!msg.text || msg.text.trim() === '')) {
+        msg.status = 'error';
+        msg.message = '内容加载失败';
+      }
+    });
+  }
+
+  console.log(conv.value.messages)
+
   // 从本地存储加载数据
   const storedMeta = localStorage.getItem('meta');
   if (storedMeta) {
diff --git a/web/src/components/MessageComponent.vue b/web/src/components/MessageComponent.vue
index f3761ce7..14680829 100644
--- a/web/src/components/MessageComponent.vue
+++ b/web/src/components/MessageComponent.vue
@@ -34,6 +34,10 @@
         <i>正在生成……</i>
       </div>
 
+      <div v-else-if="status === 'error'" class="err-msg" @click="$emit('retry')">
+        请求错误，请重试。{{ errorMessage }}
+      </div>
+
       <!-- 消息内容 -->
       <div v-else-if="contentHtml" v-html="contentHtml" class="message-md"></div>
 
@@ -42,11 +46,7 @@
 
       <!-- 引用组件 (ChatComponent特有) -->
       <slot name="refs"></slot>
-    </div>
-
-    <!-- 错误消息 -->
-    <div v-else-if="status === 'error'" class="err-msg" @click="$emit('retry')">
-      请求错误，请重试。{{ errorMessage }}
+      <!-- 错误消息 -->
     </div>
 
     <!-- 自定义内容 -->
@@ -161,7 +161,6 @@ const isEmptyAndLoading = computed(() => {
     text-align: left;
     margin: 0 0 16px 0;
     padding: 0px;
-    text-align: justify;
     background-color: transparent;
     border-radius: 0;
   }
@@ -195,12 +194,12 @@ const isEmptyAndLoading = computed(() => {
   }
 
   .err-msg {
-    color: #eb8080;
-    border: 1px solid #eb8080;
+    color: #d15252;
+    border: 1px solid #f19999;
     padding: 0.5rem 1rem;
     border-radius: 8px;
     text-align: left;
-    background: #FFF5F5;
+    background: #fffbfb;
     margin-bottom: 10px;
     cursor: pointer;
   }