From e2c55daaaff7cd2efae518378a5d0b256e28280e Mon Sep 17 00:00:00 2001 From: "devin-ai-integration[bot]" <158243242+devin-ai-integration[bot]@users.noreply.github.com> Date: Tue, 6 Oct 2026 10:47:26 -0700 Subject: [PATCH] fix(azure): add model router flat fee to azure provider cost tracking (#44876) * fix(azure): add model router flat fee to azure provider cost tracking Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(azure): price the azure router fee from one canonical entry Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * fix(azure): reuse the azure_ai model router fee path for azure Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> * test(azure): drop model router fee unit tests Co-Authored-By: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --------- Co-authored-by: kerry Co-authored-by: Devin AI <158243242+devin-ai-integration[bot]@users.noreply.github.com> --- litellm/cost_calculator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/cost_calculator.py b/litellm/cost_calculator.py index 8971b958c1c..86e1a8fd196 100644 --- a/litellm/cost_calculator.py +++ b/litellm/cost_calculator.py @@ -278,7 +278,7 @@ def _get_additional_costs( try: config_class = None - if custom_llm_provider == "azure_ai": + if custom_llm_provider in ("azure_ai", "azure"): from litellm.llms.azure_ai.common_utils import AzureFoundryModelInfo config_class = AzureFoundryModelInfo.get_azure_ai_config_for_model(model) @@ -1818,7 +1818,7 @@ def completion_cost( ) # Get additional costs from provider (e.g., routing fees, infrastructure costs) - if custom_llm_provider == "azure_ai" and not azure_ai_is_model_router_name(model): + if custom_llm_provider in ("azure_ai", "azure") and not azure_ai_is_model_router_name(model): model_for_additional_costs = request_model_for_cost if completion_response is not None: hidden_params = getattr(completion_response, "_hidden_params", None) or {}