Skip to main content

langfuse_client_base/apis/
mod.rs

1use std::error;
2use std::fmt;
3
4#[derive(Debug, Clone)]
5pub struct ResponseContent<T> {
6    pub status: reqwest::StatusCode,
7    pub content: String,
8    pub entity: Option<T>,
9}
10
11#[derive(Debug)]
12pub enum Error<T> {
13    Reqwest(reqwest::Error),
14    ReqwestMiddleware(reqwest_middleware::Error),
15    Serde(serde_json::Error),
16    Io(std::io::Error),
17    ResponseError(ResponseContent<T>),
18}
19
20impl<T> fmt::Display for Error<T> {
21    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
22        let (module, e) = match self {
23            Error::Reqwest(e) => ("reqwest", e.to_string()),
24            Error::ReqwestMiddleware(e) => ("reqwest-middleware", e.to_string()),
25            Error::Serde(e) => ("serde", e.to_string()),
26            Error::Io(e) => ("IO", e.to_string()),
27            Error::ResponseError(e) => ("response", format!("status code {}", e.status)),
28        };
29        write!(f, "error in {}: {}", module, e)
30    }
31}
32
33impl<T: fmt::Debug> error::Error for Error<T> {
34    fn source(&self) -> Option<&(dyn error::Error + 'static)> {
35        Some(match self {
36            Error::Reqwest(e) => e,
37            Error::ReqwestMiddleware(e) => e,
38            Error::Serde(e) => e,
39            Error::Io(e) => e,
40            Error::ResponseError(_) => return None,
41        })
42    }
43}
44
45impl<T> From<reqwest::Error> for Error<T> {
46    fn from(e: reqwest::Error) -> Self {
47        Error::Reqwest(e)
48    }
49}
50
51impl<T> From<reqwest_middleware::Error> for Error<T> {
52    fn from(e: reqwest_middleware::Error) -> Self {
53        Error::ReqwestMiddleware(e)
54    }
55}
56
57impl<T> From<serde_json::Error> for Error<T> {
58    fn from(e: serde_json::Error) -> Self {
59        Error::Serde(e)
60    }
61}
62
63impl<T> From<std::io::Error> for Error<T> {
64    fn from(e: std::io::Error) -> Self {
65        Error::Io(e)
66    }
67}
68
69pub fn urlencode<T: AsRef<str>>(s: T) -> String {
70    ::url::form_urlencoded::byte_serialize(s.as_ref().as_bytes()).collect()
71}
72
73pub fn parse_deep_object(prefix: &str, value: &serde_json::Value) -> Vec<(String, String)> {
74    if let serde_json::Value::Object(object) = value {
75        let mut params = vec![];
76
77        for (key, value) in object {
78            match value {
79                serde_json::Value::Object(_) => params.append(&mut parse_deep_object(
80                    &format!("{}[{}]", prefix, key),
81                    value,
82                )),
83                serde_json::Value::Array(array) => {
84                    for (i, value) in array.iter().enumerate() {
85                        params.append(&mut parse_deep_object(
86                            &format!("{}[{}][{}]", prefix, key, i),
87                            value,
88                        ));
89                    }
90                }
91                serde_json::Value::String(s) => {
92                    params.push((format!("{}[{}]", prefix, key), s.clone()))
93                }
94                _ => params.push((format!("{}[{}]", prefix, key), value.to_string())),
95            }
96        }
97
98        return params;
99    }
100
101    // Non-object values (e.g. array items) are emitted as a plain key/value pair.
102    let value = match value {
103        serde_json::Value::String(s) => s.clone(),
104        _ => value.to_string(),
105    };
106    vec![(prefix.to_string(), value)]
107}
108
109/// Internal use only
110/// A content type supported by this client.
111#[allow(dead_code)]
112enum ContentType {
113    Json,
114    Text,
115    Unsupported(String),
116}
117
118impl From<&str> for ContentType {
119    fn from(content_type: &str) -> Self {
120        if content_type.starts_with("application") && content_type.contains("json") {
121            return Self::Json;
122        } else if content_type.starts_with("text/plain") {
123            return Self::Text;
124        } else {
125            return Self::Unsupported(content_type.to_string());
126        }
127    }
128}
129
130pub mod annotation_queues_api;
131pub mod blob_storage_integrations_api;
132pub mod comments_api;
133pub mod dataset_items_api;
134pub mod dataset_run_items_api;
135pub mod datasets_api;
136pub mod evaluation_rules_api;
137pub mod evaluators_api;
138pub mod experiments_api;
139pub mod feedback_api;
140pub mod health_api;
141pub mod ingestion_api;
142pub mod legacy_metrics_v1_api;
143pub mod legacy_observations_v1_api;
144pub mod legacy_score_v1_api;
145pub mod llm_connections_api;
146pub mod media_api;
147pub mod metrics_api;
148pub mod models_api;
149pub mod observations_api;
150pub mod opentelemetry_api;
151pub mod organizations_api;
152pub mod projects_api;
153pub mod prompt_version_api;
154pub mod prompts_api;
155pub mod scim_api;
156pub mod score_configs_api;
157pub mod scores_api;
158pub mod scores_v3_api;
159pub mod sessions_api;
160pub mod trace_api;
161pub mod unstable_dashboard_widgets_api;
162pub mod unstable_dashboards_api;
163pub mod unstable_skills_api;
164
165pub mod configuration;