jx%FUddlZddlZddlmZddlmZmZmZmZm Z m Z m Z m Z m Z ddlZddlZddlmZmZmZmZddlmZddlmZddlmZddlmZdd lmZdd lmZm Z dd 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/m0Z0m1Z1ddl2m3Z3ddl4m5Z5ddl6m7Z7ddl8m9Z9ddl:m;Z;mZ>ddl?m@Z@mAZAmBZBmCZCddlDmEZEede@fdZFe d8deGde eGde eHde e eGe7fde eGde eGd e eGd!e eGd"e eGd#e eGd$e eGd%e d&deeee@ffd'ZIe d8d(d)deGde eGde eHde e eGe7fde eGde eGd e eGd!e eGd"e eGd#e eGd$e eGd%e d(de@fd*ZIe d8deGde eGde eHde e eGe7fde eGde eGd e eGd!e eGd"e eGd#e eGd$e eGde e@eeee@fffd+ZIede@fd,ZJe d9d-e( $ 1 #2249JKK ,0 mT *$5}5H  } 5$H   /**H  Q L+  1=X 3 "    sN4D:A4D+D,AD;D<DD:DD D7D22D77D:promptr3nqualityresponse_formatsizestyleuserapi_keyr5 api_versionr4Tc  yNr=rYr3rZr[r\r]r^r_timeoutr`r5rar9r4rNs rWrArAr&F)r4c  yrcr=rds rWrArArfrgc   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| }$tdC|xs|||||||| |$d |#}%t!dCi| }&|}'|'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| tjj8tjj:tjj<tjj>tjj@fvr/|$tCd&| tDjG| |||$|%| |&|||' S| d(k(r|d)d*l$m%}+|+jM| } |+jO| } |||%d <d#| d$}*|*jD]\}!}"|!|vs |"||!<t5j6|||| | d|||%|| ||||&%}|S| dk(s/| tjPjRk(s| tjTvr8| jd+d},tWj6|||| xs|| ||%||,||, }|S| d-k(r.| tYd.t[j6|||||%|||| | / }|S| d0k(r|%j3d1dxs1|%j3d2dxstj\xs t)d3}-|%j3d4dxs1|%j3d5dxstj^xs t)d6}.|%j3d7dxs|%j3d8dxs t)d9}/| xs*tj*xst)d:xs t)d;} taj6|||||%||-|.|/|| |< }|S| tjbvrd}0tjdD]}1|1d=| k(s |1d>}0|0 tg|| ?|d@ur2d}2|ti|tjr|}2|0jm||| | ||%|||2A }|Sd}3|ti|tnr|}3|0j7||| | ||%|||3A }|Scc}"}!w#tX$r}4tq|| |4t| Bd}4~4wwxYw)Dzv Maps the https://api.openai.com/v1/images/generations endpoint. Currently supports just Azure + OpenAI. r4Flitellm_call_idN logger_fn mock_responseproxy_server_requestazure_ad_token_provider model_infometadatalitellm_logging_objr extra_headersheaders base_model)r3r9r5dall-e-2r7r3)r_rr5rar` deployment_id organizationbase_urldefault_headersre max_retriesrZr[r]r^r3provider) r3rZr[r\r]r^r_r9provider_config) reazurerjrkrmrorppreset_cache_keystream_responser3r_optional_paramslitellm_paramsr9r9)r3rlr~AZURE_API_TYPEAZURE_API_BASEAZURE_API_VERSIONAZURE_OPENAI_API_KEY AZURE_API_KEYazure_ad_tokenAZURE_AD_TOKENzapplication/json)z Content-Typezapi-key)r3rYrer`r5rrn logging_objrmodel_responserar4rrsrz-image generation config is not supported for ) r`r3rY image_generation_provider_config(image_generation_optional_request_paramsr9rrrerazure_air)AzureFoundryModelInforw) r3rYrer`r5rrrrwr4rbedrockz!Model needs to be set for bedrock) r3rYrerrrr4rr5r` vertex_aivertex_projectvertex_ai_projectVERTEXAI_PROJECTvertex_locationvertex_ai_locationVERTEXAI_LOCATIONvertex_credentialsvertex_ai_credentialsVERTEXAI_CREDENTIALSVERTEXAI_API_BASEVERTEX_API_BASE) r3rYrerrrrrrr4r5rr|custom_handlerr3r9T) r3rYr`r5rrrrerr8r=)9localsrEupdatelitellmutilsr+r-_hidden_paramsr*itemsr) _member_map_valuesr,$get_provider_image_generation_configr.rupdate_environment_variablesmodel_call_detailsrr#r5rar` azure_keypoprrARECRAFTAIMLGEMINIFAL_AIRUNWAYMLrJrimage_generation_handler"litellm.llms.azure_ai.common_utilsr get_api_base get_api_key LITELLM_PROXYvalueopenai_compatible_providersr rKrrrr"_custom_providerscustom_provider_maprrGrrXrr)5rYr3rZr[r\r]r^r_rer`r5rar9rNrMr4rjrkrlrmrnrorprqrrrrsrtrdynamic_api_key openai_paramsrdefault_paramskvnon_default_paramsimage_generation_configrlitellm_params_dictloggingapi_typerryrrwrrrritemasync_custom_client custom_clientrVs5 rWrArAs2Y 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  !3"O(--/1G#!"GAJ04DD!-(?/ /-' /2NTm!  ( (  % %  ' '  ' '  ) ) %  '. CDWCXY$<<1H9H$72/=  !J . P,99(CH+77@G(3@0!3"O(--/1G#!"GAJ04DD!#(?/ /-' /2N\y 8 +"l&@&@&F&FF"g&I&II+1**^T*JL4EE2?!/ /-) / NlS!I -} CDD5FF/ /- /! NLu!K /##$4d;6"&&':DA6))6""45  ##$5t<7"&&';TB7**7""56  ##$8$?:"&&'>E:!"89 5##5!"565""34 5EE/ /-0 2#5 /! N@c 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 "A  B   3 $h    syE7["9 [[ B<["C[",[" B/[":A[",[";A+["'2["D["./["A ["*1["["" \+\\cKtj}|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. r3Nr9T async_callr5r6r7r8r=)r>r?rErimage_variationrBrCrDr-rFrGrHr+rIrKrrLs rWaimage_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)Nrrqr9r5r`)r3r9r5r`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`r5r3rrer9rrrz.API key is required for Topaz image variationsz/API base is required for Topaz image variations) rr`r5r3rrer9rrrrr=)rEr LiteLLMLoggingObjrr-r)r'rJr+r,get_provider_model_inforrOPENAIr!image_variationsTOPAZrDEFAULT_REQUEST_TIMEOUT)rr3rZr\r]r_rNrrqrr9rr5 llm_providerimage_variation_providerrrUr}r`s rWrrNsZZ$ '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_bodyrer9c t} |jd}|jdd}|jdddu}t|tr|n|g}|jd}i}t|t r|j |t| t r|j | |r t |} tdi|}t|xst| \}} }}tj|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. rqrjNrFTrsrr{z image edit is not supported for )r3image_edit_provider_configimage_edit_optional_paramsrr) r3rrYr"image_edit_optional_request_paramsr9rrrrrre _is_asyncrr8r=)rrErrGlistrHrr&r-rr,get_provider_image_edit_configrr)rJr0'get_requested_image_edit_optional_paramget_optional_params_image_editrrimage_edit_handlerrrKr)rrYr3rrZr[r\r]r_rrrrrer9rN local_varsrqrjrimagesheaders_from_kwargsmerged_extra_headersrrSrrimage_edit_request_paramsrVs rW image_editrs0JW 17r?rr-rErGrrrrBrCrDrFrIrKr)rr3rYrrZr[r\r]r_rrrrrer9rNrrOrSrrPrQrRrTrUrVs rW aimage_editr!sN>J- %%'#|  &+2+C+CjnnZ&F, (A"Aq%UD1w  + 3   &&(#CGGT2"2249JKK   } -**H%HL+  $$ 3 (    sY DB7C3C/C3$C1%C3*D+C3.D/C31C33 D<DDD) NNNNNNNiXNNNN)rur/rNN) NNNNNNNNNNNN) NNNNNNNNNNN)Pr>rB functoolsrtypingrrrrrr r r r httpxrr rrrlitellm.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.mainrrrrr r!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.rr0rXstrintrArrfloatTimeoutrrr=rgrWrsWWW GG:H5SKPPI-(:';$;8DC5 )5 5 5 v  CG%) !"!%  C=}eC!>>? @  c]  3-  C= 3-c]sm#T]"sC&'# .  CG%) !"!%',  C=}eC!>>? @  c]  3-  C= 3-c]sm#U^"# . CG%) !"!%q q C=q }q eC!>>? @ q c] q 3- q  C=q  3-q c]q smq #q   c3 %&(q q h 3 }3 3 l 27 \ \ \ \./ \ 3- \ 3- \\\~ CG%)/3,0+/59)-#o DO+ ,o o  C=o  3- o } o eC!>>? @ o c]o  3-o  3-o DcN+o $sCx.)o c3h(o eE5==01 2o ""##o & =)Cm$;< <='o o d CG%)/3,0+/59)-#L DO+ ,L L  L  3- L } L eC!>>? @ L c]L  3-L  3-L DcN+L $sCx.)L c3h(L eE5==01 2L ""##L &'L L rg