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) => { 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 = ({ 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) },