diff --git a/crates/language_models/src/provider/google.rs b/crates/language_models/src/provider/google.rs index b4753a763661ecd8eccf75fc5702cce6370571ac..c6631c52c815b241b09090eadd878f70aef0bfab 100644 --- a/crates/language_models/src/provider/google.rs +++ b/crates/language_models/src/provider/google.rs @@ -599,6 +599,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 }