use http::StatusCode;
use serde::{Deserialize, Serialize};
use crate::gemini::generate_content::types::{
GeminiCandidate, GeminiModelStatus, GeminiPromptFeedback, GeminiUsageMetadata,
};
use crate::gemini::types::{GeminiApiErrorResponse, GeminiResponseHeaders};
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize, Default)]
pub struct ResponseBody {
#[serde(default, skip_serializing_if = "Option::is_none")]
pub candidates: Option<Vec<GeminiCandidate>>,
#[serde(
rename = "promptFeedback",
default,
skip_serializing_if = "Option::is_none"
)]
pub prompt_feedback: Option<GeminiPromptFeedback>,
#[serde(
rename = "usageMetadata",
default,
skip_serializing_if = "Option::is_none"
)]
pub usage_metadata: Option<GeminiUsageMetadata>,
#[serde(
rename = "modelVersion",
default,
skip_serializing_if = "Option::is_none"
)]
pub model_version: Option<String>,
#[serde(
rename = "responseId",
default,
skip_serializing_if = "Option::is_none"
)]
pub response_id: Option<String>,
#[serde(
rename = "modelStatus",
default,
skip_serializing_if = "Option::is_none"
)]
pub model_status: Option<GeminiModelStatus>,
}
#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
#[serde(untagged)]
#[allow(clippy::large_enum_variant)]
pub enum GeminiGenerateContentResponse {
Success {
#[serde(with = "crate::gemini::types::status_code_serde")]
stats_code: StatusCode,
headers: GeminiResponseHeaders,
body: ResponseBody,
},
Error {
#[serde(with = "crate::gemini::types::status_code_serde")]
stats_code: StatusCode,
headers: GeminiResponseHeaders,
body: GeminiApiErrorResponse,
},
}