Ë
    Ýª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)
z1
Custom cost calculator for Cohere rerank models
é    )Ú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
    Úcohere)r   Úcustom_llm_providerÚinput_cost_per_query)ç        r   r   r   )r   r   Ú
model_infoÚprompt_costs       úX/var/www/openai/venv/lib/python3.12/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   