AI/open-webui
1
0
Fork 1
mirror of https://git.mirrors.martin98.com/https://github.com/open-webui/open-webui synced 2025-08-15 14:15:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: external reranker

Browse Source
This commit is contained in:
Timothy Jaeryang Baek 2025-05-10 19:33:34 +04:00
parent 32b27749ef
commit 6b5f99bf66
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
backend/open_webui/retrieval/utils.py
View File

@ -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