phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

add debugging to BaseImageGenTest

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-12 16:18:22 -08:00
parent c15359911a
commit a6ce4254f0
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/image_gen_tests/base_image_generation_test.py
View file

@ -42,6 +42,7 @@ class BaseImageGenTest(ABC):
print(response) print(response)
assert response._hidden_params["response_cost"] is not None assert response._hidden_params["response_cost"] is not None
print("response_cost", response._hidden_params["response_cost"])
from openai.types.images_response import ImagesResponse from openai.types.images_response import ImagesResponse
ImagesResponse.model_validate(response.model_dump()) ImagesResponse.model_validate(response.model_dump())