From de897ce3897af0f94e992b8e23bffca4303dfeee Mon Sep 17 00:00:00 2001 From: Krish Dholakia Date: Sat, 22 Nov 2025 15:54:23 -0800 Subject: [PATCH] fix(vertex_ai/image_generation_handler.py): fix default sample count to be 1 - same as openai (#16403) reduce cost for default flow --- .../image_generation/image_generation_handler.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/litellm/llms/vertex_ai/image_generation/image_generation_handler.py b/litellm/llms/vertex_ai/image_generation/image_generation_handler.py index 4ffe557f1b..04be4de8e3 100644 --- a/litellm/llms/vertex_ai/image_generation/image_generation_handler.py +++ b/litellm/llms/vertex_ai/image_generation/image_generation_handler.py @@ -45,17 +45,18 @@ class VertexImageGeneration(VertexLLM): Transform the optional params to the format expected by the Vertex AI API. For example, "aspect_ratio" is transformed to "aspectRatio". """ + default_params = { + "sampleCount": 1, + } if optional_params is None: - return { - "sampleCount": 1, - } + return default_params def snake_to_camel(snake_str: str) -> str: """Convert snake_case to camelCase""" components = snake_str.split("_") return components[0] + "".join(word.capitalize() for word in components[1:]) - transformed_params = {} + transformed_params = default_params.copy() for key, value in optional_params.items(): if "_" in key: camel_case_key = snake_to_camel(key)