From da75d0ca1e3afe977bcd8f556ef00812073813e4 Mon Sep 17 00:00:00 2001 From: Timothy Jaeryang Baek Date: Sat, 24 May 2025 02:13:54 +0400 Subject: [PATCH] chore: format --- backend/open_webui/retrieval/utils.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/backend/open_webui/retrieval/utils.py b/backend/open_webui/retrieval/utils.py index b7b4912ac..97a89880c 100644 --- a/backend/open_webui/retrieval/utils.py +++ b/backend/open_webui/retrieval/utils.py @@ -134,16 +134,16 @@ def query_doc_with_hybrid_search( if hybrid_bm25_weight <= 0: ensemble_retriever = EnsembleRetriever( - retrievers=[vector_search_retriever], weights=[1.] + retrievers=[vector_search_retriever], weights=[1.0] ) elif hybrid_bm25_weight >= 1: ensemble_retriever = EnsembleRetriever( - retrievers=[bm25_retriever], weights=[1.] + retrievers=[bm25_retriever], weights=[1.0] ) else: ensemble_retriever = EnsembleRetriever( 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(