e9b74651c2
- Modify saveMessagesImmediately to accept messages parameter - Pass updated message list to all save calls instead of relying on closure - Ensure complete message history is saved including the last message - Fix timing issue where old message state was being saved This fixes the issue where the last conversation was not being persisted to localStorage.
102 lines
3.6 KiB
JavaScript
102 lines
3.6 KiB
JavaScript
import { useCallback, useState } from 'react';
|
||
import { Toast, Modal } from '@douyinfe/semi-ui';
|
||
import { useTranslation } from 'react-i18next';
|
||
import { getTextContent, buildApiPayload, createLoadingAssistantMessage } from '../utils/messageUtils';
|
||
import { MESSAGE_ROLES } from '../utils/constants';
|
||
|
||
export const useMessageEdit = (
|
||
setMessage,
|
||
inputs,
|
||
parameterEnabled,
|
||
sendRequest,
|
||
saveMessages
|
||
) => {
|
||
const { t } = useTranslation();
|
||
const [editingMessageId, setEditingMessageId] = useState(null);
|
||
const [editValue, setEditValue] = useState('');
|
||
|
||
const handleMessageEdit = useCallback((targetMessage) => {
|
||
const editableContent = getTextContent(targetMessage);
|
||
setEditingMessageId(targetMessage.id);
|
||
setEditValue(editableContent);
|
||
}, []);
|
||
|
||
const handleEditSave = useCallback(() => {
|
||
if (!editingMessageId || !editValue.trim()) return;
|
||
|
||
setMessage(prevMessages => {
|
||
const messageIndex = prevMessages.findIndex(msg => msg.id === editingMessageId);
|
||
if (messageIndex === -1) return prevMessages;
|
||
|
||
const targetMessage = prevMessages[messageIndex];
|
||
let newContent;
|
||
|
||
if (Array.isArray(targetMessage.content)) {
|
||
newContent = targetMessage.content.map(item =>
|
||
item.type === 'text' ? { ...item, text: editValue.trim() } : item
|
||
);
|
||
} else {
|
||
newContent = editValue.trim();
|
||
}
|
||
|
||
const updatedMessages = prevMessages.map(msg =>
|
||
msg.id === editingMessageId ? { ...msg, content: newContent } : msg
|
||
);
|
||
|
||
// 处理用户消息编辑后的重新生成
|
||
if (targetMessage.role === MESSAGE_ROLES.USER) {
|
||
const hasSubsequentAssistantReply = messageIndex < prevMessages.length - 1 &&
|
||
prevMessages[messageIndex + 1].role === MESSAGE_ROLES.ASSISTANT;
|
||
|
||
if (hasSubsequentAssistantReply) {
|
||
Modal.confirm({
|
||
title: t('消息已编辑'),
|
||
content: t('检测到该消息后有AI回复，是否删除后续回复并重新生成？'),
|
||
okText: t('重新生成'),
|
||
cancelText: t('仅保存'),
|
||
onOk: () => {
|
||
const messagesUntilUser = updatedMessages.slice(0, messageIndex + 1);
|
||
setMessage(messagesUntilUser);
|
||
// 编辑后保存（重新生成的情况），传入更新后的消息列表
|
||
setTimeout(() => saveMessages(messagesUntilUser), 0);
|
||
|
||
setTimeout(() => {
|
||
const payload = buildApiPayload(messagesUntilUser, null, inputs, parameterEnabled);
|
||
setMessage(prevMsg => [...prevMsg, createLoadingAssistantMessage()]);
|
||
sendRequest(payload, inputs.stream);
|
||
}, 100);
|
||
},
|
||
onCancel: () => {
|
||
setMessage(updatedMessages);
|
||
// 编辑后保存（仅保存的情况），传入更新后的消息列表
|
||
setTimeout(() => saveMessages(updatedMessages), 0);
|
||
}
|
||
});
|
||
return prevMessages;
|
||
}
|
||
}
|
||
|
||
// 编辑后保存（普通情况），传入更新后的消息列表
|
||
setTimeout(() => saveMessages(updatedMessages), 0);
|
||
return updatedMessages;
|
||
});
|
||
|
||
setEditingMessageId(null);
|
||
setEditValue('');
|
||
Toast.success({ content: t('消息已更新'), duration: 2 });
|
||
}, [editingMessageId, editValue, t, inputs, parameterEnabled, sendRequest, setMessage, saveMessages]);
|
||
|
||
const handleEditCancel = useCallback(() => {
|
||
setEditingMessageId(null);
|
||
setEditValue('');
|
||
}, []);
|
||
|
||
return {
|
||
editingMessageId,
|
||
editValue,
|
||
setEditValue,
|
||
handleMessageEdit,
|
||
handleEditSave,
|
||
handleEditCancel
|
||
};
|
||
};
|