From b285ac9046623a9446d65d2ce0343fcaea4a847a Mon Sep 17 00:00:00 2001
From: Ravi Theja <ravi03071991@gmail.com>
Date: Tue, 12 Mar 2024 19:51:26 +0530
Subject: [PATCH] Add cohere command-R model (#11852)
* Add cohere command-R model
* bump version
---
.../llama-index-llms-cohere/llama_index/llms/cohere/utils.py | 1 +
.../llms/llama-index-llms-cohere/pyproject.toml | 2 +-
2 files changed, 2 insertions(+), 1 deletion(-)
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 3e0cf81449..e1c3e24859 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 83fbd4e558..3c1fc4bd2b 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"
--
GitLab