use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct RunObject {
#[serde(rename = "id")]
pub id: String,
#[serde(rename = "object")]
pub object: Object,
#[serde(rename = "created_at")]
pub created_at: i32,
#[serde(rename = "thread_id")]
pub thread_id: String,
#[serde(rename = "assistant_id")]
pub assistant_id: String,
#[serde(rename = "status")]
pub status: Status,
#[serde(rename = "required_action")]
pub required_action: Option<Box<models::RunObjectRequiredAction>>,
#[serde(rename = "last_error")]
pub last_error: Option<Box<models::RunObjectLastError>>,
#[serde(rename = "expires_at")]
pub expires_at: i32,
#[serde(rename = "started_at")]
pub started_at: i32,
#[serde(rename = "cancelled_at")]
pub cancelled_at: i32,
#[serde(rename = "failed_at")]
pub failed_at: i32,
#[serde(rename = "completed_at")]
pub completed_at: i32,
#[serde(rename = "incomplete_details")]
pub incomplete_details: Option<Box<models::RunObjectIncompleteDetails>>,
#[serde(rename = "model")]
pub model: String,
#[serde(rename = "instructions")]
pub instructions: String,
#[serde(rename = "tools")]
pub tools: Vec<models::AssistantObjectToolsInner>,
#[serde(rename = "metadata", deserialize_with = "Option::deserialize")]
pub metadata: Option<std::collections::HashMap<String, String>>,
#[serde(rename = "usage", deserialize_with = "Option::deserialize")]
pub usage: Option<Box<models::RunCompletionUsage>>,
#[serde(rename = "temperature", skip_serializing_if = "Option::is_none")]
pub temperature: Option<f64>,
#[serde(rename = "top_p", skip_serializing_if = "Option::is_none")]
pub top_p: Option<f64>,
#[serde(rename = "max_prompt_tokens")]
pub max_prompt_tokens: i32,
#[serde(rename = "max_completion_tokens")]
pub max_completion_tokens: i32,
#[serde(rename = "truncation_strategy")]
pub truncation_strategy: Box<models::TruncationObject>,
#[serde(rename = "tool_choice")]
pub tool_choice: Box<models::AssistantsApiToolChoiceOption>,
#[serde(rename = "parallel_tool_calls")]
pub parallel_tool_calls: bool,
#[serde(rename = "response_format")]
pub response_format: Box<models::AssistantsApiResponseFormatOption>,
}
impl RunObject {
pub fn new(
id: String,
object: Object,
created_at: i32,
thread_id: String,
assistant_id: String,
status: Status,
required_action: Option<models::RunObjectRequiredAction>,
last_error: Option<models::RunObjectLastError>,
expires_at: i32,
started_at: i32,
cancelled_at: i32,
failed_at: i32,
completed_at: i32,
incomplete_details: Option<models::RunObjectIncompleteDetails>,
model: String,
instructions: String,
tools: Vec<models::AssistantObjectToolsInner>,
metadata: Option<std::collections::HashMap<String, String>>,
usage: Option<models::RunCompletionUsage>,
max_prompt_tokens: i32,
max_completion_tokens: i32,
truncation_strategy: models::TruncationObject,
tool_choice: models::AssistantsApiToolChoiceOption,
parallel_tool_calls: bool,
response_format: models::AssistantsApiResponseFormatOption,
) -> RunObject {
RunObject {
id,
object,
created_at,
thread_id,
assistant_id,
status,
required_action: required_action.map(Box::new),
last_error: last_error.map(Box::new),
expires_at,
started_at,
cancelled_at,
failed_at,
completed_at,
incomplete_details: incomplete_details.map(Box::new),
model,
instructions,
tools,
metadata,
usage: usage.map(Box::new),
temperature: None,
top_p: None,
max_prompt_tokens,
max_completion_tokens,
truncation_strategy: Box::new(truncation_strategy),
tool_choice: Box::new(tool_choice),
parallel_tool_calls,
response_format: Box::new(response_format),
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Object {
#[serde(rename = "thread.run")]
ThreadRun,
}
impl Default for Object {
fn default() -> Object {
Self::ThreadRun
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Status {
#[serde(rename = "queued")]
Queued,
#[serde(rename = "in_progress")]
InProgress,
#[serde(rename = "requires_action")]
RequiresAction,
#[serde(rename = "cancelling")]
Cancelling,
#[serde(rename = "cancelled")]
Cancelled,
#[serde(rename = "failed")]
Failed,
#[serde(rename = "completed")]
Completed,
#[serde(rename = "incomplete")]
Incomplete,
#[serde(rename = "expired")]
Expired,
}
impl Default for Status {
fn default() -> Status {
Self::Queued
}
}
impl std::fmt::Display for RunObject {
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),
}
}
}