use serde::{Deserialize, Serialize};
#[derive(Serialize)]
pub(crate) struct ChatRequest {
pub(crate) model: String,
pub(crate) messages: Vec<OpenAiMessage>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) tools: Option<Vec<OpenAiTool>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) tool_choice: Option<&'static str>,
pub(crate) stream: bool,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) stream_options: Option<ChatStreamOptions>,
}
#[derive(Serialize)]
pub(crate) struct ChatStreamOptions {
pub(crate) include_usage: bool,
}
#[derive(Serialize, Deserialize)]
pub(crate) struct OpenAiMessage {
pub(crate) role: String,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) content: Option<serde_json::Value>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) tool_calls: Option<Vec<OpenAiToolCall>>,
#[serde(skip_serializing_if = "Option::is_none")]
pub(crate) tool_call_id: Option<String>,
}
#[derive(Serialize, Deserialize)]
pub(crate) struct OpenAiToolCall {
pub(crate) id: String,
#[serde(rename = "type")]
pub(crate) kind: String,
pub(crate) function: OpenAiFunctionCall,
}
#[derive(Serialize, Deserialize)]
pub(crate) struct OpenAiFunctionCall {
pub(crate) name: String,
pub(crate) arguments: String,
}
#[derive(Serialize)]
pub(crate) struct OpenAiTool {
#[serde(rename = "type")]
pub(crate) kind: &'static str,
pub(crate) function: OpenAiToolFunction,
}
#[derive(Serialize)]
pub(crate) struct OpenAiToolFunction {
pub(crate) name: String,
pub(crate) description: String,
pub(crate) parameters: serde_json::Value,
pub(crate) strict: bool,
}
#[derive(Deserialize)]
pub(crate) struct ChatResponse {
pub(crate) choices: Vec<Choice>,
}
#[derive(Deserialize)]
pub(crate) struct Choice {
pub(crate) message: ChatResponseMessage,
}
#[derive(Deserialize)]
pub(crate) struct ChatResponseMessage {
pub(crate) content: Option<String>,
pub(crate) tool_calls: Option<Vec<OpenAiToolCall>>,
}