UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-02-22 05:08:14 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy J. Baek 2024-07-05 21:20:27 -07:00
parent 55b7c30028
commit 983fe4b2bc
1 changed files with 14 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
src/lib/components/chat/Chat.svelte
View File

@ -126,39 +126,29 @@
})(); })();
} }
const chatEventHandler = async (data) => { const chatEventHandler = async (event) => {
if (data.chat_id === $chatId) { if (event.chat_id === $chatId) {
await tick(); await tick();
console.log(data); console.log(event);
let message = history.messages[data.message_id]; let message = history.messages[event.message_id];
const type = data?.data?.type ?? null; const type = event?.data?.type ?? null;
const payload = data?.data?.data ?? null; const data = event?.data?.data ?? null;
if (!type || !payload) {
console.log("Data and type fields must be provided.", data); if (type === 'status') {
return;
}
const status_keys = ["done", "description"];
const citation_keys = ["document", "url", "title"];
if (type === "status" && status_keys.every(key => key in payload)) {
if (message.statusHistory) { if (message.statusHistory) {
message.statusHistory.push(payload); message.statusHistory.push(data);
} else { } else {
message.statusHistory = [payload]; message.statusHistory = [data];
} }
} else if (type === "citation" && citation_keys.every(key => key in payload)) { } else if (type === 'citation') {
const citation = {
document: [payload.document],
metadata: [{source: payload.url}],
source: {name: payload.title}
};
if (message.citations) { if (message.citations) {
message.citations.push(citation); message.citations.push(data);
} else { } else {
message.citations = [citation]; message.citations = [data];
} }
} else { } else {
console.log("Unknown message type", data); console.log('Unknown message type', data);
} }
messages = messages; messages = messages;