diff --git a/tests/local_testing/test_custom_logger.py b/tests/local_testing/test_custom_logger.py index ee212658c8..baba409368 100644 --- a/tests/local_testing/test_custom_logger.py +++ b/tests/local_testing/test_custom_logger.py @@ -160,7 +160,7 @@ def test_completion_azure_stream_moderation_failure(): ] try: response = completion( - model="azure/chatgpt-v-3", + model="azure/gpt-4.1-nano", messages=messages, mock_response="Exception: content_filter_policy", stream=True, @@ -195,7 +195,7 @@ def test_async_custom_handler_stream(): async def test_1(): nonlocal complete_streaming_response response = await litellm.acompletion( - model="azure/chatgpt-v-3", messages=messages, stream=True + model="azure/gpt-4.1-nano", messages=messages, stream=True ) async for chunk in response: complete_streaming_response += ( @@ -239,7 +239,7 @@ def test_azure_completion_stream(): complete_streaming_response = "" response = litellm.completion( - model="azure/chatgpt-v-3", messages=messages, stream=True + model="azure/gpt-4.1-nano", messages=messages, stream=True ) for chunk in response: complete_streaming_response += chunk["choices"][0]["delta"]["content"] or ""