ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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: