From 161b33c72539a6e982207349942a436df95399b7 Mon Sep 17 00:00:00 2001
From: francois-caddet <francois.caddet@icloud.com>
Date: Tue, 27 Aug 2024 09:25:59 +0200
Subject: [PATCH] feat(constants): update model constants (#17)

* feat(Model)!: Update model constants

- Add new models Mistral Nemo and Codestral Mamba
- Add aliases to models constant to be deserialized seemlessly from the versioned forme (e.g.: "mistral-large-2407")
- The commonly named Mistral Large 2 is now `constants::Model::MistralLargeLatest` when the old one is `Model::MistralLarge`

* tests(Model): Update the tests to check the new model variants.
---
 src/v1/constants.rs        | 14 ++++++++++----
 tests/v1_constants_test.rs |  3 +++
 2 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/src/v1/constants.rs b/src/v1/constants.rs
index e060664..52c0976 100644
--- a/src/v1/constants.rs
+++ b/src/v1/constants.rs
@@ -10,16 +10,22 @@ pub enum Model {
     OpenMixtral8x7b,
     #[serde(rename = "open-mixtral-8x22b")]
     OpenMixtral8x22b,
+    #[serde(rename = "open-mistral-nemo", alias = "open-mistral-nemo-2407")]
+    OpenMistralNemo,
     #[serde(rename = "mistral-tiny")]
     MistralTiny,
-    #[serde(rename = "mistral-small-latest")]
+    #[serde(rename = "mistral-small-latest", alias = "mistral-small-2402")]
     MistralSmallLatest,
-    #[serde(rename = "mistral-medium-latest")]
+    #[serde(rename = "mistral-medium-latest", alias = "mistral-medium-2312")]
     MistralMediumLatest,
-    #[serde(rename = "mistral-large-latest")]
+    #[serde(rename = "mistral-large-latest", alias = "mistral-large-2407")]
     MistralLargeLatest,
-    #[serde(rename = "codestral-latest")]
+    #[serde(rename = "mistral-large-2402")]
+    MistralLarge,
+    #[serde(rename = "codestral-latest", alias = "codestral-2405")]
     CodestralLatest,
+    #[serde(rename = "open-codestral-mamba")]
+    CodestralMamba,
 }
 
 #[derive(Clone, Debug, Eq, PartialEq, Deserialize, Serialize)]
diff --git a/tests/v1_constants_test.rs b/tests/v1_constants_test.rs
index afd6678..7903acf 100644
--- a/tests/v1_constants_test.rs
+++ b/tests/v1_constants_test.rs
@@ -11,11 +11,14 @@ fn test_model_constant() {
         Model::OpenMistral7b,
         Model::OpenMixtral8x7b,
         Model::OpenMixtral8x22b,
+        Model::OpenMistralNemo,
         Model::MistralTiny,
         Model::MistralSmallLatest,
         Model::MistralMediumLatest,
         Model::MistralLargeLatest,
+        Model::MistralLarge,
         Model::CodestralLatest,
+        Model::CodestralMamba,
     ];
 
     let client = Client::new(None, None, None, None).unwrap();