From 4a3ffdffd03732ab95d560948cb7404c33808dce Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Wed, 11 Feb 2026 11:52:49 -0800 Subject: [PATCH] test_azure_openai_with_prompt_cache_key --- tests/llm_translation/test_azure_openai.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/tests/llm_translation/test_azure_openai.py b/tests/llm_translation/test_azure_openai.py index 3fd908f86d7..ed66683f9c4 100644 --- a/tests/llm_translation/test_azure_openai.py +++ b/tests/llm_translation/test_azure_openai.py @@ -728,3 +728,15 @@ def test_azure_with_content_safety_error(): assert e.provider_specific_fields["innererror"]["code"] == "ResponsibleAIPolicyViolation" assert e.provider_specific_fields["innererror"]["content_filter_result"]["violence"]["filtered"] is True assert e.provider_specific_fields["innererror"]["content_filter_result"]["violence"]["severity"] == "high" + + +async def test_azure_openai_with_prompt_cache_key(): + """ + E2E test for Azure OpenAI with prompt cache key param on /chat/completions API. + """ + litellm._turn_on_debug() + response = litellm.completion( + model="azure/gpt-4o", + messages=[{"role": "user", "content": "What is the weather in San Francisco?"}], + prompt_cache_key="test_streaming_azure_openai", + ) \ No newline at end of file