doubleo7/src/models.rs

use serde::Deserialize;

#[derive(Deserialize)]
pub(crate) struct ChatLogprobs {
    pub(crate) content: Vec<TokenLogprob>,
}

#[derive(Deserialize)]
pub(crate) struct TokenLogprob {
    pub(crate) top_logprobs: Vec<TopLogprob>,
}

#[derive(Deserialize)]
pub(crate) struct TopLogprob {
    pub(crate) token: String,
    pub(crate) logprob: f64,
}

#[derive(Deserialize)]
pub(crate) struct Prompts {
    pub(crate) shieldstral: ShieldstralPrompts,
    pub(crate) gemma: GemmaPrompts,
}

#[derive(Deserialize)]
pub(crate) struct ShieldstralPrompts {
    pub(crate) system: String,
    pub(crate) prompt_template: String,
}

#[derive(Deserialize)]
pub(crate) struct GemmaPrompts {
    pub(crate) preamble: String,
    pub(crate) evil_seed_prompt: String,
    pub(crate) revise_template: String,
}