From 69d19a1e9584e426a55222058cf97427698b93b9 Mon Sep 17 00:00:00 2001
From: Ravi Theja <ravi03071991@gmail.com>
Date: Thu, 4 Apr 2024 20:28:38 +0530
Subject: [PATCH] Add support for Cohere Command R+ model (#12581)
---
.../llama-index-llms-cohere/llama_index/llms/cohere/utils.py | 1 +
.../llms/llama-index-llms-cohere/pyproject.toml | 4 ++--
2 files changed, 3 insertions(+), 2 deletions(-)
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 b588510cec..a95b7a5b3f 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 4c486980e3..de7400b8b1 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"
--
GitLab