From a3da59f2bed92a24f6d4ec60a34ce3fe25995b72 Mon Sep 17 00:00:00 2001 From: Cursor Agent Date: Mon, 26 Oct 2026 13:37:12 +0000 Subject: [PATCH] test(openai): point image extra_headers coverage at with_raw_response Main now reads provider headers off the raw response, so the proxy credential cases mock that call and pass a real ImageResponse. Co-authored-by: Zsanz3 --- .../test_openai_image_generation_extra_headers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/unit/llms/openai/image_generation/test_openai_image_generation_extra_headers.py b/tests/unit/llms/openai/image_generation/test_openai_image_generation_extra_headers.py index d1a9297267f..a0195ac3484 100644 --- a/tests/unit/llms/openai/image_generation/test_openai_image_generation_extra_headers.py +++ b/tests/unit/llms/openai/image_generation/test_openai_image_generation_extra_headers.py @@ -275,7 +275,7 @@ class TestImageGenerationExtraHeaders: await openai_chat_completions.aimage_generation( prompt="A white cat", data={"model": "gpt-image-2", "prompt": "A white cat", **optional_params}, - model_response=MagicMock(), + model_response=ImageResponse(), timeout=60.0, logging_obj=mock_logging_obj, api_key="test-key",