baibot-withmcp/src/agent/provider/entity/image.rs

use mxlink::mime;

#[derive(Default)]
pub struct ImageGenerationParams {
    pub size_override: Option<String>,

    pub cheaper_model_switching_allowed: bool,

    pub cheaper_quality_switching_allowed: bool,
}

impl ImageGenerationParams {
    pub fn with_size_override(mut self, value: Option<String>) -> Self {
        self.size_override = value;
        self
    }

    pub fn with_cheaper_model_switching_allowed(mut self, value: bool) -> Self {
        self.cheaper_model_switching_allowed = value;
        self
    }

    pub fn with_cheaper_quality_switching_allowed(mut self, value: bool) -> Self {
        self.cheaper_quality_switching_allowed = value;
        self
    }
}

pub struct ImageGenerationResult {
    pub bytes: Vec<u8>,
    pub mime_type: mime::Mime,
    pub revised_prompt: Option<String>,
}

#[derive(Default)]
pub struct ImageEditParams {
}

pub struct ImageEditResult {
    pub bytes: Vec<u8>,
    pub mime_type: mime::Mime,
}

pub struct ImageSource {
    pub filename: String,
    pub bytes: Vec<u8>,
    pub mime_type: mime::Mime,
}

impl ImageSource {
    pub fn new(filename: String, bytes: Vec<u8>, mime_type: mime::Mime) -> Self {
        Self { filename, bytes, mime_type }
    }
}

impl Into<async_openai::types::ImageInput> for ImageSource {
    fn into(self) -> async_openai::types::ImageInput {
        async_openai::types::ImageInput::from_vec_u8(
            self.filename,
            self.bytes,
        )
    }
}