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