use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct Observation {
#[serde(rename = "id")]
pub id: String,
#[serde(rename = "traceId", deserialize_with = "Option::deserialize")]
pub trace_id: Option<String>,
#[serde(rename = "type")]
pub r#type: String,
#[serde(rename = "name", deserialize_with = "Option::deserialize")]
pub name: Option<String>,
#[serde(rename = "startTime")]
pub start_time: chrono::DateTime<chrono::FixedOffset>,
#[serde(rename = "endTime", deserialize_with = "Option::deserialize")]
pub end_time: Option<chrono::DateTime<chrono::FixedOffset>>,
#[serde(
rename = "completionStartTime",
deserialize_with = "Option::deserialize"
)]
pub completion_start_time: Option<chrono::DateTime<chrono::FixedOffset>>,
#[serde(rename = "model", deserialize_with = "Option::deserialize")]
pub model: Option<String>,
#[serde(rename = "modelParameters", deserialize_with = "Option::deserialize")]
pub model_parameters: Option<serde_json::Value>,
#[serde(rename = "input", deserialize_with = "Option::deserialize")]
pub input: Option<serde_json::Value>,
#[serde(rename = "version", deserialize_with = "Option::deserialize")]
pub version: Option<String>,
#[serde(rename = "metadata", deserialize_with = "Option::deserialize")]
pub metadata: Option<serde_json::Value>,
#[serde(rename = "output", deserialize_with = "Option::deserialize")]
pub output: Option<serde_json::Value>,
#[serde(rename = "usage")]
pub usage: Box<models::Usage>,
#[serde(rename = "level")]
pub level: models::ObservationLevel,
#[serde(rename = "statusMessage", deserialize_with = "Option::deserialize")]
pub status_message: Option<String>,
#[serde(
rename = "parentObservationId",
deserialize_with = "Option::deserialize"
)]
pub parent_observation_id: Option<String>,
#[serde(rename = "promptId", deserialize_with = "Option::deserialize")]
pub prompt_id: Option<String>,
#[serde(rename = "usageDetails")]
pub usage_details: std::collections::HashMap<String, i32>,
#[serde(rename = "costDetails")]
pub cost_details: std::collections::HashMap<String, f64>,
#[serde(rename = "environment")]
pub environment: String,
}
impl Observation {
pub fn new(
id: String,
trace_id: Option<String>,
r#type: String,
name: Option<String>,
start_time: chrono::DateTime<chrono::FixedOffset>,
end_time: Option<chrono::DateTime<chrono::FixedOffset>>,
completion_start_time: Option<chrono::DateTime<chrono::FixedOffset>>,
model: Option<String>,
model_parameters: Option<serde_json::Value>,
input: Option<serde_json::Value>,
version: Option<String>,
metadata: Option<serde_json::Value>,
output: Option<serde_json::Value>,
usage: models::Usage,
level: models::ObservationLevel,
status_message: Option<String>,
parent_observation_id: Option<String>,
prompt_id: Option<String>,
usage_details: std::collections::HashMap<String, i32>,
cost_details: std::collections::HashMap<String, f64>,
environment: String,
) -> Observation {
Observation {
id,
trace_id,
r#type,
name,
start_time,
end_time,
completion_start_time,
model,
model_parameters,
input,
version,
metadata,
output,
usage: Box::new(usage),
level,
status_message,
parent_observation_id,
prompt_id,
usage_details,
cost_details,
environment,
}
}
}