From 4270847b5d50ddc5a3eae01c17f9cb35e06e9f28 Mon Sep 17 00:00:00 2001
From: GeekMaster <yangjian102621@gmail.com>
Date: Wed, 17 Sep 2025 16:18:16 +0800
Subject: [PATCH] optimize regenerate for mobile chat page

---
 web/src/components/mobile/ChatReply.vue | 17 +++---
 web/src/views/ChatPlus.vue              |  2 +-
 web/src/views/mobile/ChatSession.vue    | 69 ++++++++++---------------
 3 files changed, 38 insertions(+), 50 deletions(-)

diff --git a/web/src/components/mobile/ChatReply.vue b/web/src/components/mobile/ChatReply.vue
index 91f34fe3..d54e74d9 100644
--- a/web/src/components/mobile/ChatReply.vue
+++ b/web/src/components/mobile/ChatReply.vue
@@ -42,16 +42,15 @@
 </template>
 
 <script setup>
-import { computed, onMounted, onUnmounted, ref } from 'vue'
-import { showImagePreview } from 'vant'
-import Thinking from '../Thinking.vue'
+import { processContent } from '@/utils/libs'
+import Clipboard from 'clipboard'
 import hl from 'highlight.js'
 import MarkdownIt from 'markdown-it'
 import emoji from 'markdown-it-emoji'
 import mathjaxPlugin from 'markdown-it-mathjax3'
-import { processContent } from '@/utils/libs'
-import Clipboard from 'clipboard'
-import { showNotify } from 'vant'
+import { showImagePreview, showNotify } from 'vant'
+import { computed, onMounted, onUnmounted, ref } from 'vue'
+import Thinking from '../Thinking.vue'
 
 const props = defineProps({
   content: {
@@ -85,6 +84,10 @@ const props = defineProps({
     type: String,
     default: '',
   },
+  messageIndex: {
+    type: Number,
+    default: -1,
+  },
 })
 
 const emits = defineEmits(['regenerate'])
@@ -125,7 +128,7 @@ const contentRef = ref(null)
 
 // 处理重新生成
 const handleRegenerate = () => {
-  emits('regenerate', props.messageId)
+  emits('regenerate', props.messageIndex)
 }
 
 const clipboard = ref(null)
diff --git a/web/src/views/ChatPlus.vue b/web/src/views/ChatPlus.vue
index 85476bf1..bf46ccfb 100644
--- a/web/src/views/ChatPlus.vue
+++ b/web/src/views/ChatPlus.vue
@@ -1178,7 +1178,7 @@ const reGenerate = function (messageIndex) {
   }
 
   if (messageIndex === -1 || isNaN(messageIndex)) {
-    ElMessage.error('找不到要编辑的消息')
+    ElMessage.error('找不到要重新生成消息')
     return
   }
 
diff --git a/web/src/views/mobile/ChatSession.vue b/web/src/views/mobile/ChatSession.vue
index e160d3bf..a40dbdca 100644
--- a/web/src/views/mobile/ChatSession.vue
+++ b/web/src/views/mobile/ChatSession.vue
@@ -28,11 +28,17 @@
             error-text="请求失败，点击重新加载"
             @load="onLoad"
           >
-            <van-cell v-for="item in chatData" :key="item" :border="false" class="message-line">
+            <van-cell
+              v-for="(item, index) in chatData"
+              :key="item"
+              :border="false"
+              class="message-line"
+            >
               <chat-prompt
                 v-if="item.type === 'prompt'"
                 :content="item.content"
                 :icon="item.icon"
+                :message-index="index"
               />
               <chat-reply
                 v-else-if="item.type === 'reply'"
@@ -43,6 +49,7 @@
                 :is-generating="isGenerating"
                 :show-actions="item.showAction"
                 :error="item.error"
+                :message-index="index"
                 @regenerate="handleRegenerate"
               />
             </van-cell>
@@ -62,7 +69,7 @@
               rows="1"
               :autosize="{ maxHeight: 100, minHeight: 20 }"
               show-word-limit
-              @keyup.enter="sendMessage"
+              @keyup.enter="sendMessage(0)"
             >
               <template #left-icon>
                 <van-uploader
@@ -76,7 +83,7 @@
                 </van-uploader>
               </template>
               <template #button>
-                <van-button size="small" type="primary" v-if="!isGenerating" @click="sendMessage"
+                <van-button size="small" type="primary" v-if="!isGenerating" @click="sendMessage(0)"
                   >发送</van-button
                 >
               </template>
@@ -115,20 +122,18 @@
 <script setup>
 import ChatPrompt from '@/components/mobile/ChatPrompt.vue'
 import ChatReply from '@/components/mobile/ChatReply.vue'
-import { checkSession } from '@/store/cache'
+import MobileFileList from '@/components/mobile/MobileFileList.vue'
+import { checkSession, getClientId } from '@/store/cache'
 import { getUserToken } from '@/store/session'
 import { useSharedStore } from '@/store/sharedata'
-import { showMessageError, showLoading, closeLoading } from '@/utils/dialog'
+import { closeLoading, showLoading, showMessageError } from '@/utils/dialog'
 import { httpGet, httpPost } from '@/utils/http'
-import MobileFileList from '@/components/mobile/MobileFileList.vue'
-import { processContent, randString, renderInputText, UUID } from '@/utils/libs'
+import { randString, renderInputText, UUID } from '@/utils/libs'
 import { fetchEventSource } from '@microsoft/fetch-event-source'
-import hl from 'highlight.js'
 import 'highlight.js/styles/a11y-dark.css'
 import { showImagePreview, showNotify, showToast } from 'vant'
-import { nextTick, onMounted, onUnmounted, ref, watch } from 'vue'
+import { nextTick, onMounted, ref, watch } from 'vue'
 import { useRouter } from 'vue-router'
-import { getClientId } from '@/store/cache'
 
 const winHeight = ref(0)
 const navBarRef = ref(null)
@@ -434,7 +439,7 @@ const sendSSERequest = async (message) => {
 }
 
 // 发送消息
-const sendMessage = () => {
+const sendMessage = (messageId = 0) => {
   if (isGenerating.value) {
     showToast('AI 正在作答中，请稍后...')
     return
@@ -480,6 +485,7 @@ const sendMessage = () => {
     prompt: prompt.value,
     stream: stream.value,
     files: files.value,
+    last_msg_id: messageId || 0,
   })
 
   previousText.value = prompt.value
@@ -504,47 +510,26 @@ const stopGenerate = function () {
 }
 
 // 处理从ChatReply组件触发的重新生成
-const handleRegenerate = (messageId) => {
+const handleRegenerate = (messageIndex) => {
   if (isGenerating.value) {
     showToast('AI 正在作答中，请稍后...')
     return
   }
 
-  console.log('messageId', messageId)
-  console.log('chatData.value', chatData.value)
-
-  // 判断 messageId 是整数
-  if (messageId !== '' && isNaN(messageId)) {
-    showToast('消息 ID 不合法，无法重新生成')
+  if (messageIndex === -1 || isNaN(messageIndex)) {
+    showToast('找不到要重新生成消息')
     return
   }
 
-  chatData.value = chatData.value.filter((item) => item.id < messageId && !item.isHello)
+  // 找到该消息的ID
+  const messageId = chatData.value[messageIndex].id
+  // 移除该消息之后的所有消息
+  chatData.value = chatData.value.slice(0, messageIndex)
   const userPrompt = chatData.value.pop()
+  prompt.value = userPrompt.content.text
 
-  // 添加空回复消息
-  const _role = getRoleById(roleId.value)
-  chatData.value.push({
-    chat_id: chatId,
-    role_id: roleId.value,
-    type: 'reply',
-    id: randString(32),
-    icon: _role['icon'],
-    content: {
-      text: '',
-    },
-  })
-
-  // 发送 SSE 请求
-  sendSSERequest({
-    user_id: loginUser.value.id,
-    role_id: roleId.value,
-    model_id: modelId.value,
-    chat_id: chatId.value,
-    last_msg_id: messageId,
-    prompt: userPrompt.content.text,
-    stream: stream.value,
-    files: [],
+  nextTick(() => {
+    sendMessage(messageId)
   })
 }