重写重新生成的逻辑

web/src/components/ChatReply.vue

@@ -26,7 +26,7 @@
               </el-tooltip>
             </span>
             <span v-if="!readOnly" class="flex">
-              <span class="bar-item" @click="reGenerate()">
+              <span class="bar-item" @click="reGenerate(data.id)">
                 <el-tooltip class="box-item" effect="dark" content="重新生成" placement="bottom">
                   <el-icon><Refresh /></el-icon>
                 </el-tooltip>
@@ -79,7 +79,7 @@
               </el-tooltip>
             </span>
             <span v-if="!readOnly" class="flex">
-              <span class="bar-item bg" @click="reGenerate()">
+              <span class="bar-item bg" @click="reGenerate(data.id)">
                 <el-tooltip class="box-item" effect="dark" content="重新生成" placement="bottom">
                   <el-icon><Refresh /></el-icon>
                 </el-tooltip>
@@ -223,8 +223,8 @@ const stopSynthesis = () => {
 }
 
 // 重新生成
-const reGenerate = () => {
-  emits('regen')
+const reGenerate = (messageId) => {
+  emits('regen', messageId)
 }
 </script>
 

web/src/components/ChatSetting.vue

@@ -60,9 +60,7 @@ const props = defineProps({
   show: Boolean,
 })
 
-const showDialog = computed(() => {
-  return props.show
-})
+const showDialog = ref(props.show)
 const emits = defineEmits(['hide'])
 const close = function () {
   emits('hide', false)

web/src/views/ChatPlus.vue

@@ -845,7 +845,7 @@ const sendSSERequest = async (message) => {
 }
 
 // 发送消息
-const sendMessage = () => {
+const sendMessage = (messageId) => {
   if (!isLogin.value) {
     console.log('未登录')
     store.setShowLoginDialog(true)
@@ -873,6 +873,7 @@ const sendMessage = () => {
     },
     model: getModelValue(modelID.value),
     created_at: new Date().getTime() / 1000,
+    message_id: messageId,
   })
 
   // 添加空回复消息
@@ -1185,33 +1186,20 @@ const stopGenerate = function () {
 }
 
 // 重新生成
-const reGenerate = function () {
+const reGenerate = function (messageId) {
   // 恢复发送按钮状态
   canSend.value = true
   showStopGenerate.value = false
+  console.log(messageId)
 
-  // 查找最后的用户消息和AI回复并删除
-  if (chatData.value.length >= 2) {
-    // 从后往前找，如果最后一条是AI回复，再往前一条是用户消息
-    if (chatData.value[chatData.value.length - 1].type === 'reply') {
-      // 删除AI回复
-      chatData.value.pop()
-
-      // 如果此时最后一条是用户消息，也删除它
-      if (
-        chatData.value.length > 0 &&
-        chatData.value[chatData.value.length - 1].type === 'prompt'
-      ) {
-        // 保存用户消息内容，填入输入框
-        const userPrompt = chatData.value[chatData.value.length - 1].content
-        // 删除用户消息
-        chatData.value.pop()
-        // 填入输入框
-        prompt.value = userPrompt
-      }
-    }
-  }
-
+  chatData.value = chatData.value.filter((item) => item.id < messageId)
+  // 保存用户消息内容，填入输入框
+  const userPrompt = chatData.value[chatData.value.length - 1].content.text
+  // 删除用户消息
+  chatData.value.pop()
+  // 填入输入框
+  prompt.value = userPrompt
+  sendMessage(messageId)
   // 将光标定位到输入框并聚焦
   nextTick(() => {
     if (inputRef.value) {