use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct UsageCompletionsResult {
#[serde(rename = "object")]
pub object: Object,
#[serde(rename = "input_tokens")]
pub input_tokens: i32,
#[serde(
rename = "input_cached_tokens",
skip_serializing_if = "Option::is_none"
)]
pub input_cached_tokens: Option<i32>,
#[serde(rename = "output_tokens")]
pub output_tokens: i32,
#[serde(rename = "input_audio_tokens", skip_serializing_if = "Option::is_none")]
pub input_audio_tokens: Option<i32>,
#[serde(
rename = "output_audio_tokens",
skip_serializing_if = "Option::is_none"
)]
pub output_audio_tokens: Option<i32>,
#[serde(rename = "num_model_requests")]
pub num_model_requests: i32,
#[serde(
rename = "project_id",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub project_id: Option<Option<String>>,
#[serde(
rename = "user_id",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub user_id: Option<Option<String>>,
#[serde(
rename = "api_key_id",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub api_key_id: Option<Option<String>>,
#[serde(rename = "model", skip_serializing_if = "Option::is_none")]
pub model: Option<String>,
#[serde(
rename = "batch",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub batch: Option<Option<bool>>,
#[serde(
rename = "service_tier",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub service_tier: Option<Option<String>>,
}
impl UsageCompletionsResult {
pub fn new(
object: Object,
input_tokens: i32,
output_tokens: i32,
num_model_requests: i32,
) -> UsageCompletionsResult {
UsageCompletionsResult {
object,
input_tokens,
input_cached_tokens: None,
output_tokens,
input_audio_tokens: None,
output_audio_tokens: None,
num_model_requests,
project_id: None,
user_id: None,
api_key_id: None,
model: None,
batch: None,
service_tier: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Object {
#[serde(rename = "organization.usage.completions.result")]
OrganizationUsageCompletionsResult,
}
impl Default for Object {
fn default() -> Object {
Self::OrganizationUsageCompletionsResult
}
}
impl std::fmt::Display for UsageCompletionsResult {
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),
}
}
}