uPj UddlmZddlmZddlmZddlZddlmZdZ ee e d<dZ ee e d <ed d d d dddZ eee e fe d<de dee efde dzfdZde dee efdzdedzfdZ dde dedee efdzdefdZy))Mapping)MappingProxyType)FinalN) ImageResponsehigh!FAL_KEYED_PRICING_DEFAULT_QUALITYz 1024-x-768FAL_TEXT_TO_IMAGE_DEFAULT_SIZEz 1024-x-1024z 512-x-512z 768-x-1024z 576-x-1024z 1024-x-576) square_hdsquare portrait_4_3 portrait_16_9 landscape_4_3landscape_16_9FAL_NAMED_IMAGE_SIZESmodeloptional_paramsreturncZ|jd}||jdrdStSt|trJ|jd}|jd}t|t rt|t r|d|Syt|t rtj|Sy)N image_sizez/editwidthheightz-x-)getendswithr isinstancerintstrr)rrrrrs h/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/fal_ai/cost_calculator.py _keyed_sizers'++L9J~~g.tR4RR*g&!~~g."x0 eS !j&=WCx( (*c"$((44 cB|yt||}|y|jd}t|tr|dk7r|nt}t j jd|d|d|}|y|jd}t|ttfr t|SdS)Nrrqualityautozfal_ai//output_cost_per_image) rrrrrlitellm model_costrfloat)rrsize raw_qualityr" keyed_entry keyed_costs r_keyed_cost_per_imager-%sE?KD |(,,Y7K!+s3 v8M St !++//''!D65'0RSK#(?@J *:U| D5 N$Nrimage_responsect|tstdt||j t j jjd}|jrt|jnd}t||}|||zSt j|t j jj}|jdxsd}||zS)z1 fal.ai image generation cost calculator z6image_response must be of type ImageResponse got type=r$rr!)rcustom_llm_providerr%g)rr ValueErrortype removeprefixr& LlmProvidersFAL_AIvaluedatalenr-get_model_infor)rr.r num_imageskeyed_cost_per_image _model_infor%s rcost_calculatorr=6s nm 4QRVWeRfQghii   '"6"6"="="C"C!DAF GE9G9L9LS!4!45RSJ"7eUd"e'#j00 //#0077==K+6//:Q*R*YVY : --r)N)collections.abcrtypesrtypingrr&litellm.types.utilsrrr__annotations__r robjectrr(r-r=rrrEs#"-06!5:6-9c 92B"$%%&  3uWS#X./  s WS&[-A cDj OOwsF{7Kd7ROW\_cWcO(48. ..S&[)D0. .r