
    hT                     R    d dl mZ d dlZd dlmZ 	 	 d
dededee   dee   def
d	Zy)    )OptionalN)ImageResponsemodelimage_responsesizeoptional_paramsreturnc                 V   t        j                         j                  |       rn1|xs i }|j                  dd      }|dkD  rdnd}|xs d}| d| d|  } t        j                  | d	      }|j                  d
      xs d}d}|j
                  rt        |j
                        }||z  S )zg
    Bedrock image generation cost calculator

    Handles both Stability 1 and Stability 3 models
    )r   steps2   z	max-stepsz50-stepsz1024-x-1024/bedrock)r   custom_llm_provideroutput_cost_per_imageg        r   )litellmAmazonStability3Config_is_stability_3_modelgetget_model_infodatalen)	r   r   r   r   _stepsr   _model_infor   
num_imagess	            f/var/www/Befach/backend/env/lib/python3.12/site-packages/litellm/llms/bedrock/image/cost_calculator.pycost_calculatorr      s     %%'==E=J */R !$$Wb1%{
 $}&%%)((%K
 $/??3J#K#RsJ,,-
 :--    )NN)	typingr   r   litellm.types.utilsr   strdictfloatr    r   r   <module>r$      sQ      - &*	$.$.!$. 3-$. d^	$.
 $.r   