diff --git a/tests/llm_translation/test_azure_ai.py b/tests/llm_translation/test_azure_ai.py index c56e2a7a36..91cbcda059 100644 --- a/tests/llm_translation/test_azure_ai.py +++ b/tests/llm_translation/test_azure_ai.py @@ -297,7 +297,7 @@ def test_completion_azure(): litellm.set_verbose = False ## Test azure call response = completion( - model="azure/gpt-4.1-mini", + model="azure_ai/gpt-4.1-mini", api_base=os.getenv("AZURE_AI_API_BASE"), messages=[ {