feat(chat-window): update message handling
This commit is contained in:
parent
b5ee8386e7
commit
8ecf3b4e99
1 changed files with 23 additions and 25 deletions
|
|
@ -363,20 +363,18 @@ const ChatWindow = ({ id }: { id?: string }) => {
|
||||||
|
|
||||||
if (data.type === 'sources') {
|
if (data.type === 'sources') {
|
||||||
sources = data.data;
|
sources = data.data;
|
||||||
if (!added) {
|
setMessages((prevMessages) => [
|
||||||
setMessages((prevMessages) => [
|
...prevMessages,
|
||||||
...prevMessages,
|
{
|
||||||
{
|
content: '',
|
||||||
content: '',
|
messageId: data.messageId,
|
||||||
messageId: data.messageId,
|
chatId: chatId!,
|
||||||
chatId: chatId!,
|
role: 'assistant',
|
||||||
role: 'assistant',
|
sources: sources,
|
||||||
sources: sources,
|
createdAt: new Date(),
|
||||||
createdAt: new Date(),
|
},
|
||||||
},
|
]);
|
||||||
]);
|
added = true;
|
||||||
added = true;
|
|
||||||
}
|
|
||||||
setMessageAppeared(true);
|
setMessageAppeared(true);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -394,20 +392,20 @@ const ChatWindow = ({ id }: { id?: string }) => {
|
||||||
},
|
},
|
||||||
]);
|
]);
|
||||||
added = true;
|
added = true;
|
||||||
|
setMessageAppeared(true);
|
||||||
|
} else {
|
||||||
|
setMessages((prev) =>
|
||||||
|
prev.map((message) => {
|
||||||
|
if (message.messageId === data.messageId) {
|
||||||
|
return { ...message, content: message.content + data.data };
|
||||||
|
}
|
||||||
|
|
||||||
|
return message;
|
||||||
|
}),
|
||||||
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
setMessages((prev) =>
|
|
||||||
prev.map((message) => {
|
|
||||||
if (message.messageId === data.messageId) {
|
|
||||||
return { ...message, content: message.content + data.data };
|
|
||||||
}
|
|
||||||
|
|
||||||
return message;
|
|
||||||
}),
|
|
||||||
);
|
|
||||||
|
|
||||||
recievedMessage += data.data;
|
recievedMessage += data.data;
|
||||||
setMessageAppeared(true);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
if (data.type === 'messageEnd') {
|
if (data.type === 'messageEnd') {
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue