From 5abd5851e503d7ac03b9ee8e8773b8a09ebf76ce Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Mon, 13 Apr 2026 18:18:43 -0700 Subject: [PATCH] fix(streaming): extend role-only chunk preservation to azure_ai provider --- litellm/litellm_core_utils/streaming_handler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/litellm_core_utils/streaming_handler.py b/litellm/litellm_core_utils/streaming_handler.py index 058539f6c50..6f0bb8f96a9 100644 --- a/litellm/litellm_core_utils/streaming_handler.py +++ b/litellm/litellm_core_utils/streaming_handler.py @@ -839,7 +839,7 @@ class CustomStreamWrapper: not self.sent_first_chunk and hasattr(model_response.choices[0].delta, "role") and model_response.choices[0].delta.role is not None - and self.custom_llm_provider == "azure" + and self.custom_llm_provider in ("azure", "azure_ai") ) ): return True