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 81

Merge e020fab1e6 into 82ca10cc2d

Browse source
This commit is contained in:
Luke Schaefer 2026-10-17 08:01:55 -07:00 • committed by GitHub
commit 7240af8ecb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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)