From 89874a66c399f796f396ba1c268a856b3de13d08 Mon Sep 17 00:00:00 2001 From: "anil.kumar.peesa" Date: Thu, 9 Oct 2025 12:36:37 +0530 Subject: [PATCH] fix: Image Generation and Edits - ensure response_format is not present in extra_body --- tests/test_litellm/test_utils.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 6e20e956068..61412cc9189 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -42,6 +42,8 @@ def test_get_optional_params_image_gen(): ) assert optional_params is not None assert "response_format" not in optional_params + if 'extra_body' in optional_params: + assert "response_format" not in optional_params['extra_body'] assert optional_params["n"] == 3