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 b588510cec1a19e1c1baabb31ee90fcd72bbb9af..a95b7a5b3f101b8d13ba2f4dc25fd00b30b9aaf1 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 @@ -12,6 +12,7 @@ from tenacity import ( COMMAND_MODELS = { "command-r": 128000, + "command-r-plus": 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 4c486980e327388920e55ac2e6b9b0e0ab8b4395..de7400b8b1e31012093658d42d28e0a3bc7394c3 100644 --- a/llama-index-integrations/llms/llama-index-llms-cohere/pyproject.toml +++ b/llama-index-integrations/llms/llama-index-llms-cohere/pyproject.toml @@ -27,12 +27,12 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-llms-cohere" readme = "README.md" -version = "0.1.5" +version = "0.1.6" [tool.poetry.dependencies] python = ">=3.8.1,<4.0" llama-index-core = "^0.10.1" -cohere = "^5.1.1" +cohere = "^5.1.2" [tool.poetry.group.dev.dependencies] ipython = "8.10.0"