+jQ6dZddlZddlmZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZmZmZmZmZddlmZddlmZddlZddlZddlZddl Z!ddl"Z!ddl"m#Z#dd l$m%Z%dd l&m'Z'dd l m(Z)ddl*Z*dd l+m,Z,dd l+m-Z-ddl.Z.ddl/m0Z0ddl1Z1ddl2m3Z3ddl2m4Z4ddl2m5Z5ddl6m7Z7ddl6m8Z8ddl6m9Z:ddl6m;Z;ddl6mZ?n#e@$r ddlAm>Z?YnwxYwdZB ddlCZCdZBn #eD$rYnwxYwer ddlEmFZFddlGmHZHe jIdZJdZKdZLd ZMdZNd!ZOd"d#hZPGd$d%eQZRd&eeSfd'ZTd(eUeSeSfd&dfd)ZVd*e7d+e7d&e7fd,ZWd(eUeSeSfd-eeeXeYfd&dfd.ZZd/eSd0eSd&eSfd1Z[d2eeSdfd&ee#eSffd3Z\d4e#d&e#fd5Z]d6eeeXeYfd&eeXfd7Z^eGd8d9Z_Gd:d;Z9dZbd!ZcdZdd?Zed*ee;d&e3jffd@ZgGdAdBe*jhZiGdCdDe*jjZkGdEdFZldGdFd4e!jmjnj#d&eSfdHZodGdFd4e!jmjnj#d&eSfdIZpdS)JzBase 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)AuthorizedSession) exceptions) BaseModel)ValidationError)CaseInsensitiveDict)_common)errors)version) HttpOptions)HttpOptionsOrDict) HttpResponse)HttpRetryOptions) ResourceScope)connectFT)CIMultiDictProxyAsyncAuthorizedSessionzgoogle_genai._api_clientii@useuceZdZdZdS)EphemeralTokenAPIKeyErrorz)Error raised when the API key is invalid.N)__name__ __module__ __qualname____doc__\/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/_api_client.pyr(r(bs1111r.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_keyr;hsnz~~&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_headersrEys;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_fieldskeysgetattrsetattrrE)rFrG copy_optionoptions_headerspatch_options_headershttp_options_keyskey patch_values r/patch_http_optionsrTs""$$+'-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<rUs r/populate_server_timeout_headerrZsNE.g=="%di0B&C&C"D"DG EE==r.base_urlpathct|}|jdr|jddn|j}|dr |ddn|}t ||dz|zS)N/r)r\)r r\endswith startswithr _replace)r[r\ parsed_base base_paths r/ join_url_pathres""+   " "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)rf credentialsloaded_project_ids r/ load_authrosc#);#6#6> ?$7$$ +   G  J   g r.rmcLddlm}|||S)Nr)Request)google.auth.transport.requestsrqrefresh)rmrqs r/ refresh_authrts4444444 ggii   r.timeoutc|r|dz }nd}|S)z Converts the timeout to seconds.g@@Nr-)rurUs r/get_timeout_in_secondsrws( !6) r.ceZdZUeeefed<eed<eed<eeeefefed<dZ e e ed<dS) HttpRequestr<urlmethoddataNru) r)r*r+dictrW__annotations__r objectbytesrurfloatr-r.r/ryrysk S> ((( +++ d3;&''''!'8E?!!!!!r.ryc zeZdZ ddeeeefejdefdee efdee e ffdZ ddZ de fd Z ede fd Zdee ddffd Zdee fd Zdeee e fddffd ZdeeefddfdZdeefdZdeefdZede de fdZdS)rNr<zCIMultiDictProxy[str]response_stream byte_streamcttr|_nttjr&fdD|_nvtt r$dD|_n=tj dkr%fdD|_d|_ ||_ ||_ dS)Ncbi|]+}|d|,Sz, )joinget_list.0rRr<s r/ z)HttpResponse.__init__..sB47#tyy))#..//r.ci|]\}}|| Sr-r-)rrRvalues r/rz)HttpResponse.__init__.. sCCCZS%c5CCCr.r cbi|]+}|d|,Sr)rgetallrs r/rz)HttpResponse.__init__.. s@25#tyy,,--r.) isinstancer}r<httpxHeadersrKritemstyper) status_coderr)selfr<rrs ` r/__init__zHttpResponse.__init__s'4  dll GU] + + ;B<<>>dll G0 1 1CC7==??CCCdll g #5 5 59@dl D*D"Dr.r0c8||_|SN)async_segmentssegment_iteratorrs r/ __aiter__zHttpResponse.__aiter__s //11D Kr.cpK |jd{VS#t$rtwxYwr)r __anext__ StopIterationStopAsyncIterationrs r/rzHttpResponse.__anext__sS(2244 4 4 4 4 4 44  s#5ct|jtsdS|jr |jdsdS||jdS)Nr)rrlist_load_json_from_responsers r/jsonzHttpResponse.jsonsY 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.segments*s$&-- 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_segments6s $&-- 3'DD%6;Cd++E222CCCCCDD  %RR44663333333++E2222222766s A2Bc#Kt|jtr|jEd{VdS|j gEd{VdStd)Nz8Byte segments are not supported for streaming responses.)rrrrlrs r/ byte_segmentszHttpResponse.byte_segmentsCso$"D))!!!!!!!!!!  !mmmmmmmmm  D  r.response_payloadct|dt|D]'}tjt ||||<(dS)Nr)delattrdircopydeepcopyrL)rr attributes r/ _copy_to_dictzHttpResponse._copy_to_dictNsX D$%%%YYLL $(M'$ 2J2J$K$Ky!!LLr.c#Kt|jtjsIt|jtjs*t dt |jjdd}d}g}t|jtjr|j}n|jd}|D]}|s|rd |Vg} | dr+| |tdd `|D]}|d kr|d z }|d kr|d z}||z }|dkr|Vd}|r|V|rd |Vd Sd S) z,Iterates over chunks retrieved from the API.zFExpected self.response_stream to be an httpx.Response object, but got .rrT)decode_unicode data: N{r}) rrrResponserequests TypeErrorrr) iter_linesrraappendlen)rrbalance data_bufferrliners r/rz"HttpResponse._iter_response_streamTs 4'88 d*H,= > >  <$.//8 < < <   EGK$&77M,7799oo,77t7LLo    +&& & & &+  " "4H /000 ! 88 Q,'' #XX Q,' tme A   kkk# IIk " """"""##r.c6Kt|jtjp%tot|jt j}|s*tdt|jj dd}d}g}t|jtjr |j 23d{V}|s|rd |WVg}&| dr+| |tddf|D]}|dkr|d z }|d kr|d z}||z }|dkr|WVd}6|r|WV|rd |WV|jd{VdS#|jd{VwxYwtr_t|jt jrA |jjt$ d{V}|sn|d }|s|rd |WVg}u| 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)rrrr has_aiohttpaiohttpClientResponserrr) aiter_linesrrarraclosecontentreadlineREAD_BUFFER_SIZEdecoderstriprelease)ris_valid_responserrrrr line_bytess r/rz#HttpResponse._aiter_response_streams6"4#7HHP 4#79OPP   54'((1 5 5 5   EGK$&77R'!,.::<<       $  IIk******k __X & &    tCMMNN3 4 4 4    aCxxlggclg 4-% \\KKKKE1=4  ++++  ' +&& & & & &"))+++++++++++d"))++++++++++ -' g4-'*'! !19BB.C        *  ""7++2244$  IIk******k __X & &    tCMMNN3 4 4 4    aCxxlggclg 4-% \\KKKKEC! F  ++++  ' +&& & & & & $$&&&&&$$&&&&[-'-'-'-'s&F 4E :B1F !F-DK99Lresponsec 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)r0r)r)r*r+r r}rWrrrrrrrrpropertyrrrrrrrrrrr classmethodrr-r.r/rrs*.'+## sCx. - !  #S#X#e$####:s   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 1#Xc]1#1#1#1#fb'M#,>b'b'b'b'Hcc;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)stopreraiserct|tjr |jvp%t|tjtjfSr)rrAPIErrorcoderTimeoutException ConnectError)rretriable_codess r/zretry_args..s@Av//MAFo4ME A.0BC D Dr.)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_logr7loggingINFO)rFrrrrs @r/ retry_argsr s _/22t D DD G  $W%5%H I I$-I1I/  %EEEE  %  )#;';   //2? ^ ,}    $/ EE   r.c4eZdZdZdeddffd ZddZxZS)SyncHttpxClientzSync httpx client.kwargsr0Nc f|ddtjdi|dSzInitializes the httpx client.follow_redirectsTNr- setdefaultsuperrrr  __class__s r/rzSyncHttpxClient.__init__/= ($///EGGvr.c |jrdSn#t$rYnwxYw |dS#t$rYdSwxYw)zCloses the httpx client.N) is_closed Exceptionclosers r/__del__zSyncHttpxClient.__del__4ss      d   jjlllll     dd s 3 AAr0Nr)r*r+r,rrr __classcell__rs@r/rr,sest         r.rc4eZdZdZdeddffd ZddZxZS)AsyncHttpxClientzAsync httpx client.r r0Nc f|ddtjdi|dSr r rs r/rzAsyncHttpxClient.__init__Drr.c |jrdSn#t$rYnwxYw tj|dS#t$rYdSwxYwr)rrasyncioget_running_loop create_taskrrs r/rzAsyncHttpxClient.__del__Is      d     ,,T[[]];;;;;     dd s 8A A%$A%rrrs@r/rrAsest         r.rc`eZdZdZ d8deedeedeejj j deedeedee f d Z d efd Z d efd Zd ed fdZeded eejejffdZeded ejfdZeded ejfdZd efdZd efdZd efdZd ejfdZd eeeffdZ d9dedede!ee"fdee d e#f dZ$ d:de#ded e%fdZ& d;de#dee ded e%fd Z' d:de#ded e%fd!Z( d;de#dee ded e%fd"Z)d ejfd#Z* d9dedede!ee"fdee d e+f d$Z, d9dedede!ee"fdee d e-e+ddff d%Z. d9dedede!ee"fdee d e+f d&Z/ d9dedede!ee"fdee d ef d'Z0dd(d)eee1j2fd*ed+e3dee d e%f d,Z4dd(d-e1j2d*ed+e3dee d e%f d.Z5dd(dedee d eee6ffd/Z7dd(d)eee1j2fd*ed+e3dee d e%f d0Z8dd(d-ee1j2e9j:efd*ed+e3dee d e%f d1Z;dd(dedee d eee6ffd2Zd|j$|_|jj#|j|jj#d<|rtK|j| |_n%|jj#tM|jj#|'|j\}}||_(d|_)|*rd|_+n/|jj,r|jj,|_+nt[d&i||_+|.rd|_/n/|jj0r|jj0|_/ntcd&i||_/|jj0r|jj0|_/ntcd&i||_/d|_2|3r ddl4}|jj5r|jj5|_2|6|j|_7|.rPd |j7d!<|jj8tsd"nts|jj8|j7d#<d$|j7d%<n#tt$rYnwxYwtw|jj<}|=|j|_>tj@d&i||_AtjBd&i||_CdS)'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.zNProject/location and API key are mutually exclusive in the client initializer.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.comrfz>*GNN(CTJJn  '+1133}Dl  %'--//=@   $& ,,  B    #&  #$   8            )]],%%,8!,!;L!I!I 888/00a78 ,'1, LL,+ 6&/  E   *..!7>>K:>>"94@@L!##K)kDL, DM)kDL#D$D%>++D48DBFD' }b0   A    KW  B      x [   B    K[  M    $ ,  ) ) m#,## #"$--  ! * *+< = = #"$-l*,*&* %.d$;$;$;! T\  *)$ !\;dmL   )=  J    , -833$4&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%)%?%? &&"K"%6D!<@D !!##:d  (:-:d*99[99d ""$$G!%d  .G!%!3!Fd!1!F!F4E!F!Fd ,G!%!3!Fd!1!F!F4E!F!FdaeD      , D"&"4"C$   ( (); < < /  & & ( ( H;?$ 1% 8!)1INI I I 7< (8 8   12D E GH$ 12B C      d0>??L"<>>>Ds+D D;&D66D;B$Z77 [[r0cttdsdSt|jo>tjo+tjo|jjp |jj S)Nshould_use_client_certF) hasattrrboolr%rmhas_default_client_cert_sourcer@rOrirs r/rMz#BaseApiClient._use_google_auth_syncZss 41 2 2 U    ' ) )  / 1 1    + Mt/A/M   r.cttoN|joGttdo2t jot jo |jj S)Nrm) rorr%rnrrmrpr@rRrs r/rPz$BaseApiClient._use_google_auth_asyncfsm  6 M 6 D2 3 3 6  ' ) ) 6  / 1 1  6 "5 5   r.)zaiohttp.ClientSessionr"cK|j_|rK ddlm}ddlm}Gdd|}||||_|jS#t $rYnwxYw|s|j*|jjs|jj rRGddtj }Gd d tj }||d d t |_|jS)z#Returns the aiohttp client session.Nrrr!c 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)rrurs r/rzQBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.__init__}s$ GG     !DLLLr.requestr{rzr<r0NcK|jd{V}d||d<|jjr'|jjjr|jjj|d<dSdSdS)NBearer Authorizationx-goog-user-project)rw_async_access_tokenr?quota_project_id)rrxr{rzr<tokens r/before_requestzWBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.before_requests,::<<<<<<<  ,+<+,,,     r.cB|jjsdS|jjj SNF)rwr?expiredrs r/validzNBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.valids%<, U|088 8r.)rur$) r)r*r+r,rrrWr}rrrorrrs@r/_RefreshableAsyncCredentialsrtzs Z Z " " " " " "   *- 47 BFsCx.       9T 9 9 9 8 9 9 9 9 9r.rc"eZdZefdeddfdZdS)@BaseApiClient._get_aiohttp_session..AiohttpClientSession _warningsr0NcH|js|dd}|j|j|d<dSdSdS)NzUnclosed client session)client_sessionmessagesource_traceback)closed_source_traceback)rrcontexts r/rzHBaseApiClient._get_aiohttp_session..AiohttpClientSession.__del__sR C"&4G%1,0,Bg()))  C C 21r.r)r*r+r:rrr-r.r/AiohttpClientSessionrsG+3 C CS C C C C C C Cr.rc"eZdZefdeddfdZdS)?BaseApiClient._get_aiohttp_session..AiohttpTCPConnectorrr0Nc|jrdS|jsdSd|jD}|||dd}|j |j|d<dSdS)Nc,g|]}t|Sr-)repr)rrs r/ z[BaseApiClient._get_aiohttp_session..AiohttpTCPConnector.__del__..s999q477999r.zUnclosed connector) connector connectionsrr)_closed_connsvalues_closer)rrconnsrs r/rzGBaseApiClient._get_aiohttp_session..AiohttpTCPConnector.__del__s \  F  F99DK$6$6$8$8999% ++---"-  '  # /*.*@G& ' ' '0 /r.rr-r.r/AiohttpTCPConnectorrsG+3 A AS A A A A A A Ar.r)limitT)r trust_env read_bufsize)rSrPgoogle.auth.aio.credentialsr"google.auth.aio.transport.sessionsr"rXr_looprr ClientSession TCPConnectorr)rAsyncCredentialsr"rrrs r/_get_aiohttp_sessionz"BaseApiClient._get_aiohttp_sessionps  $)D)D)F)F$! OOOOOOMMMMMM 9 9 9 9 9+; 9 9 94!7 67S7STX7Y7Y Z Z$$        & & ( (( %  ' &  & 0 0 2 2 & C C C C C!6 C C CAAAAA 4AAA$32''a000'd   s9A A&%A&rFcd|j}|j}|r|n|r|nd}|sbtjt jdtjt jd}dttj dtj dtj ffd }||||||fS) zEnsures 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. Returns: A tuple of sync/async httpx client args. verifyN SSL_CERT_FILE SSL_CERT_DIRcafilecapathargsctxr0c|r|s|pi}||<|}|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. )r6rinspect signaturerClientr parametersrr copied_argsrRrs r/ _maybe_setz7BaseApiClient._ensure_httpx_ssl_ctx.._maybe_sets&))   ""V IIKKk!!### g' (=>>I I I# r.) rirjr6r4create_default_contextr4r5certifiwhererr StringDict SSLContext)rFr async_argsrrrs @r/rJz#BaseApiClient._ensure_httpx_ssl_ctxsF  D*J   Z^^F+++     &AA//   c w)* ^  4  4 :s## r.cd|j}|r|nd}|sbtjtjdt jtjd}dttj dtj dtj ffd }|||S) Ensures 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. Returns: An async aiohttp ClientSession._request args. r4Nrrrrrr0c|r|s|pi}||<|}|D]/}|tjtjjjvr||=0|Sr)r6rrrrr_requestrrs r/rz9BaseApiClient._ensure_aiohttp_ssl_ctx.._maybe_sets&))   ""V IIKKk!!### $W%:%CDDO P P# r. rjr6r4rr4r5rrrrrrrFrrrrs @r/rVz%BaseApiClient._ensure_aiohttp_ssl_ctxsF*J$. 8*.. DC   &AA//   c w)* ^  8 :j# & &&r.cd|j}|r|nd}|sbtjtjdt jtjd}dttj dtj dtj ffd }|||S) rr4Nrrrrrr0c|r|s|pi}||<|}|D]+}|tjtjvr |dkr||=,|S)rr4)r6rrr ws_connectrrs r/rz;BaseApiClient._ensure_websocket_ssl_ctx.._maybe_setWs&))   ""V IIKKk!!### g' 33> > >3%<<# r.rrs @r/rZz'BaseApiClient._ensure_websocket_ssl_ctx;sF*J$. 8*.. DC    &AA//   c w)* ^  2 :j# & &&r.crto0|jjpidduo |jjduS)N transport)rr@rjr6rRrs r/rTzBaseApiClient._use_aiohttprsJ  <   1 7R < <[ I I   <  2d : r.c|jr|jp|j}|jr |s|jSt |jj}|dS)Nwss)scheme) rfr'r&r8r r@r[rbgeturl)rrh url_partss r/_websocket_base_urlz!BaseApiClient._websocket_base_url|sl<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.r1N0Could not resolve API token from the environment)rCr?rorfget_token_from_credentials RuntimeError)rrfs 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#A22A69A6cK|jp|jtj|_|j4d{V|jtj|_ dddd{Vn#1d{VswxYwY|jS)a[Lazily 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, preventing race conditions when accessing or refreshing credentials. The lock is created on the first call to this method. An internal async lock is used to protect the creation of the main authentication lock to ensure it's a singleton within the client instance. Returns: The asyncio.Lock instance for asynchronous authentication operations. N)rDrEr rBrs r/_get_async_auth_lockz"BaseApiClient._get_async_auth_locks $ , 4.5lnn+411111111  ("),..$  0111111111111111111111111111   s A)) A36A3cK|js|d{V}|4d{V|jsRetrieves the access token for the credentials asynchronously.Nr1r)r?rr  to_threadrorf async_get_token_from_credentialsr)rasync_auth_lockrfs r/r}z!BaseApiClient._async_access_tokens   #"7799999999o # # # # # # # #  #-4->. . . ( ( ( ( ( ( $$ W #"DL " # # # # # # # # # # # # # # # # # # # # # # # # # # # M3    K L LLsAB B B  http_methodr\ 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)_)ra)rrRs r/rz0BaseApiClient._build_request..s)PPPcCNN3r COLLECTIONrHr[rlrnrrrecursive_dict_updater&rer(rwrur<rZry) rrr\rr(keys_to_deleterRpatched_http_optionsquery_vertex_base_modelsversioned_pathr[rzrUs r/_build_requestzBaseApiClient._build_requestsRQP\%6%6%8%8PPPN s   0 L+ . . 1      2   :< H H  "/$ & 5 OO6 7 7 !"&  J ,, J) J\ J"] J  J%<'((C B B$- B B BT Id'/!4zznn.:CCTCCn %-#, . . / //%.h $l33 +# ,7 C  * L*!]* <*  6  % & &Hd++     * 8M2CML-...$$"$ !!##"G   ! !c ! )#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}rrYrrr\rr]rrr(rparameter_modelrkrs r/rzBaseApiClient._requestjs  ? d + + + % % % % %  &?!/"?@@ !11L11uT'v>>> ;;t)< @ @@r.c lKd}|jrZ|js|jrLd|d{V|jd<|jr |jjr|jj|jd<|jrNt|jts-tj |j d}n|j}|r| r`|d{V|_|j}|rtt%j}|j|d{V|jjr5d|vr1d|vr|dd}n|dd } |jjd|j||j|t3j|j d |jd{V}n#t2jt2jt2jt2j tBj"f$r}tGj$d tKj&d dzd{VtN(d|z|)|j*|_|d{V|_|jjd|j||j|t3j|j d |jd{V}Yd}~nd}~wwxYwtVj,-|d{Vt]|dr|j/}ta|j|S|j12|j|j||j|j}|j13||d{V} tVj,-| d{Vta| j| S| r|d{V|_|j}|rtt%j}|j|d{V|jjr5d|vr1d|vr|dd}n|dd } |jjd|j||j|t3j|j d |jd{V}tVj,-|d{V|} t]| dr| j/} ta| j| 4d{VgS#t2jt2jt2jt2j tBj"f$rF}tGj$d tKj&d dzd{VtN(d|z|)|j*|_|d{V|_|jjd|j||j|t3j|j d |jd{V}tVj,-|d{V|} t]| dr| j/} ta| j| 4d{VgcYd}~Sd}~wwxYw|j1|j|j|j||jd{V} tVj,-| d{Vta| j| j4gS)Nrzr{r|rrrrrrtotalrrr z!Retrying due to aiohttp error: %s _responserrr{rzr<rrur-)5r%rfr'r}r<r?r~r|rrrrencoderTrrSrzrPrrrrrrxr{r ClientTimeoutrurWClientConnectorErrorClientConnectorDNSError ClientOSErrorServerDisconnectedErrorauth_exceptionsTransportErrorr sleeprandomrandintr7rFrVr@rrraise_for_async_responsernr rrQrrr) rrrr|rzrrrrclient_responseunwrapped_responseunwrapped_retry_responses r/_async_request_oncez!BaseApiClient._async_request_once~sB !D } $, $- 6$2244444444 6 6?+   t0A   . 23!  )5 1 1!z,+,,33G<<  QM    DF&*&?&?&A&A A A A A A A  & & ( ( I#>@@ %<<           " + I0@C0G0GCKK*,IccKK 02GHHc 8408 !("*+,2FGGG   7         ((  (  +  !  +  *      a&.A"6"6677 7 7 7 7 7 7 7 ++9A= > > >**4+=>>  1)-(A(A(C(C"C"C"C"C"C"C$ 8408 !("*+,2FGGG   7         (((((( 0o66x@@@@@@@@@ 8[ ) ) ('(H,h7770>>&  ( ( ?  !% 8 = = !>! !       o66GGGGGGGGGO3_EEE    JM&*&?&?&A&A A A A A A A  & & ( ( I#>@@ %<<           " + I0@C0G0GCKK*,IccKK 02GHHc0 8408 !("*+,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= > > >**4+=>>  1)-(A(A(C(C"C"C"C"C"C"C$ 8408 !("*+,2FGGG   7         (88BB B B B B B B B*2 " -{ ; ; J'?'I $&.-22444444445        9 D!% 8 @ @&  ( ( !A! !       o66GGGGGGGGGO3o6J5KLLLs:AG F F FFr.cK|||||}|||dd{Vfd}|S)NTr6cK23d{V}tj|} |drbtj|}tj|did|d{Vn7#tjj $r t d|zYnwxYwtj |WV6dS)Nr+r,rr-r&)rrrarrrraise_error_asyncr6r/rr7r0r(r<)rr2r3rs r/async_generatorz=BaseApiClient.async_request_streamed..async_generatorsA!IIIIIIIZ&&    " "; / / J//J/33w++//77        |+    ,,AJN    $   h&6ZHHHHHHHH"sC+A7B1C C)rr )rrr\rr(rr<rs @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. r>rbN)rioIOBase _upload_fdopen)rr?r@rAr(files r/ upload_filezBaseApiClient.upload_files,)RY'' __ Z<  4  D *k                     sA**A.1A.rHcd}|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. rr[NrnetlocTupload , finalizerur<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) r@rr}r6rLr r rbrrLread CHUNK_SIZErrurwupdaterWrZMAX_RETRY_COUNTrNrxr<INITIAL_RETRY_DELAYDELAY_MULTIPLIERtimerrlrrrrr)rrHr@rAr(offsetr[rc parsed_upload file_chunk chunk_sizeupload_commandrurU user_headersupload_headers retry_countr delay_secondss r/rFzBaseApiClient._upload_fds[(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 r6r\rr(Nrrr)rr|rrrrrNrxr{rzr<rurrrrrWr)rr\r(rr|rs r/ download_filezBaseApiClient.download_file%s&& 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. r>NrC)rrDrE_async_upload_fdanyioPathrG)rr?r@rAr(rHfds r/async_upload_filezBaseApiClient.async_upload_fileLs*)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|tjr|t } n |t d{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|j d || |t3j|  d{V}|jdrnL  Y + + +, 266<"  0<Cl+++%3$'KK!*oo     '~7IJJJ O++088$+2DEEE 9        (  ! !"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 |j |tj|jd{V}t"j|d{Vt)|j |d{VgjdS|j |j |j |j ||j d{V}t"j|d{Vt)|j |gjdS) rhr6riNrr rrjrr)rr|rrrrrrTrrSrxr{rzr<rrrurrrrrWrrQ)rr\r(rr|rrs r/async_download_filez!BaseApiClient.async_download_file8sD&& Dr 'L!D!  )5 1 1!z,+,,33G<<  $($=$=$?$???????d,44$&'l.BCCC 5h O 4 4X > >>>>>>>>   x}})>)>)>)>)>)>(?A #6>>$&& ?o O 4 4_ E EEEEEEEE   !0D0D0F0F/GAr.response_modelcdSrr-)rrus r/_verify_responsezBaseApiClient._verify_responsersDr.c|jjs |jr|j|jr|jdSdS)zCloses the API client.N)r@rOrNrrLrs r/rzBaseApiClient.closeusa   *!t/A!     ' $$&&&&&''r.cK|jjs|jd{V|jr-|jjs#|jd{VdSdSdS)zCloses the API async client.N)r@rRrQrrSrUrrs r/rzBaseApiClient.aclose~s   0.  $ + + - -------- *T%7%F*  ! ' ' ) ))))))))))****r.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)r@rOrrr r!r"rrs r/rzBaseApiClient.__del__s   ,     d     ,,T[[]];;;;;     dd s # 008A.. A<;A<)NNNNNNr)Frr)Br)r*r+r,rrorWrirjrmrrrrMrPr r staticmethodrr rrrJrVrZrTrrr rBrrr}r}rryrrrrrr r$r(rxrr4r8r=rDrEintrIrFrrkrqrn AsyncFilermrtrrwrrrr-r.r/r$r$UsL@@"&#CG# $26@?@?@?}@?FK3?@ @? } @?  @?./@?@?@?@?D T    dS! >?S!S!S!S!j?? W !3 34???<?B1'{1'w7I1'1'1'<1'f4'4'9K4'4'4'<4'lD535555 OS O O O O!',!!!!2MsCxMMMM837 iii if% i ./ i  iiii\;;;; ;;;;@37 AAA./A A  AAAA*7<eMeM%eM/3eMeMeMeMeMT37 ./   *'*<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$rucv|js|jst||jstd|jS)=Refreshes the authentication token for the given credentials.r)rrrtr)rurms r/rrsK  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 rrtr)rurmrs r/rrsU ; 1;"7799999999O;;;;;;;;  ;K$5; k:::::::::;;;;;;;;;;;;;;;;;;;;;;;;;;;  K I J JJ  s/A44 A>A>)qr,r collections.abcrr dataclassesrrrDrrrXr4rr4r@rAr]typingrrrrr r r urllib.parser r r:rnr google.authrigoogle.auth.credentialsrgoogle.auth.transportrrrrrrrpydanticrrrrequests.structuresrrrrrrtypesrrrr(rrwebsockets.asyncio.clientrrModuleNotFoundErrorwebsockets.clientrrrX multidictr rr" getLoggerr7rXrrZr[r\rGrlr(rWr;r}rErTrr|rZrerortrwryrrrrrrrrrr AsyncClientrr$rjrmrrr-r.r/rs" %%%%%% !!!!!!   VVVVVVVVVVVVVVVVVV!!!!!!###### //////&&&&&&<<<<<<555555 $$$$$$333333$$$$$$222222###### 6=======666555555556  ...++$H((((((GGGGGG  5 6 6  !4L22222 222 :#::::"8DcN8t8888@ )48E #s(^E19%s :K1LE EEEEGCGsGsGGGG%T *u[#5E/F"kk  eE3J' (  e_     """"""" "~~~~~~~~J  !12 w7I    F     el   *     u(   (B B B B B B B B J2   (4       (4 s$C## C10C17C>>DD