gemini: Fix issue when deserializing tool call (#29363)

Bennet Bo Fenner and Agus Zubiaga created 8 months ago

Fixes a regression introduced in #29322

Release Notes:

- N/A

Co-authored-by: Agus Zubiaga <[e-mail]>

Change summary

crates/google_ai/src/google_ai.rs             | 1 -
crates/language_models/src/provider/google.rs | 5 ++---
2 files changed, 2 insertions(+), 4 deletions(-)

Detailed changes

crates/google_ai/src/google_ai.rs

@@ -334,7 +334,6 @@ pub struct CountTokensResponse {
 #[derive(Debug, Serialize, Deserialize)]
 pub struct FunctionCall {
     pub name: String,
-    pub raw_args: String,
     pub args: serde_json::Value,
 }
 

crates/language_models/src/provider/google.rs

@@ -396,7 +396,6 @@ pub fn into_google(
                     Some(Part::FunctionCallPart(google_ai::FunctionCallPart {
                         function_call: google_ai::FunctionCall {
                             name: tool_use.name.to_string(),
-                            raw_args: tool_use.raw_input,
                             args: tool_use.input,
                         },
                     }))
@@ -540,8 +539,8 @@ pub fn map_to_language_model_completion_events(
                                                     is_input_complete: true,
                                                     raw_input: function_call_part
                                                         .function_call
-                                                        .raw_args
-                                                        .clone(),
+                                                        .args
+                                                        .to_string(),
                                                     input: function_call_part.function_call.args,
                                                 },
                                             )));