fix(vertex_ai/image_generation_handler.py): fix default sample count to be 1 - same as openai (#16403)

reduce cost for default flow
This commit is contained in:
Krish Dholakia
2025-11-22 15:54:23 -08:00
committed by GitHub
parent 478f9b0072
commit de897ce389
@@ -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)