Add new Cohere models (#834)

* Add new Cohere models

Added `command-r` and `command-r-plus` models

* [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci

---------

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

Author: srdas

packages/jupyter-ai-magics/jupyter_ai_magics/providers.py

@@ -549,8 +549,15 @@ 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")