From e50e76bbd5179ead90906ed96bee6e08398b4f8c Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Sebasti=C3=A1n=20Est=C3=A9vez?= Date: Mon, 1 Apr 2024 13:24:03 -0400 Subject: [PATCH] support cohere_chat in get_api_key --- litellm/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/utils.py b/litellm/utils.py index 600d80599..ae775c3a6 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5700,7 +5700,7 @@ def get_api_key(llm_provider: str, dynamic_api_key: Optional[str]): elif llm_provider == "baseten": api_key = api_key or litellm.baseten_key or get_secret("BASETEN_API_KEY") # cohere - elif llm_provider == "cohere": + elif llm_provider == "cohere" or llm_provider == "cohere_chat": api_key = api_key or litellm.cohere_key or get_secret("COHERE_API_KEY") # huggingface elif llm_provider == "huggingface":