lattice-tune 0.2.2

Training infrastructure for Lattice neural models
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

//! Teacher model configuration

mod config;
mod security;

pub use config::{TeacherConfig, TeacherConfigBuilder};
pub use security::EndpointSecurity;

#[cfg(feature = "serde")]
use serde::{Deserialize, Serialize};

/// Supported teacher model providers
#[derive(Debug, Clone, PartialEq, Eq)]
#[cfg_attr(feature = "serde", derive(Serialize, Deserialize))]
#[cfg_attr(feature = "serde", serde(rename_all = "snake_case"))]
pub enum TeacherProvider {
    /// Anthropic Claude models
    Claude,
    /// OpenAI GPT models
    OpenAI,
    /// Google Gemini models
    Gemini,
    /// Local model (e.g., via Ollama)
    Local,
    /// Custom provider
    Custom(String),
}

impl std::fmt::Display for TeacherProvider {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            TeacherProvider::Claude => write!(f, "claude"),
            TeacherProvider::OpenAI => write!(f, "openai"),
            TeacherProvider::Gemini => write!(f, "gemini"),
            TeacherProvider::Local => write!(f, "local"),
            TeacherProvider::Custom(name) => write!(f, "custom:{name}"),
        }
    }
}

#[cfg(test)]
mod tests;