use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct Batch {
#[serde(rename = "id")]
pub id: String,
#[serde(rename = "object")]
pub object: Object,
#[serde(rename = "endpoint")]
pub endpoint: String,
#[serde(rename = "model", skip_serializing_if = "Option::is_none")]
pub model: Option<String>,
#[serde(rename = "errors", skip_serializing_if = "Option::is_none")]
pub errors: Option<Box<models::BatchErrors>>,
#[serde(rename = "input_file_id")]
pub input_file_id: String,
#[serde(rename = "completion_window")]
pub completion_window: String,
#[serde(rename = "status")]
pub status: Status,
#[serde(rename = "output_file_id", skip_serializing_if = "Option::is_none")]
pub output_file_id: Option<String>,
#[serde(rename = "error_file_id", skip_serializing_if = "Option::is_none")]
pub error_file_id: Option<String>,
#[serde(rename = "created_at")]
pub created_at: i32,
#[serde(rename = "in_progress_at", skip_serializing_if = "Option::is_none")]
pub in_progress_at: Option<i32>,
#[serde(rename = "expires_at", skip_serializing_if = "Option::is_none")]
pub expires_at: Option<i32>,
#[serde(rename = "finalizing_at", skip_serializing_if = "Option::is_none")]
pub finalizing_at: Option<i32>,
#[serde(rename = "completed_at", skip_serializing_if = "Option::is_none")]
pub completed_at: Option<i32>,
#[serde(rename = "failed_at", skip_serializing_if = "Option::is_none")]
pub failed_at: Option<i32>,
#[serde(rename = "expired_at", skip_serializing_if = "Option::is_none")]
pub expired_at: Option<i32>,
#[serde(rename = "cancelling_at", skip_serializing_if = "Option::is_none")]
pub cancelling_at: Option<i32>,
#[serde(rename = "cancelled_at", skip_serializing_if = "Option::is_none")]
pub cancelled_at: Option<i32>,
#[serde(rename = "request_counts", skip_serializing_if = "Option::is_none")]
pub request_counts: Option<Box<models::BatchRequestCounts>>,
#[serde(rename = "usage", skip_serializing_if = "Option::is_none")]
pub usage: Option<Box<models::BatchUsage>>,
#[serde(
rename = "metadata",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub metadata: Option<Option<std::collections::HashMap<String, String>>>,
}
impl Batch {
pub fn new(
id: String,
object: Object,
endpoint: String,
input_file_id: String,
completion_window: String,
status: Status,
created_at: i32,
) -> Batch {
Batch {
id,
object,
endpoint,
model: None,
errors: None,
input_file_id,
completion_window,
status,
output_file_id: None,
error_file_id: None,
created_at,
in_progress_at: None,
expires_at: None,
finalizing_at: None,
completed_at: None,
failed_at: None,
expired_at: None,
cancelling_at: None,
cancelled_at: None,
request_counts: None,
usage: None,
metadata: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Object {
#[serde(rename = "batch")]
Batch,
}
impl Default for Object {
fn default() -> Object {
Self::Batch
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Status {
#[serde(rename = "validating")]
Validating,
#[serde(rename = "failed")]
Failed,
#[serde(rename = "in_progress")]
InProgress,
#[serde(rename = "finalizing")]
Finalizing,
#[serde(rename = "completed")]
Completed,
#[serde(rename = "expired")]
Expired,
#[serde(rename = "cancelling")]
Cancelling,
#[serde(rename = "cancelled")]
Cancelled,
}
impl Default for Status {
fn default() -> Status {
Self::Validating
}
}
impl std::fmt::Display for Batch {
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),
}
}
}