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

Merge pull request #20053 from BerriAI/litellm_gemini_edit_jan_30

Browse source 
fix aspectRatio mapping in image edit
This commit is contained in:
Sameer Kankute 2026-01-30 17:01:46 +05:30 • committed by GitHub
commit 5277a0cf82
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 5 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
litellm/llms/gemini/image_edit/transformation.py
View file

@ -106,7 +106,10 @@ class GeminiImageEditConfig(BaseImageEditConfig):
generation_config: Dict[str, Any] = {}
if "aspectRatio" in image_edit_optional_request_params:
generation_config["aspectRatio"] = image_edit_optional_request_params[
# Move aspectRatio into imageConfig inside generationConfig
if "imageConfig" not in generation_config:
generation_config["imageConfig"] = {}
generation_config["imageConfig"]["aspectRatio"] = image_edit_optional_request_params[
"aspectRatio"
]

2
tests/test_litellm/llms/gemini/image_edit/test_gemini_image_edit_transformation.py
View file

@ -61,7 +61,7 @@ class TestGeminiImageEditTransformation:
assert base64.b64decode(inline_data["data"]) == image_bytes
generation_config = request_body["generationConfig"]
assert generation_config["aspectRatio"] == "16:9"
assert generation_config["imageConfig"]["aspectRatio"] == "16:9"
def test_transform_image_edit_request_multiple_images(self) -> None:
image_one = BytesIO(b"image_one")