allms 0.41.0

One Library to rule them aLLMs
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

use serde::{Deserialize, Serialize};

// Enum of supported Completions APIs
#[derive(Deserialize, Serialize, Debug, Clone, Eq, PartialEq, Default)]
pub enum GoogleApiEndpoints {
    #[default]
    GoogleStudio,
    GoogleVertex,
}

impl GoogleApiEndpoints {
    /// Parses a string into `GoogleApiEndpoints`.
    ///
    /// Supported formats (case-insensitive):
    /// - `"google-studio"` -> `GoogleApiEndpoints::GoogleStudio`
    /// - `"google-vertex"` -> `GoogleApiEndpoints::GoogleVertex`
    ///
    /// Returns default for others.
    #[allow(clippy::should_implement_trait)]
    pub fn from_str(s: &str) -> Self {
        let s_lower = s.to_lowercase();
        match s_lower.as_str() {
            "google-studio" => GoogleApiEndpoints::GoogleStudio,
            "google-vertex" => GoogleApiEndpoints::GoogleVertex,
            _ => GoogleApiEndpoints::default(),
        }
    }
}