providers-kit 0.0.2

Portable enum-dispatch Kit over multiple model provider APIs.
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

use serde::{Deserialize, Serialize};

use crate::config::GeminiServiceTier;

#[derive(Clone, Copy, Debug, PartialEq, Eq, Serialize, Deserialize)]
#[serde(rename_all = "SCREAMING_SNAKE_CASE")]
pub(crate) enum GeminiNativeServiceTier {
    Standard,
    Flex,
    Priority,
}

impl GeminiNativeServiceTier {
    pub(crate) fn as_rest_value(self) -> &'static str {
        match self {
            Self::Standard => "standard",
            Self::Flex => "flex",
            Self::Priority => "priority",
        }
    }
}

impl From<GeminiServiceTier> for GeminiNativeServiceTier {
    fn from(value: GeminiServiceTier) -> Self {
        match value {
            GeminiServiceTier::Standard => Self::Standard,
            GeminiServiceTier::Flex => Self::Flex,
            GeminiServiceTier::Priority => Self::Priority,
        }
    }
}

pub(crate) fn render_service_tier(value: GeminiServiceTier) -> Option<String> {
    match value {
        GeminiServiceTier::Standard => None,
        GeminiServiceTier::Flex => Some(GeminiNativeServiceTier::Flex.as_rest_value().to_string()),
        GeminiServiceTier::Priority => Some(
            GeminiNativeServiceTier::Priority
                .as_rest_value()
                .to_string(),
        ),
    }
}