UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-03-12 23:38:36 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

use data field

Browse Source
This commit is contained in:
Michael Poluektov 2024-07-04 12:21:09 +01:00
parent 4e433d9015
commit 0527755600
1 changed files with 13 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -133,29 +133,24 @@
let message = history.messages[data.message_id];
const type = data?.data?.type ?? null;
if (type === "status") {
const status = {
done: data?.data?.done ?? null,
description: data?.data?.status ?? 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", "metadata", "source"];
if (type === "status" && status_keys.every(key => key in payload)) {
if (message.statusHistory) {
message.statusHistory.push(status);
message.statusHistory.push(payload);
} else {
message.statusHistory = [status];
message.statusHistory = [payload];
}
} else if (type === "citation") {
console.log(data);
const citation = {
document: data?.data?.document ?? null,
metadata: data?.data?.metadata ?? null,
source: data?.data?.source ?? null
};
} else if (type === "citation" && citation_keys.every(key => key in payload)) {
if (message.citations) {
message.citations.push(citation);
message.citations.push(payload);
} else {
message.citations = [citation];
message.citations = [payload];
}
} else {
console.log("Unknown message type", data);