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)