From 8a880ccd2f07bd696a9ebc03abae83d3287b08b1 Mon Sep 17 00:00:00 2001 From: Bryan Low Date: Mon, 19 May 2025 13:52:47 +1000 Subject: [PATCH] fix cohere rerank provider (#10822) --- litellm/utils.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index be5a526c49..d2cf36f24c 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -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: