OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-19 12:51:35 +00:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

simplify citation API

Browse Source
This commit is contained in:
Michael Poluektov 2024-07-04 18:50:09 +01:00
parent 0527755600
commit 55b7c30028
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -139,7 +139,7 @@
return; return;
} }
const status_keys = ["done", "description"]; const status_keys = ["done", "description"];
const citation_keys = ["document", "metadata", "source"]; const citation_keys = ["document", "url", "title"];
if (type === "status" && status_keys.every(key => key in payload)) { if (type === "status" && status_keys.every(key => key in payload)) {
if (message.statusHistory) { if (message.statusHistory) {
message.statusHistory.push(payload); message.statusHistory.push(payload);
@ -147,10 +147,15 @@
message.statusHistory = [payload]; message.statusHistory = [payload];
} }
} else if (type === "citation" && citation_keys.every(key => key in payload)) { } else if (type === "citation" && citation_keys.every(key => key in payload)) {
const citation = {
document: [payload.document],
metadata: [{source: payload.url}],
source: {name: payload.title}
};
if (message.citations) { if (message.citations) {
message.citations.push(payload); message.citations.push(citation);
} else { } else {
message.citations = [payload]; message.citations = [citation];
} }
} else { } else {
console.log("Unknown message type", data); console.log("Unknown message type", data);