1use crate::models;
12use serde::{Deserialize, Serialize};
13#[cfg(not(feature = "ahash"))]
14use std::collections::HashMap;
15#[cfg(feature = "ahash")]
16use ahash::HashMap;
17
18#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
19#[cfg_attr(feature="bon", derive(bon::Builder))]
20pub struct Observation {
21 #[serde(rename = "id")]
23 pub id: String,
24 #[serde(rename = "traceId", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
26 pub trace_id: Option<Option<String>>,
27 #[serde(rename = "type")]
29 pub r#type: String,
30 #[serde(rename = "name", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
32 pub name: Option<Option<String>>,
33 #[serde(rename = "startTime")]
35 pub start_time: String,
36 #[serde(rename = "endTime", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
38 pub end_time: Option<Option<String>>,
39 #[serde(rename = "completionStartTime", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
41 pub completion_start_time: Option<Option<String>>,
42 #[serde(rename = "model", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
44 pub model: Option<Option<String>>,
45 #[serde(rename = "modelParameters", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
47 pub model_parameters: Option<Option<HashMap<String, models::MapValue>>>,
48 #[serde(rename = "input", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
50 pub input: Option<Option<serde_json::Value>>,
51 #[serde(rename = "version", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
53 pub version: Option<Option<String>>,
54 #[serde(rename = "metadata", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
56 pub metadata: Option<Option<serde_json::Value>>,
57 #[serde(rename = "output", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
59 pub output: Option<Option<serde_json::Value>>,
60 #[serde(rename = "usage", skip_serializing_if = "Option::is_none")]
61 pub usage: Option<models::Usage>,
62 #[serde(rename = "level")]
63 pub level: models::ObservationLevel,
64 #[serde(rename = "statusMessage", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
66 pub status_message: Option<Option<String>>,
67 #[serde(rename = "parentObservationId", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
69 pub parent_observation_id: Option<Option<String>>,
70 #[serde(rename = "promptId", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
72 pub prompt_id: Option<Option<String>>,
73 #[serde(rename = "usageDetails", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
75 pub usage_details: Option<Option<HashMap<String, i32>>>,
76 #[serde(rename = "costDetails", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
78 pub cost_details: Option<Option<HashMap<String, f64>>>,
79 #[serde(rename = "environment", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
81 pub environment: Option<Option<String>>,
82}
83
84impl Observation {
85 pub fn new(id: String, r#type: String, start_time: String, level: models::ObservationLevel) -> Observation {
86 Observation {
87 id,
88 trace_id: None,
89 r#type,
90 name: None,
91 start_time,
92 end_time: None,
93 completion_start_time: None,
94 model: None,
95 model_parameters: None,
96 input: None,
97 version: None,
98 metadata: None,
99 output: None,
100 usage: None,
101 level,
102 status_message: None,
103 parent_observation_id: None,
104 prompt_id: None,
105 usage_details: None,
106 cost_details: None,
107 environment: None,
108 }
109 }
110}
111