langfuse_client/models/
open_ai_completion_usage_schema.rs1use crate::models;
12use serde::{Deserialize, Serialize};
13#[cfg(not(feature = "ahash"))]
14use std::collections::HashMap;
15#[cfg(feature = "ahash")]
16use ahash::HashMap;
17
18#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
20#[cfg_attr(feature="bon", derive(bon::Builder))]
21pub struct OpenAiCompletionUsageSchema {
22 #[serde(rename = "prompt_tokens")]
23 pub prompt_tokens: i32,
24 #[serde(rename = "completion_tokens")]
25 pub completion_tokens: i32,
26 #[serde(rename = "total_tokens")]
27 pub total_tokens: i32,
28 #[serde(rename = "prompt_tokens_details", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
29 pub prompt_tokens_details: Option<Option<HashMap<String, i32>>>,
30 #[serde(rename = "completion_tokens_details", default, with = "::serde_with::rust::double_option", skip_serializing_if = "Option::is_none")]
31 pub completion_tokens_details: Option<Option<HashMap<String, i32>>>,
32}
33
34impl OpenAiCompletionUsageSchema {
35 pub fn new(prompt_tokens: i32, completion_tokens: i32, total_tokens: i32) -> OpenAiCompletionUsageSchema {
37 OpenAiCompletionUsageSchema {
38 prompt_tokens,
39 completion_tokens,
40 total_tokens,
41 prompt_tokens_details: None,
42 completion_tokens_details: None,
43 }
44 }
45}
46