From 13a1e866e6199d9d0a511d57a0abac75a58d06a1 Mon Sep 17 00:00:00 2001
From: kangfenmao <kangfenmao@qq.com>
Date: Thu, 2 Jan 2025 15:42:47 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E9=87=8D=E6=96=B0=E5=8F=91=E9=80=81?=
 =?UTF-8?q?=E6=8C=89=E9=92=AE=E6=97=A0=E5=8F=8D=E5=BA=94=20#587?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/renderer/src/pages/home/Messages/MessageMenubar.tsx | 9 +++++++--
 src/renderer/src/pages/home/Messages/Messages.tsx       | 1 +
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/renderer/src/pages/home/Messages/MessageMenubar.tsx b/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
index 2a7883ceff..03d1beabbb 100644
--- a/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
+++ b/src/renderer/src/pages/home/Messages/MessageMenubar.tsx
@@ -14,7 +14,7 @@ import TextEditPopup from '@renderer/components/Popups/TextEditPopup'
 import { EVENT_NAMES, EventEmitter } from '@renderer/services/EventService'
 import { translateText } from '@renderer/services/TranslateService'
 import { Message, Model } from '@renderer/types'
-import { removeTrailingDoubleSpaces } from '@renderer/utils'
+import { removeTrailingDoubleSpaces, uuid } from '@renderer/utils'
 import { Button, Dropdown, Popconfirm, Tooltip } from 'antd'
 import dayjs from 'dayjs'
 import { FC, useCallback, useMemo, useState } from 'react'
@@ -92,7 +92,12 @@ const MessageMenubar: FC<Props> = (props) => {
         translatedContent: undefined
       })
     }
-  }, [assistantModel?.id, message.id, model?.id, onGetMessages])
+
+    if (!nextMessage) {
+      onDeleteMessage?.(message)
+      EventEmitter.emit(EVENT_NAMES.SEND_MESSAGE, { ...message, id: uuid() })
+    }
+  }, [assistantModel?.id, message, model?.id, onDeleteMessage, onGetMessages])
 
   const onEdit = useCallback(async () => {
     let resendMessage = false
diff --git a/src/renderer/src/pages/home/Messages/Messages.tsx b/src/renderer/src/pages/home/Messages/Messages.tsx
index f9cbdfd83b..9d48516163 100644
--- a/src/renderer/src/pages/home/Messages/Messages.tsx
+++ b/src/renderer/src/pages/home/Messages/Messages.tsx
@@ -136,6 +136,7 @@ const Messages: FC<Props> = ({ assistant, topic, setActiveTopic }) => {
     (message: Message) => {
       const _messages = messages.filter((m) => m.id !== message.id)
       setMessages(_messages)
+      setDisplayMessages(_messages)
       db.topics.update(topic.id, { messages: _messages })
       deleteMessageFiles(message)
     },