From b653042e90741d2bd5a666b0ba33577b79f67744 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Sun, 30 Mar 2025 15:16:08 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E5=8E=86=E5=8F=B2=E8=AE=B0?=
 =?UTF-8?q?=E5=BD=95=E6=B6=88=E6=81=AF=EF=BC=88=E8=8E=B7=E5=BE=97=E6=9B=B4?=
 =?UTF-8?q?=E5=8A=A0=E5=87=86=E7=A1=AE=E7=9A=84=E5=8E=86=E5=8F=B2=E6=B6=88?=
 =?UTF-8?q?=E6=81=AF=EF=BC=8C=E5=90=8E=E9=9D=A2=E8=80=83=E8=99=91=E5=B0=86?=
 =?UTF-8?q?=20conv.history=20=E7=A7=BB=E9=99=A4=EF=BC=89?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 web/src/components/ChatComponent.vue | 49 ++++++++++++++++++++--------
 1 file changed, 35 insertions(+), 14 deletions(-)

diff --git a/web/src/components/ChatComponent.vue b/web/src/components/ChatComponent.vue
index 3187c1ea..6088b629 100644
--- a/web/src/components/ChatComponent.vue
+++ b/web/src/components/ChatComponent.vue
@@ -119,10 +119,10 @@
           请求错误，请重试。{{ message.message }}
         </div>
         <div v-if="message.isStoppedByUser" class="retry-hint">
-          你停止生成了本次回答 
+          你停止生成了本次回答
           <span class="retry-link" @click="retryStoppedMessage(message)">重新编辑问题</span>
         </div>
-          <RefsComponent v-if="message.role=='received' && message.status=='finished'" :message="message" :conv="conv" @regenerateMessage="regenerateMessage" /> 
+          <RefsComponent v-if="message.role=='received' && message.status=='finished'" :message="message" :conv="conv" @regenerateMessage="regenerateMessage" />
       </div>
     </div>
     <div class="bottom">
@@ -176,10 +176,10 @@
           </div>
           <div class="options__right">
             <a-tooltip :title="isStreaming ? '停止回答' : ''">
-              <a-button 
-                size="large" 
-                @click="handleSendOrStop" 
-                :disabled="(!conv.inputText && !isStreaming)" 
+              <a-button
+                size="large"
+                @click="handleSendOrStop"
+                :disabled="(!conv.inputText && !isStreaming)"
                 type="link"
               >
                 <template #icon>
@@ -271,7 +271,7 @@ const meta = reactive(JSON.parse(localStorage.getItem('meta')) || {
   selectedKB: null,
   stream: true,
   summary_title: false,
-  history_round: 5,
+  history_round: 20,
   db_id: null,
   fontSize: 'default',
   wideScreen: false,
@@ -303,6 +303,24 @@ const renderMarkdown = (msg) => {
   }
 }
 
+// 从 message 中获取 history 信息，每个消息都是 {role, content} 的格式
+const getHistory = () => {
+  const history = conv.value.messages.map((msg) => {
+    if (msg.text) {
+      return {
+        role: msg.role === 'sent' ? 'user' : 'assistant',
+        content: msg.text
+      }
+    }
+  }).reduce((acc, cur) => {
+    if (cur) {
+      acc.push(cur)
+    }
+    return acc
+  }, [])
+  return history.slice(-meta.history_round)
+}
+
 const useDatabase = (index) => {
   const selected = opts.databases[index]
   console.log(selected)
@@ -494,14 +512,17 @@ const fetchChatResponse = (user_input, cur_res_id) => {
   const controller = new AbortController();
   const signal = controller.signal;
 
+  const params = {
+    query: user_input,
+    history: getHistory(),
+    meta: meta,
+    cur_res_id: cur_res_id,
+  }
+  console.log(params)
+
   fetch('/api/chat/', {
     method: 'POST',
-    body: JSON.stringify({
-      query: user_input,
-      history: conv.value.history,
-      meta: meta,
-      cur_res_id: cur_res_id,
-    }),
+    body: JSON.stringify(params),
     headers: {
       'Content-Type': 'application/json'
     },
@@ -971,7 +992,7 @@ const regenerateMessage = (message) => {
     padding-right: 0;
     text-align: justify;
     position: relative;
-    
+
 
   }