feat(constants): add OpenMixtral8x22b, MistralTiny & CodestralLatest to Model enum

src/v1/constants.rs

@@ -8,12 +8,18 @@ pub enum Model {
     OpenMistral7b,
     #[serde(rename = "open-mixtral-8x7b")]
     OpenMixtral8x7b,
+    #[serde(rename = "open-mixtral-8x22b")]
+    OpenMixtral8x22b,
+    #[serde(rename = "mistral-tiny")]
+    MistralTiny,
     #[serde(rename = "mistral-small-latest")]
     MistralSmallLatest,
     #[serde(rename = "mistral-medium-latest")]
     MistralMediumLatest,
     #[serde(rename = "mistral-large-latest")]
     MistralLargeLatest,
+    #[serde(rename = "codestral-latest")]
+    CodestralLatest,
 }
 
 #[derive(Clone, Debug, Eq, PartialEq, Deserialize, Serialize)]

tests/v1_constants_test.rs

@@ -0,0 +1,45 @@
+use jrest::expect;
+use mistralai_client::v1::{
+    chat::{ChatMessage, ChatParams},
+    client::Client,
+    constants::Model,
+};
+
+mod setup;
+
+#[test]
+fn test_model_constant() {
+    setup::setup();
+
+    let models = vec![
+        Model::OpenMistral7b,
+        Model::OpenMixtral8x7b,
+        Model::OpenMixtral8x22b,
+        Model::MistralTiny,
+        Model::MistralSmallLatest,
+        Model::MistralMediumLatest,
+        Model::MistralLargeLatest,
+        Model::CodestralLatest,
+    ];
+
+    let client = Client::new(None, None, None, None).unwrap();
+
+    let messages = vec![ChatMessage::new_user_message("A number between 0 and 100?")];
+    let options = ChatParams {
+        temperature: 0.0,
+        random_seed: Some(42),
+        ..Default::default()
+    };
+
+    for model in models {
+        let response = client
+            .chat(model.clone(), messages.clone(), Some(options.clone()))
+            .unwrap();
+
+        expect!(response.model).to_be(model);
+        expect!(response.object).to_be("chat.completion".to_string());
+        expect!(response.choices.len()).to_be(1);
+        expect!(response.choices[0].index).to_be(0);
+        expect!(response.choices[0].message.content.len()).to_be_greater_than(0);
+    }
+}