diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py index c6e9835e9..8a04efd47 100644 --- a/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py +++ b/packages/jupyter-ai-magics/jupyter_ai_magics/providers.py @@ -550,7 +550,14 @@ class CohereProvider(BaseProvider, Cohere): id = "cohere" name = "Cohere" # 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"] + 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")