o pkj3c@svdZddlZddlmZddlZddlmZddlZddlZddl Z ddl Z ddl Z ddl Z ddl Z ddlZddlZddlZddlZddlmZmZmZmZmZmZmZmZmZddlmZddlmZddlZddl Z ddl!Z!ddl"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/m0Z0d dl/m1Z1d dl/m2Z2d dl3m4Z4d dl3m5Z5d dl3m6Z7d dl3m8Z8d dl3m9Z9zddl:m;Z<Wne=yddl>m;Z<YnwdZ?zddl@Z@dZ?Wn eAyYnwzddlBZBWn eAy dZBYnwer"ddlCmDZDddlEmFZFddlGmHZHe IdZJdZKdZLdZMd ZNd ZOd!d"hZPeBdur>e*jQfne*jQeBjQfZReBdurMe*jSfne*jSeBjSfZTeBdur^e*jUe*jVfn e*jUe*jVeBjUeBjVfZWGd#d$d$eXZYd%eeZfd&d'Z[d(e\eZeZfd%dfd)d*Z]d+e4d,e4d%e4fd-d.Z^d(e\eZeZfd/eee_e`fd%dfd0d1Zad2eZd3eZd%eZfd4d5Zbd6eeZdfd%ee%eZffd7d8Zcd9e%d%e%fd:d;ZddZeeGd?d@d@ZfGdAdBdBZ6dCZgdDZhdEZid Zjd ZkdFZld+ee8d%e0jmfdGdHZnGdIdJdJe*joZpGdKdLdLe*jqZrGdMdNdNZsdOdNd9e#jtjuj%d%eZfdPdQZvdOdNd9e#jtjuj%d%eZfdRdSZwdS)TzBase 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_CHECKINGTupleUnioncastoverload)urlparse) urlunparse) Credentials)mtls) exceptions) BaseModel)ValidationError)_common)errors)version) HttpOptions)HttpOptionsOrDict) HttpResponse)HttpRetryOptions) ResourceScope)connectFTAuthorizedSession)CIMultiDictProxy)CaseInsensitiveDictzgoogle_genai._api_clientii@useuc@seZdZdZdS)EphemeralTokenAPIKeyErrorz)Error raised when the API key is invalid.N)__name__ __module__ __qualname____doc__r+r+V/home/djax/ivt_ai_plugin/venv/lib/python3.10/site-packages/google/genai/_api_client.pyr&{sr&returncCs:tjdd}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_keyr+r+r,get_env_api_keys r7headerscCsdtj}dtjd}|d|}d|vr,||dvr,|d|d|d<nd|vrB||dvrB|dd||d<nd|vrJ||d<d|vra||dvra|d|d|d<dSd|vrx||dvrx|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)r8 library_labellanguage_labelversion_header_valuer+r+r,append_library_version_headerss*       r@options patch_optionscCs|}|jpi}|jp i}i|||_tj}|D]!}|dkr#qt||d}|dur4t|||qt||t||q|jdurHt|j|S)Nr8) model_copyr8r model_fieldskeysgetattrsetattrr@)rArB copy_optionoptions_headerspatch_options_headershttp_options_keyskey patch_valuer+r+r,patch_http_optionss&      rNtimeout_in_secondscCs*|rd|vrtt||d<dSdSdS)z8Populates the server timeout header in the headers dict.zX-Server-TimeoutN)strmathceil)r8rOr+r+r,populate_server_timeout_headers rSbase_urlpathcCsZt|}|jdr|jddn|j}|dr|ddn|}t|j|d|dS)N/r)rU)r rUendswith startswithr_replace)rTrU parsed_base base_pathr+r+r, join_url_paths r]projectcCs0tjjdgd\}}|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)r^ credentialsloaded_project_idr+r+r, load_auths rfrdcCsddlm}|||S)Nr)Request)google.auth.transport.requestsrgrefresh)rdrgr+r+r, refresh_auths  rjtimeoutcCs|r|d}|Sd}|S)z Converts the timeout to seconds.g@@Nr+)rkrOr+r+r,get_timeout_in_secondss rlc@sVeZdZUeeefed<eed<eed<eeeefefed<dZ e e ed<dS) HttpRequestr8urlmethoddataNrk) r'r(r)dictrP__annotations__r objectbytesrkrfloatr+r+r+r,rms rmc @seZdZ  d deeeefejddfdeeefdeee ffddZ d!d d Z d efd d Z e d efddZd eeddffddZd eefddZd eee efddffddZdeeefd dfddZd eefddZd eefddZeded efddZdS)"rNr8zCIMultiDictProxy[str]r!response_stream byte_streamcsttr |_n?ttrfddD|_n-t}dur5t|jjr5dd D|_nt j dkrHfddD|_d|_ ||_ ||_dS)Nci|] }|d|qSz, )joinget_list.0rLr8r+r, z)HttpResponse.__init__..cSsi|]\}}||qSr+r+)r}rLvaluer+r+r,r&sr crxry)rzgetallr|r~r+r,r(r) isinstancerqr8_HTTPX_HEADERS_TYPESrErloaded_requests structuresr!itemstyper' status_codervrw)selfr8rvrwrequests_moduler+r~r,__init__s*      zHttpResponse.__init__r-cCs||_|SN)async_segmentssegment_iteratorrr+r+r, __aiter__1s zHttpResponse.__aiter__cs(z |jIdHWStytwr)r __anext__ StopIterationStopAsyncIterationrr+r+r,r5s  zHttpResponse.__anext__cCs4t|jtsdS|jr|jdsdS||jdS)Nr)rrvlist_load_json_from_responserr+r+r,json;s zHttpResponse.jsonccsht|jtr|jD] }|r||niVq dS|jdur%gEdHdS|D]}||Vq)dSr)rrvrr_iter_response_stream)rchunkr+r+r,segmentsGs    zHttpResponse.segmentscCst|jtr|jD] }|r||niVq dS|jdur-g2z 3dHW}|Vq 6dS|2z 3dHW}||Vq16dSr)rrvrr_aiter_response_stream)rrcr+r+r,rSs   zHttpResponse.async_segmentsccs>t|jtr|jEdHdS|jdurgEdHdStd)Nz8Byte segments are not supported for streaming responses.)rrwrrcrr+r+r, byte_segments`s  zHttpResponse.byte_segmentsresponse_payloadcCs0t|dt|D] }tt||||<q dS)Nr)delattrdircopydeepcopyrF)rr attributer+r+r, _copy_to_dictks  zHttpResponse._copy_to_dictccs$t}t|jts"|durt|j|js"tdt|jjdd}d}g}t|jtr4|j }n|jj dd}|D]C}|sL|rKd |Vg}q=| d r]| |t d dq=|D]}|d krj|d 7}q_|d krr|d 8}q_||7}|dkr|Vd}q=|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})rrrrv_HTTPX_RESPONSE_TYPESResponse TypeErrorrr' iter_linesrzrYappendlen)rrrbalance data_bufferrvlinerr+r+r,rqsV        z"HttpResponse._iter_response_streamc Cst|jtptot|jtj}|stdt|jjdd}d}g}t|jtrzq|j}| 2zG3dHW}|sE|rDd |Vg}q2| drV| |t ddq2|D]}|dkrc|d 7}qX|d krk|d 8}qX||7}|dkry|Vd}q26|r|V|rd |VW|IdHdSW|IdHdS|IdHwtrEt|jtjrGz z |jjjtd IdH}Wnty|jjIdH}Ynw|snL|d }|s|rd |Vg}q| dr| |t ddq|D]}|dkr|d 7}q|d kr|d 8}q||7}|dkr|Vd}q|r&|V|r7d |VW|jdSW|jdS|jwdSdS)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)rrvr has_aiohttpaiohttpClientResponserrr' aiter_linesrzrYrraclosecontentreadlineREAD_BUFFER_SIZEdecoderstriprelease) ris_valid_responserrrrvrr line_bytesr+r+r,rs     "       (z#HttpResponse._aiter_response_streamresponsec Cs:zt|WStjy}z td||d}~ww)zFLoads JSON from the response, or raises an error if the parsing fails.z0Failed to parse response as JSON. Raw response: N)rloadsJSONDecodeErrorrUnknownApiResponseError)clsrer+r+r,rs z%HttpResponse._load_json_from_response)NN)r-r)r'r(r)r rqrPhttpxHeadersrrtrrrpropertyrrrrrrrsrrrr classmethodrr+r+r+r,rs8    !   7lrg?gN@)iiiiiics|dur tdddS|jdkrd|_t|jpt}|jp ttfdd}tj|jp0t |j p4t |j p8t |jpCs zretry_args..)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_logr3loggingINFO)rArrrr+rr, retry_args2s*     rc4eZdZdZdeddffdd Zd ddZZS) SyncHttpxClientzSync httpx client.kwargsr-Nc "|ddtjdi|dSzInitializes the httpx client.follow_redirectsTNr+ setdefaultsuperrrr __class__r+r,rX zSyncHttpxClient.__init__cCsHz|jrWdSWn tyYnwz|WdSty#YdSw)zCloses the httpx client.N) is_closed Exceptioncloserr+r+r,__del__]s  zSyncHttpxClient.__del__r-Nr'r(r)r*rrr __classcell__r+r+rr,rUrcr) AsyncHttpxClientzAsync httpx client.rr-Nc rrrrrr+r,rmrzAsyncHttpxClient.__init__cCsRz|jrWdSWn tyYnwz t|WdSty(YdSwr)rrasyncioget_running_loop create_taskrrr+r+r,rrs  zAsyncHttpxClient.__del__rrr+r+rr,rjrrc@sleZdZUdZdZeeed<dZee ed<      dfdeedee dee j j j dee dee d eef d d Zd efd dZd efddZded efddZed eefddZd efddZe dgdeded eejejffddZe dgdeded ejfddZe dgdeded ejfddZd efd d!Zd e fd"d#Z d e fd$d%Z!d e"j#fd&d'Z$d e%e effd(d)Z& dhd*e d+e d,e'e e(fd eed e)f d-d.Z* dgd/e)d0ed e+fd1d2Z,  did/e)d eed0ed e+fd3d4Z- dgd/e)d0ed e+fd5d6Z.  did/e)d eed0ed e+fd7d8Z/d ejfd9d:Z0 dhd*e d+e d,e'e e(fd eed e1f d;d<Z2 dhd*e d+e d,e'e e(fd eed e3e1ddff d=d>Z4 dhd*e d+e d,e'e e(fd eed e1f d?d@Z5 dhd*e d+e d,e'e e(fd eed ef dAdBZ6ddCdDe%e e7j8fdEe dFe9d eed e+f dGdHZ:ddCdIe7j8dEe dFe9d eed e+f dJdKZ;eedddLdMd+e d eedNee%e e?j@e e7j8fdOe9d ee=f dTdQZ>ddCdDe%e e7j8fdEe dFe9d eed e+f dUdVZAddCdIe%e7j8eBjCefdEe dFe9d eed e+f dWdXZDej?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.comr^z._RefreshableAsyncCredentialszJAdapter to use the client's sync credentials in an AsyncAuthorizedSession.clientr cst||_dSr)rr_client)rrerr+r,rs  zQBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.__init__requestrornr8r-NcsL|jIdH}d||d<|jjr"|jjjr$|jjj|d<dSdSdS)NBearer Authorizationx-goog-user-project)rf_async_access_tokenr%quota_project_id)rrgrornr8tokenr+r+r,before_requests zWBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.before_requestcSs|jjsdS|jjj SNF)rfr%expiredrr+r+r,valids zNBaseApiClient._get_aiohttp_session.._RefreshableAsyncCredentials.valid)rer ) r'r(r)r*rrrPrqrnrr0rqrr+r+rr,_RefreshableAsyncCredentialss   rrc@"eZdZefdeddfddZdS)z@BaseApiClient._get_aiohttp_session..AiohttpClientSession _warningsr-NcSs0|js|dd}|jdur|j|d<dSdSdS)NzUnclosed client session)client_sessionmessagesource_traceback)r\_source_traceback)rrtcontextr+r+r,rs zHBaseApiClient._get_aiohttp_session..AiohttpClientSession.__del__r'r(r)r rrr+r+r+r,AiohttpClientSessionr{c@rs)z?BaseApiClient._get_aiohttp_session..AiohttpTCPConnectorrtr-NcSsX|jrdS|js dSdd|jD}|||dd}|jdur*|j|d<dSdS)NcSsg|]}t|qSr+)repr)r}rr+r+r, sz[BaseApiClient._get_aiohttp_session..AiohttpTCPConnector.__del__..zUnclosed connector) connector connectionsrvrw)r^_connsvalues_closerx)rrtconnsryr+r+r,rs zGBaseApiClient._get_aiohttp_session..AiohttpTCPConnector.__del__rzr+r+r+r,AiohttpTCPConnectorr|r)limitT)r trust_env read_bufsize)r&r`rrr)r9r2r_r6rUrVrAsyncAuthorizedSessionr=r ClientSession TCPConnectorr) rrcr[rUAsyncCredentialsrrrr{rr+r+r,_get_aiohttp_sessionsF       z"BaseApiClient._get_aiohttp_sessionFrAc sd|j}|j}|r|n|r|nd}|dur\|rFttdd}|durFz|}WntyE}z td|WYd}~nd}~ww|dur\tj t j dt t j dd}dttjd tjd tjffd d }||||||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-csb|pi}|s||<d|vrd|d<|}|D]}|ttjjjvr.||=q|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. rkN)rr2inspect signaturerClientr parametersrr copied_argsrLrr+r, _maybe_set\s   z7BaseApiClient._ensure_httpx_ssl_ctx.._maybe_set)rNrOr2rFrrr3r4rcreate_default_contextr0r1certifiwhererr StringDict SSLContext)rAr r async_argsr get_ctx_fnrrr+rr,r/-sD     z#BaseApiClient._ensure_httpx_ssl_ctxc d|j}|r |nd}|durR|r._maybe_setrOr2rFrrr3r4rrr0r1rrrrrrrAr rrrrrr+rr,r;|s4    z%BaseApiClient._ensure_aiohttp_ssl_ctxc r) rrNrrrrrrrr-csZ|r|s|p i}||<|}|D]}|ttjvr*|dkr*||=q|S)rr)r2rrr ws_connectrrrr+r,rs  z;BaseApiClient._ensure_websocket_ssl_ctx.._maybe_setrrr+rr,r?s4    z'BaseApiClient._ensure_websocket_ssl_ctxcCs&to|jjpidduo|jjduS)N transport)rr&rOr2r8rr+r+r,r:s zBaseApiClient._use_aiohttpcCs>|jr|jp|j}|jr|s|jSt|jj}|jddS)Nwss)scheme) r^rr r r r&rTrZgeturl)rrM url_partsr+r+r,_websocket_base_url s   z!BaseApiClient._websocket_base_urlcCsh|j'|jst|jd\|_}|js||_|jr&t||jWdStd1s-wYdS)z/Retrieves the access token for the credentials.rN0Could not resolve API token from the environment)r)r%rfr^get_token_from_credentialsra)rr^r+r+r, _access_tokens  zBaseApiClient._access_tokencsVt}|j||jvrt|j|<|j|WdS1s$wYdS)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)rrr)r*r(rbr+r+r,_get_async_auth_lock!s $z"BaseApiClient._get_async_auth_lockc s|js>|IdH}|4IdH"|js)tjt|jdIdH\|_}|js)||_WdIdHn 1IdHs9wY|jrJt||jIdHStd)z>Retrieves the access token for the credentials asynchronously.Nrr)r%rr to_threadrfr^ async_get_token_from_credentialsra)rasync_auth_lockr^r+r+r,rk.s"(  z!BaseApiClient._async_access_token http_methodrU request_dictc Csdd|D}|D]}||=q |r)t|trt|j|}n t|jt|}n|j}d}|jr<|dkr<|dr.Fr2z publishers/Tz projects/z /locations/rVzBase URL must be set. extra_bodyz auth_tokens/z4Ephemeral tokens can only be used with the live API.zRequest headers must be set.rornr8rprk)rErrrNr&r"r rYr^rr r$r COLLECTIONr-rTrcrRrrrecursive_dict_updater r]r&rlrkr8rSrm) rrrUrrkeys_to_deleterLpatched_http_optionsquery_vertex_base_modelsversioned_pathrTrnrOr+r+r,_build_requestEs          zBaseApiClient._build_request http_requeststreamc Csd}|jr2|js |jr2|js2d||jd<|jr&|jjr&|jj|jd<|jr/t |jnd}n|jrJt |jt sG|jrDt |jnd}n|j}| rt|j}|jdurpddlm}||jdd|_t}|j||jjrd|vrd |vr|d d }n|dd }|jj|j|||j|j|d }n|jj|j|j||j|jd}|jj||d}tj !|t"|j|r|S|j#gS)Nrhrirjrrrmax_refresh_attemptsgoogleapis.comsandboxsandbox.googleapis.commtls.sandbox.googleapis.commtls.googleapis.comrornrpr8rkrrornrr8rkr)$r r^rr rr8r%rlrprdumpsrrtr3rPrnr2rhrrdefault_client_cert_sourceconfigure_mtls_channel_is_mtlsreplacergroupperrkr4 build_requestsendrrraise_for_responsertext) rrrrprnrclient_cert_sourcer httpx_requestr+r+r, _request_oncesf        zBaseApiClient._request_oncecCs^|r't|trtdi|n|}|jr't|j}tjdi|}||j||S||j||SNr+) rrqrr>rrrArrBrrrrparameter_modelrPrr+r+r,rs zBaseApiClient._requestc sd}|jr*|js |jr*|js*d|IdH|jd<|jr*|jjr*|jj|jd<|jr@t |jt s=t |j d}n|j}|r4|r |IdH}|j}|rzt}||IdH|jrzd|vrzd|vrt|dd}n|dd }z|jd|j||j|tj|jd d |jIdH}Wn]tjtjtjtj t!j"fy}zGt#$d t%&d dIdHt'(d||j)|j*t+|jd|_|IdH}|jd|j||j|tj|jd d |jIdH}WYd}~nd}~wwt,j-.|IdHt/|dr|j0}t1|j|S|j2j3|j|j||j|jd} |j2j4| |dIdH} t,j-.| IdHt1| j| S|r(|IdH}|j}|rot}||IdH|jrod|vrod|vri|dd}n|dd }z:|jd|j||j|tj|jd d |jIdH}t,j-.|IdH|} t/| dr| j0} t1| j| 5IdHgWStjtjtjtj t!j"fy'}zgt#$d t%&d dIdHt'(d||j)|j*t+|jd|_|IdH}|jd|j||j|tj|jd d |jIdH}t,j-.|IdH|} t/| dr| j0} t1| j| 5IdHgWYd}~Sd}~ww|j2j|j|j|j||jdIdH} t,j-.| IdHt1| j| j5gS)Nrhrirjrrrrrrtotalrrr z!Retrying due to aiohttp error: %sr _responserrrornr8rrkr+)6r r^rr rkr8r%rlrprrtrrencoder:rrnr6rrrrrrgror ClientTimeoutrkr<ClientConnectorErrorClientConnectorDNSError ClientOSErrorServerDisconnectedErrorauth_exceptionsTransportErrorrsleeprandomrandintr3r+r;r&r0rrraise_for_async_responserRrrr7rrr) rrrrpr[rnrrrrclient_responseunwrapped_responseunwrapped_retry_responser+r+r,_async_request_onces4                      % z!BaseApiClient._async_request_oncecsl|r+t|trtdi|n|}|jr+t|j}tjdi|}||j||IdHS||j||IdHSr) rrqrr>rrrCrrDrr+r+r,_async_requests  zBaseApiClient._async_requestcCs$t|jtr |j}|S|j}|Sr)rr&r model_dump)rcopiedr+r+r,get_read_only_http_optionss  z(BaseApiClient.get_read_only_http_optionscCsB|||||}|j||dd}|jr|jdnd}t|j|dS)NFrrrr8body)rrrvSdkHttpResponser8)rrrUrrrr response_bodyr+r+r,rgszBaseApiClient.requestc cs|||||}|j||dd}|D]=}t|}z|dr6t|} tj | di d| |Wntj j yIt d|Ynwt|j|dVqdS)NTr {"error":errorr1Failed to decode chunk that contains an error: %sr)rrrrrrYrrr raise_errorr2decoderrr3debugrr8) rrrUrrrsession_responser chunk_dump chunk_jsonr+r+r,request_streameds0    zBaseApiClient.request_streamedcsJ|||||}|j||ddIdH}|jr|jdnd}t|j|dS)NFrrrrrr)rrrvrr8)rrrUrrrresultrr+r+r, async_requests zBaseApiClient.async_requestcs:|||||}|j||ddIdHfdd}|S)NTr c s2zD3dHW}t|}z|dr+t|}tj|did|IdHWntjj y>t d|Ynwt j |dVq6dS)Nrrrrr)rrrYrrrraise_error_asyncr2rrr3rrr8)rrrrr+r,async_generators(    z=BaseApiClient.async_request_streamed..async_generator)rr)rrrUrrrrr+r r,async_request_streamed s z$BaseApiClient.async_request_streamedr file_path upload_url upload_sizecCs^t|tjr|j||||dSt|d}|j||||dWdS1s(wYdS)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)rioIOBase _upload_fdopen)rrrrrfiler+r+r, upload_file-s  $zBaseApiClient.upload_filercCsd}|r|n|j}t|tr|dnt|dd}|r.t|}t|}t|j|j|j d} | t } d} | r._write_chunksrrrrrrrrrTrr1rornrpr8rkrr))rrPr0rCrRrcrrrprtrrrr3rnr2rhrr%rrrrrrgrorr8rkrrr iter_contentrr rr4rr iter_bytesr') rrUrr7r1rrprFrnrrrrr+rEr,r:s               c st|tjr|j||||dIdHSt|}|dIdH}|4IdH|j||||dIdHWdIdHS1IdHsCwYdS)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)rrr_async_upload_fdanyioPathr)rrrrrrfdr+r+r,async_upload_file=s   0zBaseApiClient.async_upload_filecsZd}|r|n|j}t|tr|dnt|dd}|r/t|}t|}t|j|j|j d}| r-| IdH} t|t j rH|t} n|tIdH} d} | rXt| } d} | ||krd| d7} t|trn|dn|j} | durt|jtr|jjn|jj} t| }t|tr|d int|d ipi}|rt|ni}|| t|t| d t||d}d}|tkr| jd || |tj|d d IdH}|jdrntt|}|d7}t|IdH|tks|| 7}|dur|jddkrn ||krtdqr2r?rr@r-cstttjfr0td}|2z 3dHW}||q6WddS1s)wYdSdurTtdrV|2z3dHW}|}t|rP|IdHq;6dSdSdSrA) rrPr0rCrr<rRr isawaitable)r@rDrresrEr+r,rFe s  "   z8BaseApiClient.async_download_file.._write_chunksrrrrrrrrrTrrGrr+),rrPr0rCrRrcrrrprtrrrrr:rrnr6rrrrrrgror8rrrkr<rrrrr iter_chunkedrr r'r7rr aiter_bytesr)rrUrr7r1rrprFr[rnrr raw_responserrr+rEr,rQ? s               response_modelcCr8rr+)rrXr+r+r,_verify_response szBaseApiClient._verify_responsecCs0|jjs |jr |j|jr|jdSdS)zCloses the API client.N)r&r5r4rr2rr+r+r,r s  zBaseApiClient.closec s|jjs |jIdH|jrV|jjsXzt}Wn ty%d}Ynwt |j D],\}}||kr=| IdHq-| rUz t | |Wq-tyTYq-wq-dSdSdS)zCloses the API async client.N)r&r8r7rr9r`rrrarrr is_runningrun_coroutine_threadsafe)r current_looprcr[r+r+r,r s*   zBaseApiClient.aclosecCsVz |jjs |Wn tyYnwz t|WdSty*YdSw)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&r5rrrrr rrr+r+r,r s  zBaseApiClient.__del__)NNNNNN)Frror)Kr'r(r)r*r rr0rrr rPr`rardrrrr3r6rr_rrdr staticmethodrr rrr/r;r?r:rrrr(rr rkrqrsrmrrrrrrrrrgrrr rrrintrrr rtr:r0rCrOrL AsyncFilerKrQrrYrrrr+r+r+r,r ~sd        d N @ C      n B  1       "    )  & c    }  ' L    }  r recCs(|js|js t||jstd|jS)=Refreshes the authentication token for the given credentials.r)rprmrjra)rerdr+r+r,r s rc s|js|js9|IdH}|4IdH|js|js$tt|IdHWdIdHn 1IdHs4wY|js@td|jS)r`Nr)rprmrrrrjra)rerdrr+r+r,r s  (r)xr*rcollections.abcrr dataclassesrrrrrrQr0rrr;r'r-typingrrrrrr r r r urllib.parser rr rLr google.authr`google.auth.credentialsrgoogle.auth.transportrrrrpydanticrrrrrrrtypesrrrrrrwebsockets.asyncio.clientrrModuleNotFoundErrorwebsockets.clientrrr=httpx2rhr multidictr requests.structuresr! getLoggerr3r(rr*r+r,r,rrrrTimeoutException ConnectErrorrrcr&rPr7rqr@rNrur^rSr]rfrjrlrmrrrrrrrrrr AsyncClientrr rardrrr+r+r+r,s(  ,                             "    #