UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-01-18 00:30:51 +00:00
Code Issues Actions 5 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;
}
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 (message.statusHistory) {
message.statusHistory.push(payload);
@ -147,10 +147,15 @@
message.statusHistory = [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) {
message.citations.push(payload);
message.citations.push(citation);
} else {
message.citations = [payload];
message.citations = [citation];
}
} else {
console.log("Unknown message type", data);