From a155fd97a3a611d6ac8e4a7a28e8888d35f7cb8b Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <wenjie.zhang@stu.jiangnan.edu.cn>
Date: Tue, 25 Feb 2025 14:45:49 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BC=98=E5=8C=96=E5=93=8D=E5=BA=94=E9=80=BB?=
 =?UTF-8?q?=E8=BE=91?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/routers/chat_router.py           | 1 +
 web/src/components/ChatComponent.vue | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/routers/chat_router.py b/src/routers/chat_router.py
index 12a60100..7395f9f2 100644
--- a/src/routers/chat_router.py
+++ b/src/routers/chat_router.py
@@ -54,6 +54,7 @@ def chat_post(
                 yield make_chunk(message=f"Retriever error: {e}", status="error")
                 return
 
+        yield make_chunk(status="generating")
         messages = history_manager.get_history_with_msg(modified_query, max_rounds=meta.get('history_round'))
         history_manager.add_user(query)  # 注意这里使用原始查询
         logger.debug(f"Web history: {history_manager.messages}")
diff --git a/web/src/components/ChatComponent.vue b/web/src/components/ChatComponent.vue
index 7dc7fb8e..92d644f8 100644
--- a/web/src/components/ChatComponent.vue
+++ b/web/src/components/ChatComponent.vue
@@ -77,6 +77,7 @@
           <div></div>
         </div>
         <div v-else-if="message.status == 'searching' && isStreaming" class="searching-msg"><i>正在检索……</i></div>
+        <div v-else-if="message.status == 'generating' && isStreaming" class="searching-msg"><i>正在生成……</i></div>
         <div
           v-else-if="(message.text.length == 0 && message.status!='reasoning') || message.status == 'error' || (message.status != 'finished' && !isStreaming)"
           class="err-msg"
@@ -216,7 +217,6 @@ const meta = reactive(JSON.parse(localStorage.getItem('meta')) || {
   use_graph: false,
   use_web: false,
   graph_name: "neo4j",
-  // use_rewrite_query: "off",
   selectedKB: null,
   stream: true,
   summary_title: false,