Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #13751 from open-webui/dev

Browse Source 
fix: external reranker
This commit is contained in:
Tim Jaeryang Baek
2025-05-10 19:33:59 +04:00
committed by GitHub
parent ef301aa16b 6b5f99bf66
commit 4191ec692e
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/open_webui/retrieval/utils.py
+3 -1
View File
@@ -818,7 +818,9 @@ class RerankCompressor(BaseDocumentCompressor):
) )
scores = util.cos_sim(query_embedding, document_embedding)[0] 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: if self.r_score:
docs_with_scores = [ docs_with_scores = [
(d, s) for d, s in docs_with_scores if s >= self.r_score (d, s) for d, s in docs_with_scores if s >= self.r_score