language-model-type 0.1.0

A Rust crate providing an enumeration for various language model types used in machine learning applications, enabling precise model specification through enums.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59

// ---------------- [ File: language-model-type/src/language_model_type.rs ]
crate::ix!();

/// Supported model types.
#[derive(Copy,Clone,Debug, Serialize, Deserialize)]
#[serde(rename_all = "lowercase")]
pub enum LanguageModelType {
    Gpt3_5Turbo,
    Gpt4o,
    Gpt4oMini,
    Gpt4Turbo,
    O1Preview,
    O1Mini,
    O1,
}

impl std::fmt::Display for LanguageModelType {

    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            LanguageModelType::Gpt3_5Turbo => write!(f, "gpt-3.5-turbo"),
            LanguageModelType::Gpt4o       => write!(f, "gpt-4o"),
            LanguageModelType::Gpt4oMini   => write!(f, "gpt-4o-mini"),
            LanguageModelType::Gpt4Turbo   => write!(f, "gpt-4-turbo"),
            LanguageModelType::O1Preview   => write!(f, "o1-preview"),
            LanguageModelType::O1Mini      => write!(f, "o1-mini"),
            LanguageModelType::O1          => write!(f, "o1"),
        }
    }
}

pub mod model_type {

    use super::*;

    pub fn serialize<S>(value: &LanguageModelType, serializer: S) -> Result<S::Ok, S::Error>
    where
        S: Serializer,
    {
        serializer.serialize_str(&value.to_string())
    }

    pub fn deserialize<'de, D>(deserializer: D) -> Result<LanguageModelType, D::Error>
    where
        D: Deserializer<'de>,
    {
        let s: String = Deserialize::deserialize(deserializer)?;
        match s.as_ref() {
            "gpt-3.5-turbo" => Ok(LanguageModelType::Gpt3_5Turbo),
            "gpt-4o"        => Ok(LanguageModelType::Gpt4o),
            "gpt-4o-mini"   => Ok(LanguageModelType::Gpt4oMini),
            "gpt-4-turbo"   => Ok(LanguageModelType::Gpt4Turbo),
            "o1-preview"    => Ok(LanguageModelType::O1Preview),
            "o1-mini"       => Ok(LanguageModelType::O1Mini),
            "o1"            => Ok(LanguageModelType::O1),
            _               => Err(serde::de::Error::custom("unknown model type")),
        }
    }
}