feat: enhance message handling in NewChatPage by updating message IDs upon saving, ensuring consistent state management and improved user experience

This commit is contained in:
Anish Sarkar 2026-03-07 04:29:52 +05:30
parent 232618f995
commit f431f43191

View file

@ -557,7 +557,17 @@ export default function NewChatPage() {
role: "user",
content: persistContent,
})
.then(() => {
.then((savedMessage) => {
const newUserMsgId = `msg-${savedMessage.id}`;
setMessages((prev) =>
prev.map((m) => (m.id === userMsgId ? { ...m, id: newUserMsgId } : m))
);
setMessageDocumentsMap((prev) => {
const docs = prev[userMsgId];
if (!docs) return prev;
const { [userMsgId]: _, ...rest } = prev;
return { ...rest, [newUserMsgId]: docs };
});
if (isNewThread) {
queryClient.invalidateQueries({ queryKey: ["threads", String(searchSpaceId)] });
}