From 02ed4aefb869f4f6a7635192d69f0370d7a1492d Mon Sep 17 00:00:00 2001
From: versecafe <147033096+versecafe@users.noreply.github.com>
Date: Wed, 7 May 2025 18:57:15 -0700
Subject: [PATCH] mistral: Add new Mistral medium model (#30171)

Release Notes:

- Added `mistral-medium` to the Mistral provider.
---
 crates/mistral/src/mistral.rs | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/crates/mistral/src/mistral.rs b/crates/mistral/src/mistral.rs
index ac99f37100..3dbe3a5d88 100644
--- a/crates/mistral/src/mistral.rs
+++ b/crates/mistral/src/mistral.rs
@@ -50,6 +50,8 @@ pub enum Model {
     CodestralLatest,
     #[serde(rename = "mistral-large-latest", alias = "mistral-large-latest")]
     MistralLargeLatest,
+    #[serde(rename = "mistral-medium-latest", alias = "mistral-medium-latest")]
+    MistralMediumLatest,
     #[serde(rename = "mistral-small-latest", alias = "mistral-small-latest")]
     MistralSmallLatest,
     #[serde(rename = "open-mistral-nemo", alias = "open-mistral-nemo")]
@@ -77,6 +79,7 @@ impl Model {
         match id {
             "codestral-latest" => Ok(Self::CodestralLatest),
             "mistral-large-latest" => Ok(Self::MistralLargeLatest),
+            "mistral-medium-latest" => Ok(Self::MistralMediumLatest),
             "mistral-small-latest" => Ok(Self::MistralSmallLatest),
             "open-mistral-nemo" => Ok(Self::OpenMistralNemo),
             "open-codestral-mamba" => Ok(Self::OpenCodestralMamba),
@@ -88,6 +91,7 @@ impl Model {
         match self {
             Self::CodestralLatest => "codestral-latest",
             Self::MistralLargeLatest => "mistral-large-latest",
+            Self::MistralMediumLatest => "mistral-medium-latest",
             Self::MistralSmallLatest => "mistral-small-latest",
             Self::OpenMistralNemo => "open-mistral-nemo",
             Self::OpenCodestralMamba => "open-codestral-mamba",
@@ -99,6 +103,7 @@ impl Model {
         match self {
             Self::CodestralLatest => "codestral-latest",
             Self::MistralLargeLatest => "mistral-large-latest",
+            Self::MistralMediumLatest => "mistral-medium-latest",
             Self::MistralSmallLatest => "mistral-small-latest",
             Self::OpenMistralNemo => "open-mistral-nemo",
             Self::OpenCodestralMamba => "open-codestral-mamba",
@@ -112,6 +117,7 @@ impl Model {
         match self {
             Self::CodestralLatest => 256000,
             Self::MistralLargeLatest => 131000,
+            Self::MistralMediumLatest => 128000,
             Self::MistralSmallLatest => 32000,
             Self::OpenMistralNemo => 131000,
             Self::OpenCodestralMamba => 256000,