scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

fix cohere rerank provider (#10822)

Browse source
This commit is contained in:
Bryan Low 2025-05-19 13:52:47 +10:00 • committed by GitHub
parent fd40bc6ddd
commit 8a880ccd2f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/utils.py
View file

@ -6489,7 +6489,10 @@ class ProviderConfigManager:
api_base: Optional[str],
present_version_params: List[str],
) -> BaseRerankConfig:
if litellm.LlmProviders.COHERE == provider:
if (
litellm.LlmProviders.COHERE == provider
or litellm.LlmProviders.COHERE_CHAT == provider
):
if should_use_cohere_v1_client(api_base, present_version_params):
return litellm.CohereRerankConfig()
else: