Skip to content

Commit

Permalink
[pre-commit.ci] auto fixes from pre-commit.com hooks
Browse files Browse the repository at this point in the history
for more information, see https://pre-commit.ci
  • Loading branch information
pre-commit-ci[bot] committed Jun 14, 2024
1 parent 131a158 commit 6b06a4a
Showing 1 changed file with 8 additions and 1 deletion.
9 changes: 8 additions & 1 deletion packages/jupyter-ai-magics/jupyter_ai_magics/providers.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 6b06a4a

Please sign in to comment.