mistral saba support (#17847)

vbump

Author: logan-markewich

llama-index-integrations/llms/llama-index-llms-mistralai/llama_index/llms/mistralai/utils.py

@@ -4,17 +4,18 @@
     "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 = (

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"