tdpeuter/open-webui
tdpeuter/open-webui
1
Fork 0
forked from open-webui/open-webui
Code Issues Pull requests 1 Projects 1 Releases 2 Packages 1 Activity Actions

Recursively set delete=true flag for all children messages

Browse source
This commit is contained in:
Danny Liu 2024-02-20 19:47:30 -08:00
parent 7c4d818f2d
commit c42ac39dc4
1 changed files with 10 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/lib/components/chat/Messages.svelte
View file

@ -223,15 +223,18 @@
}, 100);
};
const deleteMessagePair = async (messageId) => {
history.messages[messageId].deleted = true;
history.messages[history.messages[messageId].childrenIds[0]].deleted = true;
const deleteMessageAndDescendants = async (messageId: string) => {
if (history.messages[messageId]) {
history.messages[messageId].deleted = true;
const responseId = history.messages[messageId].childrenIds[0];
if (history.messages[responseId].childrenIds.length === 0) {
await cancelChatCompletion(localStorage.token, chatId);
for (const childId of history.messages[messageId].childrenIds) {
await deleteMessageAndDescendants(childId);
}
}
};
const triggerDeleteMessageRecursive = async (messageId: string) => {
await deleteMessageAndDescendants(messageId);
await updateChatById(localStorage.token, chatId, { history });
await chats.set(await getChatList(localStorage.token));
};
@ -252,7 +255,7 @@
>
{#if message.role === 'user'}
<UserMessage
on:delete={() => deleteMessagePair(message.id)}
on:delete={() => triggerDeleteMessageRecursive(message.id)}
user={$user}
{message}
isFirstMessage={messageIdx === 0}