langfuse_rs/apis/
dataset_items_api.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 - Postman collection: https://cloud.langfuse.com/generated/postman/collection.json
5 *
6 * The version of the OpenAPI document:
7 *
8 * Generated by: https://openapi-generator.tech
9 */
10
11use super::{configuration, Error};
12use crate::{apis::ResponseContent, models};
13use reqwest;
14use serde::{Deserialize, Serialize};
15
16/// struct for typed errors of method [`dataset_items_create`]
17#[derive(Debug, Clone, Serialize, Deserialize)]
18#[serde(untagged)]
19pub enum DatasetItemsCreateError {
20	Status400(serde_json::Value),
21	Status401(serde_json::Value),
22	Status403(serde_json::Value),
23	Status404(serde_json::Value),
24	Status405(serde_json::Value),
25	UnknownValue(serde_json::Value),
26}
27
28/// struct for typed errors of method [`dataset_items_get`]
29#[derive(Debug, Clone, Serialize, Deserialize)]
30#[serde(untagged)]
31pub enum DatasetItemsGetError {
32	Status400(serde_json::Value),
33	Status401(serde_json::Value),
34	Status403(serde_json::Value),
35	Status404(serde_json::Value),
36	Status405(serde_json::Value),
37	UnknownValue(serde_json::Value),
38}
39
40/// struct for typed errors of method [`dataset_items_list`]
41#[derive(Debug, Clone, Serialize, Deserialize)]
42#[serde(untagged)]
43pub enum DatasetItemsListError {
44	Status400(serde_json::Value),
45	Status401(serde_json::Value),
46	Status403(serde_json::Value),
47	Status404(serde_json::Value),
48	Status405(serde_json::Value),
49	UnknownValue(serde_json::Value),
50}
51
52/// Create a dataset item
53pub async fn dataset_items_create(
54	configuration: &configuration::Configuration,
55	create_dataset_item_request: models::CreateDatasetItemRequest,
56) -> Result<models::DatasetItem, Error<DatasetItemsCreateError>> {
57	// add a prefix to parameters to efficiently prevent name collisions
58	let p_create_dataset_item_request = create_dataset_item_request;
59
60	let uri_str = format!("{}/api/public/dataset-items", configuration.base_path);
61	let mut req_builder = configuration.client.request(reqwest::Method::POST, &uri_str);
62
63	if let Some(ref user_agent) = configuration.user_agent {
64		req_builder = req_builder.header(reqwest::header::USER_AGENT, user_agent.clone());
65	}
66	if let Some(ref auth_conf) = configuration.basic_auth {
67		req_builder = req_builder.basic_auth(auth_conf.0.to_owned(), auth_conf.1.to_owned());
68	};
69	req_builder = req_builder.json(&p_create_dataset_item_request);
70
71	let req = req_builder.build()?;
72	let resp = configuration.client.execute(req).await?;
73
74	let status = resp.status();
75
76	if !status.is_client_error() && !status.is_server_error() {
77		let content = resp.text().await?;
78		serde_json::from_str(&content).map_err(Error::from)
79	} else {
80		let content = resp.text().await?;
81		let entity: Option<DatasetItemsCreateError> = serde_json::from_str(&content).ok();
82		Err(Error::ResponseError(ResponseContent { status, content, entity }))
83	}
84}
85
86/// Get a dataset item
87pub async fn dataset_items_get(configuration: &configuration::Configuration, id: &str) -> Result<models::DatasetItem, Error<DatasetItemsGetError>> {
88	// add a prefix to parameters to efficiently prevent name collisions
89	let p_id = id;
90
91	let uri_str = format!(
92		"{}/api/public/dataset-items/{id}",
93		configuration.base_path,
94		id = crate::apis::urlencode(p_id)
95	);
96	let mut req_builder = configuration.client.request(reqwest::Method::GET, &uri_str);
97
98	if let Some(ref user_agent) = configuration.user_agent {
99		req_builder = req_builder.header(reqwest::header::USER_AGENT, user_agent.clone());
100	}
101	if let Some(ref auth_conf) = configuration.basic_auth {
102		req_builder = req_builder.basic_auth(auth_conf.0.to_owned(), auth_conf.1.to_owned());
103	};
104
105	let req = req_builder.build()?;
106	let resp = configuration.client.execute(req).await?;
107
108	let status = resp.status();
109
110	if !status.is_client_error() && !status.is_server_error() {
111		let content = resp.text().await?;
112		serde_json::from_str(&content).map_err(Error::from)
113	} else {
114		let content = resp.text().await?;
115		let entity: Option<DatasetItemsGetError> = serde_json::from_str(&content).ok();
116		Err(Error::ResponseError(ResponseContent { status, content, entity }))
117	}
118}
119
120/// Get dataset items
121pub async fn dataset_items_list(
122	configuration: &configuration::Configuration,
123	dataset_name: Option<&str>,
124	source_trace_id: Option<&str>,
125	source_observation_id: Option<&str>,
126	page: Option<i32>,
127	limit: Option<i32>,
128) -> Result<models::PaginatedDatasetItems, Error<DatasetItemsListError>> {
129	// add a prefix to parameters to efficiently prevent name collisions
130	let p_dataset_name = dataset_name;
131	let p_source_trace_id = source_trace_id;
132	let p_source_observation_id = source_observation_id;
133	let p_page = page;
134	let p_limit = limit;
135
136	let uri_str = format!("{}/api/public/dataset-items", configuration.base_path);
137	let mut req_builder = configuration.client.request(reqwest::Method::GET, &uri_str);
138
139	if let Some(ref param_value) = p_dataset_name {
140		req_builder = req_builder.query(&[("datasetName", &param_value.to_string())]);
141	}
142	if let Some(ref param_value) = p_source_trace_id {
143		req_builder = req_builder.query(&[("sourceTraceId", &param_value.to_string())]);
144	}
145	if let Some(ref param_value) = p_source_observation_id {
146		req_builder = req_builder.query(&[("sourceObservationId", &param_value.to_string())]);
147	}
148	if let Some(ref param_value) = p_page {
149		req_builder = req_builder.query(&[("page", &param_value.to_string())]);
150	}
151	if let Some(ref param_value) = p_limit {
152		req_builder = req_builder.query(&[("limit", &param_value.to_string())]);
153	}
154	if let Some(ref user_agent) = configuration.user_agent {
155		req_builder = req_builder.header(reqwest::header::USER_AGENT, user_agent.clone());
156	}
157	if let Some(ref auth_conf) = configuration.basic_auth {
158		req_builder = req_builder.basic_auth(auth_conf.0.to_owned(), auth_conf.1.to_owned());
159	};
160
161	let req = req_builder.build()?;
162	let resp = configuration.client.execute(req).await?;
163
164	let status = resp.status();
165
166	if !status.is_client_error() && !status.is_server_error() {
167		let content = resp.text().await?;
168		serde_json::from_str(&content).map_err(Error::from)
169	} else {
170		let content = resp.text().await?;
171		let entity: Option<DatasetItemsListError> = serde_json::from_str(&content).ok();
172		Err(Error::ResponseError(ResponseContent { status, content, entity }))
173	}
174}