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"])