use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct CreateChatCompletionResponse {
#[serde(rename = "id")]
pub id: String,
#[serde(rename = "choices")]
pub choices: Vec<models::CreateChatCompletionResponseChoicesInner>,
#[serde(rename = "created")]
pub created: i32,
#[serde(rename = "model")]
pub model: String,
#[serde(
rename = "service_tier",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub service_tier: Option<Option<models::ServiceTier>>,
#[serde(rename = "system_fingerprint", skip_serializing_if = "Option::is_none")]
pub system_fingerprint: Option<String>,
#[serde(rename = "object")]
pub object: Object,
#[serde(rename = "usage", skip_serializing_if = "Option::is_none")]
pub usage: Option<Box<models::CompletionUsage>>,
}
impl CreateChatCompletionResponse {
pub fn new(
id: String,
choices: Vec<models::CreateChatCompletionResponseChoicesInner>,
created: i32,
model: String,
object: Object,
) -> CreateChatCompletionResponse {
CreateChatCompletionResponse {
id,
choices,
created,
model,
service_tier: None,
system_fingerprint: None,
object,
usage: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Object {
#[serde(rename = "chat.completion")]
ChatCompletion,
}
impl Default for Object {
fn default() -> Object {
Self::ChatCompletion
}
}
impl std::fmt::Display for CreateChatCompletionResponse {
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),
}
}
}