uPj |dZddlmZddlmZddlmZddlmZddl m Z  dde d ed e dzd e dzd e e e ff d Zy)zf Helper util for handling azure openai-specific cost calculation - e.g.: prompt caching, audio tokens )Final)verbose_logger)generic_cost_per_token)Usage)get_model_infoNmodelusageresponse_time_ms service_tierreturnct|d}d|vr?|d:|8tjd||jd|d}|d|zdz }||fSt ||d|S)aU Calculates the cost per token for a given model, prompt tokens, and completion tokens. Input: - model: str, the model name without provider prefix - usage: LiteLLM Usage block, containing caching and audio token information Returns: Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd azure)rcustom_llm_provideroutput_cost_per_secondzr$sg +Q%( &)# + + +dl+* +  5%< +r