diff --git a/tests/local_testing/test_custom_logger.py b/tests/local_testing/test_custom_logger.py index 1b627d56717..1fdfb78c580 100644 --- a/tests/local_testing/test_custom_logger.py +++ b/tests/local_testing/test_custom_logger.py @@ -211,7 +211,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 += ( @@ -255,7 +257,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 ""