chore: format
This commit is contained in:
@@ -134,16 +134,16 @@ def query_doc_with_hybrid_search(
|
|||||||
|
|
||||||
if hybrid_bm25_weight <= 0:
|
if hybrid_bm25_weight <= 0:
|
||||||
ensemble_retriever = EnsembleRetriever(
|
ensemble_retriever = EnsembleRetriever(
|
||||||
retrievers=[vector_search_retriever], weights=[1.]
|
retrievers=[vector_search_retriever], weights=[1.0]
|
||||||
)
|
)
|
||||||
elif hybrid_bm25_weight >= 1:
|
elif hybrid_bm25_weight >= 1:
|
||||||
ensemble_retriever = EnsembleRetriever(
|
ensemble_retriever = EnsembleRetriever(
|
||||||
retrievers=[bm25_retriever], weights=[1.]
|
retrievers=[bm25_retriever], weights=[1.0]
|
||||||
)
|
)
|
||||||
else:
|
else:
|
||||||
ensemble_retriever = EnsembleRetriever(
|
ensemble_retriever = EnsembleRetriever(
|
||||||
retrievers=[bm25_retriever, vector_search_retriever],
|
retrievers=[bm25_retriever, vector_search_retriever],
|
||||||
weights=[hybrid_bm25_weight, 1. - hybrid_bm25_weight]
|
weights=[hybrid_bm25_weight, 1.0 - hybrid_bm25_weight],
|
||||||
)
|
)
|
||||||
|
|
||||||
compressor = RerankCompressor(
|
compressor = RerankCompressor(
|
||||||
|
|||||||
Reference in New Issue
Block a user