Merge pull request #13751 from open-webui/dev

fix: external reranker
2025-05-10 19:33:59 +04:00
parent ef301aa16b 6b5f99bf66
commit 4191ec692e
1 changed files with 3 additions and 1 deletions
--- a/backend/open_webui/retrieval/utils.py
+++ b/backend/open_webui/retrieval/utils.py
@@ -818,7 +818,9 @@ class RerankCompressor(BaseDocumentCompressor):
            )
            scores = util.cos_sim(query_embedding, document_embedding)[0]

-        docs_with_scores = list(zip(documents, scores.tolist()))
+        docs_with_scores = list(
+            zip(documents, scores.tolist() if not isinstance(scores, list) else scores)
+        )
        if self.r_score:
            docs_with_scores = [
                (d, s) for d, s in docs_with_scores if s >= self.r_score