Browse Source

Revert the ordering change with chromadb, not necessary with reranker results

Marko Henning 4 months ago
parent
commit
5f48af5b91
1 changed files with 1 additions and 6 deletions
  1. 1 6
      backend/open_webui/retrieval/utils.py

+ 1 - 6
backend/open_webui/retrieval/utils.py

@@ -146,12 +146,7 @@ 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)
+            sorted_items = sorted(zip(distances, metadatas, documents), key=lambda x: x[0], reverse=True)
             sorted_items = sorted_items[:k]
             distances, documents, metadatas = map(list, zip(*sorted_items))
         result = {