We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent fd40bc6 commit 8a880ccCopy full SHA for 8a880cc
1 file changed
litellm/utils.py
@@ -6489,7 +6489,10 @@ def get_provider_rerank_config(
6489
api_base: Optional[str],
6490
present_version_params: List[str],
6491
) -> BaseRerankConfig:
6492
- if litellm.LlmProviders.COHERE == provider:
+ if (
6493
+ litellm.LlmProviders.COHERE == provider
6494
+ or litellm.LlmProviders.COHERE_CHAT == provider
6495
+ ):
6496
if should_use_cohere_v1_client(api_base, present_version_params):
6497
return litellm.CohereRerankConfig()
6498
else:
0 commit comments