UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-03-16 10:28:28 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

fix: uploaded files leaking to other user chats issue

Browse Source 
#4601
This commit is contained in:
Timothy J. Baek 2024-08-23 15:02:23 +02:00
parent 7b91be21b4
commit 0a5a2e67e8
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
backend/apps/rag/utils.py
View File

@ -149,6 +149,7 @@ def query_collection(
): ):
results = [] results = []
for collection_name in collection_names: for collection_name in collection_names:
if collection_name:
try: try:
result = query_doc( result = query_doc(
collection_name=collection_name, collection_name=collection_name,
@ -159,6 +160,9 @@ def query_collection(
results.append(result) results.append(result)
except Exception: except Exception:
pass pass
else:
pass
return merge_and_sort_query_results(results, k=k) return merge_and_sort_query_results(results, k=k)
@ -257,7 +261,7 @@ def get_rag_context(
collection_names = ( collection_names = (
file["collection_names"] file["collection_names"]
if file["type"] == "collection" if file["type"] == "collection"
else [file["collection_name"]] else [file["collection_name"]] if file["collection_name"] else []
) )
collection_names = set(collection_names).difference(extracted_collections) collection_names = set(collection_names).difference(extracted_collections)