use http::StatusCode;
use serde::{Deserialize, Serialize};
use crate::gemini::count_tokens::types::GeminiModalityTokenCount;
use crate::gemini::types::{GeminiApiErrorResponse, GeminiResponseHeaders};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct ResponseBody {
#[serde(rename = "totalTokens")]
pub total_tokens: u64,
#[serde(
rename = "cachedContentTokenCount",
default,
skip_serializing_if = "Option::is_none"
)]
pub cached_content_token_count: Option<u64>,
#[serde(
rename = "promptTokensDetails",
default,
skip_serializing_if = "Option::is_none"
)]
pub prompt_tokens_details: Option<Vec<GeminiModalityTokenCount>>,
#[serde(
rename = "cacheTokensDetails",
default,
skip_serializing_if = "Option::is_none"
)]
pub cache_tokens_details: Option<Vec<GeminiModalityTokenCount>>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(untagged)]
pub enum GeminiCountTokensResponse {
Success {
#[serde(with = "crate::gemini::types::status_code_serde")]
stats_code: StatusCode,
headers: GeminiResponseHeaders,
body: ResponseBody,
},
Error {
#[serde(with = "crate::gemini::types::status_code_serde")]
stats_code: StatusCode,
headers: GeminiResponseHeaders,
body: GeminiApiErrorResponse,
},
}