diff --git a/llama-index-integrations/llms/llama-index-llms-cohere/llama_index/llms/cohere/utils.py b/llama-index-integrations/llms/llama-index-llms-cohere/llama_index/llms/cohere/utils.py index 3e0cf814494581c2b19ecdc0414acc4739ca8b31..e1c3e24859ecbdcf1659ec53b81b58d7087ae5c5 100644 --- a/llama-index-integrations/llms/llama-index-llms-cohere/llama_index/llms/cohere/utils.py +++ b/llama-index-integrations/llms/llama-index-llms-cohere/llama_index/llms/cohere/utils.py @@ -11,6 +11,7 @@ from tenacity import ( ) COMMAND_MODELS = { + "command-r": 128000, "command": 4096, "command-nightly": 4096, "command-light": 4096, diff --git a/llama-index-integrations/llms/llama-index-llms-cohere/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-cohere/pyproject.toml index 83fbd4e5580bd6a637a8f2370cc33c8e22d3c4a6..3c1fc4bd2b2760346c4dcc12ade79e16a6ef0a57 100644 --- a/llama-index-integrations/llms/llama-index-llms-cohere/pyproject.toml +++ b/llama-index-integrations/llms/llama-index-llms-cohere/pyproject.toml @@ -27,7 +27,7 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-llms-cohere" readme = "README.md" -version = "0.1.2" +version = "0.1.3" [tool.poetry.dependencies] python = ">=3.8.1,<4.0"