OpenWebUI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-06-10 00:17:52 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

enh: provide source name in context if available

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-24 00:52:09 +04:00
parent 8b5e89eada
commit 80eaa24ca8
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -919,6 +919,7 @@ async def process_chat_payload(request, form_data, user, metadata, model):
for doc_context, doc_meta in zip( for doc_context, doc_meta in zip(
source["document"], source["metadata"] source["document"], source["metadata"]
): ):
source_name = source.get("source", {}).get("name", None)
citation_id = ( citation_id = (
doc_meta.get("source", None) doc_meta.get("source", None)
or source.get("source", {}).get("id", None) or source.get("source", {}).get("id", None)
@ -926,7 +927,11 @@ async def process_chat_payload(request, form_data, user, metadata, model):
) )
if citation_id not in citation_idx: if citation_id not in citation_idx:
citation_idx[citation_id] = len(citation_idx) + 1 citation_idx[citation_id] = len(citation_idx) + 1
context_string += f'<source id="{citation_idx[citation_id]}">{doc_context}</source>\n' context_string += (
f'<source id="{citation_idx[citation_id]}"'
+ (f' name="{source_name}"' if source_name else "")
+ f">{doc_context}</source>\n"
)
context_string = context_string.strip() context_string = context_string.strip()
prompt = get_last_user_message(form_data["messages"]) prompt = get_last_user_message(form_data["messages"])