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