scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-24 05:45:11 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 71

fix(huggingface): allow truncate / truncation_direction passthrough in rerank

Browse source
This commit is contained in:
Luke Schaefer 2026-10-09 21:57:49 +00:00
parent 231627ecd9
commit e020fab1e6
1 changed files with 6 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/huggingface/rerank/transformation.py
View file

@ -85,6 +85,8 @@ class HuggingFaceRerankConfig(BaseRerankConfig):
"documents",
"top_n",
"return_documents",
"truncate",
"truncation_direction",
]
def map_cohere_rerank_params(
@ -115,6 +117,10 @@ class HuggingFaceRerankConfig(BaseRerankConfig):
optional_rerank_params["texts"] = v
elif k == "query" and v is not None:
optional_rerank_params["query"] = v
elif k == "truncate" and v is not None and isinstance(v, bool):
optional_rerank_params["truncate"] = v
elif k == "truncation_direction" and v is not None:
optional_rerank_params["truncation_direction"] = v
return OptionalRerankParams(**optional_rerank_params)