use std::collections::BTreeMap;
use serde::{Deserialize, Serialize};
use crate::gemini::live::types::{GeminiBidiGenerateContentClientMessage, HttpMethod};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
pub struct GeminiLiveConnectRequest {
pub method: HttpMethod,
pub path: PathParameters,
pub query: QueryParameters,
pub headers: RequestHeaders,
#[serde(default, skip_serializing_if = "Option::is_none")]
pub body: Option<RequestBody>,
}
impl Default for GeminiLiveConnectRequest {
fn default() -> Self {
Self {
method: HttpMethod::Get,
path: PathParameters::default(),
query: QueryParameters::default(),
headers: RequestHeaders::default(),
body: None,
}
}
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct PathParameters {
#[serde(default)]
pub rpc: GeminiLiveRpcMethod,
}
#[derive(Debug, Clone, Copy, PartialEq, Eq, Serialize, Deserialize, Default)]
pub enum GeminiLiveRpcMethod {
#[default]
#[serde(rename = "google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContent")]
BidiGenerateContent,
#[serde(
rename = "google.ai.generativelanguage.v1beta.GenerativeService.BidiGenerateContentConstrained"
)]
BidiGenerateContentConstrained,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct QueryParameters {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub key: Option<String>,
#[serde(
rename = "access_token",
default,
skip_serializing_if = "Option::is_none"
)]
pub access_token: Option<String>,
}
#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, Default)]
pub struct RequestHeaders {
#[serde(
rename = "Authorization",
default,
skip_serializing_if = "Option::is_none"
)]
pub authorization: Option<String>,
#[serde(flatten, default, skip_serializing_if = "BTreeMap::is_empty")]
pub extra: BTreeMap<String, String>,
}
pub type RequestBody = GeminiBidiGenerateContentClientMessage;