scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-29 01:42:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 79

test(provider): justify AIPG client interception

Browse source
This commit is contained in:
halfaipg 2026-08-29 00:39:36 -04:00
parent 0e1430595f
commit f492ae7f28
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
tests/llm_translation/test_aipg.py
View file

@ -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",