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

2024-06-22 13:22:57 +02:00
parent 0df67b1b25
commit ecd0c3028f
2 changed files with 51 additions and 0 deletions
--- a/src/v1/constants.rs
+++ b/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)]