langfuse_client_base/models/
open_ai_response_usage_schema.rs1use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
16pub struct OpenAiResponseUsageSchema {
17 #[serde(rename = "input_tokens")]
18 pub input_tokens: i32,
19 #[serde(rename = "output_tokens")]
20 pub output_tokens: i32,
21 #[serde(rename = "total_tokens")]
22 pub total_tokens: i32,
23 #[serde(
24 rename = "input_tokens_details",
25 default,
26 with = "::serde_with::rust::double_option",
27 skip_serializing_if = "Option::is_none"
28 )]
29 pub input_tokens_details: Option<Option<std::collections::HashMap<String, i32>>>,
30 #[serde(
31 rename = "output_tokens_details",
32 default,
33 with = "::serde_with::rust::double_option",
34 skip_serializing_if = "Option::is_none"
35 )]
36 pub output_tokens_details: Option<Option<std::collections::HashMap<String, i32>>>,
37}
38
39impl OpenAiResponseUsageSchema {
40 pub fn new(
42 input_tokens: i32,
43 output_tokens: i32,
44 total_tokens: i32,
45 ) -> OpenAiResponseUsageSchema {
46 OpenAiResponseUsageSchema {
47 input_tokens,
48 output_tokens,
49 total_tokens,
50 input_tokens_details: None,
51 output_tokens_details: None,
52 }
53 }
54}