o ưi@s:ddlmZddlZddlmZdededefddZdS) )AnyN) ImageResponsemodelimage_responsereturncCsZtj|tjjjd}|dpd}d}t|tr$|jr t |j}||St dt |)z2 Recraft image generation cost calculator )rZcustom_llm_provideroutput_cost_per_imagegrz6image_response must be of type ImageResponse got type=) litellmZget_model_infoZ LlmProvidersZRECRAFTvalueget isinstancerdatalen ValueErrortype)rrZ _model_inforZ num_imagesr[/home/app/Keep/.python/lib/python3.10/site-packages/litellm/llms/recraft/cost_calculator.pycost_calculators  r)typingrrZlitellm.types.utilsrstrfloatrrrrrs