use std::collections::BTreeMap;
use serde::{Deserialize, Serialize};
use crate::gemini::embeddings::types::{GeminiContent, GeminiTaskType, HttpMethod};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct GeminiEmbedContentRequest {
pub method: HttpMethod,
pub path: PathParameters,
pub query: QueryParameters,
pub headers: RequestHeaders,
pub body: RequestBody,
}
impl Default for GeminiEmbedContentRequest {
fn default() -> Self {
Self {
method: HttpMethod::Post,
path: PathParameters::default(),
query: QueryParameters::default(),
headers: RequestHeaders::default(),
body: RequestBody::default(),
}
}
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct PathParameters {
pub model: String,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct QueryParameters {}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct RequestHeaders {
#[serde(flatten, default, skip_serializing_if = "BTreeMap::is_empty")]
pub extra: BTreeMap<String, String>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct RequestBody {
pub content: GeminiContent,
#[serde(rename = "taskType", default, skip_serializing_if = "Option::is_none")]
pub task_type: Option<GeminiTaskType>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub title: Option<String>,
#[serde(
rename = "outputDimensionality",
default,
skip_serializing_if = "Option::is_none"
)]
pub output_dimensionality: Option<u32>,
}
impl Default for RequestBody {
fn default() -> Self {
Self {
content: GeminiContent {
parts: Vec::new(),
role: None,
},
task_type: None,
title: None,
output_dimensionality: None,
}
}
}