From b351c543dc610b492d9d8b8222daf49408f6f012 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Wed, 21 May 2025 21:37:08 -0700 Subject: [PATCH] fix: test --- tests/image_gen_tests/base_image_generation_test.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/tests/image_gen_tests/base_image_generation_test.py b/tests/image_gen_tests/base_image_generation_test.py index 98c00f305eb..c3a5cfb2251 100644 --- a/tests/image_gen_tests/base_image_generation_test.py +++ b/tests/image_gen_tests/base_image_generation_test.py @@ -68,10 +68,15 @@ class BaseImageGenTest(ABC): assert logged_standard_logging_payload is not None assert logged_standard_logging_payload["response_cost"] is not None assert logged_standard_logging_payload["response_cost"] > 0 - + import openai from openai.types.images_response import ImagesResponse - response_dict = response.model_dump() + # print openai version + print("openai version=", openai.__version__) + + response_dict = dict(response) + if "usage" in response_dict: + response_dict["usage"] = dict(response_dict["usage"]) print("response usage=", response_dict.get("usage")) ImagesResponse.model_validate(response_dict)