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}