use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct ComputerToolCallOutput {
#[serde(rename = "type")]
pub r#type: Type,
#[serde(rename = "id", skip_serializing_if = "Option::is_none")]
pub id: Option<String>,
#[serde(rename = "call_id")]
pub call_id: String,
#[serde(
rename = "acknowledged_safety_checks",
skip_serializing_if = "Option::is_none"
)]
pub acknowledged_safety_checks: Option<Vec<models::ComputerCallSafetyCheckParam>>,
#[serde(rename = "output")]
pub output: Box<models::ComputerScreenshotImage>,
#[serde(rename = "status", skip_serializing_if = "Option::is_none")]
pub status: Option<Status>,
}
impl ComputerToolCallOutput {
pub fn new(
r#type: Type,
call_id: String,
output: models::ComputerScreenshotImage,
) -> ComputerToolCallOutput {
ComputerToolCallOutput {
r#type,
id: None,
call_id,
acknowledged_safety_checks: None,
output: Box::new(output),
status: None,
}
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Type {
#[serde(rename = "computer_call_output")]
ComputerCallOutput,
}
impl Default for Type {
fn default() -> Type {
Self::ComputerCallOutput
}
}
#[derive(Clone, Copy, Debug, Eq, PartialEq, Ord, PartialOrd, Hash, Serialize, Deserialize)]
pub enum Status {
#[serde(rename = "in_progress")]
InProgress,
#[serde(rename = "completed")]
Completed,
#[serde(rename = "incomplete")]
Incomplete,
}
impl Default for Status {
fn default() -> Status {
Self::InProgress
}
}
impl std::fmt::Display for ComputerToolCallOutput {
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),
}
}
}