UniqAI/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui synced 2025-04-29 02:32:34 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

Add documentation on chroma special case

Browse Source
This commit is contained in:
Marko Henning 2025-03-18 16:44:58 +01:00
parent ba676b7ed6
commit 5ab789e83e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -147,6 +147,8 @@ def query_doc_with_hybrid_search(
# retrieve only min(k, k_reranker) items, sort and cut by distance if k < k_reranker
if k < k_reranker:
if VECTOR_DB == "chroma":
# Chroma uses unconventional cosine similarity, so we don't need to reverse the results
# https://docs.trychroma.com/docs/collections/configure#configuring-chroma-collections
sorted_items = sorted(zip(distances, metadatas, documents), key=lambda x: x[0], reverse=False)
else:
sorted_items = sorted(zip(distances, metadatas, documents), key=lambda x: x[0], reverse=True)