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

fix: test

Browse source
This commit is contained in:
Ishaan Jaff 2025-05-21 21:30:34 -07:00
parent 38c0074cfb
commit f810a4c036
2 changed files with 5 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm/types/utils.py
View file

@ -1567,7 +1567,7 @@ class ImageUsage(BaseLiteLLMOpenAIResponseObject):
from openai.types.images_response import ImagesResponse as OpenAIImageResponse
class ImageResponse(BaseLiteLLMOpenAIResponseObject):
class ImageResponse(OpenAIImageResponse, BaseLiteLLMOpenAIResponseObject):
_hidden_params: dict = {}
usage: ImageUsage
@ -1609,12 +1609,7 @@ class ImageResponse(BaseLiteLLMOpenAIResponseObject):
output_tokens=0,
total_tokens=0,
)
#######################
self.created = created
self.data = _data
self.usage = _usage
#######################
super().__init__(created=created, data=_data, usage=_usage) # type: ignore
self._hidden_params = hidden_params or {}
def __contains__(self, key):

4
tests/image_gen_tests/base_image_generation_test.py
View file

@ -71,7 +71,9 @@ class BaseImageGenTest(ABC):
from openai.types.images_response import ImagesResponse
ImagesResponse.model_validate(dict(response))
response_dict = response.model_dump()
print("response usage=", response_dict.get("usage"))
ImagesResponse.model_validate(response_dict)
for d in response.data:
assert isinstance(d, Image)