diff options
Diffstat (limited to 'src/client/vertexai.rs')
| -rw-r--r-- | src/client/vertexai.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs index 9907a7f..4d06934 100644 --- a/src/client/vertexai.rs +++ b/src/client/vertexai.rs @@ -201,7 +201,7 @@ pub(crate) fn gemini_build_body( body["safetySettings"] = safety_settings; } - if let Some(v) = model.max_output_tokens { + if let Some(v) = model.max_tokens_param() { body["generationConfig"]["maxOutputTokens"] = v.into(); } if let Some(v) = temperature { |