use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct CreateSpeechRequest {
#[serde(rename = "model")]
pub model: String,
#[serde(rename = "input")]
pub input: String,
#[serde(rename = "instructions", skip_serializing_if = "Option::is_none")]
pub instructions: Option<String>,
#[serde(rename = "voice")]
pub voice: Box<models::VoiceIdsOrCustomVoice>,
#[serde(rename = "response_format", skip_serializing_if = "Option::is_none")]
pub response_format: Option<ResponseFormat>,
#[serde(rename = "speed", skip_serializing_if = "Option::is_none")]
pub speed: Option<f64>,
#[serde(rename = "stream_format", skip_serializing_if = "Option::is_none")]
pub stream_format: Option<StreamFormat>,
}
impl CreateSpeechRequest {
pub fn new(
model: String,
input: String,
voice: models::VoiceIdsOrCustomVoice,
) -> CreateSpeechRequest {
CreateSpeechRequest {
model,
input,
instructions: None,
voice: Box::new(voice),
response_format: None,
speed: None,
stream_format: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum ResponseFormat {
#[serde(rename = "mp3")]
Mp3,
#[serde(rename = "opus")]
Opus,
#[serde(rename = "aac")]
Aac,
#[serde(rename = "flac")]
Flac,
#[serde(rename = "wav")]
Wav,
#[serde(rename = "pcm")]
Pcm,
}
impl Default for ResponseFormat {
fn default() -> ResponseFormat {
Self::Mp3
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum StreamFormat {
#[serde(rename = "sse")]
Sse,
#[serde(rename = "audio")]
Audio,
}
impl Default for StreamFormat {
fn default() -> StreamFormat {
Self::Sse
}
}
impl std::fmt::Display for CreateSpeechRequest {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
match serde_json::to_string(self) {
Ok(s) => write!(f, "{}", s),
Err(_) => Err(std::fmt::Error),
}
}
}