uPjN$UddlmZmZmZddlZddlmZddlmZddl m Z ddl m Z m Z ddlmZmZer ddlZddlmZeZneZd Zeeed fed <Gd d eZy)) TYPE_CHECKINGAnyFinalN)BaseImageGenerationConfig)get_secret_str) AimlImageGenerationRequestParams)AllMessageValues#OpenAIImageGenerationOptionalParams) ImageObject ImageResponse)Logging)zopenai/.!OPENAI_STYLE_IMAGE_MODEL_PREFIXEScjeZdZUdZeed<dZeed<ededefdZ dede e fdZ d e d e ded ede f d Z d ded zded zded e de ded zdefdZ d!de dede ed e de ded zded zde fdZdeded e de de de f dZ d!dedej(dedede d e de ddded zded zdefdZy )"AimlImageGenerationConfigzhttps://api.aimlapi.comDEFAULT_BASE_URLzv1/images/generationsIMAGE_GENERATION_ENDPOINTmodelreturnc,|jtS)z OpenAI image models routed through AI/ML API (e.g. ``openai/gpt-image-2``) use the upstream OpenAI request schema, not the flux-style schema used by the rest of the AI/ML catalog. ) startswithr)rs v/home/oem/CyberBrain26-unified/.venv/lib/python3.12/site-packages/litellm/llms/aiml/image_generation/transformation.py_is_openai_style_modelz0AimlImageGenerationConfig._is_openai_style_model!s ABBc4|j|rgdSgdS)z? https://api.aimlapi.com/v1/images/generations )nsizequalityresponse_format output_format background moderationoutput_compression)rrr)r)selfrs rget_supported_openai_paramsz5AimlImageGenerationConfig.get_supported_openai_params*s$  & &u -  0/rnon_default_paramsoptional_params drop_paramsc |j|}|j|}|D]}||vr||vr|rtd|d|d|d|r ||||</|dk(r |||d<=|dk(r |||d<K|d k(rJ||}t|tr/d |vr+t t |jd \} } | | d |d <||d <||||<|S) Nz Parameter z is not supported for model z. Supported parameters are z6. Set drop_params=True to drop unsupported parameters.r num_imagesrrrx)widthheight image_size)r$r ValueError isinstancestrmapintsplit) r#r%r&rr'supported_paramsis_openai_stylek size_valuer+r,s rmap_openai_paramsz+AimlImageGenerationConfig.map_openai_params;s6#'"B"B5"I!%!rNapi_baseapi_keylitellm_paramsstreamc|xstdxs |j}|jd}|jd}|d|j}|S)z6 Get the complete url for the request AIML_API_BASE/z/v1)rrrstrip removesuffixr)r#r9r:rr&r;r< complete_urls rget_complete_urlz*AimlImageGenerationConfig.get_complete_urlfs[%`(G`4K`K` #**3/ #007 &q)G)G(HI rheadersmessagescr|xstdxs td}|s tdd||d<d|d<|S)N AIML_API_KEY AIMLAPI_KEYz&AIML_API_KEY or AIMLAPI_KEY is not setzBearer Authorizationzapplication/jsonz Content-Type)rr.) r#rDrrEr&r;r:r9 final_api_keys rvalidate_environmentz.AimlImageGenerationConfig.validate_environmentzsQ  V~n5 V 9V EF F%,]O#< "4rpromptcd|j|r||d|Std||d|}t|S)z Transform the image generation request to the AI/ML image generation request body https://api.aimlapi.com/v1/images/generations )rrL)rLr)rrdict)r#rrLr&r;rD"aiml_image_generation_request_bodys r"transform_image_generation_requestz"''..#%) %e +0995E+F 5(Ne,C"''..#%*YYz%:%Weii>W $+0995E+F dC &9 h8O+O'1)< !V+"''..#%+N%; $ f_"''..#%) &u  @# &&x0 E>"''..#%) %e  $u,"''..#%*>%: $ G && NqcR(44$,,'  sG H+G==H)N)NN)__name__ __module__ __qualname__rr0__annotations__r staticmethodboolrrfr r$rOr8rCr rKrQhttpxResponser LiteLLMLoggingObjrmrNrrrrs5c5%<s<CcCdCC00>a9b0") )) )  )  )d#*t     t  6##'(     t* (888 8  8  8 8B#!%WWnnW& W ' W  WWW-WtW$;W Wrr)typingrrrrt5litellm.llms.base_llm.image_generation.transformationrlitellm.secret_managers.mainrlitellm.types.llms.aimlrlitellm.types.llms.openair r litellm.types.utilsr r tiktoken*litellm.litellm_core_utils.litellm_loggingr _LiteLLMLoggingObjrvrtupler0rqrrNrrrsd,, 8D;X*=I!5sCx#9H_ 9_r