Bläddra i källkod

fix: reranking

Timothy Jaeryang Baek 1 månad sedan
förälder
incheckning
fbff4e19de
1 ändrade filer med 1 tillägg och 1 borttagningar
  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,