tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-06 16:24:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

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
parent 8b9d5018b6 3c451e945a
commit 5277a0cf82
2 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/llms/gemini/image_edit/transformation.py
+4 -1
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"
]
tests/test_litellm/llms/gemini/image_edit/test_gemini_image_edit_transformation.py
+1 -1
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")