Mirrors/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-16 11:57:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

refac

Browse Source
This commit is contained in:
Timothy Jaeryang Baek
2025-08-21 21:48:21 +04:00
parent 5a66f69460
commit 60b8cfb9fa
1 changed files with 28 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -952,6 +952,7 @@ class RerankCompressor(BaseDocumentCompressor):
) -> Sequence[Document]:
reranking = self.reranking_function is not None
scores = None
if reranking:
scores = self.reranking_function(
[(query, doc.page_content) for doc in documents]
@@ -965,8 +966,12 @@ class RerankCompressor(BaseDocumentCompressor):
)
scores = util.cos_sim(query_embedding, document_embedding)[0]
if scores:
docs_with_scores = list(
zip(documents, scores.tolist() if not isinstance(scores, list) else scores)
zip(
documents,
scores.tolist() if not isinstance(scores, list) else scores,
)
)
if self.r_score:
docs_with_scores = [
@@ -984,3 +989,8 @@ class RerankCompressor(BaseDocumentCompressor):
)
final_results.append(doc)
return final_results
else:
log.warning(
"No valid scores found, check your reranking function. Returning original documents."
)
return documents