From 69b27954db17edef923fff3a084c5250d320d33e Mon Sep 17 00:00:00 2001
From: Logan <logan.markewich@live.com>
Date: Mon, 17 Feb 2025 17:10:57 -0600
Subject: [PATCH] mistral saba support (#17847)
vbump
---
.../llama_index/llms/mistralai/utils.py | 11 ++++++-----
.../llms/llama-index-llms-mistralai/pyproject.toml | 2 +-
2 files changed, 7 insertions(+), 6 deletions(-)
diff --git a/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py b/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py
index 92ec3de41..a4faed40f 100644
--- a/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py
+++ b/llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py
@@ -4,17 +4,18 @@ MISTRALAI_MODELS: Dict[str, int] = {
"mistral-tiny": 32000,
"mistral-small": 32000,
"mistral-medium": 32000,
- "mistral-large": 32000,
+ "mistral-large": 131000,
+ "mistral-saba-latest": 32000,
"open-mixtral-8x7b": 32000,
"open-mistral-7b": 32000,
"open-mixtral-8x22b": 64000,
"mistral-small-latest": 32000,
"mistral-medium-latest": 32000,
"mistral-large-latest": 32000,
- "codestral-latest": 32000,
- "open-mistral-nemo-latest": 128000,
- "ministral-8b-latest": 128000,
- "ministral-3b-latest": 128000,
+ "codestral-latest": 256000,
+ "open-mistral-nemo-latest": 131000,
+ "ministral-8b-latest": 131000,
+ "ministral-3b-latest": 131000,
}
MISTRALAI_FUNCTION_CALLING_MODELS = (
diff --git a/llama-index-integrations/llms/llama-index-llms-mistralai/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-mistralai/pyproject.toml
index 764889d1e..a999420fe 100644
--- a/llama-index-integrations/llms/llama-index-llms-mistralai/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-mistralai/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
license = "MIT"
name = "llama-index-llms-mistralai"
readme = "README.md"
-version = "0.3.2"
+version = "0.3.3"
[tool.poetry.dependencies]
python = ">=3.9,<4.0"
--
GitLab