o
    ưit                     @   sR   d Z ddlmZmZ ddlmZ erddlmZ dedddee	e	f fd	d
Z
dS )zN
Helper util for handling amazon nova cost calculation
- e.g.: prompt caching
    )TYPE_CHECKINGTuplegeneric_cost_per_token)Usagemodelusager   returnc                 C   s   t | |ddS )z
    Calculates the cost per token for a given model, prompt tokens, and completion tokens.
    Follows the same logic as Anthropic's cost per token calculation.
    Zamazon_nova)r   r   Zcustom_llm_providerr   )r   r    r
   `/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/amazon_nova/cost_calculation.pycost_per_token   s   r   N)__doc__typingr   r   Z.litellm.litellm_core_utils.llm_cost_calc.utilsr   Zlitellm.types.utilsr   strfloatr   r
   r
   r
   r   <module>   s    "