Skip to content

Commit 8a880cc

Browse files
authored
fix cohere rerank provider (BerriAI#10822)
1 parent fd40bc6 commit 8a880cc

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

litellm/utils.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6489,7 +6489,10 @@ def get_provider_rerank_config(
64896489
api_base: Optional[str],
64906490
present_version_params: List[str],
64916491
) -> BaseRerankConfig:
6492-
if litellm.LlmProviders.COHERE == provider:
6492+
if (
6493+
litellm.LlmProviders.COHERE == provider
6494+
or litellm.LlmProviders.COHERE_CHAT == provider
6495+
):
64936496
if should_use_cohere_v1_client(api_base, present_version_params):
64946497
return litellm.CohereRerankConfig()
64956498
else:

0 commit comments

Comments
 (0)