mirror of
https://github.com/tiennm99/litellm.git
synced 2026-08-11 16:26:07 +00:00
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:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user