gproxy_protocol/gemini/count_tokens/
request.rs

1use std::collections::BTreeMap;
2
3use serde::{Deserialize, Serialize};
4
5use crate::gemini::count_tokens::types::{GeminiContent, GeminiGenerateContentRequest, HttpMethod};
6
7/// Request descriptor for Gemini `models.countTokens` endpoint.
8#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
9pub struct GeminiCountTokensRequest {
10    /// HTTP method.
11    pub method: HttpMethod,
12    /// Path parameters.
13    pub path: PathParameters,
14    /// Query parameters.
15    pub query: QueryParameters,
16    /// Request headers.
17    pub headers: RequestHeaders,
18    /// Request body.
19    pub body: RequestBody,
20}
21
22impl Default for GeminiCountTokensRequest {
23    fn default() -> Self {
24        Self {
25            method: HttpMethod::Post,
26            path: PathParameters::default(),
27            query: QueryParameters::default(),
28            headers: RequestHeaders::default(),
29            body: RequestBody::default(),
30        }
31    }
32}
33
34#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
35pub struct PathParameters {
36    /// Resource name in form `models/{model}`.
37    pub model: String,
38}
39
40/// Proxy-side request model does not carry query parameters.
41#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
42pub struct QueryParameters {}
43
44/// Proxy-side request model does not carry auth headers.
45#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
46pub struct RequestHeaders {
47    #[serde(flatten, default, skip_serializing_if = "BTreeMap::is_empty")]
48    pub extra: BTreeMap<String, String>,
49}
50
51#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
52pub struct RequestBody {
53    /// Prompt input content.
54    ///
55    /// This is ignored when `generate_content_request` is present.
56    #[serde(default, skip_serializing_if = "Option::is_none")]
57    pub contents: Option<Vec<GeminiContent>>,
58    /// Full generation request used for counting tokens.
59    ///
60    /// This field is mutually exclusive with `contents`.
61    #[serde(
62        rename = "generateContentRequest",
63        default,
64        skip_serializing_if = "Option::is_none"
65    )]
66    pub generate_content_request: Option<GeminiGenerateContentRequest>,
67}
gproxy_protocol/gemini/count_tokens/request.rs

gproxy_protocol/gemini/count_tokens/
request.rs
