ó
    Ù^íh@  ã            	       óD   • S r SSKJr  SSKJr  S	S\S\S\\\4   4S jjrg)
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   • [        U SS9nSU;  d  US   c  gUS   U-  nUS4$ )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       Ú_/home/james-whalen/.local/lib/python3.13/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   