
    g!                     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                 :   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}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        )litellmAmazonStability3Config_is_stability_3_modelgetget_model_infolendata)	r   r   r   r   _stepsr   _model_infor   
num_imagess	            _/var/www/openai/venv/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   