use serde::{Deserialize, Serialize};
use serde_json::Value;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct OpenAIRequest {
pub model: String,
#[serde(default)]
pub messages: Vec<OpenAIMessage>,
#[serde(default)]
pub prompt: Option<String>,
#[serde(default)]
pub stream: bool,
#[serde(default)]
pub n: Option<u32>,
#[serde(rename = "max_tokens")]
pub max_tokens: Option<u32>,
pub temperature: Option<f64>,
#[serde(rename = "top_p")]
pub top_p: Option<f64>,
pub stop: Option<Value>,
pub response_format: Option<ResponseFormat>,
#[serde(default)]
pub tools: Option<Vec<Value>>,
#[serde(rename = "tool_choice")]
pub tool_choice: Option<Value>,
#[serde(rename = "parallel_tool_calls")]
pub parallel_tool_calls: Option<bool>,
pub instructions: Option<String>,
pub input: Option<Value>,
#[serde(default)]
pub size: Option<String>,
#[serde(default)]
pub quality: Option<String>,
#[serde(default, rename = "personGeneration")]
pub person_generation: Option<String>,
#[serde(default)]
pub thinking: Option<ThinkingConfig>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ThinkingConfig {
#[serde(rename = "type")]
pub thinking_type: Option<String>,
#[serde(rename = "budget_tokens")]
pub budget_tokens: Option<u32>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ResponseFormat {
pub r#type: String,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(untagged)]
pub enum OpenAIContent {
String(String),
Array(Vec<OpenAIContentBlock>),
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
#[serde(tag = "type")]
pub enum OpenAIContentBlock {
#[serde(rename = "text", alias = "input_text")]
Text { text: String },
#[serde(rename = "image_url")]
ImageUrl { image_url: OpenAIImageUrl },
#[serde(rename = "audio_url")]
AudioUrl { audio_url: AudioUrlContent },
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct OpenAIImageUrl {
pub url: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub detail: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq)]
pub struct AudioUrlContent {
pub url: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct OpenAIMessage {
pub role: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub content: Option<OpenAIContent>,
#[serde(skip_serializing_if = "Option::is_none")]
pub reasoning_content: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub tool_calls: Option<Vec<ToolCall>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub tool_call_id: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub name: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ToolCall {
pub id: String,
pub r#type: String,
pub function: ToolFunction,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ToolFunction {
pub name: String,
pub arguments: String,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct OpenAIResponse {
pub id: String,
pub object: String,
pub created: u64,
pub model: String,
pub choices: Vec<Choice>,
#[serde(skip_serializing_if = "Option::is_none")]
pub usage: Option<OpenAIUsage>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Choice {
pub index: u32,
pub message: OpenAIMessage,
pub finish_reason: Option<String>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct OpenAIUsage {
pub prompt_tokens: u32,
pub completion_tokens: u32,
pub total_tokens: u32,
#[serde(skip_serializing_if = "Option::is_none")]
pub prompt_tokens_details: Option<PromptTokensDetails>,
#[serde(skip_serializing_if = "Option::is_none")]
pub completion_tokens_details: Option<CompletionTokensDetails>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct PromptTokensDetails {
#[serde(skip_serializing_if = "Option::is_none")]
pub cached_tokens: Option<u32>,
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct CompletionTokensDetails {
#[serde(skip_serializing_if = "Option::is_none")]
pub reasoning_tokens: Option<u32>,
}
#[cfg(test)]
mod tests {
use super::*;
use serde_json::json;
#[test]
fn openai_request_preserves_sampling_precision_with_f64() {
let req: OpenAIRequest = serde_json::from_value(json!({
"model": "gpt-5.3-codex",
"temperature": 0.123456789123,
"top_p": 0.987654321987
}))
.expect("request should deserialize");
let temp = req.temperature.expect("temperature");
let top_p = req.top_p.expect("top_p");
assert!((temp - 0.123456789123).abs() < 1e-12);
assert!((top_p - 0.987654321987).abs() < 1e-12);
let out = serde_json::to_value(req).expect("serialize");
assert!((out["temperature"].as_f64().unwrap() - 0.123456789123).abs() < 1e-12);
assert!((out["top_p"].as_f64().unwrap() - 0.987654321987).abs() < 1e-12);
}
}