Skip to main content

gproxy_protocol/protocol/gemini/
count_tokens.rs

1use std::collections::BTreeMap;
2
3use serde::{Deserialize, Serialize};
4
5use super::common::{Content, ExtraFields, ModalityTokenCount};
6use super::generate_content::GenerateContentRequest;
7
8#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
9#[serde(rename_all = "camelCase")]
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 = "BTreeMap::is_empty", flatten)]
18    pub extra: ExtraFields,
19}
20
21#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
22#[serde(rename_all = "camelCase")]
23pub struct CountTokensResponse {
24    #[serde(skip_serializing_if = "Option::is_none")]
25    pub total_tokens: Option<i32>,
26    #[serde(skip_serializing_if = "Option::is_none")]
27    pub cached_content_token_count: Option<i32>,
28    #[serde(default, skip_serializing_if = "Vec::is_empty")]
29    pub prompt_tokens_details: Vec<ModalityTokenCount>,
30    #[serde(default, skip_serializing_if = "Vec::is_empty")]
31    pub cache_tokens_details: Vec<ModalityTokenCount>,
32    #[serde(default, skip_serializing_if = "BTreeMap::is_empty", flatten)]
33    pub extra: ExtraFields,
34}