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 92ec3de419f3b70fa5daa1ad9f257b6aaaeedcdb..a4faed40f46d29181a8cf44eff75f5aabe87cc74 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 764889d1e93cf552c075b232f02b182762b20865..a999420fe1df2bb02fdc28011f6d8975c9c1444b 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"