From 33f80f3f77fe2e4df60f4245b1e46a3d4f98e6f9 Mon Sep 17 00:00:00 2001 From: ryan-crabbe-berri Date: Tue, 2 Jun 2026 17:14:08 -0700 Subject: [PATCH] Use CI_CD_DEFAULT_AZURE_MODEL env var in live azure CI tests Replace the hardcoded azure/gpt-4.1-mini deployment name at the two live azure call sites in test_custom_logger.py with the CI_CD_DEFAULT_AZURE_MODEL env var, so the deployment can be updated from CircleCI project settings instead of editing code. The mocked call and all router config fixtures are left unchanged. --- tests/local_testing/test_custom_logger.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/tests/local_testing/test_custom_logger.py b/tests/local_testing/test_custom_logger.py index 6af2ff7e964..f2dd1474b60 100644 --- a/tests/local_testing/test_custom_logger.py +++ b/tests/local_testing/test_custom_logger.py @@ -213,7 +213,9 @@ def test_async_custom_handler_stream(): async def test_1(): nonlocal complete_streaming_response response = await litellm.acompletion( - model="azure/gpt-4.1-mini", messages=messages, stream=True + model=f"azure/{os.environ.get('CI_CD_DEFAULT_AZURE_MODEL', 'gpt-4.1-mini')}", + messages=messages, + stream=True, ) async for chunk in response: complete_streaming_response += ( @@ -257,7 +259,9 @@ def test_azure_completion_stream(): complete_streaming_response = "" response = litellm.completion( - model="azure/gpt-4.1-mini", messages=messages, stream=True + model=f"azure/{os.environ.get('CI_CD_DEFAULT_AZURE_MODEL', 'gpt-4.1-mini')}", + messages=messages, + stream=True, ) for chunk in response: complete_streaming_response += chunk["choices"][0]["delta"]["content"] or ""