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