use serde::{Deserialize, Serialize};
use super::{Content, ModalityTokenCount, TaskType};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct EmbedContentRequest {
#[serde(skip_serializing_if = "Option::is_none")]
pub model: Option<String>,
pub content: Content,
#[serde(skip_serializing_if = "Option::is_none")]
pub task_type: Option<TaskType>,
#[serde(skip_serializing_if = "Option::is_none")]
pub title: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub output_dimensionality: Option<i32>,
#[serde(skip_serializing_if = "Option::is_none")]
pub embed_content_config: Option<EmbedContentConfig>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct EmbedContentConfig {
#[serde(skip_serializing_if = "Option::is_none")]
pub title: Option<String>,
#[serde(skip_serializing_if = "Option::is_none")]
pub task_type: Option<TaskType>,
#[serde(skip_serializing_if = "Option::is_none")]
pub auto_truncate: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub output_dimensionality: Option<i32>,
#[serde(skip_serializing_if = "Option::is_none")]
pub document_ocr: Option<bool>,
#[serde(skip_serializing_if = "Option::is_none")]
pub audio_track_extraction: Option<bool>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct EmbedContentResponse {
#[serde(skip_serializing_if = "Option::is_none")]
pub embedding: Option<ContentEmbedding>,
#[serde(skip_serializing_if = "Option::is_none")]
pub usage_metadata: Option<EmbeddingUsageMetadata>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct BatchEmbedContentsRequest {
pub requests: Vec<EmbedContentRequest>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct BatchEmbedContentsResponse {
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub embeddings: Vec<ContentEmbedding>,
#[serde(skip_serializing_if = "Option::is_none")]
pub usage_metadata: Option<EmbeddingUsageMetadata>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct ContentEmbedding {
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub values: Vec<f32>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub shape: Vec<i32>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
#[serde(rename_all = "camelCase")]
#[derive(gproxy_protocol_macros::WireBuilder)]
#[cfg_attr(not(feature = "exhaustive"), non_exhaustive)]
pub struct EmbeddingUsageMetadata {
#[serde(skip_serializing_if = "Option::is_none")]
pub prompt_token_count: Option<i32>,
#[serde(default, skip_serializing_if = "Vec::is_empty")]
pub prompt_token_details: Vec<ModalityTokenCount>,
#[serde(default, skip_serializing_if = "serde_json::Map::is_empty", flatten)]
pub rest: serde_json::Map<String, serde_json::Value>,
}