langfuse_client_base/models/
dataset_item.rs1use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize, bon::Builder)]
15pub struct DatasetItem {
16 #[serde(rename = "id")]
17 pub id: String,
18 #[serde(rename = "status")]
19 pub status: models::DatasetStatus,
20 #[serde(rename = "input", deserialize_with = "Option::deserialize")]
22 pub input: Option<serde_json::Value>,
23 #[serde(rename = "expectedOutput", deserialize_with = "Option::deserialize")]
25 pub expected_output: Option<serde_json::Value>,
26 #[serde(rename = "metadata", deserialize_with = "Option::deserialize")]
28 pub metadata: Option<serde_json::Value>,
29 #[serde(
31 rename = "sourceTraceId",
32 default,
33 with = "::serde_with::rust::double_option",
34 skip_serializing_if = "Option::is_none"
35 )]
36 pub source_trace_id: Option<Option<String>>,
37 #[serde(
39 rename = "sourceObservationId",
40 default,
41 with = "::serde_with::rust::double_option",
42 skip_serializing_if = "Option::is_none"
43 )]
44 pub source_observation_id: Option<Option<String>>,
45 #[serde(rename = "datasetId")]
46 pub dataset_id: String,
47 #[serde(rename = "datasetName")]
48 pub dataset_name: String,
49 #[serde(rename = "createdAt")]
50 pub created_at: chrono::DateTime<chrono::FixedOffset>,
51 #[serde(rename = "updatedAt")]
52 pub updated_at: chrono::DateTime<chrono::FixedOffset>,
53}
54
55impl DatasetItem {
56 pub fn new(
57 id: String,
58 status: models::DatasetStatus,
59 input: Option<serde_json::Value>,
60 expected_output: Option<serde_json::Value>,
61 metadata: Option<serde_json::Value>,
62 dataset_id: String,
63 dataset_name: String,
64 created_at: chrono::DateTime<chrono::FixedOffset>,
65 updated_at: chrono::DateTime<chrono::FixedOffset>,
66 ) -> DatasetItem {
67 DatasetItem {
68 id,
69 status,
70 input,
71 expected_output,
72 metadata,
73 source_trace_id: None,
74 source_observation_id: None,
75 dataset_id,
76 dataset_name,
77 created_at,
78 updated_at,
79 }
80 }
81}