diff --git a/litellm/llms/gemini/image_edit/transformation.py b/litellm/llms/gemini/image_edit/transformation.py index 16541138217..c3ea63ad43b 100644 --- a/litellm/llms/gemini/image_edit/transformation.py +++ b/litellm/llms/gemini/image_edit/transformation.py @@ -106,7 +106,10 @@ def transform_image_edit_request( # type: ignore[override] 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 021cfaeff5e..9cd746cfde4 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 @@ def test_transform_image_edit_request(self) -> None: 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")