OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-05-22 22:04:27 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Merge pull request #12562 from almajo/fix/agrimm/correct_knowledge_base_citations

Browse Source 
fix: Correct Citations for knowledge base models
This commit is contained in:
Tim Jaeryang Baek 2025-04-10 08:48:47 -07:00 committed by GitHub
commit ef14d0b46d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
backend/open_webui/utils/middleware.py
View File

@ -897,11 +897,15 @@ async def process_chat_payload(request, form_data, user, metadata, model):
# If context is not empty, insert it into the messages
if len(sources) > 0:
context_string = ""
for source_idx, source in enumerate(sources):
citated_file_idx = {}
for _, source in enumerate(sources, 1):
if "document" in source:
for doc_idx, doc_context in enumerate(source["document"]):
for doc_context, doc_meta in zip(source["document"], source['metadata']):
file_id = doc_meta.get('file_id')
if file_id not in citated_file_idx:
citated_file_idx[file_id] = len(citated_file_idx) + 1
context_string += (
f'<source id="{source_idx + 1}">{doc_context}</source>\n'
f'<source id="{citated_file_idx[file_id]}">{doc_context}</source>\n'
)
context_string = context_string.strip()