use std::collections::BTreeMap;
use serde::{Deserialize, Serialize};
use crate::openai::create_image::types::{
HttpMethod, OpenAiImageBackground, OpenAiImageModel, OpenAiImageModeration,
OpenAiImageOutputFormat, OpenAiImageQuality, OpenAiImageResponseFormat, OpenAiImageSize,
OpenAiImageStyle,
};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct OpenAiCreateImageRequest {
pub method: HttpMethod,
pub path: PathParameters,
pub query: QueryParameters,
pub headers: RequestHeaders,
pub body: RequestBody,
}
impl Default for OpenAiCreateImageRequest {
fn default() -> Self {
Self {
method: HttpMethod::Post,
path: PathParameters::default(),
query: QueryParameters::default(),
headers: RequestHeaders::default(),
body: RequestBody::default(),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct PathParameters {}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct QueryParameters {}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct RequestHeaders {
#[serde(flatten, default, skip_serializing_if = "BTreeMap::is_empty")]
pub extra: BTreeMap<String, String>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
pub struct RequestBody {
pub prompt: String,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub background: Option<OpenAiImageBackground>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub model: Option<OpenAiImageModel>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub moderation: Option<OpenAiImageModeration>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub n: Option<u32>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub output_compression: Option<u8>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub output_format: Option<OpenAiImageOutputFormat>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub partial_images: Option<u32>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub quality: Option<OpenAiImageQuality>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub response_format: Option<OpenAiImageResponseFormat>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub size: Option<OpenAiImageSize>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub stream: Option<bool>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub style: Option<OpenAiImageStyle>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub user: Option<String>,
}