Skip to content
Snippets Groups Projects
Unverified Commit 69d19a1e authored by Ravi Theja's avatar Ravi Theja Committed by GitHub
Browse files

Add support for Cohere Command R+ model (#12581)

parent f03db8da
No related branches found
No related tags found
No related merge requests found
...@@ -12,6 +12,7 @@ from tenacity import ( ...@@ -12,6 +12,7 @@ from tenacity import (
COMMAND_MODELS = { COMMAND_MODELS = {
"command-r": 128000, "command-r": 128000,
"command-r-plus": 128000,
"command": 4096, "command": 4096,
"command-nightly": 4096, "command-nightly": 4096,
"command-light": 4096, "command-light": 4096,
......
...@@ -27,12 +27,12 @@ exclude = ["**/BUILD"] ...@@ -27,12 +27,12 @@ exclude = ["**/BUILD"]
license = "MIT" license = "MIT"
name = "llama-index-llms-cohere" name = "llama-index-llms-cohere"
readme = "README.md" readme = "README.md"
version = "0.1.5" version = "0.1.6"
[tool.poetry.dependencies] [tool.poetry.dependencies]
python = ">=3.8.1,<4.0" python = ">=3.8.1,<4.0"
llama-index-core = "^0.10.1" llama-index-core = "^0.10.1"
cohere = "^5.1.1" cohere = "^5.1.2"
[tool.poetry.group.dev.dependencies] [tool.poetry.group.dev.dependencies]
ipython = "8.10.0" ipython = "8.10.0"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment