gproxy_protocol/gemini/embeddings/
request.rs

1use std::collections::BTreeMap;
2
3use serde::{Deserialize, Serialize};
4
5use crate::gemini::embeddings::types::{GeminiContent, GeminiTaskType, HttpMethod};
6
7/// Request descriptor for Gemini `models.embedContent` endpoint.
8#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
9pub struct GeminiEmbedContentRequest {
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 GeminiEmbedContentRequest {
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/// Embed request body.
52#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
53pub struct RequestBody {
54    /// The content to embed. Only `parts.text` is counted.
55    pub content: GeminiContent,
56    /// Optional embedding task type.
57    #[serde(rename = "taskType", default, skip_serializing_if = "Option::is_none")]
58    pub task_type: Option<GeminiTaskType>,
59    /// Optional document title, used with `RETRIEVAL_DOCUMENT`.
60    #[serde(default, skip_serializing_if = "Option::is_none")]
61    pub title: Option<String>,
62    /// Optional reduced embedding dimension.
63    #[serde(
64        rename = "outputDimensionality",
65        default,
66        skip_serializing_if = "Option::is_none"
67    )]
68    pub output_dimensionality: Option<u32>,
69}
70
71impl Default for RequestBody {
72    fn default() -> Self {
73        Self {
74            content: GeminiContent {
75                parts: Vec::new(),
76                role: None,
77            },
78            task_type: None,
79            title: None,
80            output_dimensionality: None,
81        }
82    }
83}
gproxy_protocol/gemini/embeddings/request.rs

gproxy_protocol/gemini/embeddings/
request.rs
