diff --git a/crates/google_ai/src/google_ai.rs b/crates/google_ai/src/google_ai.rs index 68a36ac8ff..c6472dfd68 100644 --- a/crates/google_ai/src/google_ai.rs +++ b/crates/google_ai/src/google_ai.rs @@ -202,6 +202,7 @@ pub enum Part { InlineDataPart(InlineDataPart), FunctionCallPart(FunctionCallPart), FunctionResponsePart(FunctionResponsePart), + ThoughtPart(ThoughtPart), } #[derive(Debug, Serialize, Deserialize)] @@ -235,6 +236,13 @@ pub struct FunctionResponsePart { pub function_response: FunctionResponse, } +#[derive(Debug, Serialize, Deserialize)] +#[serde(rename_all = "camelCase")] +pub struct ThoughtPart { + pub thought: bool, + pub thought_signature: String, +} + #[derive(Debug, Serialize, Deserialize)] #[serde(rename_all = "camelCase")] pub struct CitationSource { diff --git a/crates/language_models/src/provider/google.rs b/crates/language_models/src/provider/google.rs index b95d94506c..718a7ba7ea 100644 --- a/crates/language_models/src/provider/google.rs +++ b/crates/language_models/src/provider/google.rs @@ -620,6 +620,7 @@ impl GoogleEventMapper { ))); } Part::FunctionResponsePart(_) => {} + Part::ThoughtPart(_) => {} }); } }