避免生图占位时丢失用户消息

This commit is contained in:
Your Name 2026-02-05 20:15:12 +08:00
parent 4c24001c2f
commit 21a6280db8

View File

@ -36,6 +36,9 @@ export function useAutoResume({
}, [autoResume, initialMessages.at, resumeStream]);
useEffect(() => {
if (!autoResume) {
return;
}
if (!dataStream) {
return;
}
@ -43,11 +46,16 @@ export function useAutoResume({
return;
}
const dataPart = dataStream[0];
const dataPart = dataStream[dataStream.length - 1];
if (dataPart.type === "data-appendMessage") {
const message = JSON.parse(dataPart.data);
setMessages([...initialMessages, message]);
setMessages((prev) => {
if (prev.some((m) => m.id === message.id)) {
return prev;
}
return [...prev, message];
});
}
}, [dataStream, initialMessages, setMessages]);
}, [autoResume, dataStream, setMessages]);
}