fix(deps): update rust crate strum to 0.27.0

* 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.

CHANGELOG.md

@@ -1,3 +1,33 @@
+## [0.14.0](https://github.com/ivangabriele/mistralai-client-rs/compare/v0.13.0...v) (2024-08-27)
+
+### Features
+
+* **constants:** update model constants ([#17](https://github.com/ivangabriele/mistralai-client-rs/issues/17)) ([161b33c](https://github.com/ivangabriele/mistralai-client-rs/commit/161b33c72539a6e982207349942a436df95399b7))
+## [0.13.0](https://github.com/ivangabriele/mistralai-client-rs/compare/v0.12.0...v) (2024-08-21)
+
+### ⚠ BREAKING CHANGES
+
+* **client:** `v1::model_list::ModelListData` struct has been updated.
+
+### Bug Fixes
+
+* **client:** remove the `Content-Type` from the headers of the reqwest builders. ([#14](https://github.com/ivangabriele/mistralai-client-rs/issues/14)) ([9bfbf2e](https://github.com/ivangabriele/mistralai-client-rs/commit/9bfbf2e9dc7b48103ac56923fb8b3ac9a5e2d9cf)), closes [#13](https://github.com/ivangabriele/mistralai-client-rs/issues/13)
+* **client:** update ModelListData struct following API changes ([2114916](https://github.com/ivangabriele/mistralai-client-rs/commit/2114916941e1ff5aa242290df5f092c0d4954afc))
+## [0.12.0](https://github.com/ivangabriele/mistralai-client-rs/compare/v0.11.0...v) (2024-07-24)
+
+### Features
+
+* implement the Debug trait for Client ([#11](https://github.com/ivangabriele/mistralai-client-rs/issues/11)) ([3afeec1](https://github.com/ivangabriele/mistralai-client-rs/commit/3afeec1d586022e43c7b10906acec5e65927ba7d))
+* mark Function trait as Send ([#12](https://github.com/ivangabriele/mistralai-client-rs/issues/12)) ([8e9f7a5](https://github.com/ivangabriele/mistralai-client-rs/commit/8e9f7a53863879b2ad618e9e5707b198e4f3b135))
+## [0.11.0](https://github.com/ivangabriele/mistralai-client-rs/compare/v0.10.0...v) (2024-06-22)
+
+### Features
+
+* **constants:** add OpenMixtral8x22b, MistralTiny & CodestralLatest to Model enum ([ecd0c30](https://github.com/ivangabriele/mistralai-client-rs/commit/ecd0c3028fdcfab32b867eb1eed86182f5f4ab81))
+
+### Bug Fixes
+
+* **chat:** implement Clone trait for ChatParams & ResponseFormat ([0df67b1](https://github.com/ivangabriele/mistralai-client-rs/commit/0df67b1b2571fb04b636ce015a2daabe629ff352))
 ## [0.10.0](https://github.com/ivangabriele/mistralai-client-rs/compare/v0.9.0...v) (2024-06-07)
 
 ### ⚠ BREAKING CHANGES

Cargo.toml

@@ -2,7 +2,7 @@
 name = "mistralai-client"
 description = "Mistral AI API client library for Rust (unofficial)."
 license = "Apache-2.0"
-version = "0.10.0"
+version = "0.14.0"
 
 edition = "2021"
 rust-version = "1.76.0"
@@ -23,7 +23,7 @@ log = "0.4.21"
 reqwest = { version = "0.12.0", features = ["json", "blocking", "stream"] }
 serde = { version = "1.0.197", features = ["derive"] }
 serde_json = "1.0.114"
-strum = "0.26.1"
+strum = "0.27.0"
 thiserror = "1.0.57"
 tokio = { version = "1.36.0", features = ["full"] }
 tokio-stream = "0.1.14"

src/v1/chat.rs

@@ -46,7 +46,7 @@ pub enum ChatMessageRole {
 /// The format that the model must output.
 ///
 /// See the [API documentation](https://docs.mistral.ai/api/#operation/createChatCompletion) for more information.
-#[derive(Debug, Serialize, Deserialize)]
+#[derive(Clone, Debug, Serialize, Deserialize)]
 pub struct ResponseFormat {
     #[serde(rename = "type")]
     pub type_: String,
@@ -65,7 +65,7 @@ impl ResponseFormat {
 /// The parameters for the chat request.
 ///
 /// See the [API documentation](https://docs.mistral.ai/api/#operation/createChatCompletion) for more information.
-#[derive(Debug)]
+#[derive(Clone, Debug)]
 pub struct ChatParams {
     /// The maximum number of tokens to generate in the completion.
     ///

src/v1/client.rs

@@ -10,6 +10,7 @@ use std::{
 
 use crate::v1::{chat, chat_stream, constants, embedding, error, model_list, tool, utils};
 
+#[derive(Debug)]
 pub struct Client {
     pub api_key: String,
     pub endpoint: String,
@@ -383,7 +384,6 @@ impl Client {
         let request_builder = request
             .bearer_auth(&self.api_key)
             .header("Accept", "application/json")
-            .header("Content-Type", "application/json")
             .header("User-Agent", user_agent);
 
         request_builder
@@ -398,7 +398,6 @@ impl Client {
         let request_builder = request
             .bearer_auth(&self.api_key)
             .header("Accept", "application/json")
-            .header("Content-Type", "application/json")
             .header("User-Agent", user_agent);
 
         request_builder
@@ -413,7 +412,6 @@ impl Client {
         let request_builder = request
             .bearer_auth(&self.api_key)
             .header("Accept", "text/event-stream")
-            .header("Content-Type", "application/json")
             .header("User-Agent", user_agent);
 
         request_builder

src/v1/constants.rs

@@ -8,12 +8,24 @@ pub enum Model {
     OpenMistral7b,
     #[serde(rename = "open-mixtral-8x7b")]
     OpenMixtral8x7b,
-    #[serde(rename = "mistral-small-latest")]
+    #[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", 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 = "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)]

src/v1/model_list.rs

@@ -9,6 +9,7 @@ pub struct ModelListResponse {
     pub data: Vec<ModelListData>,
 }
 
+/// See: https://docs.mistral.ai/api/#tag/models
 #[derive(Clone, Debug, Deserialize, Serialize)]
 pub struct ModelListData {
     pub id: String,
@@ -16,27 +17,21 @@ pub struct ModelListData {
     /// Unix timestamp (in seconds).
     pub created: u32,
     pub owned_by: String,
-    pub permission: Vec<ModelListDataPermission>,
-    // TODO Check this prop (seen in API responses but undocumented).
-    // pub root: ???,
-    // TODO Check this prop (seen in API responses but undocumented).
-    // pub parent: ???,
+    pub root: Option<String>,
+    pub archived: bool,
+    pub name: String,
+    pub description: String,
+    pub capabilities: ModelListDataCapabilies,
+    pub max_context_length: u32,
+    pub aliases: Vec<String>,
+    /// ISO 8601 date (`YYYY-MM-DDTHH:MM:SSZ`).
+    pub deprecation: Option<String>,
 }
 
 #[derive(Clone, Debug, Deserialize, Serialize)]
-pub struct ModelListDataPermission {
-    pub id: String,
-    pub object: String,
-    /// Unix timestamp (in seconds).
-    pub created: u32,
-    pub allow_create_engine: bool,
-    pub allow_sampling: bool,
-    pub allow_logprobs: bool,
-    pub allow_search_indices: bool,
-    pub allow_view: bool,
-    pub allow_fine_tuning: bool,
-    pub organization: String,
-    pub is_blocking: bool,
-    // TODO Check this prop (seen in API responses but undocumented).
-    // pub group: ???,
+pub struct ModelListDataCapabilies {
+    pub completion_chat: bool,
+    pub completion_fim: bool,
+    pub function_calling: bool,
+    pub fine_tuning: bool,
 }

src/v1/tool.rs

@@ -1,6 +1,6 @@
 use async_trait::async_trait;
 use serde::{Deserialize, Serialize};
-use std::{any::Any, collections::HashMap};
+use std::{any::Any, collections::HashMap, fmt::Debug};
 
 // -----------------------------------------------------------------------------
 // Definitions
@@ -133,6 +133,12 @@ pub enum ToolChoice {
 // Custom
 
 #[async_trait]
-pub trait Function {
+pub trait Function: Send {
     async fn execute(&self, arguments: String) -> Box<dyn Any + Send>;
 }
+
+impl Debug for dyn Function {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "Function()")
+    }
+}

tests/v1_client_new_test.rs

@@ -1,6 +1,11 @@
 use jrest::expect;
 use mistralai_client::v1::{client::Client, error::ClientError};
 
+#[derive(Debug)]
+struct _Foo {
+    _client: Client,
+}
+
 #[test]
 fn test_client_new_with_none_params() {
     let maybe_original_mistral_api_key = std::env::var("MISTRAL_API_KEY").ok();

tests/v1_constants_test.rs

@@ -0,0 +1,44 @@
+use jrest::expect;
+use mistralai_client::v1::{
+    chat::{ChatMessage, ChatParams},
+    client::Client,
+    constants::Model,
+};
+
+#[test]
+fn test_model_constant() {
+    let models = vec![
+        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();
+
+    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);
+    }
+}

tests/v1_tool_test.rs

@@ -0,0 +1,7 @@
+use mistralai_client::v1::client::Client;
+
+trait _Trait: Send {}
+struct _Foo {
+    _dummy: Client,
+}
+impl _Trait for _Foo {}