diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index e24f53e56..c6e9835e9 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -549,8 +549,8 @@ def is_api_key_exc(cls, e: Exception): class CohereProvider(BaseProvider, Cohere): id = "cohere" name = "Cohere" - # Source: https://docs.cohere.com/reference/generate - models = ["command", "command-nightly", "command-light", "command-light-nightly"] + # Source: https://docs.cohere.com/reference/generate; https://docs.cohere.com/docs/models + models = ["command", "command-nightly", "command-light", "command-light-nightly", "command-r-plus", "command-r"] model_id_key = "model" pypi_package_deps = ["cohere"] auth_strategy = EnvAuthStrategy(name="COHERE_API_KEY")