o ưi@s:ddlmZddlZddlmZdededefddZdS) )AnyN) ImageResponsemodelimage_responsereturncCsZtj|tjjjd}|dpd}d}t|tr$|jr t |j}||St dt |)z RunwayML image generation cost calculator. RunwayML charges per image generated, not per pixel. Pricing is stored in model_prices_and_context_window.json with output_cost_per_image. )rZcustom_llm_provideroutput_cost_per_imagegrz7image_response must be of type ImageResponse, got type=) litellmZget_model_infoZ LlmProvidersZRUNWAYMLvalueget isinstancerdatalen ValueErrortype)rrZ _model_inforZ num_imagesr\/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/runwayml/cost_calculator.pycost_calculators    r)typingrrZlitellm.types.utilsrstrfloatrrrrrs