fix(streaming): extend role-only chunk preservation to azure_ai provider

This commit is contained in:
Ishaan Jaffer 2026-04-13 18:18:43 -07:00
parent 63eb58a82b
commit 5abd5851e5
No known key found for this signature in database

View file

@ -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