UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2024-11-24 04:54:12 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

refac: file handling

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2024-11-22 10:35:59 -08:00
parent 9aac02824d
commit 35d75e733d
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
backend/open_webui/main.py
View File

@ -698,12 +698,13 @@ class ChatCompletionMiddleware(BaseHTTPMiddleware):
if "document" in source: if "document" in source:
for doc_idx, doc_context in enumerate(source["document"]): for doc_idx, doc_context in enumerate(source["document"]):
metadata = source.get("metadata") metadata = source.get("metadata")
doc_source_id = None
if metadata: if metadata:
doc_source_id = metadata[doc_idx].get("source", source_id) doc_source_id = metadata[doc_idx].get("source", source_id)
if source_id: if source_id:
context_string += f"<source><source_id>{doc_source_id}</source_id><source_context>{doc_context}</source_context></source>\n" context_string += f"<source><source_id>{doc_source_id if doc_source_id is not None else source_id}</source_id><source_context>{doc_context}</source_context></source>\n"
else: else:
# If there is no source_id, then do not include the source_id tag # If there is no source_id, then do not include the source_id tag
context_string += f"<source><source_context>{doc_context}</source_context></source>\n" context_string += f"<source><source_context>{doc_context}</source_context></source>\n"