ti.sil/open-webui
1
0
Fork 0
mirror of https://github.com/open-webui/open-webui.git synced 2025-12-12 04:15:25 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

refac: rerank

Browse source
This commit is contained in:
Timothy Jaeryang Baek 2025-11-09 21:33:50 -05:00
parent e76f77bcb7
commit bc739de024
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
backend/open_webui/retrieval/utils.py
View file

@ -168,7 +168,7 @@ def query_doc_with_hybrid_search(
): ):
log.warning(f"query_doc_with_hybrid_search:no_docs {collection_name}") log.warning(f"query_doc_with_hybrid_search:no_docs {collection_name}")
return {"documents": [], "metadatas": [], "distances": []} return {"documents": [], "metadatas": [], "distances": []}
# Now safely check the documents content after confirming attributes exist # Now safely check the documents content after confirming attributes exist
if ( if (
not collection_result.documents not collection_result.documents
@ -516,11 +516,13 @@ def get_reranking_function(reranking_engine, reranking_model, reranking_function
if reranking_function is None: if reranking_function is None:
return None return None
if reranking_engine == "external": if reranking_engine == "external":
return lambda sentences, user=None: reranking_function.predict( return lambda query, documents, user=None: reranking_function.predict(
sentences, user=user [(query, doc.page_content) for doc in documents], user=user
) )
else: else:
return lambda sentences, user=None: reranking_function.predict(sentences) return lambda query, documents, user=None: reranking_function.predict(
[(query, doc.page_content) for doc in documents]
)
def get_sources_from_items( def get_sources_from_items(
@ -1064,9 +1066,7 @@ class RerankCompressor(BaseDocumentCompressor):
scores = None scores = None
if reranking: if reranking:
scores = self.reranking_function( scores = self.reranking_function(query, documents)
[(query, doc.page_content) for doc in documents]
)
else: else:
from sentence_transformers import util from sentence_transformers import util