diff --git a/tests/llm_translation/test_azure_openai.py b/tests/llm_translation/test_azure_openai.py index 2463304d2d2..1da380b57a2 100644 --- a/tests/llm_translation/test_azure_openai.py +++ b/tests/llm_translation/test_azure_openai.py @@ -736,7 +736,10 @@ def test_azure_openai_with_prompt_cache_key(): """ litellm._turn_on_debug() response = litellm.completion( - model="azure/gpt-4o", + model="azure/gpt-4.1-mini", + api_key=os.getenv("AZURE_API_KEY"), + api_base=os.getenv("AZURE_API_BASE"), + api_version="2024-12-01-preview", messages=[{"role": "user", "content": "What is the weather in San Francisco?"}], prompt_cache_key="test_streaming_azure_openai", ) \ No newline at end of file