scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-05 08:07:05 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 137

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

Browse source 
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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 5 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/llms/vertex_ai/image_generation/image_generation_handler.py
View file

@ -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)