diff --git a/litellm/types/utils.py b/litellm/types/utils.py index 0eae96d066d..0b87235d6a4 100644 --- a/litellm/types/utils.py +++ b/litellm/types/utils.py @@ -2551,8 +2551,8 @@ class ImageResponse(OpenAIImageResponse, BaseLiteLLMOpenAIResponseObject): model_config = ConfigDict(extra="allow", protected_namespaces=()) @field_serializer("data") - def _serialize_image_data(self, data: list[OpenAIImage]) -> list[dict[str, object]]: - return [image.model_dump() for image in data] + def _serialize_image_data(self, data: list[OpenAIImage] | None) -> list[dict[str, object]] | None: + return None if data is None else [image.model_dump() for image in data] def __init__( self,