From 80eaa24ca8a30fac27fc95db560c4e72e64f53d1 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 24 May 2025 00:52:09 +0400 Subject: [PATCH] enh: provide source name in context if available --- backend/open_webui/utils/middleware.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/backend/open_webui/utils/middleware.py b/backend/open_webui/utils/middleware.py index 6b629c12d..564b7aab6 100644 --- a/backend/open_webui/utils/middleware.py +++ b/backend/open_webui/utils/middleware.py @@ -919,6 +919,7 @@ async def process_chat_payload(request, form_data, user, metadata, model): for doc_context, doc_meta in zip( source["document"], source["metadata"] ): + source_name = source.get("source", {}).get("name", None) citation_id = ( doc_meta.get("source", 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: citation_idx[citation_id] = len(citation_idx) + 1 - context_string += f'{doc_context}\n' + context_string += ( + f'{doc_context}\n" + ) context_string = context_string.strip() prompt = get_last_user_message(form_data["messages"])