Li:dZddlZddlmZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZmZmZmZmZddlmZddlmZddlZddlZddlZ ddl!Z ddl!m"Z"ddl#Z#dd l$m%Z%dd l$m&Z&ddl'Z'd d l(m)Z)d d l(m*Z*d dl(m+Z+d dl,m-Z-d dl,m.Z.d dl,m/Z0d dl,m1Z1 ddl2m3Z4dZ7 ddl8Z8dZ7erddl:m;Z;e jxdZ=dZ>dZ?dZ@d ZAdZBGddeCZDdeeEfdZFd eGeEeEfddfd!ZHd"e-d#e-de-fd$ZId eGeEeEfd%eeeJeKfddfd&ZLd'eEd(eEdeEfd)ZMd*eeEdfdee"eEffd+ZNd,e"de"fd-ZOd.eeeJeKfdeeJfd/ZPeGd0d1ZQGd2d3Z/d4ZRd5ZSd6ZTdZUd ZVd7ZWd"ee1de)jfd8ZYGd9d:e#jZ[Gd;dZ^y#e5$r ddl6m3Z4Y#wxYw#e9$rY&wxYw)?zBase 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 AsyncIteratorIteratorOptionalTuple TYPE_CHECKINGUnion)urlparse) urlunparse) Credentials) BaseModel)ValidationError)_common)errors)version) HttpOptions)HttpOptionsOrDict) HttpResponse)HttpRetryOptionsconnectFT)CIMultiDictProxyzgoogle_genai._api_clientii@ceZdZdZy)EphemeralTokenAPIKeyErrorz)Error raised when the API key is invalid.N)__name__ __module__ __qualname____doc__V/var/www/html/eduruby.in/venv/lib/python3.12/site-packages/google/genai/_api_client.pyrrWs1r%rreturnctjjdd}tjjdd}|r|rtj d|xs|xsdS)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_keyr2]sXzz~~&6=zz~~&6=. NNO  91 9T9r%headerscdtj}dtjjdz}|d|}d|vr||dvr|d|dz|d<n d|vr||d<d|vr||dvr|d|dz|d<yd|vr||d<yy)z1Appends the telemetry header to the headers dict.zgoogle-genai-sdk/z gl-python/r z user-agentzx-goog-api-clientN)r __version__syssplit)r3 library_labellanguage_labelversion_header_values r&append_library_version_headersr<ns%g&9&9%:;-#++"3"3"5a"88.)?!N+;<g gl&; ;34A69NNGL7"0GLW$ g.A&B B "W-@%AA  ')#7G  *r%options patch_optionsc ||j}|jxsi}|jxsi}i|||_tjj }|D]<}|dk(r t ||d}|t |||&t ||t ||>|jt|j|S)Nr3) model_copyr3r model_fieldskeysgetattrsetattrr<)r=r> copy_optionoptions_headerspatch_options_headershttp_options_keyskey patch_values r&patch_http_optionsrKs""$+''-2/'//52+ "..335 7c i-d3K k3 , k3 567$";#6#67 r%timeout_in_secondscV|r'd|vr"ttj||d<yyy)z8Populates the server timeout header in the headers dict.zX-Server-TimeoutN)strmathceil)r3rLs r&populate_server_timeout_headerrQs1.g="%dii0B&C"DG >r%base_urlpathct|}|jjdr|jddn |j}|jdr|ddn|}t |j |dz|zS)N/r)rS)r rSendswith startswithr _replace)rRrS parsed_base base_paths r& join_url_pathr\s~"+    " "3 's    __S)abt$ K((i#o.D(E FFr%projectcttjjdg\}}|s|}|s td||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)r] credentialsloaded_project_ids r& load_authrfsQ#);;#6#6> ?$7$ +  G  J  g r%rdc>ddlm}|j||S)Nr)Request)google.auth.transport.requestsrhrefresh)rdrhs r& refresh_authrks4 gi r%timeoutc|r|dz }|Sd}|S)z Converts the timeout to seconds.g@@Nr$)rlrLs r&get_timeout_in_secondsrns* !6)  r%cheZdZUeeefed<eed<eed<eeeefefed<dZ e e ed<y) HttpRequestr3urlmethoddataNrl) r r!r"dictrN__annotations__r objectbytesrlrfloatr$r%r&rprpsA S> ( + d3;&''!'8E?!r%rpc 2eZdZ ddeeeefejdfdeeefdeee ffdZ ddZ defd Z e defd Zdeeddffd Zdeefd Zdeee efddffd ZdeeefddfdZdeefdZdeefdZededefdZy)rNr3zCIMultiDictProxy[str]response_stream byte_streamc t|tr||_nt|tjrC|j Dcic]#}|dj |j|%c}|_nZt|jdk(rB|j Dcic]#}|dj |j|%c}|_d|_ ||_ ||_ ycc}wcc}w)Nz, r) isinstancertr3httpxHeadersrBjoinget_listtyper getall status_coderzr{)selfr3rzr{rIs r&__init__zHttpResponse.__init__s '4 dl GU]] +;B<<>47#tyy))#./ /dl g  #5 59@25#tyy,- -dl D*D"Ds (C& (C+r'c0|j|_|SN)async_segmentssegment_iteratorrs r& __aiter__zHttpResponse.__aiter__s //1D Kr%czK |jjd{S7#t$rtwxYwwr)r __anext__ StopIterationStopAsyncIterationrs r&rzHttpResponse.__anext__s:((224 44 4  s ;(&(;(8;c^|jdsy|j|jdS)Nr)rz_load_json_from_responsers r&jsonzHttpResponse.jsons/    "   ( ()=)=a)@ AAr%c#Kt|jtr)|jD]}|r|j|niy|j gEd{y|j D]}|j|y7.wr)r~rzlistr_iter_response_stream)rchunks r&segmentszHttpResponse.segments s$&&-''D%6;d++E2CD   %mm--/3%++E223sABB/Bc8Kt|jtr*|jD]}|r|j|niy|jg23d{}||j 23d{}|j|786y76ywr)r~rzrr_aiter_response_stream)rrcs r&rzHttpResponse.async_segmentss$&&-''D%6;d++E2CD   %44633++E22 R36sNABBBBB4B8B9B<BBBBBc#Kt|jtr|jEd{y|j gEd{ytd7'7w)Nz8Byte segments are not supported for streaming responses.)r~r{rrcrs r& byte_segmentszHttpResponse.byte_segments"sQ$""D)!!!!   !mm  D  "s!*AAAAAAresponse_payloadct|dt|D]$}tjt ||||<&y)Nr)delattrdircopydeepcopyrC)rr attributes r& _copy_to_dictzHttpResponse._copy_to_dict-s= D$%YL $(MM'$ 2J$Ky!Lr%c#Kt|jtjs,t dt |jj dd}d}|jjD]U}|s|jdr|tdd(|D]}|dk(r|dz }|d k(s|dz}||z }|dk(sP|d}W|r|yyw) z,Iterates over chunks retrieved from the API.zFExpected self.response_stream to be an httpx.Response object, but got .rrdata: N{r}) r~rzrResponse TypeErrorrr iter_linesrXlen)rrbalanceliners r&rz"HttpResponse._iter_response_stream3s d**ENN ;  $../889 <  EG$$//1   "3x=>"" ! 8 Q,' #X Q,'   tme A -2 k sB,C/CCc$Kt|jtjxs,txr$t|jt j }|s,tdt|jjdd}d}t|jtjrz |jj23d{}|s |jdr|tdd/|D]}|dk(r|dz }|d k(s|dz}||z }|dk(sW|d}_trt|jt j r |jjjd{}|snt|j!d j#}|sR|jdr|tddu|D]}|dk(r|dz }|d k(s|dz}||z }|dk(r|d}|r||jj%yyy7N6|r||jjd{7y#|jjd{7wxYw7#|jj%wxYww) 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 rrrNrrrrzutf-8)r~rzrr has_aiohttpaiohttpClientResponserrr aiter_linesrXraclosecontentreadlinedecoderstriprelease)ris_valid_responserrrr line_bytess r&rz#HttpResponse._aiter_response_streamYs"4#7#7HP 4#7#79O9OP   4''(11 2! 5  EG$&&7,..::<  $ __X &s8}~& &   aCxlgclg   4-% \KE  g44#'!1199BBDD* ""7+224$ __X &s8}~& &   aCxlgclg   4-% \KE9< + $$&M= <. +""))+++d""))+++E@ $$&sB)J,IH H HU'nccr%rg?gN@)iiiiiic|tjdddStj|jxst}|jxst tj fd}tj|jxst|jxst|jxst|jxst}||d|tj t"t$j&dS)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. rT)stopreraisecXt|tjxr|jvSr)r~rAPIErrorcode)rretriable_codess r&zretry_args..s 1foo.L166_3Lr%)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_logr.loggingINFO)r=rrrrs @r& retry_argsrs _//2t DD  $ $W%5%5%H I$--I1I/  % %L %  ) )##;';    //2? ^^ ,}  $// E  r%c4eZdZdZdeddffd ZddZxZS)SyncHttpxClientzSync httpx client.kwargsr'Nc H|jddt|di|yzInitializes the httpx client.follow_redirectsTNr$ setdefaultsuperrrr __class__s r&rzSyncHttpxClient.__init__$ ($/ Gvr%c |jry |jy#t$rYwxYw#t$rYywxYw)zCloses the httpx client.N) is_closed Exceptionclosers r&__del__zSyncHttpxClient.__del__sK    jjl        s "1 .. ==r'Nr r!r"r#rrr __classcell__rs@r&rrsst  r%rc4eZdZdZdeddffd ZddZxZS)AsyncHttpxClientzAsync httpx client.rr'Nc H|jddt|di|yrrrs r&rzAsyncHttpxClient.__init__rr%c |jry tjj |j y#t$rY>wxYw#t$rYywxYwr)rrasyncioget_running_loop create_taskrrs r&rzAsyncHttpxClient.__del__ s_     ,,T[[];        s" A1A AA AArrrs@r&rrsst  r%rceZdZdZ d4deedeedeejjjdeedeedee f d Z d5d Z ed ed eej$ej$ffd Zed ed ej$fdZed ed ej$fdZd efdZd efdZd efdZd ej4fdZd eeeffdZ d6dededeee fdee d e!f dZ" d7de!ded e#fdZ$ d8de!dee ded e#fdZ% d7de!ded e#fdZ& d8de!dee ded e#fdZ'd ej$fdZ( d6dededeee fdee d e)f d Z* d6dededeee fdee d e+e)ddff d!Z, d6dededeee fdee d e)f d"Z- d6dededeee fdee d ef d#Z.dd$d%eee/j`fd&ed'e1dee d e#f d(Z2dd$d)e/j`d&ed'e1dee d e#f d*Z3dd$dedee d eee4ffd+Z5dd$d%eee/j`fd&ed'e1dee d e#f d,Z6dd$d)ee/j`e7jpefd&ed'e1dee d e#f d-Z9dd$dedee d eee4ffd.Z:d/ejvd dfd0Zd9d3Z?y): BaseApiClientz8Client for calling HTTP APIs sending and receiving JSON.Nvertexaiapi_keyrdr]location http_optionsc ||_d|_|j7tjj ddj dvrd|_|s|r |r t d|r |r t dt}t|tr tj|}nt|tr|}tjj dd} tjj d d} t} |xs| |_ |xs| |_|xs| |_||_t|_t%j&|_d|_d|_|jr|r| rt.j1d d|_no| s| r&|r$t.j1d d|_ d|_nE|s|r| rt.j1d d|_n"| s| r| rt.j1d d|_|j2r |j2nd|_|js|js|jsd|_|js?|js3|js't5d\}|_ |j s||_|jxr |jxs |j} | s|js t d|js|jdk(rd|j"_n[|jr0|r|s,|s*|j|j"_d|_ d|_nd|jd|j"_d|j"_n9|js t dd|j"_d|j"_ddi|j"_|jrX|jj;|_|j"j8#|j|j"j8d<|rt=|j"||_n5|j"j8t?|j"j8|jA|j"\} }||_!|j"jDr|j"jD|_#ntIdi| |_#|j"jJr|j"jJ|_&ntOdi||_&|jQr% ddl)}|jU|j"|_+d|_-t]|j"j^}|ja|j"|_1tejfdi||_4tejjdi||_6y#t$r}t d|d}~wwxYw#tX$rYwxYw)NGOOGLE_GENAI_USE_VERTEXAI0)true1TzNProject/location and API key are mutually exclusive in the client initializer.zICredentials and API key are mutually exclusive in the client initializer.zInvalid http_optionsGOOGLE_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.globalr]zK::>>"94@L!#K)kDL, DM)kDL#D$D%>>+D48DBFD' }}   A  KW  B   x[  B  K[  M  $ , , ! ) )  ]]4<<8L8L"$-ll,,&&$-d$;! T\  )$ !\\;dmmL  )=)=    (2(J#   'h7'+&:&:#  t}}o%@ A #(1d$ \\ G  %Qd!'/d$"02D!ED ||\\'')dl   # # /7;||""#34-   4d    # # /&t'9'9'A'AB%)%?%? &"K"%6D! &&--::d*9[9d ,,!%!3!3!F!Fd!1!F4E!Fd  262N2N   3 />BDd00>>?L"<>Dm8/0a78Z   s* W'$W# W  WW # W/.W/r'cK|j|jjr5tjtjddt |_|jSw)z#Returns the aiohttp client session.r)limitT) connector trust_env read_bufsize)r/closedr ClientSession TCPConnectorREAD_BUFFER_SIZErs r&_get_aiohttp_sessionz"BaseApiClient._get_aiohttp_sessionsU $(=(=(D(D%33((q1'd   sA$A&r=cd|j}|j}|r|jn|r|jnd}|sdtjt j jdtjt j jd}dttjdtjdtjffd }||||||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_DIRcafilecapathargsctxr'c$|r|js|xsij}||<|j}|jD]?}|tjtj j jvs=||=A|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. )r-rinspect signaturerClientr parametersrPrQ copied_argsrIrJs r& _maybe_setz7BaseApiClient._ensure_httpx_ssl_ctx.._maybe_sets&)   "V IIKk!!## g'' (=(=>II I#r%) r<r=r-sslcreate_default_contextr+r,certifiwhererr StringDict SSLContext)r=rP async_argsrQr[rJs @r&r%z#BaseApiClient._ensure_httpx_ssl_ctxsF   D**J    ^^F+     & &A/ c w))* ^^   4 4:s# r%cd|j}|r|jnd}|sdtjtj jdt jtj jd}dttjdtjdtjffd }|||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. r\NrKrLrMrPrQr'c$|r|js|xsij}||<|j}|jD]?}|tjtj j jvs=||=A|SrS)r-rrUrVrrE_requestrXrYs r&r[z9BaseApiClient._ensure_aiohttp_ssl_ctx.._maybe_set@s&)   "V IIKk!!## $$W%:%:%C%CDOO P#  r% r=r-r\r]r+r,r^r_rrr`rar=rbrQr[rJs @r&r,z%BaseApiClient._ensure_aiohttp_ssl_ctx%sF**J$.*.. DC   & &A/ c w))* ^^   8 j# &&r%cd|j}|r|jnd}|sdtjtj jdt jtj jd}dttjdtjdtjffd }|||S) rdr\NrKrLrMrPrQr'c|r|js|xsij}||<|j}|jD]1}|tjtj vs)|dk7s/||=3|S)rTr\)r-rrUrV ws_connectrXrYs r&r[z;BaseApiClient._ensure_websocket_ssl_ctx.._maybe_setzs&)   "V IIKk!!## g'' 3>> >3%<#r%rgrhs @r&r1z'BaseApiClient._ensure_websocket_ssl_ctx^sF**J$.*.. DC   & &A/ c w))* ^^   2 j# &&r%ctxrE|jjxsijdduxr|jjduS)N transport)rrr=r-r)rs r&r+zBaseApiClient._use_aiohttpsS  <    1 1 7R < <[ I   <   2 2d : r%c|jxr |jxs |j}|jr|s |jSt |j j }|jdjS)Nwss)scheme) r]r r rr rrRrYgeturl)rr; url_partss r&_websocket_base_urlz!BaseApiClient._websocket_base_urlsk<<9DMMJdll $7 ! !!++445I   U  + 2 2 44r%c|j5|js1t|j\|_}|js||_|jr|jjs|jj st |j|jj s td|jj cdddStd#1swYyxYw)z/Retrieves the access token for the credentials.r0Could not resolve API token from the environmentN)rrrfr]expiredtokenrk RuntimeError)rr]s r& _access_tokenzBaseApiClient._access_tokens  O   %.t||%D"7|| $,      $ $D,=,=,C,C t(( )  &&OP P  &&OOMNNOOsCC# C##C,cxK|j{|jtj|_|j4d{|jtj|_dddd{|jS|jS7Q7#1d{7swY|jSxYww)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)r r!rrrs r&_get_async_auth_lockz"BaseApiClient._get_async_auth_locks $ , , 4.5lln+4411  ("),,.$ 11   4  11111   sHAB:BB:&B. B:9B:B:B:B7 B# !B7(B:cK|js|jd{}|4d{|jsHtjt|j d{\|_}|j s||_dddd{|jr|jj s|jjs|jd{}|4d{|jj s|jjs,tjt|jd{dddd{|jjs td|jjStd7y7q7;7#1d{7swY$xYw777|7n#1d{7swY~xYww)z>Retrieves the access token for the credentials asynchronously.Nrru) rr{r to_threadrfr]rvrwrkrx)rasync_auth_lockr]s r&_async_access_tokenz!BaseApiClient._async_access_tokens   "7799o # #  -4->->. ( $$ W"DL # #    " "$*;*;*A*A $ 9 9 ;;" E E    & &d.?.?.E.E##L$2C2CD D D E E    $ $MNN    $ $$ K LL5: # ( # # # #< E E E E E Es GF GFG5F'&F!'F' GF$AG!F=" G,F?-G0AGGG GGAGG!F'$G'F:-F0 .F:5 G?GGGG G GG http_methodrS request_dictc|jDcgc]}|jds|}}|D]}||=|rQt|trt |j |}n6t |j tj |}n |j }d}|jr|dk(r|jdrd}|jrK|jds:|s8|js |jr d|jd|jd|z}|jd|} n|jd|} |j |js td |j} t|d r,|jr tj ||j| } |j"r$|jr |js |j$r t'| | } |j$r&|j$jd r t)d t+|j,} |j. td t1|j.| t3|| |j.|| Scc}w)N_Fr-zpublishers/google/modelsTz projects/z /locations/rUzBase URL must be set. extra_bodyz auth_tokens/z4Ephemeral tokens can only be used with the live API.zRequest headers must be set.rrrqr3rsrl)rBrXr~rrKrrr r]r r#rRrchasattrrrrecursive_dict_updaterr r\rrnrlr3rQrp) rrrSrr rIkeys_to_deletepatched_http_optionsquery_vertex_base_modelsversioned_pathrRrqrLs r&_build_requestzBaseApiClient._build_requestsZ&2%6%6%8PcCNN32C2CL--.$""$ ((66$$&&&& 7m##((v(Fh oo((2    H5=]]O##++$$&&&& ,h oo((2    H5=]]Or%c|rgt|tr tdi|n|}|jr>t |j}t j di|}||j||S|j|j||SNr$) r~rtrr0rrr3rr4rrr rparameter_modelr>rs r&rfzBaseApiClient._request|s  d +  % %  & &!/"?"?@ !!1L1T''v>> ;;t))< @@r%c Kd}|jr|js |jrd|jd{|jd<|j r9|j j r#|j j |jd<|jrtj|jnd}n`|jrTt|jts.|jrtj|jnd}n |j}|r|jr|jd{|_ |jjd |j |j"|j|t%j&|j(d|j*d{}tDjFjI|d{tK|j|S|jLjO|j |j"||j|j( }|jLjQ|| d{}tDjFjI|d{tK|j|S|jr|jd{|_ |jjd |j |j"|j|t%j&|j(d|j*d{}tDjFjI|d{tK|j|jSd{gS|jLj|j |j"|j||j( d{}tDjFjI|d{tK|j|jRgS777|#t$j,t$j.t$j0t$j2f$r}t5j6dt9j:ddzd{7t<j?d |z|jA|jB|_|jd{7|_|jjd |j |j"|j|t%j&|j(d|j*d{7}Yd}~d}~wwxYw77$777R7.7 #t$j,t$j.t$j0t$j2f$r\}t5j6dt9j:ddzd{7t<j?d |z|jA|jB|_|jd{7|_|jjd |j |j"|j|t%j&|j(d|j*d{7}tDjFjI|d{7tK|j|jSd{7gcYd}~Sd}~wwxYw7X74w)Nrrrrrrr z!Retrying due to aiohttp error: %srrrr$)*r r]r rr3rrrsrrr~rwr+rHr/rrrrqr ClientTimeoutrlr-ClientConnectorErrorClientConnectorDNSError ClientOSErrorServerDisconnectedErrorrsleeprandomrandintr.r"r,rrrraise_for_async_responserr*rrr)rrrrsrrrclient_responses r&_async_request_oncez!BaseApiClient._async_request_onces)-D }}$,,$-- $22445 6?+   t00AA    . . 23/;.?.?TZZ )) *Td   ,++U32>2C2CL--.$""$   &*&?&?&A A 840088 !(("""**++L4H4HI  77   (>oo66x@@@H,,h7700>>&&   (( (( ? !% 8 8 = = !>!  oo66GGGO33_EE   &*&?&?&A A" I840088 !(("""**++L4H4HI  77   (88B B Bh..x}}1F0GH H6!% 8 8 @ @&&   (( (( !A!  oo66GGGO33o6J6J5KLLe5 !B   ( (  + +  ! !  + +    a&..A"667 7 7 ++9A= >**4+=+=>  1)-(A(A(C"C"C$ 840088 !(("""**++L4H4HI  77    ( . A  H!B  C1F  ( (  + +  ! !  + +   I  a&..A"667 7 7 ++9A= >**4+=+=>  1)-(A(A(C"C"C$ 840088 !(("""**++L4H4HI  77    (88B B Bh..x}}1F1F0GH H/ I4  Hs;]P6D ] P9 ]A+P?P<P?"]'V (B]*V+&]V=]V ]A+V V&V ,V-'V V V A ])]*&]]&]9]<P??AV/V0R31AVTA5V7U:8V=]V]]]]V V V A\>"/\9XA\9"Y%#A5\9['\9\(\9)\, * \93\>4]9\>>]]c>K|rot|tr tdi|n|}|jrFt |j}t j di|}||j||d{S|j|j||d{S7)7wr) r~rtrr0rrr5rr6rs r&_async_requestzBaseApiClient._async_request s  d +  % %  & &!/"?"?@ &&66433\6JJJ""   , K s$A-B/B0$BBBBct|jtr|jj}|S|j}|Sr)r~rr model_dump)rcopieds r&get_read_only_http_optionsz(BaseApiClient.get_read_only_http_options!s?$$$i0!!,,.f M!!f Mr%c|j||||}|j||d}|jr|jdnd}t|j|S)NFrrrr3body)rrfrzSdkHttpResponser3)rrrSrr rr response_bodys r&rzBaseApiClient.request(se&&T<L}}\<}FH'/'?'?  #R 8#3#3- HHr%c#&K|j||||}|j||d}|jD]}tj|} |j drUtj |} tjj| jdijd| |t|j |y#tjj$rtjd|zYSwxYww)NTr {"error":errorr1Failed to decode chunk that contains an error: %sr)rrfrrrrXrrr raise_errorr-decoderrr.debugrr3) rrrSrr rsession_responser chunk_dump chunk_jsons r&request_streamedzBaseApiClient.request_streamed8s&&T<L}}\<}M!**,O::e$j    -zz*-* // % %nnWb)--f5  $4$<$<: NNO\\ ) )  ?* L   s+ADA&C:D5D D DDcK|j||||}|j||dd{}|jr|jdnd}t|j|S78w)NF)rr rrrr)rrrzrr3)rrrSrr rresultrs r& async_requestzBaseApiClient.async_requestUsv&&T<L&&! U'F281G1GF**1-RM 6>> FF s,A)A'9A)cK|j||||}|j|dd{fd}|S7w)NT)rrcK23d{}tj|} |jdr]tj|}tj j |jdijd|d{tj|77!#tjj$rtjd|zYVwxYw6yw)Nrrrrr)rrrXrrrraise_error_asyncr-rrr.rrr3)rrrrs r&async_generatorz=BaseApiClient.async_request_streamed..async_generatorss!IIZZ&    " "; /J/J//33w+//7  h&6&6ZHHI   ||++  ,,AJN    "sUC/C-B.C-C/A)B2B0B2C/.C-0B225C*'C/)C**C/)rr)rrrSrr rrrs @r&async_request_streamedz$BaseApiClient.async_request_streamedfsS&&T<L((l4(PPHI$  )Qs+A?Ar  file_path upload_url upload_sizect|tjr|j||||St |d5}|j||||cdddS#1swYyxYw)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. rrbN)r~ioIOBase _upload_fdopen)rrrrr files r& upload_filezBaseApiClient.upload_filesn,)RYY' __ Z<  4  D *k     s AA$rc d} |jt}d}|r t|}d}||z|k\r|dz }|r|n |j}t |t r|j dn |j} | Ft |jt r|jjn|jj} t| } |t|t|d} t| | d} | tkrp|jjd|| || } | jj drn4tt | zz}| d z } t#j$|| tkrp||z } jj dd k7rn||kr t'd t(j*j-| | jj dd k7r t'd t/| j| j0gS)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. rupload , finalizerlzX-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.rz)read CHUNK_SIZErrr~rtr-rlrnrNrQMAX_RETRY_COUNTr(rr3INITIAL_RETRY_DELAYDELAY_MULTIPLIERtimerrcrrrrr)rrrrr offset file_chunk chunk_sizeupload_commandrlrLupload_headers retry_countr delay_secondss r&rzBaseApiClient._upload_fds(F 99Z(jj _ n f  +,&%1\t7I7Il d +   9 %## $,,d3    & &##++  2':#1"%f+ On %^5GHk / )%%--"& .       6 7 +/?/LM q  =! / ) f     4 5 A    a h OO&&x023w> O PP ((8==/ JJr%c|jd|i|}d}|jrFt|jts t j |j}n |j}|j j|j|j|j||j}tjj|t|j|j!gj"dS)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 r-rSrr Nrr{r)rrsr~rwrrr(rrrrqr3rlrrrrrr{)rrSr rrsrs r& download_filezBaseApiClient.download_files&& Dr 'L)-D  ))5 1zz,++,  !!))""   $$$$ *H OO&&x0 x}}&7 k! r%cKt|tjr|j||||d{St j |}|j dd{}|4d{|j||||d{cdddd{S7k7?7677 #1d{7swYyxYww)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. rNr)r~rr_async_upload_fdanyioPathr)rrrrr rfds r&async_upload_filezBaseApiClient.async_upload_files*)RYY'(( Z<)ZZ "d4 b  **  Kl+      !      s{3CB!-C#B#$ C.B%/C2B+ B' B+ CB)C#C%C'B+)C+B=1B4 2B=9Cc N Kd}|jr|jd{|_ t|tj r|j t}n|j td{}d}|r t|}d}||z|k\r|dz }|r|n |j}t|tr|jdn |j} | Ft|jtr|jjn|jj} t| } |t|t|d} t| | d} d} | t kr|jj#d||| t%j&|  d{} | j(jd rnj#d||| | d{}|(|j(r|j(jdrn4t*t,| zz}| d z } tAj0|| t kr||z }||j(jdd k7rn||kr t3d t4j6j9|d{|)|j(jddk7r t3dt;|j(|j<gS777z7/77y7(7$7iw)aTransfers a file asynchronously 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 finalized request. rNrrrlrrr)rrrqrsr3rlzX-Goog-Upload-Statusrrrrrrrr)!r+rHr/r~rrrrrrrtr-rlrnrNrQrrrrr3rrrrrcrrrrrr*r)rrrrr rrrrrlrLrrrrrs r&rzBaseApiClient._async_upload_fd>s+(F $($=$=$??d  dBII &yy,*!YYz22* :*!  + - L (.'3|9K9K <.   Y '&&  ?D..5  ((%%--  4G<%3%([!*o  '~7IJ O+0088$++4FG 9  (   ! !"8 9 -1A;1NO-  + m, , ,O+ *    $$%;<H  &   u ~ OO 4 4X >>>  ""#9:gE D     8==?-B,C  dBII &yy,*!YYz22* :*!  + - L (.'3|9K9K ,-   Y '%%  ?D..5  ((%%--  4G<%3$'K!*o  '~7IJ O+"&":":"B"B $( #C#  /)!))!))--.DE -1A;1NO-  + **] #!O+$ *  '''++,BCxO  &   { D OO 4 4_ EEE  %%%))*@AWL D    ! !O4H4H3Io@ 3@  -?.C3@ <Fs'T%T AT%<T=DT%TA T%%T& T%4AT%TAT%!T "AT%5T6DT%<T =A!T%AT%9T#:AT%T%T%T%T%T%T% T%#T%c 2K|jd|i|}d}|jrFt|jts t j |j}n |j}|j r|jd{|_|jj|j|j|j|tj|jd{}t j"j%|d{t'|j|j)d{gj*dS|j,j|j|j|j||jd{}t j"j%|d{t'|j|j)gj*dS7z7777h7Cw) rr-rNrrrrr)rrsr~rwrrr+rHr/rrrrqr3rrrlrrrrrr{r*)rrSr rrsrrs r&async_download_filez!BaseApiClient.async_download_files&& Dr 'L)-D  ))5 1zz,++,   $($=$=$??d,,44$$&&'' 0D0DE 5h OO 4 4X >>>    x}})>(? A #66>>$$&&&& ?o OO 4 4_ EEE   ! !0D0D0F/G A1@?*?FsnB HH A(H8H 9&HH 'HH A"H*H+&HH8H HHHHHresponse_modelcyrr$)rrs r&_verify_responsezBaseApiClient._verify_response*sr%cf|jjs|jjyy)zCloses the API client.N)rr'r(rrs r&rzBaseApiClient.close-s*    * *   +r%cK|jjs"|jjd{|jr#|jj d{yy747w)zCloses the API async client.N)rr)r*rr/rrs r&rzBaseApiClient.aclose4s^    0 0  $ $ + + ---   ! ! ' ' ))).)s!4A/A+-A/$A-%A/-A/c |jjs|j t j j |jy#t$rY>wxYw#t$rYywxYw)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)rr'rrrrrrrs r&rzBaseApiClient.__del__=sh    , ,    ,,T[[];       s"&A1A* A'&A'* A65A6)NNNNNN)r'zaiohttp.ClientSessionr)F)NFr)@r r!r"r#rboolrNr`rardrrrrH staticmethodrr rr`r%r,r1r+rsryrrr{r rrrtrvrprrrrfrrrrrrrrrrrintrrrwrrr AsyncFilerrrrrrrr$r%r&rrs@"&#CG# $26{?{?}{?FKK33??@ {? } {?  {?./{?z !?? W  !3!3 34??B6'{6'w7I7I6'6'p4'4'9K9K4'4'lD535OSO$!',,!2MsCxMH37 YY Yf% Y ./ Y  Y|--- -d37 AA./A A  A*7<zM%zM/3zMzM~37 ./   *'*<*<37 II If% I ./ I  I*37 OO Of% O ./ O $,- OD37 GG Gf% G ./ G  G,37  f%  ./   N37  sBII~&    ./   L37 MK IIMKMK MK ./ MKMKf37 % %./ %  S%Z %Z37  sBII~&    ./   N37 p "))U__S11 2pp p ./ ppl37 5 5./ 5  S%Z 5t W->-> 4 !* r%r)_r#rcollections.abcrr dataclassesrrUrrrrOr+rr\r7rrtypingrrrrr r r urllib.parser r rr^ google.authr`google.auth.credentialsrrpydanticrrrrrrrtypesrrrrrwebsockets.asyncio.clientrrkModuleNotFoundErrorwebsockets.clientrrr. multidictr getLoggerr.rrGrrrrcrrNr2rtr<rKrxrrQr\rfrkrnrprrrrrrr`rrWr AsyncClientrrr$r%r&rs" % !   VVV!# / $$2#6=  + (   5 6  2 2 :#:"8DcN8t8. )48E #s(^E19%s :K1LE EGCGsGsG%T *u[#5E/F"kk  eE3J' (  e_  "" "WW| !12w7I7IB ell * u(( (u u u6556s$6G?G# G G #G,+G,