Emil_Shanaty/open-webui-ai4me
1
0
Fork 0
Code Issues Pull Requests Actions 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
backend/open_webui/retrieval/utils.py
+11 -1
View File
@@ -952,6 +952,7 @@ class RerankCompressor(BaseDocumentCompressor):
) -> Sequence[Document]: ) -> Sequence[Document]:
reranking = self.reranking_function is not None reranking = self.reranking_function is not None
scores = None
if reranking: if reranking:
scores = self.reranking_function( scores = self.reranking_function(
[(query, doc.page_content) for doc in documents] [(query, doc.page_content) for doc in documents]
@@ -965,8 +966,12 @@ class RerankCompressor(BaseDocumentCompressor):
) )
scores = util.cos_sim(query_embedding, document_embedding)[0] scores = util.cos_sim(query_embedding, document_embedding)[0]
if scores:
docs_with_scores = list( 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: if self.r_score:
docs_with_scores = [ docs_with_scores = [
@@ -984,3 +989,8 @@ class RerankCompressor(BaseDocumentCompressor):
) )
final_results.append(doc) final_results.append(doc)
return final_results return final_results
else:
log.warning(
"No valid scores found, check your reranking function. Returning original documents."
)
return documents