UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-08 01:30:40 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

feat: use url as source name for citations

Browse Source
This commit is contained in:
Jun Siang Cheah 2024-05-11 23:48:30 +08:00
parent 77928ae141
commit 7538dc051e
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/lib/components/chat/Messages/ResponseMessage.svelte
View File

@ -453,6 +453,11 @@
citation.document.forEach((document, index) => { citation.document.forEach((document, index) => {
const metadata = citation.metadata?.[index]; const metadata = citation.metadata?.[index];
const id = metadata?.source ?? 'N/A'; const id = metadata?.source ?? 'N/A';
let source = citation?.source;
// Check if ID looks like a URL
if (id.startsWith('http://') || id.startsWith('https://')) {
source = { name: id };
}
const existingSource = acc.find((item) => item.id === id); const existingSource = acc.find((item) => item.id === id);
@ -460,7 +465,7 @@
existingSource.document.push(document); existingSource.document.push(document);
existingSource.metadata.push(metadata); existingSource.metadata.push(metadata);
} else { } else {
acc.push( { id: id, source: citation?.source, document: [document], metadata: metadata ? [metadata] : [] } ); acc.push( { id: id, source: source, document: [document], metadata: metadata ? [metadata] : [] } );
} }
}); });
return acc; return acc;