From f492ae7f283710cda2d42ac15fe91c4b2af5c83f Mon Sep 17 00:00:00 2001 From: halfaipg Date: Sat, 29 Aug 2026 00:39:36 -0400 Subject: [PATCH] test(provider): justify AIPG client interception --- tests/llm_translation/test_aipg.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tests/llm_translation/test_aipg.py b/tests/llm_translation/test_aipg.py index cedc15a4232..6f4d8111ade 100644 --- a/tests/llm_translation/test_aipg.py +++ b/tests/llm_translation/test_aipg.py @@ -134,7 +134,9 @@ def test_aipg_image_generation_uses_native_endpoint(): with ( mock.patch.dict(os.environ, {"AIPG_API_KEY": "grid-test"}, clear=True), - mock.patch("litellm.llms.openai.openai.OpenAI", return_value=mock_client) as constructor, + mock.patch( # test-quality-ok: capture the SDK-created client to verify the trusted AIPG base and env key + "litellm.llms.openai.openai.OpenAI", return_value=mock_client + ) as constructor, ): response = litellm.image_generation( model="aipg/z-image-turbo",