ó
    Ù^í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)
z1
Custom cost calculator for Cohere rerank models
é    )Ú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
Úcohere)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/cohere/cost_calculator.pyÚcost_per_queryr   
   sD   € ô   eÀÑJ€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   