From 3c451e945a358e51903e48d1b366dba0e43468a6 Mon Sep 17 00:00:00 2001 From: Sameer Kankute Date: Fri, 30 Jan 2026 15:25:18 +0530 Subject: [PATCH] fix aspectRatio mapping --- litellm/llms/gemini/image_edit/transformation.py | 5 ++++- .../image_edit/test_gemini_image_edit_transformation.py | 2 +- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/litellm/llms/gemini/image_edit/transformation.py b/litellm/llms/gemini/image_edit/transformation.py index 1654113821..c3ea63ad43 100644 --- a/litellm/llms/gemini/image_edit/transformation.py +++ b/litellm/llms/gemini/image_edit/transformation.py @@ -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" ] diff --git a/tests/test_litellm/llms/gemini/image_edit/test_gemini_image_edit_transformation.py b/tests/test_litellm/llms/gemini/image_edit/test_gemini_image_edit_transformation.py index 021cfaeff5..9cd746cfde 100644 --- a/tests/test_litellm/llms/gemini/image_edit/test_gemini_image_edit_transformation.py +++ b/tests/test_litellm/llms/gemini/image_edit/test_gemini_image_edit_transformation.py @@ -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")