diff --git a/crates/google_ai/src/google_ai.rs b/crates/google_ai/src/google_ai.rs index 68a36ac8ff1356b1d678e3b9903ad28a955368fe..c6472dfd6800e99f70b7650a29af0188617104ec 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 b95d94506c5da22557352035232b6b0c97e86ad6..718a7ba7ea00d6c6a0181f850e517d52ec7a269e 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(_) => {} }); } }