Ver Fonte

fix: reranking

Timothy Jaeryang Baek há 1 mês atrás
pai
commit
fbff4e19de
1 ficheiros alterados com 1 adições e 1 exclusões
  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,