Ë
    Ýªg@  ã            	       ó@   — d Z ddlmZ ddlmZ d	dededeeef   fd„Zy)
z|
Handles custom cost calculation for Azure AI models.

Custom cost calculation for Azure AI models only requied for rerank.
é    )ÚTuple©Úget_model_infoÚmodelÚnum_queriesÚreturnc                 óH   — t        | d¬«      }d|vs|d   €y|d   |z  }|dfS )zá
    Calculates the cost per query for a given rerank model.

    Input:
        - model: str, the model name without provider prefix

    Returns:
        Tuple[float, float] - prompt_cost_in_usd, completion_cost_in_usd
    Úazure_ai)r   Úcustom_llm_providerÚinput_cost_per_query)ç        r   r   r   )r   r   Ú
model_infoÚprompt_costs       úZ/var/www/openai/venv/lib/python3.12/site-packages/litellm/llms/azure_ai/cost_calculator.pyÚcost_per_queryr      sD   € ô   eÀÔL€Jð 	 jÑ0ØÐ,Ñ-Ð5ààÐ3Ñ4°{ÑB€Kà˜ÐÐó    N)é   )	Ú__doc__Útypingr   Úlitellm.utilsr   ÚstrÚintÚfloatr   © r   r   Ú<module>r      s5   ðñõ å (ñ˜#ð ¨Cð ¸¸eÀU¸lÑ8Kô r   