From 3058ab0b1485b981d6a42f8bdaf6990c6ea584c7 Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 11 Feb 2026 12:25:15 -0800 Subject: [PATCH] test_azure_openai_with_prompt_cache_key --- tests/llm_translation/test_azure_openai.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) 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