scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-10-10 03:28:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 85

fix: update image response handling

Browse source
This commit is contained in:
Rhys 2026-02-24 14:10:31 +07:00
parent 6134984008
commit 1630943c22
1 changed files with 4 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
litellm/llms/openai/image_edit/transformation.py
View file

@ -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,