hx%FUddlZddlZddlmZddlmZmZmZmZm Z m Z m Z m Z ddl Z ddlZddlmZmZmZmZddlmZddlmZddlmZddlmZdd lmZdd lmZmZdd l m!Z!m"Z"dd l#m$Z$dd l%m&Z&e$Z'e$e(d<ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4ddl5m6Z6ddl7m8Z8ddl9m:Z:m;Z;mm?Z?m@Z@mAZAmBZBddlCmDZDede?fdZEe d9deFde eFde eGde e eFe6fde eFde eFd e eFd!e eFd"e eFd#e eFd$e eFd%e eFd&ed'deeee?ffd(ZHe d9d)d*deFde eFde eGde e eFe6fde eFde eFd e eFd!e eFd"e eFd#e eFd$e eFd%e eFd&ed)de?fd+ZHe d9deFde eFde eGde e eFe6fde eFde eFd e eFd!e eFd"e eFd#e eFd$e eFd%e eFde e?eeee?fffd,ZHede?fd-ZIe d:d.e;deFdeGded/de eFd!e eFde?fd0ZJe d;d.e;deFde eFd1e eFde eGde e eFe6fde eFde eFd!e eFd2e eeFefd3e eeFefd4e eeFefd5e e eKe jfd6e eFde e?eeee?fffd7ZMe d( $ 1 #2249JKK ,0 mT *$5}5H  } 5$H   /**H  Q L+  1=X 3 "    sN4D:A4D+D,AD;D<DD:DD D7D22D77D:promptr2nqualityresponse_formatsizestyleuserinput_fidelityapi_keyr4 api_versionr3Tc yNr<rXr2rYrZr[r\r]r^r_timeoutr`r4rar8r3rMs rVr@r@p(F)r3c yrcr<rds rVr@r@rfrgc t}|jdd}|jdd}|jdd}|jdd}|jdd}|jdd}|jd d}|jd i}|jd }|jd d}|jd d}|jddxsi}|jdd}||j|tjj }d}|| t || | \}} }} nd}d} ||jd<gd}t} || z}!|jD"#cic] \}"}#|"|!vs |"|#}$}"}#d}%| D| tjjvr$tj|xs|t| }%td?|xs||||||||| |%d |$}&t!d?i|}'|}(|(j#|||&| d|||||did | d|(j$vr| |(j$d<| t'||S| dk(rt)dxsd})| xstj*xs t)d} | xstj,xs t)d} | xs<tj.xs*tj0xst)dxs t)d } |&j3d!dxs t)d"}*d#| d$}+|+jD]\}"}#|"|vs |#||"<t5j6||| | | |*|||&|| ||||'%}|S| dk(s| tj8vr%t;j6||| | xs|| ||&|||& }|S| d'k(r.| t=d(t?j6||| ||&|||| | ) }|S| d*k(r|&j3d+dxs1|&j3d,dxstj@xs t)d-},|&j3d.dxs1|&j3d/dxstjBxs t)d0}-|&j3d1dxs|&j3d2dxs t)d3}.| xs*tj*xst)d4xs t)d5} tEj6||| ||&||,|-|.|| |6 }|S| tjjFfvr.|%tId7| tJjM|||%|&| |'|| |8 S| tjNvrd}/tjPD]}0|0d9| k(s |0d:}/|/ tS|| ;|dd}3~3wwxYw)@zv Maps the https://api.openai.com/v1/images/generations endpoint. Currently supports just Azure + OpenAI. r3Flitellm_call_idN logger_fn mock_responseproxy_server_requestazure_ad_token_provider model_infometadatalitellm_logging_objr extra_headersheaders base_model)r2r8r4dall-e-2r6r2)r^rr4rar` deployment_id organizationbase_urldefault_headersre max_retriesrYrZr\r]r_r2provider) r2rYrZr[r\r]r^r_r8provider_config) reazurerjrkrmrorppreset_cache_keystream_responser2r^optional_paramslitellm_paramsr8r8)r2rlr~AZURE_API_TYPEAZURE_API_BASEAZURE_API_VERSIONAZURE_OPENAI_API_KEY AZURE_API_KEYazure_ad_tokenAZURE_AD_TOKENzapplication/json;)z Content-Typezapi-key)r2rXrer`r4rrn logging_objrmodel_responserar3r rsr) r2rXrer`r4rrrr3r bedrockz!Model needs to be set for bedrock) r2rXrerrrr3r r4r` vertex_aivertex_projectvertex_ai_projectVERTEXAI_PROJECTvertex_locationvertex_ai_locationVERTEXAI_LOCATIONvertex_credentialsvertex_ai_credentialsVERTEXAI_CREDENTIALSVERTEXAI_API_BASEVERTEX_API_BASE) r2rXrerrrrrrr3r4r z-image generation config is not supported for ) r2rX image_generation_provider_config(image_generation_optional_request_paramsr8rrrer r|custom_handlerr2r8T) r2rXr`r4rrrrer r7r<)/localsrDupdatelitellmutilsr*r,_hidden_paramsr)itemsr( _member_map_valuesr+$get_provider_image_generation_configr-rupdate_environment_variablesmodel_call_detailsrr"r4rar` azure_keypoprr@openai_compatible_providersrrJrrrr!RECRAFTrIrimage_generation_handler_custom_providerscustom_provider_maprrFrrWrr)4rXr2rYrZr[r\r]r^r_rer`r4rar8rMrLr3rjrkrlrmrnrorprqr rrrsrtrdynamic_api_key openai_paramsrdefault_paramskvnon_default_paramsimage_generation_configrlitellm_params_dictloggingapi_typerryrrrritemasync_custom_client custom_clientrUs4 rVr@r@s4o x **%6> **%6=JJ{D1 '-zz/4'H %zz*@$G"(**-F"MZZ d3 ::j"-17 $8  %+) 33  !  1:6:.,,+"#2&(<($$(#% !4 - " !(B(B B@SG & &'< =  $(uMR R ' )%&67B7HW7#3#3W~FV7WH7&&7!"56 3??3$$3""893"/2 -00 $2 01  !4"O(--/1G#!"GAJ04DD!-(?/ /-' /2N|Y 8 +"g&I&II4EE2?!/ /- / NR{!I -} CDD5FF/ /- /! Nt]!K /##$4d;6"&&':DA6))6""45  ##$5t<7"&&';TB7**7""56  ##$8$?:"&&'>E:!"89 5##5!"565""34 5EE/ /-0 2#5 /! NhG!  ( (%  '. #PQdPe!fgg#<<1H9H$72/=   7#<#< <37N33 #'::%)*:%;N4%,5H $&BF#%*V=M*N*0'"0!A!A!#%#1$3 3#."B ":#8< %*V[*I$*M"0!@!@!#%#1$3 3#("A "k l   3 $h    skE7W 9 WW B=W CW ,W ;W 2W :DW A W /W A W 1W W W3W..W3cKtj}|jdd}|jdd}d|d< ttg|i|}t j }t|j|}|$|"t||jdd\}}}}|jd|d{} t| tst| trt| tr td i| } | } | Stj| r | d{} | S|jd|d{} | S77$7 #t$r} |xsd}t||| || d} ~ wwxYww) a Asynchronously calls the `image_variation` function with the given arguments and keyword arguments. Parameters: - `args` (tuple): Positional arguments to be passed to the `image_variation` function. - `kwargs` (dict): Keyword arguments to be passed to the `image_variation` function. Returns: - `response` (Any): The response returned by the `image_variation` function. r2Nr8T async_callr4r5r6r7r<)r=r>rDrimage_variationrArBrCr,rErFrGr*rHrJrrKs rVaimage_variationrs  ! ! #D JJw %E **%:DAF<# 888&&(#CGGT2  &5+<+;fjjT&B, (A"Aq #2249JKK mT *j =/ -. - > > $H    /**H"11$8IJJHL+K  1=X 3 "    sy>E'A8E9D::AE<E'=ED<EE'E3D>4E9E':E<E>E E$ EE$$E'image)urlb64_jsonc *|jdd}tt|jd}tdi|} t || jdd| jdd| jdd\}} } } t | } t | }t}d}tj|xsd | }|td | dt |j| jdd}|j| jdd} |t jk(rJ| td | td tj||| ||| jdd| |i|  }ne|t j k(rR| td| tdt#j||| ||| jddxst$| |i| | }|td| dt |S#t$rtd| dt wxYw)Nr rqr8r4r`)r2r8r4r`z"Invalid image variation provider: z. Supported providers are: r{z_image variation provider has no known model info config - required for getting api keys, etc.: z/API key is required for OpenAI image variationsz0API base is required for OpenAI image variationsre) rr`r4r2rrer8rrrz.API key is required for Topaz image variationsz/API base is required for Topaz image variations) rr`r4r2rrer8rrrr r<)rDr LiteLLMLoggingObjrr,r(r&rIr*r+get_provider_model_info get_api_key get_api_baseOPENAIr image_variationsTOPAZrDEFAULT_REQUEST_TIMEOUT)rr2rYr[r\r^rMr rqrr8rr4 llm_providerimage_variation_providerrrTr}r`s rVrr"sZZ$ 'F0&**=R2ST(1&1N,>z4,PQH#D#K#KK ?NO O  OP P*;;)"&&y$7 3+)   "%F%L%L L ?MN N  NO O+<<)"&&y$7R;R 3+)  01D0EE`bCaD E   OA  01D0EE`bCaD E   s G33Hmaskrr extra_query extra_bodyrer8c t} |jd}|jdd}|jdddu}tdi|}t |xst | \}} }}t j|tj| }|td | |j|tj|}tj||| }|j||t!|d|i|| t#j$|||||| ||| | | xst&||jd  S#t($r}tj*|| |||d}~wwxYw)zW Maps the image edit functionality, similar to OpenAI's images/edits endpoint. rqrjNrFTrr{z image edit is not supported for )r2image_edit_provider_configimage_edit_optional_paramsrr ) r2rrXr"image_edit_optional_request_paramsr8rrrrrre _is_asyncr r7r<)rrDrr%r,rr+get_provider_image_edit_configrr(rIrr/'get_requested_image_edit_optional_paramget_optional_params_image_editrrGrimage_edit_handlerrrJr)rrXr2rrYrZr[r\r^rrrrrer8rM local_varsrqrjrrrRrrimage_edit_request_paramsrUs rV image_editrs0JJ 17  # & -?@S?TUV V&! " I I* U # " @ @+E+E  " 88 !:;!?+!4 9 %77'A/H 3)+'!66::h'   $$ 3 (    sD/D<< E$EE$cKt} tj}d|d<| ,tj||j dd\}} }}t tf|||||||||| | d |}tj}t |j|}|jd|d{}tj|r |d{}|S|}|S7)7 #t$r}tj|| |||d}~wwxYww)a} Asynchronously calls the `image_edit` function with the given arguments and keyword arguments. Parameters: - `args` (tuple): Positional arguments to be passed to the `image_edit` function. - `kwargs` (dict): Keyword arguments to be passed to the `image_edit` function. Returns: - `response` (Any): The response returned by the `image_edit` function. TrNrxr5) rrXrr2rYrZr[r\r^rer8r7)rr=r>rr,rDrrrArBrCrErHrJr)rr2rXrrYrZr[r\r^rrrrrer8rMrrNrRrOrPrQrSrTrUs rV aimage_editrs<>J+ %%'#|  &+2+C+CjnnZ&F, (A"Aq  + 3   &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY D B"C0C1CCCD CD CC D'DDD ) NNNNNNNNiXNNNN)rur.rNN) NNNNNNNNNNNN) NNNNNNNNNNN)Or=rA functoolsrtypingrrrrrr r r httpxrr r rrlitellm.constantsrrrlitellm.exceptionsr*litellm.litellm_core_utils.litellm_loggingr)litellm.litellm_core_utils.mock_functionsrlitellm.llms.base_llmrr&litellm.llms.custom_httpx.http_handlerrr*litellm.llms.custom_httpx.llm_http_handlerrlitellm.llms.custom_llmrr__annotations__ litellm.mainrrrrrr r!litellm.secret_managers.mainr"litellm.types.images.mainr#litellm.types.llms.openair$litellm.types.routerr%litellm.types.utilsr&r'r(r) litellm.utilsr*r+r,r-rr/rWstrintr@rrfloatTimeoutrrr<rgrVrs QQQ GG:H5SKPPI-(:';$;8DC5 )5 5 5 r  CG%)$( !"!%  C=}eC!>>? @  c]  3-  C= 3-SMc]sm# T]!$sC&'% .  CG%)$( !"!% ',!  C=}eC!>>? @  c]  3-  C= 3-SMc]sm# U^!$% , CG%)$( !"!%H H C=H }H eC!>>? @ H c] H 3- H  C=H  3-H SMH c]H smH #H  #sM)* ,!H H V 3 }3 3 l 27 \ \ \ \./ \ 3- \ 3- \\\~ CG%)/3,0+/59)-#b b b  C=b  3- b } b eC!>>? @ b c]b  3-b  3-b DcN+b $sCx.)b c3h(b eE5==01 2b ""##b & =)Cm$;< <='b b J CG%)/3,0+/59)-#J J J  J  3- J } J eC!>>? @ J c]J  3-J  3-J DcN+J $sCx.)J c3h(J eE5==01 2J ""##J &'J J rg