UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-07 09:09:53 +00:00
Code Issues Actions 5 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) => {
if (data.chat_id === $chatId) {
const chatEventHandler = async (event) => {
if (event.chat_id === $chatId) {
await tick();
console.log(data);
let message = history.messages[data.message_id];
console.log(event);
let message = history.messages[event.message_id];
const type = data?.data?.type ?? null;
const payload = data?.data?.data ?? null;
if (!type || !payload) {
console.log("Data and type fields must be provided.", data);
return;
}
const status_keys = ["done", "description"];
const citation_keys = ["document", "url", "title"];
if (type === "status" && status_keys.every(key => key in payload)) {
const type = event?.data?.type ?? null;
const data = event?.data?.data ?? null;
if (type === 'status') {
if (message.statusHistory) {
message.statusHistory.push(payload);
message.statusHistory.push(data);
} else {
message.statusHistory = [payload];
message.statusHistory = [data];
}
} else if (type === "citation" && citation_keys.every(key => key in payload)) {
const citation = {
document: [payload.document],
metadata: [{source: payload.url}],
source: {name: payload.title}
};
} else if (type === 'citation') {
if (message.citations) {
message.citations.push(citation);
message.citations.push(data);
} else {
message.citations = [citation];
message.citations = [data];
}
} else {
console.log("Unknown message type", data);
console.log('Unknown message type', data);
}
messages = messages;