babel 0.0.11

Provide Rust enums for Groq, SambaNova, Openrouter's llm model names.
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

pub trait Model {
    fn model_id(&self) -> &'static str;
}

pub trait Provider {
    type ModelType: Model;
    fn provider_name() -> &'static str;
}

macro_rules! define_provider_models {
    ($provider:ident, $enum_name:ident, {
        $(($variant:ident, $value:expr)),*
    }) => {
        // Define model enum with Debug derive
        #[derive(Debug, Clone, Copy)]
        pub enum $enum_name {
            $($variant),*
        }

        // Implement Model trait for enum
        impl Model for $enum_name {
            fn model_id(&self) -> &'static str {
                match self {
                    $(Self::$variant => $value),*
                }
            }
        }
    }
}

pub(crate) use define_provider_models;