Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

mistral: Add support for magistral-small and magistral-medium (#34983)

Browse source 
Release Notes:

- mistral: Added support for magistral-small and magistral-medium
This commit is contained in:
Peter Tripp 2025-07-23 23:13:49 -04:00 • committed by GitHub
parent 67027bb241
commit b93e1c736b
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 21 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

21
crates/mistral/src/mistral.rs
View file

@ -48,20 +48,29 @@ pub enum Model {
#[serde(rename = "codestral-latest", alias = "codestral-latest")] #[serde(rename = "codestral-latest", alias = "codestral-latest")]
#[default] #[default]
CodestralLatest, CodestralLatest,
#[serde(rename = "mistral-large-latest", alias = "mistral-large-latest")] #[serde(rename = "mistral-large-latest", alias = "mistral-large-latest")]
MistralLargeLatest, MistralLargeLatest,
#[serde(rename = "mistral-medium-latest", alias = "mistral-medium-latest")] #[serde(rename = "mistral-medium-latest", alias = "mistral-medium-latest")]
MistralMediumLatest, MistralMediumLatest,
#[serde(rename = "mistral-small-latest", alias = "mistral-small-latest")] #[serde(rename = "mistral-small-latest", alias = "mistral-small-latest")]
MistralSmallLatest, MistralSmallLatest,
#[serde(rename = "magistral-medium-latest", alias = "magistral-medium-latest")]
MagistralMediumLatest,
#[serde(rename = "magistral-small-latest", alias = "magistral-small-latest")]
MagistralSmallLatest,
#[serde(rename = "open-mistral-nemo", alias = "open-mistral-nemo")] #[serde(rename = "open-mistral-nemo", alias = "open-mistral-nemo")]
OpenMistralNemo, OpenMistralNemo,
#[serde(rename = "open-codestral-mamba", alias = "open-codestral-mamba")] #[serde(rename = "open-codestral-mamba", alias = "open-codestral-mamba")]
OpenCodestralMamba, OpenCodestralMamba,
#[serde(rename = "devstral-medium-latest", alias = "devstral-medium-latest")] #[serde(rename = "devstral-medium-latest", alias = "devstral-medium-latest")]
DevstralMediumLatest, DevstralMediumLatest,
#[serde(rename = "devstral-small-latest", alias = "devstral-small-latest")] #[serde(rename = "devstral-small-latest", alias = "devstral-small-latest")]
DevstralSmallLatest, DevstralSmallLatest,
#[serde(rename = "pixtral-12b-latest", alias = "pixtral-12b-latest")] #[serde(rename = "pixtral-12b-latest", alias = "pixtral-12b-latest")]
Pixtral12BLatest, Pixtral12BLatest,
#[serde(rename = "pixtral-large-latest", alias = "pixtral-large-latest")] #[serde(rename = "pixtral-large-latest", alias = "pixtral-large-latest")]
@ -91,6 +100,8 @@ impl Model {
"mistral-large-latest" => Ok(Self::MistralLargeLatest), "mistral-large-latest" => Ok(Self::MistralLargeLatest),
"mistral-medium-latest" => Ok(Self::MistralMediumLatest), "mistral-medium-latest" => Ok(Self::MistralMediumLatest),
"mistral-small-latest" => Ok(Self::MistralSmallLatest), "mistral-small-latest" => Ok(Self::MistralSmallLatest),
"magistral-medium-latest" => Ok(Self::MagistralMediumLatest),
"magistral-small-latest" => Ok(Self::MagistralSmallLatest),
"open-mistral-nemo" => Ok(Self::OpenMistralNemo), "open-mistral-nemo" => Ok(Self::OpenMistralNemo),
"open-codestral-mamba" => Ok(Self::OpenCodestralMamba), "open-codestral-mamba" => Ok(Self::OpenCodestralMamba),
"devstral-medium-latest" => Ok(Self::DevstralMediumLatest), "devstral-medium-latest" => Ok(Self::DevstralMediumLatest),
@ -107,6 +118,8 @@ impl Model {
Self::MistralLargeLatest => "mistral-large-latest", Self::MistralLargeLatest => "mistral-large-latest",
Self::MistralMediumLatest => "mistral-medium-latest", Self::MistralMediumLatest => "mistral-medium-latest",
Self::MistralSmallLatest => "mistral-small-latest", Self::MistralSmallLatest => "mistral-small-latest",
Self::MagistralMediumLatest => "magistral-medium-latest",
Self::MagistralSmallLatest => "magistral-small-latest",
Self::OpenMistralNemo => "open-mistral-nemo", Self::OpenMistralNemo => "open-mistral-nemo",
Self::OpenCodestralMamba => "open-codestral-mamba", Self::OpenCodestralMamba => "open-codestral-mamba",
Self::DevstralMediumLatest => "devstral-medium-latest", Self::DevstralMediumLatest => "devstral-medium-latest",
@ -123,6 +136,8 @@ impl Model {
Self::MistralLargeLatest => "mistral-large-latest", Self::MistralLargeLatest => "mistral-large-latest",
Self::MistralMediumLatest => "mistral-medium-latest", Self::MistralMediumLatest => "mistral-medium-latest",
Self::MistralSmallLatest => "mistral-small-latest", Self::MistralSmallLatest => "mistral-small-latest",
Self::MagistralMediumLatest => "magistral-medium-latest",
Self::MagistralSmallLatest => "magistral-small-latest",
Self::OpenMistralNemo => "open-mistral-nemo", Self::OpenMistralNemo => "open-mistral-nemo",
Self::OpenCodestralMamba => "open-codestral-mamba", Self::OpenCodestralMamba => "open-codestral-mamba",
Self::DevstralMediumLatest => "devstral-medium-latest", Self::DevstralMediumLatest => "devstral-medium-latest",
@ -141,6 +156,8 @@ impl Model {
Self::MistralLargeLatest => 131000, Self::MistralLargeLatest => 131000,
Self::MistralMediumLatest => 128000, Self::MistralMediumLatest => 128000,
Self::MistralSmallLatest => 32000, Self::MistralSmallLatest => 32000,
Self::MagistralMediumLatest => 40000,
Self::MagistralSmallLatest => 40000,
Self::OpenMistralNemo => 131000, Self::OpenMistralNemo => 131000,
Self::OpenCodestralMamba => 256000, Self::OpenCodestralMamba => 256000,
Self::DevstralMediumLatest => 128000, Self::DevstralMediumLatest => 128000,
@ -166,6 +183,8 @@ impl Model {
| Self::MistralLargeLatest | Self::MistralLargeLatest
| Self::MistralMediumLatest | Self::MistralMediumLatest
| Self::MistralSmallLatest | Self::MistralSmallLatest
| Self::MagistralMediumLatest
| Self::MagistralSmallLatest
| Self::OpenMistralNemo | Self::OpenMistralNemo
| Self::OpenCodestralMamba | Self::OpenCodestralMamba
| Self::DevstralMediumLatest | Self::DevstralMediumLatest
@ -184,6 +203,8 @@ impl Model {
| Self::MistralSmallLatest => true, | Self::MistralSmallLatest => true,
Self::CodestralLatest Self::CodestralLatest
| Self::MistralLargeLatest | Self::MistralLargeLatest
| Self::MagistralMediumLatest
| Self::MagistralSmallLatest
| Self::OpenMistralNemo | Self::OpenMistralNemo
| Self::OpenCodestralMamba | Self::OpenCodestralMamba
| Self::DevstralMediumLatest | Self::DevstralMediumLatest