diff --git a/tests/image_gen_tests/base_image_generation_test.py b/tests/image_gen_tests/base_image_generation_test.py index 56c4557cc..be0ecb1b4 100644 --- a/tests/image_gen_tests/base_image_generation_test.py +++ b/tests/image_gen_tests/base_image_generation_test.py @@ -42,6 +42,7 @@ class BaseImageGenTest(ABC): print(response) assert response._hidden_params["response_cost"] is not None + print("response_cost", response._hidden_params["response_cost"]) from openai.types.images_response import ImagesResponse ImagesResponse.model_validate(response.model_dump())