tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-09 02:24:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix cohere rerank provider (#10822)

Browse Source
This commit is contained in:
Bryan Low
2025-05-18 20:52:47 -07:00
committed by GitHub
parent fd40bc6ddd
commit 8a880ccd2f
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/utils.py
+4 -1
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: