|
@@ -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
|
|
# retrieve only min(k, k_reranker) items, sort and cut by distance if k < k_reranker
|
|
if k < k_reranker:
|
|
if k < k_reranker:
|
|
if VECTOR_DB == "chroma":
|
|
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)
|
|
sorted_items = sorted(zip(distances, metadatas, documents), key=lambda x: x[0], reverse=False)
|
|
else:
|
|
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)
|