Jelajahi Sumber

fix: reranking

Timothy Jaeryang Baek 1 bulan lalu
induk
melakukan
fbff4e19de
1 mengubah file dengan 1 tambahan dan 1 penghapusan
  1. 1 1
      backend/open_webui/retrieval/utils.py

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

@@ -966,7 +966,7 @@ class RerankCompressor(BaseDocumentCompressor):
             )
             scores = util.cos_sim(query_embedding, document_embedding)[0]
 
-        if scores:
+        if scores is not None:
             docs_with_scores = list(
                 zip(
                     documents,