Skip to main content

langfuse_client_base/models/
dataset_item.rs

1/*
2 * langfuse
3 *
4 * ## Authentication  Authenticate with the API using [Basic Auth](https://en.wikipedia.org/wiki/Basic_access_authentication), get API keys in the project settings:  - username: Langfuse Public Key - password: Langfuse Secret Key  ## Exports  - OpenAPI spec: https://cloud.langfuse.com/generated/api/openapi.yml
5 *
6 * The version of the OpenAPI document:
7 *
8 * Generated by: https://openapi-generator.tech
9 */
10
11use 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    /// Input data for the dataset item
21    #[serde(rename = "input", deserialize_with = "Option::deserialize")]
22    pub input: Option<serde_json::Value>,
23    /// Expected output for the dataset item
24    #[serde(rename = "expectedOutput", deserialize_with = "Option::deserialize")]
25    pub expected_output: Option<serde_json::Value>,
26    /// Metadata associated with the dataset item
27    #[serde(rename = "metadata", deserialize_with = "Option::deserialize")]
28    pub metadata: Option<serde_json::Value>,
29    /// The trace ID that sourced this dataset item
30    #[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    /// The observation ID that sourced this dataset item
38    #[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}