use std::collections::BTreeMap;
use serde::{Deserialize, Serialize};
use crate::gemini::generate_content::types::{
GeminiContent, GeminiGenerationConfig, GeminiSafetySetting, GeminiTool, GeminiToolConfig,
HttpMethod,
};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct GeminiGenerateContentRequest {
pub method: HttpMethod,
pub path: PathParameters,
pub query: QueryParameters,
pub headers: RequestHeaders,
pub body: RequestBody,
}
impl Default for GeminiGenerateContentRequest {
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, Default)]
pub struct RequestBody {
pub contents: Vec<GeminiContent>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub tools: Option<Vec<GeminiTool>>,
#[serde(
rename = "toolConfig",
default,
skip_serializing_if = "Option::is_none"
)]
pub tool_config: Option<GeminiToolConfig>,
#[serde(
rename = "safetySettings",
default,
skip_serializing_if = "Option::is_none"
)]
pub safety_settings: Option<Vec<GeminiSafetySetting>>,
#[serde(
rename = "systemInstruction",
default,
skip_serializing_if = "Option::is_none"
)]
pub system_instruction: Option<GeminiContent>,
#[serde(
rename = "generationConfig",
default,
skip_serializing_if = "Option::is_none"
)]
pub generation_config: Option<GeminiGenerationConfig>,
#[serde(
rename = "cachedContent",
default,
skip_serializing_if = "Option::is_none"
)]
pub cached_content: Option<String>,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub store: Option<bool>,
}