uPj/dZddlmZddlmZddlmZddlmZddl m Z de de fd Z de d edefd Z dde d eded zde d zde d zdeeeff dZy )zx Azure AI cost calculation helper. Handles Azure AI Foundry Model Router flat cost and other Azure AI specific pricing. )Final)verbose_logger)generic_cost_per_token)Usage)get_model_infomodelreturncD|j}d|vxs d|vxs|dk(S)a< Check if the model is Azure AI Foundry Model Router. Detects patterns like: - "azure-model-router" - "model-router" - "model_router/" - "model-router/" Args: model: The model name Returns: bool: True if this is a model router model z model-router model_routerzazure-model-router)lower)r model_lowers j/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/azure_ai/cost_calculator.py_is_azure_model_routerrs/ K [ ( pNk,I p[\pMpp prompt_tokenscrt|sytdd}|jdd}|r |dkDr||zSy)a  Calculate the flat cost for Azure AI Foundry Model Router. Args: model: The model name (should be a model router model) prompt_tokens: Number of prompt tokens Returns: float: The flat cost in USD, or 0.0 if not applicable r azure_ai)rcustom_llm_providerinput_cost_per_tokenr)rrget)rr model_inforouter_flat_cost_per_tokens r&calculate_azure_model_router_flat_costr"sJ "% ('^Q[\J(27Mq(Q!&@1&D999 rNusageresponse_time_ms request_model service_tierc d}d}t|xs|duxr t|} t||d|\}}|r^|r|n|} t | |j } | dkDr=tjd| dd |j d | |j z d d || z }||fS#t$r-}t|stjd||Yd}~d}~wwxYw) aJ Calculate the cost per token for Azure AI models. For Azure AI Foundry Model Router: - Adds a flat cost of $0.14 per million input tokens (from model_prices_and_context_window.json) - Plus the cost of the actual model used (handled by generic_cost_per_token) Args: model: str, the model name without provider prefix (from response) usage: LiteLLM Usage block response_time_ms: Optional response time in milliseconds request_model: Optional[str], the original request model name (to detect router usage) Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd Raises: ValueError: If the model is not found in the cost map and cost cannot be calculated (except for Model Router models where we return just the routing flat cost) rNr)rrrrz`Azure AI Model Router: model '%s' not in cost map, calculating routing flat cost only. Error: %srz"Azure AI Model Router flat cost: $z.6fz (u tokens × $z.9fz/token))rr Exceptionrdebugrr) rrrrr prompt_costcompletion_costis_router_requesterouter_model_for_calcrouter_flat_costs rcost_per_tokenr(;s6KO 6e< T!K&<]&K  '= *% ( $ _&8E}5"HI^`e`s`s"t a   45Ec4JK''( 5EH[H[5[\_4``gi  + +K  ''7   &e,  npuwx    sB C##C  C)rNN)__doc__typingrlitellm._loggingr.litellm.litellm_core_utils.llm_cost_calc.utilsrlitellm.types.utilsr litellm.utilsrstrboolrintfloatrtupler(rrr5s +Q%(q#q$q(#ce8&) $# H( H( H(dlH(: H( * H(  5%< H(r