This commit is contained in:
Timothy Jaeryang Baek
2025-07-14 14:05:06 +04:00
parent 0013f5c1fc
commit 18bd83413b
3 changed files with 16 additions and 4 deletions

View File

@@ -446,6 +446,8 @@ def get_embedding_function(
def get_reranking_function(reranking_engine, reranking_model, reranking_function):
if reranking_function is None:
return None
if reranking_engine == "external":
return lambda sentences, user=None: reranking_function.predict(
sentences, user=user