쎒j^PdZddlZddlmZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZmZmZmZmZddlmZddlmZddlZddlZddlZddl Z!ddl"Z!ddl"m#Z#dd l$m%Z%dd l m&Z'ddl(Z(dd l)m*Z*dd l)m+Z+ddl,Z,d dl-m.Z.d dl-m/Z/d dl-m0Z0d dl1m2Z2d dl1m3Z3d dl1m4Z5d dl1m6Z6d dl1m7Z7 ddl8m9Z:n#e;$r ddlZ>dZ=n #e?$rYnwxYw ddl@Z@n #e?$rdZ@YnwxYwerddlAmBZBddlCmDZDddlEmFZFe jGdZHdZIdZJdZKd ZLd ZMd!d"hZNe@e(jOfn e(jOe@jOfZPe@e(jQfn e(jQe@jQfZRe@e(jSe(jTfne(jSe(jTe@jSe@jTfZUGd#d$eVZWd%eeXfd&ZYd'eZeXeXfd%dfd(Z[d)e2d*e2d%e2fd+Z\d'eZeXeXfd,eee]e^fd%dfd-Z_d.eXd/eXd%eXfd0Z`d1eeXdfd%ee#eXffd2Zad3e#d%e#fd4Zbd5eee]e^fd%ee]fd6ZceGd7d8ZdGd9d:Z4d;ZedZjd)ee6d%e.jkfd?ZlGd@dAe(jmZnGdBdCe(joZpGdDdEZqdFdEd3e!jrjsj#d%eXfdGZtdFdEd3e!jrjsj#d%eXfdHZudS)IzBase client for calling HTTP APIs sending and receiving JSON. The BaseApiClient is intended to be a private module and is subject to change. N) Generator) dataclass)Any AsyncIteratorIteratorOptional TYPE_CHECKINGTupleUnion)urlparse) urlunparse) Credentials)mtls) exceptions) BaseModel)ValidationError)_common)errors)version) HttpOptions)HttpOptionsOrDict) HttpResponse)HttpRetryOptions) ResourceScope)connectFTAuthorizedSession)CIMultiDictProxy)CaseInsensitiveDictzgoogle_genai._api_clientii@useuceZdZdZdS)EphemeralTokenAPIKeyErrorz)Error raised when the API key is invalid.N)__name__ __module__ __qualname____doc__FC:\Projects\DSCons\.venv\Lib\site-packages\google\genai\_api_client.pyr&r&{s1111r,r&returnctjdd}tjdd}|r|rtd|p|pdS)zGets the API key from environment variables, prioritizing GOOGLE_API_KEY. Returns: The API key string if found, otherwise None. Empty string is considered invalid. GOOGLE_API_KEYNGEMINI_API_KEYzEBoth GOOGLE_API_KEY and GEMINI_API_KEY are set. Using GOOGLE_API_KEY.)osenvirongetloggerwarning)env_google_api_keyenv_gemini_api_keys r-get_env_api_keyr9snz~~&6==z~~&6==. NNO  91 9T9r,headerscdtj}dtjdz}|d|}d|vr||dvr|d|dz|d<n(d|vr||dvr|dd||d<n d|vr||d<d|vr||dvr|d|dz|d<dSd|vr||dvr|dd||d<dSd|vr||d<dSdS)z1Appends the telemetry header to the headers dict.zgoogle-genai-sdk/z gl-python/r z user-agentzx-goog-api-clientN)r __version__syssplit)r: library_labellanguage_labelversion_header_values r-append_library_version_headersrCs;g&9;;-#+"3"3"5"5a"88.)<9N#N#N&|4GG~GGGL7""0GLW$$ w':; ; ; """W-@%AA    W$$ (; < < < & '::.::    '))#7G    *)r,options patch_optionsc z|}|jpi}|jpi}i|||_tj}|D]M}|dkr t ||d}|t |||.t ||t ||N|jt|j|S)Nr:) model_copyr:r model_fieldskeysgetattrsetattrrC)rDrE copy_optionoptions_headerspatch_options_headershttp_options_keyskey patch_values r-patch_http_optionsrRs""$$+'-2/'/52+ ".3355 77c i-d33K k3 ,,,, k3 5 56666$";#6777 r,timeout_in_secondscb|r*d|vr(ttj||d<dSdSdS)z8Populates the server timeout header in the headers dict.zX-Server-TimeoutN)strmathceil)r:rSs r-populate_server_timeout_headerrXsNE.g=="%di0B&C&C"D"DG EE==r,base_urlpathct|}|jdr|jddn|j}|dr |ddn|}t ||dz|zS)N/r)rZ)r rZendswith startswithr _replace)rYrZ parsed_base base_paths r- join_url_pathrcs""+   " "3 ' 'kss   __S)) 3abbt$ K((i#o.D(EE F FFr,projectc|tjdg\}}|s|}|std||fS)z-Loads google auth credentials and project id.z.https://www.googleapis.com/auth/cloud-platform)scopesz@Could not resolve project using application default credentials.)googleauthdefault ValueError)rd credentialsloaded_project_ids r- load_authrmsc#);#6#6> ?$7$$ +   G  J   g r,rkcLddlm}|||S)Nr)Request)google.auth.transport.requestsrorefresh)rkros r- refresh_authrrs4444444 ggii   r,timeoutc|r|dz }nd}|S)z Converts the timeout to seconds.g@@Nr+)rsrSs r-get_timeout_in_secondsrus( !6) r,ceZdZUeeefed<eed<eed<eeeefefed<dZ e e ed<dS) HttpRequestr:urlmethoddataNrs) r'r(r)dictrU__annotations__r objectbytesrsrfloatr+r,r-rwrwsk S> ((( +++ d3;&''''!'8E?!!!!!r,rwc zeZdZ ddeeeefejddfdeeefdeee ffdZ dd Z defd Z e defd Zdeeddffd Zdeefd Zdeee efddffdZdeeefddfdZdeefdZdeefdZededefdZdS)rNr:zCIMultiDictProxy[str]r response_stream byte_streamcttr|_nttr&fdD|_nt jx} >t|jjr$d D|_n=tj dkr%fdD|_d|_ ||_ ||_dS)Ncbi|]+}|d|,Sz, )joinget_list.0rPr:s r- z)HttpResponse.__init__..sB47#tyy))#..//r,ci|]\}}|| Sr+r+)rrPvalues r-rz)HttpResponse.__init__..&sCCCZS%c5CCCr,rcbi|]+}|d|,Sr)rgetallrs r-rz)HttpResponse.__init__..(s@25#tyy,,--r,) isinstancer{r:_HTTPX_HEADERS_TYPESrIrloaded_requests structuresr itemstyper' status_coderr)selfr:rrrequests_modules ` r-__init__zHttpResponse.__init__s('4  dll G1 2 2 ;B<<>>dll#2444   +?  DC7==??CCCdll g #5 5 59@dl D*D"Dr,r.c8||_|SN)async_segmentssegment_iteratorrs r- __aiter__zHttpResponse.__aiter__1s //11D Kr,cpK |jd{VS#t$rtwxYwr)r __anext__ StopIterationStopAsyncIterationrs r-rzHttpResponse.__anext__5sS(2244 4 4 4 4 4 44  s#5ct|jtsdS|jr |jdsdS||jdS)Nr)rrlist_load_json_from_responsers r-jsonzHttpResponse.json;sY d*D 1 1 T  t';A'> R  ( ()=a)@ A AAr,c# Kt|jtr'|jD]}|r||niVdS|j gEd{VdS|D]}||VdSr)rrrr_iter_response_stream)rchunks r-segmentszHttpResponse.segmentsGs$&-- 3'DD%6;Cd++E222CCCCDD  %mmmmmmmmm--//33%++E22222233r,c*Kt|jtr(|jD]}|r||niWVdS|jg23d{V}|WV 6dS|23d{V}||WV 6dSr)rrrr_aiter_response_stream)rrcs r-rzHttpResponse.async_segmentsSs $&-- 3'DD%6;Cd++E222CCCCCDD  %RR44663333333++E2222222766s A2Bc#Kt|jtr|jEd{VdS|j gEd{VdStd)Nz8Byte segments are not supported for streaming responses.)rrrrjrs r- byte_segmentszHttpResponse.byte_segments`so$"D))!!!!!!!!!!  !mmmmmmmmm  D  r,response_payloadct|dt|D]'}tjt ||||<(dS)Nr)delattrdircopydeepcopyrJ)rr attributes r- _copy_to_dictzHttpResponse._copy_to_dictksX D$%%%YYLL $(M'$ 2J2J$K$Ky!!LLr,c#Ktj}t|jtsF|t|j|js*t dt|jjdd}d}g}t|jtr|j }n|j d}|D]}|s|rd |Vg} | d r+| |td d`|D]}|d kr|d z }|d kr|d z}||z }|dkr|Vd}|r|V|rd |VdSdS) z,Iterates over chunks retrieved from the API.NzFExpected self.response_stream to be an httpx.Response object, but got .rrT)decode_unicode data: {r})rrrr_HTTPX_RESPONSE_TYPESResponse TypeErrorrr' iter_linesrr_appendlen)rrrbalance data_bufferrliners r-rz"HttpResponse._iter_response_streamqs-//O4')>??   '4/1IJJ (  <$.//8 < < <   EGK$&(=>>M,7799oo,77t7LLo    +&& & & &+  " "4H /000 ! 88 Q,'' #XX Q,' tme A   kkk# IIk " """"""##r,cKt|jtp%tot|jtj}|s*t dt|jjdd}d}g}t|jtr |j}| 23d{V}|s|rd |WVg}&| dr+| |tddf|D]}|dkr|d z }|d kr|d z}||z }|dkr|WVd}6|r|WV|rd |WV|d{VdS#|d{VwxYwtrt|jtjrx |jjt" d{V}n4#t $r'|jjd{V}YnwxYw|sn|d }|s|rd |WVg}| dr+| |tdd|D]}|dkr|d z }|d kr|d z}||z }|dkr|WVd}|r|WV|rd |WV|jdS#|jwxYwdSdS)z;Asynchronously iterates over chunks retrieved from the API.z`Expected self.response_stream to be an httpx.Response or aiohttp.ClientResponse object, but got rrrNrrrrrT)max_line_lengthutf-8)rrr has_aiohttpaiohttpClientResponserrr' aiter_linesrr_rraclosecontentreadlineREAD_BUFFER_SIZEdecoderstriprelease) ris_valid_responserrrrrr line_bytess r-rz#HttpResponse._aiter_response_streams" 3 P 4#79OPP   54'((1 5 5 5   EGK$&(=>>X'"'#3)5577       $  IIk******k __X & &    tCMMNN3 4 4 4    aCxxlggclg 4-% \\KKKKE184  ++++  ' +&& & & & &$$&&&&&&&&&&&o$$&&&&&&&&&& 2' g42'/'&  G#3;DD 0 E  JJ G G G $3;DDFFFFFFFFJJJ G  ""7++2244$  IIk******k __X & &    tCMMNN3 4 4 4    aCxxlggclg 4-% \\KKKKEM& P  ++++  ' +&& & & & & $$&&&&&$$&&&&e2'2'2'2'sJE?,E2B1E??FL +G54L5.H&#L%H&&CLL9responsec tj|S#tj$r}tjd||d}~wwxYw)zFLoads JSON from the response, or raises an error if the parsing fails.z0Failed to parse response as JSON. Raw response: N)rloadsJSONDecodeErrorrUnknownApiResponseError)clsres r-rz%HttpResponse._load_json_from_responses\ Z ! !!    * GX G G   sA=A)NN)r.r)r'r(r)r r{rUhttpxHeadersrr~rrrpropertyrrrrrrr}rrrr classmethodrr+r,r-rrs*.'+## sCx. - !  !#S#X#e$####Bs   BC B B B 8 B 3 #tT/2 3 3 3 3 3M#$6 3 3 3 3 YuUCZ'8$'DE    LDf,=L$LLLL 5#Xc]5#5#5#5#nj'M#,>j'j'j'j'Xcc;r,rg?gN@)iiiiiic|tjdddS|jdkrd|_tj|jpt}|jpt tjfd}tj|jpt|j pt|j pt|jpt}||d|tjt"t$jdS) aReturns the retry args for the given http retry options. Args: options: The http retry options to use for the retry configuration. If None, the 'never retry' stop strategy will be used. Returns: The arguments passed to the tenacity.(Async)Retrying constructor. NrT)stopreraiserctt|tjr |jvpt|tSr)rrAPIErrorcode_HTTPX_TRANSIENT_EXC)rretriable_codess r-zretry_args..Cs5Av//MAFo4M- A+ , ,r,)initialmaxexp_basejitter)rretryrwait before_sleep)tenacitystop_after_attemptattempts_RETRY_ATTEMPTShttp_status_codes_RETRY_HTTP_STATUS_CODESretry_if_exceptionwait_exponential_jitter initial_delay_RETRY_INITIAL_DELAY max_delay_RETRY_MAX_DELAYr_RETRY_EXP_BASEr _RETRY_JITTERbefore_sleep_logr5loggingINFO)rDrrrrs @r- retry_argsr2s _/22t D DD G  $W%5%H I I$-I1I/  %----  %  )#;';   //2? ^ ,}    $/ EE   r,c4eZdZdZdeddffd ZddZxZS)SyncHttpxClientzSync httpx client.kwargsr.Nc f|ddtjdi|dSzInitializes the httpx client.follow_redirectsTNr+ setdefaultsuperrrr  __class__s r-rzSyncHttpxClient.__init__X= ($///EGGvr,c |jrdSn#t$rYnwxYw |dS#t$rYdSwxYw)zCloses the httpx client.N) is_closed Exceptionclosers r-__del__zSyncHttpxClient.__del__]ss      d   jjlllll     dd s 3 AAr.Nr'r(r)r*rrr __classcell__rs@r-r r Usest         r,r c4eZdZdZdeddffd ZddZxZS)AsyncHttpxClientzAsync httpx client.r r.Nc f|ddtjdi|dSr rrs r-rzAsyncHttpxClient.__init__mrr,c |jrdSn#t$rYnwxYw tj|dS#t$rYdSwxYwr)rrasyncioget_running_loop create_taskrrs r-rzAsyncHttpxClient.__del__rs      d     ,,T[[]];;;;;     dd s 8A A%$A%rrrs@r-rrjsest         r,rceZdZdZ d:deedeedeejj j deedeedee f d Z d efd Z d efd Zd ed efdZed eefdZd efdZe d;deded eejejffdZe d;deded ejfdZe d;deded ejfdZd efdZd efdZd efdZd ej fdZ!d e"eeffdZ# ddd*dedee d e"ee9ffd4Z?d5ej@d dfd6ZAd>d7ZBd>d8ZCd>d9ZDdS)? BaseApiClientz8Client for calling HTTP APIs sending and receiving JSON.Nvertexaiapi_keyrkrdlocation http_optionsc||_d|_|jtjdd}tjdd}d} ||dv} d} ||dv} | | | | krt jd| | |_n | | |_|s|r|jstd|r|rtdt} t|tr8 tj |} n@#t$r} td| d} ~ wwxYw|rtj|tr|} | jr| jstdtjd d} tjd d}t%}|p| |_|p||_|p||_||_t|_t1j|_i|_|jr,|r$|r"t8d d|_n|r1|s/|s-| s|r)t8d d|_d|_nW|s|r&|s$|r"t8d d|_n-|s+|s)|s'| s|r#|r!t8dd|_| jr| jnd|_|js7|js0|jsd|_n!|jdrd|_|js4|js-|js&t?d\}|_|js||_|jr|jp|j}|s|jstd|jr|jr |jdkr|js d|j_n|jt@vr|jsd|jd|j_n\|jr@|jds&|r|s"|s |j|j_d|_d|_nd|jd|j_d|j_!n.|jstdd|j_d|j_!ddi|j_"|jr>|j#|_|jj"|j|jj"d<|rtI|j| |_n%|jj"tK|jj"|&|jtO|j\}}||_(d|_)|*rd|_+n/|jj,r|jj,|_+nt[d'i||_+|.rd|_/n/|jj0r|jj0|_/ntcd'i||_/|jj0r|jj0|_/ntcd'i||_/i|_2|3r d dl4}|5|jtO|j|_6|.rPd!|j6d"<|jj7tqd#ntq|jj7|j6d$<d%|j6d&<n#tr$rYnwxYwtu|jj;}|<|jtO|j|_=t}j?d'i||_@t}jAd'i||_BdS)(NGOOGLE_GENAI_USE_ENTERPRISEGOOGLE_GENAI_USE_VERTEXAI)true1zWarning: Both GOOGLE_GENAI_USE_ENTERPRISE and GOOGLE_GENAI_USE_VERTEXAI are set with conflicting values. The value of GOOGLE_GENAI_USE_ENTERPRISE will be used.z-Gemini API does not support project/location.zICredentials and API key are mutually exclusive in the client initializer.zInvalid http_optionsz9base_url must be set when base_url_resource_scope is set.GOOGLE_CLOUD_PROJECTGOOGLE_CLOUD_LOCATIONzoThe user provided Google Cloud credentials will take precedence over the API key from the environment variable.zrThe user provided Vertex AI API key will take precedence over the project/location from the environment variables.zqThe user provided project/location will take precedence over the Vertex AI API key from the environment variable.zyThe project/location from the environment variables will take precedence over the API key from the environment variables.globalz.googleapis.comrdz>*GNN(CTJJn  '+1133}Dl  %'--//=@   $& ,,  B    #&  #$  H8HT]H F G GGw     )]],%%,8!,!;L!I!I 888/00a78 ,'1, LL,+ 6&/  E   *..!7>>K:>>"94@@L!##K)kDL, DM)kDL#D$D%>++DLND }m0 ''  A        +    B     x[  B        +      M    $ ,  ) ) m#,## #"$--  ! * *+< = = #"$-l*,*&* %.d$;$;$;! T\  *)$ !\;dmL   )=  J    < $  26-82K2K$3L&K## -4 4 4& 5 F$- E E E ##   &//0ABB  ' ,3 '+&:#   Bt} A A A #(1d$$ \  >   %Qd!'/d$"02D!ED |D\''))dl  # /7;|"#34C-  4d   # /&t'9'ABBB%)%?%? dm$$&@&&"K"%6D!>BD !!##:d  (:-:d*99[99d ""$$G!%d  .G!%!3!Fd!1!F!F4E!F!Fd ,G!%!3!Fd!1!F!F4E!F!Fd.0D     ( ("dm,, )   /  & & ( ( H;?$ 1% 8!)1INI I I 7< (8 8   12D E GH$ 12B C      d0>??L"<< dm$$=D#33l33DK .>>>>Ds+ D!! E+D;;E7B[ [ [ r.cttdsdSt|jo>tjo+tjo|jjp |jj S)Nshould_use_client_certF) hasattrrrLr&rnhas_default_client_cert_sourcerBrQrjrs r-rOz#BaseApiClient._use_google_auth_syncss 41 2 2 U    ' ) )  / 1 1    + Mt/A/M   r,cF ddl}|d}|j}d}n#ttf$rd}YnwxYwt t oP|oN|joGttdo2tj otj o |j j S)Nrgoogle.auth.aio.credentialsTFrn) importlib import_modulerrYAttributeErrorrLrr&rorrnrprBrT)rrsaio_creds_module_has_google_auth_aios r-rRz$BaseApiClient._use_google_auth_asyncs""001NOO  &a   ("""!"  6  6 M 6 D2 3 3 6  ' ) )  6  / 1 1  6 "5 5  s "%;;sessionct|drt|jSt|dr.t|jdrt|jjSdS)z&Returns True if the session is closed.closed _auth_request_closedF)rorLr{r|r})rrys r-_is_session_closedz BaseApiClient._is_session_closedsjw!!" '. ! !!w((1Wy..1''/ 0 00 5r,c|jjr |jjS tj}|j|S#t $rYdSwxYw)zUReturns the aiohttp client session for the current loop (for backward compatibility).N)rBaiohttp_clientr!r"rUr4 RuntimeErrorrloops r-_aiohttp_sessionzBaseApiClient._aiohttp_sessionsj  (/   ..  % ' 'd  # ' ' - --  TTs,A AAcK|jjr |jjStj}|j5|j|}|||rd}dddn #1swxYwY||rl ddl }| dj }| dj }Gdd|}|||}n]#t$rYnQwxYwGddtj}Gd d tj}||d d t" }|r)|j5||j|<dddn #1swxYwY|S)z#Returns the aiohttp client session.Nrrrz"google.auth.aio.transport.sessionsc leZdZdZd fd Zdedededeeefd d f d Ze d e fd Z xZ S)HBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentialszJAdapter to use the client's sync credentials in an AsyncAuthorizedSession.clientr%cVt||_dSr)rr_client)rrrs r-rzQBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.__init__s$gg   #dlllr,requestryrxr:r.NcK|jd{V}d||d<|jjr'|jjjr|jjj|d<dSdSdS)NBearer Authorizationx-goog-user-project)r_async_access_tokenrAquota_project_id)rrryrxr:tokens r-before_requestzWBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.before_requests!L<<>>>>>>>>e):5):):go&,+l/@ L->-... r,cB|jjsdS|jjj SNF)rrAexpiredrs r-validzNBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.valids%\.u2:::r,)rr%) r'r(r)r*rrrUr{rrrLrrrs@r-_RefreshableAsyncCredentialsrs \ \ $ $ $ $ $ $      c3h        " ;t ; ; ;X ; ; ; ; ;r,rc"eZdZefdeddfdZdS)@BaseApiClient._get_aiohttp_session..AiohttpClientSession _warningsr.NcH|js|dd}|j|j|d<dSdSdS)NzUnclosed client session)client_sessionmessagesource_traceback)r{_source_traceback)rrcontexts r-rzHBaseApiClient._get_aiohttp_session..AiohttpClientSession.__del__sR; E$(6g'3.2.D*+++  E E 43r,r'r(r)r<rrr+r,r-AiohttpClientSessionrsG-5 E Es E$ E E E E E Er,rc"eZdZefdeddfdZdS)?BaseApiClient._get_aiohttp_session..AiohttpTCPConnectorrr.Nc|jrdS|jsdSd|jD}|||dd}|j |j|d<dSdS)Nc,g|]}t|Sr+)repr)rrs r- z[BaseApiClient._get_aiohttp_session..AiohttpTCPConnector.__del__..s;;;T!WW;;;r,zUnclosed connector) connector connectionsrr)r}_connsvalues_closer)rrconnsrs r-rzGBaseApiClient._get_aiohttp_session..AiohttpTCPConnector.__del__ s| f; f;;dk&8&8&:&:;;;E KKMMM!$/G %1,0,Bg()))21r,rr+r,r-AiohttpTCPConnectorr sG-5 C Cs C$ C C C C C Cr,r)limitT)r trust_env read_bufsize)rBrr!r"rErUr4r~rRrsrtrAsyncAuthorizedSessionrYr ClientSession TCPConnectorr) rrryrsAsyncCredentialsrrrrs r-_get_aiohttp_sessionz"BaseApiClient._get_aiohttp_sessions (/   ..  # % %D &**400g  !8!8!A!A   $ $ & &M &     &445RSS_ #,#:#:;_#`#`#w  ; ; ; ; ;-= ; ; ;<+*+G+G+M+MNN''    $  E E E E E7#8 E E E C C C C C'"6 C C C&'&))222)    1  ! 1 1)0$  & 1 1 1 1 1 1 1 1 1 1 1 1 1 1 1 Ns64A55A9<A9AC22 C?>C? E//E36E3FrDcd|j}|j}|r|n|r|nd}||rVttdd}|> |}n2#t $r%}t d|Yd}~nd}~wwxYw|btj tj dtj tj d}dttjd tjd tjffd }||||||fS) a,Ensures the SSL context is present in the HTTPX client args. Creates a default SSL context if one is not provided. Args: options: The http options to check for SSL context. vertexai: Whether Vertex AI is enabled. Returns: A tuple of sync/async httpx client args. verifyNget_default_ssl_context6Failed to get default SSL context from google-auth: %s SSL_CERT_FILE SSL_CERT_DIRcafilecapathargsctxr.c*|pi}|s||<d|vrd|d<|}|D]/}|tjtjjjvr||=0|S)-Sets the SSL context in the client args if not set. Does not override the SSL context if it is already set. Args: args: The client args to to check for SSL context. ctx: The SSL context to set. Returns: The client args with the SSL context included. rsN)rr4inspect signaturerClientr parametersrr copied_argsrPrs r- _maybe_setz7BaseApiClient._ensure_httpx_ssl_ctx.._maybe_setYsjb   d XXf  V $  YIIKKk!!### g' (=>>I I I# r,)rjrkr4rJrrr5r6r6create_default_contextr2r3certifiwhererr StringDict SSLContext) rDr&r async_argsr get_ctx_fnrrrs @r-rKz#BaseApiClient._ensure_httpx_ssl_ctx*s"F  D*J   Z^^F+++   { XT#>/7=??CC:>>.11    w)* ^  8  4 :s## s A** B4BBcFd|j}|r|nd}||rVttdd}|> |}n2#t$r%}t d|Yd}~nd}~wwxYw|btjtj dtj tj d}dttjd tjd tjffd }|||S) 1Ensures the SSL context is present in the async client args. Creates a default SSL context if one is not provided. Args: options: The http options to check for SSL context. vertexai: Whether Vertex AI is enabled. Returns: An async aiohttp ClientSession._request args. r6Nrrrrrrrr.c|r|s|pi}||<|}|D]/}|tjtjjjvr||=0|S)r)r4rrrrr_requestrrs r-rz9BaseApiClient._ensure_aiohttp_ssl_ctx.._maybe_sets&))   ""V IIKKk!!### $W%:%CDDO P P# r,rkr4rJrrr5r6r6rr2r3rrrrrrrDr&rrrrrrs @r-rWz%BaseApiClient._ensure_aiohttp_ssl_ctxysO F*J$. 8*.. DC { XT#>/7=??CC:>>.11    w)* ^  8 :j# & && A A;A66A;cFd|j}|r|nd}||rVttdd}|> |}n2#t$r%}t d|Yd}~nd}~wwxYw|btjtj dtj tj d}dttjd tjd tjffd }|||S) rr6Nrrrrrrrr.c|r|s|pi}||<|}|D]+}|tjtjvr |dkr||=,|S)rr6)r4rrr ws_connectrrs r-rz;BaseApiClient._ensure_websocket_ssl_ctx.._maybe_sets&))   ""V IIKKk!!### g' 33> > >3%<<# r,rrs @r-r[z'BaseApiClient._ensure_websocket_ssl_ctxsQ"F*J$. 8*.. DC { XT#>/7=??CC:>>.11    w)* ^  2 :j# & &&rcrto0|jjpidduo |jjduS)N transport)rrBrkr4rTrs r-rVzBaseApiClient._use_aiohttpsJ  <   1 7R < <[ I I   <  2d : r,c|jr|jp|j}|jr |s|jSt |jj}|dS)Nwss)scheme) rdr(r'r:r rBrYr`geturl)rri url_partss r-_websocket_base_urlz!BaseApiClient._websocket_base_urlsl<9DMJdl "$7" !!+455I   U  + + 2 2 4 44r,c|j5|js+t|j\|_}|js||_|jr!t ||jcdddSt d#1swxYwYdS)z/Retrieves the access token for the credentials.r2N0Could not resolve API token from the environment)rErArmrdget_token_from_credentialsr)rrds r- _access_tokenzBaseApiClient._access_tokens  O O  !%.t|%D%D%D"7| ! $, O)$0ABB O O O O O O O OMNNN O O O O O O O O O OsAA2#A22A69A6cKtj}|j5||jvrtj|j|<|j|cdddS#1swxYwYdS)aLazily initializes and returns an asyncio.Lock for async authentication. This method ensures that a single `asyncio.Lock` instance is created and shared among all asynchronous operations that require authentication within the same event loop. N)r!r"rErFrDrs r-_get_async_auth_lockz"BaseApiClient._get_async_auth_locks  # % %D ** T+ + +'.|~~t$  #D )******************s1AA"AcK|js|d{V}|4d{V|jsRetrieves the access token for the credentials asynchronously.Nr2r)rArr! to_threadrmrd async_get_token_from_credentialsr)rasync_auth_lockrds r-rz!BaseApiClient._async_access_token+s   #"7799999999o # # # # # # # #  #-4->. . . ( ( ( ( ( ( $$ W #"DL " # # # # # # # # # # # # # # # # # # # # # # # # # # # M3    K L LLsAB B B  http_methodrZ request_dictcd|D}|D]}||=|rSt|trt|j|}n/t|jtj|}n|j}d}|jr|dkr|drd}|jrW|dsB|s@|js|j r2|j r|j tj ksd|jd|j d|z}|jd|} n |jd|} |j|jst!d |j} t#|d r!|jrt'j||j| } |j r|jr|j s|jr7|j tj krt-| |} n=t-| | } n,|j r%|j tj krt-| |} |jr)|jd rt/d t1|j} |jt!d t7|j| t9|| |j|| S)Nc<g|]}|d|S)rw)r_)rrPs r-rz0BaseApiClient._build_request..Js)PPPcCNN3r&r_rdr(r:r@r COLLECTIONrIrYrjrorrrecursive_dict_updater'rcr&rursr:rXrw) rrrZrr)keys_to_deleterPpatched_http_optionsquery_vertex_base_modelsversioned_pathrYrxrSs r-_build_requestzBaseApiClient._build_requestBsQQP\%6%6%8%8PPPN s   0 L+ . . 1      2   :< H H  "/$ & 5 OOM * * !"&  J ,, J) J\ J"] J  J%<'((C B B$- B B BT Id'/!4zznn.:CCTCCn %-#, . . / //%.h $l33 +# ,7 C  * L*!]* <*  6  % & &Hd++     * 8Mj !|tE|j|r|n|j#gS)Nrrrrrr)max_refresh_attemptsgoogleapis.comsandboxsandbox.googleapis.commtls.sandbox.googleapis.commtls.googleapis.com)ryrxrzr:rsrryrxrr:rsr)$r&rdr(r'rr:rArrzrdumpsrr~rOrUrxrNrprrdefault_client_cert_sourceconfigure_mtls_channel_is_mtlsreplacerryupperrsrP build_requestsendrrraise_for_responsertext) rrrrzrxrclient_cert_sourcer httpx_requests r- _request_oncezBaseApiClient._request_onces )-D } #$, #$- # #.N8J8J8L8L.N.Nl?+  t0A   . 23/;.? ITZ ) * * *Tdd  #,+U33 #2>2CML-...$$"$ !!##$G   ! !c ! )DDDDDD#4#4  !"$ $ $  "<>>  77      ! *E/?3/F/F    &(E  ## ,.CDD#)11$**,,&& 2hh(66$&& 7m#((v(FFh O&&x000 fA((8=/  r,c|rat|tr tdi|n|}|jr7t |j}t jdi|}||j||S||j||SNr+) rr{rrZrrr]rr^rrr)rparameter_modelrlrs r-rzBaseApiClient._requests  ? d + + + % % % % %  &?!/"?@@ !11L11uT'v>>> ;;t)< @ @@r,c RKd}|jra|js|jrS|jsLd|d{V|jd<|jr |jjr|jj|jd<|jrNt|jts-tj |j d}n|j}|r|rV|d{V}|j}|rjt%j}||d{V|jr5d|vr1d|vr|dd}n|dd } |jd|j||j|t3j|j d |jd{V}n##t2jt2jt2jt2j tBj"f$r}tGj$d tKj&d dzd{VtN(d|z|)|j*tW|j|_|d{V}|jd|j||j|t3j|j d |jd{V}Yd}~nd}~wwxYwtXj-.|d{Vt_|dr|j0}tc|j|S|j23|j|j||j|j} |j24| |d{V} tXj-.| d{Vtc| j| S|r|d{V}|j}|rjt%j}||d{V|jr5d|vr1d|vr|dd}n|dd } |jd|j||j|t3j|j d |jd{V}tXj-.|d{V|} t_| dr| j0} tc| j| 5d{VgS#t2jt2jt2jt2j tBj"f$rP}tGj$d tKj&d dzd{VtN(d|z|)|j*tW|j|_|d{V}|jd|j||j|t3j|j d |jd{V}tXj-.|d{V|} t_| dr| j0} tc| j| 5d{VgcYd}~Sd}~wwxYw|j2|j|j|j||jd{V} tXj-.| d{Vtc| j| j5gS)Nrrrrrrrrrtotalrrr z!Retrying due to aiohttp error: %sr5 _responserrryrxr:rrsr+)6r&rdr(r'rr:rArrzrr~rrencoderVrrxrRrr r r r rryr ClientTimeoutrsrXClientConnectorErrorClientConnectorDNSError ClientOSErrorServerDisconnectedErrorauth_exceptionsTransportErrorr!sleeprandomrandintr5rGrWrBrLrrraise_for_async_responserorrrSrrr) rrrrzryrxrrrrclient_responseunwrapped_responseunwrapped_retry_responses r-_async_request_oncez!BaseApiClient._async_request_oncesQ !D } $, $-  6$2244444444 6 6?+   t0A   . 23!  )5 1 1!z,+,,33G<<  WM    GF1133333333  & & ( ( I#>@@ ..            I"2c"9"9CKK*,IccKK 02GHHc" *7? !("*+,2FGGG   7         ((  (  +  !  +  *      a&.A"6"6677 7 7 7 7 7 7 7 ++9A= > > >**$ ..+  13355555555'*7? !("*+,2FGGG   7         ((((((% 6o66x@@@@@@@@@ 8[ ) ) ('(H,h7770>>&  ( ( ?  !% 8 = = !>! !       o66GGGGGGGGGO3_EEE    MM1133333333  & & ( ( I#>@@ ..            I"2c"9"9CKK*,IccKK 02GHHc3 *7? !("*+,2FGGG   7         (88BB B B B B B B B$,  ' 5 5 >!3!=  (1C1H1H1J1J+J+J+J+J+J+J*K     (  +  !  +  *  " " "  a&.A"6"6677 7 7 7 7 7 7 7 ++9A= > > >**$ ..+  13355555555'*7? !("*+,2FGGG   7         (88BB B B B B B B B*2 " -{ ; ; J'?'I $&.-22444444445        ?" J!% 8 @ @&  ( ( !A! !       o66GGGGGGGGGO3o6J5KLLLs::AF<< F F FFr,cK|||||}|||dd{Vfd}|S)NTrFcK23d{V}tj|} |drbtj|}tj|did|d{Vn7#tjj $r t d|zYnwxYwtj |WV6dS)Nr;r<rr=r6)rrr_rrrraise_error_asyncr4r?rr5r@r8r:)rrBrCrs r-async_generatorz=BaseApiClient.async_request_streamed..async_generatorsA!IIIIIIIZ&&    " "; / / J//J/33w++//77        |+    ,,AJN    $   h&6ZHHHHHHHH"sC+A7B1C C)rr0)rrrZrr)rrLrs @r-async_request_streamedz$BaseApiClient.async_request_streameds&&T<L((! T)HIIIII$ ?  r,r) file_path upload_url upload_sizect|tjr|||||St |d5}|||||cdddS#1swxYwYdS)a%Transfers a file to the given URL. Args: file_path: The full path to the file or a file like object inherited from io.BytesIO. If the local file path is not found, an error will be raised. upload_url: The URL to upload the file to. upload_size: The size of file content to be uploaded, this will have to match the size requested in the resumable upload request. http_options: The http options to use for the request. returns: The HttpResponse object from the finalize request. rNrbN)rioIOBase _upload_fdopen)rrOrPrQr)files r- upload_filezBaseApiClient.upload_file*s,)RY'' __ Z<  4  D *k                     sA**A.1A.rXcd}|r|n|j}t|tr|dnt |dd}|rLt |}t |}t ||j|j } | t} d} | rt| } d} | |z|kr| dz } t|tr|dn|j } | 2t|jtr |jj n |jj } t| } t|tr|d int |d ipi}|rt|ni}|| t!|t!| d t#|| d}|t$krp|jd ||| | }|jd rn6t,t.|zz}|dz }t1j||t$kp|| z }|jd dkrn||krt5dt6j||jd dkrt5dt=|j|jgS)aTransfers a file to the given URL. Args: file: A file like object inherited from io.BytesIO. upload_url: The URL to upload the file to. upload_size: The size of file content to be uploaded, this will have to match the size requested in the resumable upload request. http_options: The http options to use for the request. returns: The HttpResponse object from the finalize request. rrYNrnetlocTupload , finalizersr:zX-Goog-Upload-CommandzX-Goog-Upload-OffsetzContent-LengthPOSTrx-goog-upload-statusractiveFAll content has been uploaded, but the upload status is not finalized.final6Failed to upload file: Upload status is not finalized.r) rBrr{r4rJr r r`rr\read CHUNK_SIZErrsruupdaterUrXMAX_RETRY_COUNTrPrr:INITIAL_RETRY_DELAYDELAY_MULTIPLIERtimer'rjrrrrr)rrXrPrQr)offsetrYra parsed_upload file_chunk chunk_sizeupload_commandrsrS user_headersupload_headers retry_countr delay_secondss r-rVzBaseApiClient._upload_fdJs[(F#/G<> O P PP (8=/ J J JJr,c|d|i|}d}|jr;t|jtst j|j}n|j}|j|j|j |j ||j }tj |t|j |gjdS)Downloads the file data. Args: path: The request path with query params. http_options: The http options to use for the request. returns: The file bytes r4rZrr)Nrrr)rrzrr~rrrPrryrxr:rsrrrrrgr)rrZr)rrzrs r- download_filezBaseApiClient.download_files&& Dr 'L)-D!  )5 1 1!z,+,, !))"  $$ *H O&&x000 x}}&7   ! r,cpKt|tjr|||||d{VSt j|}|dd{V}|4d{V|||||d{Vcdddd{VS#1d{VswxYwYdS)aTransfers a file asynchronously to the given URL. Args: file_path: The full path to the file. If the local file path is not found, an error will be raised. upload_url: The URL to upload the file to. upload_size: The size of file content to be uploaded, this will have to match the size requested in the resumable upload request. http_options: The http options to use for the request. returns: The HttpResponse object from the finalize request. rNNrS)rrTrU_async_upload_fdanyioPathrW)rrOrPrQr)rXfds r-async_upload_filezBaseApiClient.async_upload_files*)RY''  (( Z<)Z " "d4 b        **  Kl+                                      s3B%% B/2B/c Kd}|r|n|j}t|tr|dnt |dd}|rLt |}t |}t ||j|j }| r| d{V} t|tj r|t} n |td{V} d} | rt!| } d} | |z|kr| dz } t|tr|dn|j} | 2t|jtr |jjn |jj} t%| }t|tr|d int |d ipi}|rt|ni}|| t)|t)| d t+||d}d}|t,kr| d || |t1j|  d{V}|jdrnL  Y + + +, 266<"  0<Cl+++%3$'KK!*oo     '~7IJJJ O++"??$+2DEEE +        (  ! !"8 9 9  -1A;1NO-  + m,, , , , , , , ,O++ *   $$%;<<HH  &    B H O 4 4X > >>>>>>>>  ""#9::gEE D     8==??-B-B-B-B-B-B,C E dBI & & 3yy,,**!YYz22222222*  ':*!  + - - L (.,-- &L  Y ' ' '%  ?D.55.d ((%-  4G<<,-- >L  Y + + +, 266<"  0<Cl+++%3$'KK!*oo     '~7IJJJ O++"&":"B"B $( #C# #       /)!)*!)--.DEE* -1A;1NO-  + *] # # #!O++$ *  ''++,BCCxOO  &    EE N O 4 4_ E EEEEEEEE  %%))*@AAWLL D     !O4H3Ir,c K|d|i|}d}|jrNt|jts-t j|jd}n|j}|r|d{V}| |j |j |j |tj|jd{V}t j|d{Vt'|j |d{VgjdS|j |j |j |j ||j d{V}t j|d{Vt'|j |gjdS) rxr4ryNrrrrzrr)rrzrr~rrrrVrrryrxr:rr rsrrr*rrgrrS)rrZr)rrzryrr+s r-async_download_filez!BaseApiClient.async_download_files<&& Dr 'L!D!  )5 1 1!z,+,,33G<<  //11111111g$&'l.BCCC 'h O 4 4X > >>>>>>>>   x}})>)>)>)>)>)>(?A #6>>$&& ?o O 4 4_ E EEEEEEEE   !0D0D0F0F/GAr,response_modelcdSrr+)rrs r-_verify_responsezBaseApiClient._verify_responsesDr,c|jjs |jr|j|jr|jdSdS)zCloses the API client.N)rBrQrPrrNrs r-rzBaseApiClient.closesa   *!t/A!     ' $$&&&&&''r,cK|jjs|jd{V|jr|jjs t j}n#t$rd}YnwxYwt|j D]w\}}||kr| d{V&| r9 t j | |c#t$rYowxYwtdSdSdS)zCloses the API async client.N)rBrTrSrrUrr!r"rrrr is_runningrun_coroutine_threadsafe)r current_looprrys r-rzBaseApiClient.aclose s`   0.  $ + + - --------  d&8&G /11    6 < < > >??-$ <            __      ,W]]__d C C C C    D       s$A A%$A% 'C11 C>=C>c |jjs|n#t$rYnwxYw t j|dS#t$rYdSwxYw)zCloses the API client when the object is garbage collected. ADK uses this client so cannot rely on the genai.[Async]Client.__del__ for cleanup. N)rBrQrrr!r"r#rrs r-rzBaseApiClient.__del__ s   ,     d     ,,T[[]];;;;;     dd s # 008A.. A<;A<)NNNNNN)Frrr)Er'r(r)r*rrLrUrgrhrkrrrrOrRrr~rrr staticmethodrr rrrKrWr[rVrrr!rDrr rr{r}rwrrrrr.r0r4r8rrrDrHrMrTrUintrYrVr~r{rr~ AsyncFiler}rrrrrrr+r,r-r%r%~s@@"&#CG# $26K?K?K?}K?FK3?@ K? } K?  K?./K?K?K?K?Z T    d(        }    8 b bbbbHLLLL W !3 34LLL<L\>'>'>'>'>'>'>'<>'@A'A'A'A'A'A'A'<A'FD535555 OS O O O O *', * * * *MsCxMMMM837 iii if% i ./ i  iiii\==== ====D37 AAA./A A  AAAA*7<kMkM%kM/3kMkMkMkMkM`37 ./   *'*<37 III If% I ./ I  IIII*37 OOO Of% O ./ O $,- OOOOD37 GGG Gf% G ./ G  GGGG,37 !!! !f% ! ./ !  !!!!R37    sBI~&    ./      L37 aKaKaK IaKaK aK ./ aKaKaKaKaKN37 %%% %./ %  S%Z %%%%Z37    sBI~&    ./      N37 III ")U_S11 2II I ./ IIIII^37 555 5./ 5  S%Z 5555t W-> 4    ''''*      r,r%rcv|js|jst||jstd|jS)=Refreshes the authentication token for the given credentials.r)rrrrr)rrks r-rr. sK  1  K I J JJ  r,cDK|js|jst|d{V}|4d{V|js|js tjt |d{Vdddd{Vn#1d{VswxYwY|jst d|jS)rNr)rrrr!rrrr)rrkrs r-rr; sU ; 1;"7799999999O;;;;;;;;  ;K$5; k:::::::::;;;;;;;;;;;;;;;;;;;;;;;;;;;  K I J JJ  s/A44 A>A>)vr*r!collections.abcrr dataclassesrrrTrrrVr2r(r6r>rCrmtypingrrrrr r r urllib.parser r r<r~r google.authrggoogle.auth.credentialsrgoogle.auth.transportrrr%rpydanticrrrrrrrtypesrrrr8rrwebsockets.asyncio.clientrrModuleNotFoundErrorwebsockets.clientrrrYhttpx2rpr multidictrrequests.structuresr getLoggerr5rhrrjrkrlrHrrrrTimeoutException ConnectErrorrrjr&rUr9r{rCrRrrrXrcrmrrrurwrrrrrrrrrr  AsyncClientrr%rhrkrrr+r,r-rsJ" %%%%%% !!!!!!   VVVVVVVVVVVVVVVVVV!!!!!!###### //////&&&&&&555555 $$$$$$$$$$$$222222###### 6=======666555555556  ...++$---- &&&6>>>>>>((((((555555  5 6 6  !4L U^enfo-NU]U]FN,K ~ U/00    22222 222 :#::::"8DcN8t8888@ )48E #s(^E19%s :K1LE EEEEGCGsGsGGGG%T *u[#5E/F"kk  eE3J' (  e_     """"""" "NNNNNNNNj  !12 w7I    F     el   *     u(   (m m m m m m m m `5   (4       (4 s6 C C! C!'C..C65C6:C??D D