
    ^h!                     V    S SK Jr  S SKrS SKJr    S
S\S\S\\   S\\   S\4
S	 jjrg)    )OptionalN)ImageResponsemodelimage_responsesizeoptional_paramsreturnc                 Z   [         R                  " 5       R                  U S9(       a  O<U=(       d    0 nUR                  SS5      nUS:  a  SOSnU=(       d    SnU SU SU  3n [         R                  " U SS	9nUR                  S
5      =(       d    Sn[        UR                  5      nXx-  $ )z[
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	            d/home/james-whalen/.local/lib/python3.13/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   