use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
pub struct ExperimentItem {
#[serde(rename = "id")]
pub id: String,
#[serde(rename = "traceId")]
pub trace_id: 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 = "level")]
pub level: models::ObservationLevel,
#[serde(rename = "environment")]
pub environment: String,
#[serde(rename = "experimentId")]
pub experiment_id: String,
#[serde(rename = "experimentName")]
pub experiment_name: String,
#[serde(rename = "experimentItemId")]
pub experiment_item_id: String,
#[serde(
rename = "experimentDatasetId",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub experiment_dataset_id: Option<Option<String>>,
#[serde(
rename = "experimentItemVersion",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub experiment_item_version: Option<Option<chrono::DateTime<chrono::FixedOffset>>>,
#[serde(
rename = "input",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub input: Option<Option<serde_json::Value>>,
#[serde(
rename = "output",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub output: Option<Option<serde_json::Value>>,
#[serde(
rename = "expectedOutput",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub expected_output: Option<Option<serde_json::Value>>,
#[serde(
rename = "metadata",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub metadata: Option<Option<std::collections::HashMap<String, serde_json::Value>>>,
#[serde(
rename = "experimentItemMetadata",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub experiment_item_metadata:
Option<Option<std::collections::HashMap<String, serde_json::Value>>>,
#[serde(
rename = "experimentMetadata",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub experiment_metadata: Option<Option<std::collections::HashMap<String, serde_json::Value>>>,
#[serde(
rename = "experimentDescription",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub experiment_description: Option<Option<String>>,
#[serde(
rename = "scores",
default,
with = "::serde_with::rust::double_option",
skip_serializing_if = "Option::is_none"
)]
pub scores: Option<Option<Vec<models::ScoreV3>>>,
}
impl ExperimentItem {
pub fn new(
id: String,
trace_id: String,
start_time: chrono::DateTime<chrono::FixedOffset>,
end_time: Option<chrono::DateTime<chrono::FixedOffset>>,
level: models::ObservationLevel,
environment: String,
experiment_id: String,
experiment_name: String,
experiment_item_id: String,
) -> ExperimentItem {
ExperimentItem {
id,
trace_id,
start_time,
end_time,
level,
environment,
experiment_id,
experiment_name,
experiment_item_id,
experiment_dataset_id: None,
experiment_item_version: None,
input: None,
output: None,
expected_output: None,
metadata: None,
experiment_item_metadata: None,
experiment_metadata: None,
experiment_description: None,
scores: None,
}
}
}