Skip to main content

bloop_sdk/
types.rs

1use serde::{Serialize, Deserialize};
2
3#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq)]
4#[serde(rename_all = "snake_case")]
5pub enum SpanType {
6    Generation,
7    Tool,
8    Retrieval,
9    Custom,
10}
11
12#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq)]
13#[serde(rename_all = "snake_case")]
14pub enum SpanStatus {
15    Ok,
16    Error,
17}
18
19#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq)]
20#[serde(rename_all = "snake_case")]
21pub enum TraceStatus {
22    Running,
23    Completed,
24    Error,
25}
26
27#[derive(Debug, Clone, Serialize, Deserialize)]
28pub struct SpanData {
29    pub id: String,
30    pub span_type: SpanType,
31    pub name: String,
32    pub input_tokens: i64,
33    pub output_tokens: i64,
34    pub cost: f64,
35    pub latency_ms: i64,
36    pub status: SpanStatus,
37    pub started_at: i64,
38    #[serde(skip_serializing_if = "Option::is_none")]
39    pub ended_at: Option<i64>,
40    #[serde(skip_serializing_if = "Option::is_none")]
41    pub model: Option<String>,
42    #[serde(skip_serializing_if = "Option::is_none")]
43    pub provider: Option<String>,
44    #[serde(skip_serializing_if = "Option::is_none")]
45    pub parent_span_id: Option<String>,
46    #[serde(skip_serializing_if = "Option::is_none")]
47    pub time_to_first_token_ms: Option<i64>,
48    #[serde(skip_serializing_if = "Option::is_none")]
49    pub error_message: Option<String>,
50    #[serde(skip_serializing_if = "Option::is_none")]
51    pub input: Option<String>,
52    #[serde(skip_serializing_if = "Option::is_none")]
53    pub output: Option<String>,
54    #[serde(skip_serializing_if = "Option::is_none")]
55    pub metadata: Option<serde_json::Value>,
56}
57
58#[derive(Debug, Clone, Serialize, Deserialize)]
59pub struct TraceData {
60    pub id: String,
61    pub name: String,
62    pub status: TraceStatus,
63    pub started_at: i64,
64    #[serde(skip_serializing_if = "Option::is_none")]
65    pub ended_at: Option<i64>,
66    pub spans: Vec<SpanData>,
67    #[serde(skip_serializing_if = "Option::is_none")]
68    pub session_id: Option<String>,
69    #[serde(skip_serializing_if = "Option::is_none")]
70    pub user_id: Option<String>,
71    #[serde(skip_serializing_if = "Option::is_none")]
72    pub input: Option<String>,
73    #[serde(skip_serializing_if = "Option::is_none")]
74    pub output: Option<String>,
75    #[serde(skip_serializing_if = "Option::is_none")]
76    pub metadata: Option<serde_json::Value>,
77    #[serde(skip_serializing_if = "Option::is_none")]
78    pub prompt_name: Option<String>,
79    #[serde(skip_serializing_if = "Option::is_none")]
80    pub prompt_version: Option<String>,
81}