github-starred/open-webui
2
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2026-04-29 12:09:30 -05:00
Code Issues 786 Packages Projects Releases 126 Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2026-03-07 20:42:21 -06:00
parent e24299e66d
commit 2bff50f736
1 changed files with 14 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -890,21 +890,24 @@ def handle_responses_streaming_event(
return current_output, None return current_output, None
def get_source_context(sources: list, include_content: bool = True) -> str: def get_source_context(
sources: list, source_id_map: dict = None, include_content: bool = True
) -> str:
""" """
Build <source> tag context string from citation sources. Build <source> tag context string from citation sources.
""" """
context_string = "" context_string = ""
citation_idx = {} if source_id_map is None:
source_id_map = {}
for source in sources: for source in sources:
for doc, meta in zip(source.get("document", []), source.get("metadata", [])): for doc, meta in zip(source.get("document", []), source.get("metadata", [])):
src_id = meta.get("source") or source.get("source", {}).get("id") or "N/A" src_id = meta.get("source") or source.get("source", {}).get("id") or "N/A"
if src_id not in citation_idx: if src_id not in source_id_map:
citation_idx[src_id] = len(citation_idx) + 1 source_id_map[src_id] = len(source_id_map) + 1
src_name = source.get("source", {}).get("name") src_name = source.get("source", {}).get("name")
body = doc if include_content else "" body = doc if include_content else ""
context_string += ( context_string += (
f'<source id="{citation_idx[src_id]}"' f'<source id="{source_id_map[src_id]}"'
+ (f' name="{src_name}"' if src_name else "") + (f' name="{src_name}"' if src_name else "")
+ f">{body}</source>\n" + f">{body}</source>\n"
) )
@@ -929,7 +932,7 @@ def apply_source_context_to_messages(
if not sources or not user_message: if not sources or not user_message:
return messages return messages
context = get_source_context(sources, include_content) context = get_source_context(sources, include_content=include_content)
context = context.strip() context = context.strip()
if not context: if not context:
@@ -4491,10 +4494,13 @@ async def streaming_chat_response_handler(response, ctx):
# Build context: file sources with content, # Build context: file sources with content,
# tool sources as citation markers only. # tool sources as citation markers only.
source_id_map = {}
source_context = get_source_context( source_context = get_source_context(
metadata.get("sources", []) metadata.get("sources", []), source_id_map
) + get_source_context( ) + get_source_context(
all_tool_call_sources, include_content=False all_tool_call_sources,
source_id_map,
include_content=False,
) )
source_context = source_context.strip() source_context = source_context.strip()
if source_context: if source_context: