Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 03:47:49 +01:00
Code Issues Packages Projects Releases Wiki Activity

chore: format

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-03-31 17:59:21 -07:00
parent 0bc5441d72
commit 391dd33da3
51 changed files with 513 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
backend/open_webui/retrieval/utils.py
View File

@@ -299,7 +299,10 @@ def query_collection_with_hybrid_search(
log.exception(f"Failed to fetch collection {collection_name}: {e}")
collection_results[collection_name] = None
log.info(f"Starting hybrid search for {len(queries)} queries in {len(collection_names)} collections...")
log.info(
f"Starting hybrid search for {len(queries)} queries in {len(collection_names)} collections..."
)
def process_query(collection_name, query):
try:
result = query_doc_with_hybrid_search(
@@ -317,7 +320,11 @@ def query_collection_with_hybrid_search(
log.exception(f"Error when querying the collection with hybrid_search: {e}")
return None, e
tasks = [(collection_name, query) for collection_name in collection_names for query in queries]
tasks = [
(collection_name, query)
for collection_name in collection_names
for query in queries
]
with ThreadPoolExecutor() as executor:
future_results = [executor.submit(process_query, cn, q) for cn, q in tasks]
@@ -330,8 +337,10 @@ def query_collection_with_hybrid_search(
results.append(result)
if error and not results:
raise Exception("Hybrid search failed for all collections. Using Non-hybrid search as fallback.")
raise Exception(
"Hybrid search failed for all collections. Using Non-hybrid search as fallback."
)
return merge_and_sort_query_results(results, k=k)