Timothy Jaeryang Baek преди 4 месеца
родител
ревизия
da75d0ca1e
променени са 1 файла, в които са добавени 3 реда и са изтрити 3 реда
  1. 3 3
      backend/open_webui/retrieval/utils.py

+ 3 - 3
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(