gproxy_protocol/gemini/
count_tokens.rs

1use serde::{Deserialize, Serialize};
2
3use super::common::{Content, ModalityTokenCount};
4use super::generate_content::GenerateContentRequest;
5
6#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
7#[serde(rename_all = "camelCase")]
8#[derive(gproxy_protocol_macros::WireBuilder)]
9#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
10pub struct CountTokensRequest {
11    #[serde(skip_serializing_if = "Option::is_none")]
12    pub model: Option<String>,
13    #[serde(default, skip_serializing_if = "Vec::is_empty")]
14    pub contents: Vec<Content>,
15    #[serde(skip_serializing_if = "Option::is_none")]
16    pub generate_content_request: Option<GenerateContentRequest>,
17    #[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
18    pub rest: serde_json::Map<String, serde_json::Value>,
19}
20
21#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
22#[serde(rename_all = "camelCase")]
23#[derive(gproxy_protocol_macros::WireBuilder)]
24#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
25pub struct CountTokensResponse {
26    #[serde(skip_serializing_if = "Option::is_none")]
27    pub total_tokens: Option<i32>,
28    #[serde(skip_serializing_if = "Option::is_none")]
29    pub cached_content_token_count: Option<i32>,
30    #[serde(default, skip_serializing_if = "Vec::is_empty")]
31    pub prompt_tokens_details: Vec<ModalityTokenCount>,
32    #[serde(default, skip_serializing_if = "Vec::is_empty")]
33    pub cache_tokens_details: Vec<ModalityTokenCount>,
34    #[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
35    pub rest: serde_json::Map<String, serde_json::Value>,
36}
gproxy_protocol/gemini/count_tokens.rs

gproxy_protocol/gemini/
count_tokens.rs
