From 1630943c22481d7503c057a661e2ed2dc868a683 Mon Sep 17 00:00:00 2001 From: Rhys Date: Tue, 24 Feb 2026 14:10:31 +0700 Subject: [PATCH] fix: update image response handling --- litellm/llms/openai/image_edit/transformation.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/litellm/llms/openai/image_edit/transformation.py b/litellm/llms/openai/image_edit/transformation.py index a1e5375d098..2bc0684efa7 100644 --- a/litellm/llms/openai/image_edit/transformation.py +++ b/litellm/llms/openai/image_edit/transformation.py @@ -14,7 +14,7 @@ from litellm.types.images.main import ( ) from litellm.types.llms.openai import FileTypes from litellm.types.router import GenericLiteLLMParams -from litellm.utils import ImageResponse +from litellm.utils import ImageResponse, LiteLLMResponseObjectHandler from ..common_utils import OpenAIError @@ -157,7 +157,9 @@ class OpenAIImageEditConfig(BaseImageEditConfig): raise OpenAIError( message=raw_response.text, status_code=raw_response.status_code ) - return ImageResponse(**raw_response_json) + return LiteLLMResponseObjectHandler.convert_to_image_response( + response_object=raw_response_json, + ) def validate_environment( self,