From 1d8c8ad750b519d10312b05423223028c9fad3bf Mon Sep 17 00:00:00 2001 From: Ishaan Jaffer Date: Tue, 13 Jan 2026 14:11:07 -0800 Subject: [PATCH] fix:_check_provider_match --- litellm/utils.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/litellm/utils.py b/litellm/utils.py index 7c7591cdbab..7f889c16bb5 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -5040,6 +5040,11 @@ def _check_provider_match(model_info: dict, custom_llm_provider: Optional[str]) custom_llm_provider == "litellm_proxy" ): # litellm_proxy is a special case, it's not a provider, it's a proxy for the provider return True + elif custom_llm_provider == "azure_ai" and model_info["litellm_provider"] in ("azure", "openai"): + # Azure AI also works with azure models + # as a last attempt if the model is not on Azure AI, Azure then fallback to OpenAI cost + # tracking the cost is better than attributing 0 cost to it. + return True else: return False