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

refac

Browse source
This commit is contained in:
Timothy J. Baek 2024-03-06 00:33:47 -08:00
parent 03907f9a8a
commit a9d74b669e
1 changed files with 15 additions and 33 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -224,42 +224,24 @@
};
const messageDeleteHandler = async (messageId) => {
const messageToDelete = history.messages[messageId];
const messageParentId = messageToDelete.parentId;
const messageChildrenIds = messageToDelete.childrenIds ?? [];
const message = history.messages[messageId];
const parentId = message.parentId;
const childrenIds = message.childrenIds ?? [];
const grandchildrenIds = [];
const hasSibling = messageChildrenIds.some(childId => history.messages[childId]?.childrenIds?.length > 0);
// Iterate through childrenIds to find grandchildrenIds
for (const childId of childrenIds) {
const childMessage = history.messages[childId];
const grandChildrenIds = childMessage.childrenIds ?? [];
grandchildrenIds.push(...grandChildrenIds);
}
messageChildrenIds.forEach((childId) => {
const child = history.messages[childId];
if (child && child.childrenIds) {
if (child.childrenIds.length === 0 && !hasSibling) { // if last prompt/response pair
history.messages[messageParentId].childrenIds = []
history.currentId = messageParentId;
}
else {
child.childrenIds.forEach((grandChildId) => {
if (history.messages[grandChildId]) {
history.messages[grandChildId].parentId = messageParentId;
history.messages[messageParentId].childrenIds.push(grandChildId);
}
});
}
}
history.messages[parentId].childrenIds.push(...grandchildrenIds);
history.messages[parentId].childrenIds = history.messages[parentId].childrenIds.filter(
(id) => id !== messageId
);
// remove response
history.messages[messageParentId].childrenIds = history.messages[messageParentId].childrenIds
.filter((id) => id !== childId);
});
// remove prompt
history.messages[messageParentId].childrenIds = history.messages[messageParentId].childrenIds
.filter((id) => id !== messageId);
await updateChatById(localStorage.token, chatId, {
messages: messages,
history: history
});
await updateChatById(localStorage.token, chatId, { messages, history });
};
</script>