diff --git a/crates/language_models/src/provider/google.rs b/crates/language_models/src/provider/google.rs index b79fcb2e86..11517abc18 100644 --- a/crates/language_models/src/provider/google.rs +++ b/crates/language_models/src/provider/google.rs @@ -628,6 +628,7 @@ impl GoogleEventMapper { // responds with `finish_reason: STOP` if wants_to_use_tool { self.stop_reason = StopReason::ToolUse; + events.push(Ok(LanguageModelCompletionEvent::Stop(StopReason::ToolUse))); } events }