+j"Z,ddlZddlZddlmZmZmZddlZddlZddl m Z ddl m Z ddl mZddlmZmZddlmZmZdd lmZmZdd lmZmZdd lmZmZdd lmZdd l m!Z!m"Z"ddl#m$Z$m%Z%ddl&m'Z'm(Z(ddl)m*Z*m+Z+ddl,m-Z-m.Z.m/Z/ddl0Z0ddl1Z1ddl m2Z2ddl3m4Z4ddl5m6Z6m7Z7m8Z8ddl3m5Z5ddl9m:Z;mZ>m?Z?m@Z@mAZAdaBdaCGdde5jDZDGdde5jEZEGddZFGdd ejGZHGd!d"ZIdS)#N)AnyOptionalUnion) BaseApiClient) get_base_url)ReplayApiClient) AsyncBatchesBatches) AsyncCachesCaches) AsyncChatsChats)AsyncFileSearchStoresFileSearchStores) AsyncFilesFiles) AsyncLive) AsyncModelsModels)AsyncOperations Operations) AsyncTokensTokens) AsyncTuningsTunings) HttpOptionsHttpOptionsDictHttpRetryOptions) has_aiohttp)_common)AsyncGeminiNextGenAPIClientDEFAULT_MAX_RETRIESGeminiNextGenAPIClient) _interactions)AsyncInteractionsResourceInteractionsResource)WebhooksResourceAsyncWebhooksResourceAgentsResourceAsyncAgentsResourceFcjeZdZdZdefdZdefdZdedzfdZ dedzfdZ de eeffd Z dS) "AsyncGeminiNextGenAPIClientAdapter*Adapter for the Gemini NextGen API Client. api_clientc||_dSN _api_clientselfr/s W/home/wildlama/visual-decline/.venv/lib/python3.11/site-packages/google/genai/client.py__init__z+AsyncGeminiNextGenAPIClientAdapter.__init__7!Dreturnc|jjpdSNFr3vertexair5s r6 is_vertex_aiz/AsyncGeminiNextGenAPIClientAdapter.is_vertex_ai:   $ --r9Nc|jjSr1r3projectr?s r6 get_projectz.AsyncGeminiNextGenAPIClientAdapter.get_project=   ##r9c|jjSr1r3locationr?s r6 get_locationz/AsyncGeminiNextGenAPIClientAdapter.get_location@   $$r9cK|jjrd|jjiS|jd{V}dd|i}|jjx}r|jr |j|d<|SNzx-goog-api-key AuthorizationzBearer zx-goog-user-project)r3api_key_async_access_token _credentialsquota_project_idr5 access_tokenheaderscredss r6async_get_auth_headersz9AsyncGeminiNextGenAPIClientAdapter.async_get_auth_headersCs : 0 8 99)==????????L///G --u@  @).)?%& Nr9) __name__ __module__ __qualname____doc__rr7boolr@strrErJdictrWr9r6r-r-5s22""""".D....$3:$$$$%C$J%%%% DcN      r9r-cjeZdZdZdefdZdefdZdedzfdZ dedzfdZ de eeffd Z dS) GeminiNextGenAPIClientAdapterr.r/c||_dSr1r2r4s r6r7z&GeminiNextGenAPIClientAdapter.__init__Rr8r9r:c|jjpdSr<r=r?s r6r@z*GeminiNextGenAPIClientAdapter.is_vertex_aiUrAr9Nc|jjSr1rCr?s r6rEz)GeminiNextGenAPIClientAdapter.get_projectXrFr9c|jjSr1rHr?s r6rJz*GeminiNextGenAPIClientAdapter.get_location[rKr9c|jjrd|jjiS|j}dd|i}|jjx}r|jr |j|d<|SrM)r3rO _access_tokenrQrRrSs r6get_auth_headersz.GeminiNextGenAPIClientAdapter.get_auth_headers^s : 0 8 99#1133L///G --u@  @).)?%& Nr9) rXrYrZr[rr7r\r@r]rErJr^rhr_r9r6raraPs22""""".D....$3:$$$$%C$J%%%% S#X      r9raceZdZdZdefdZedefdZede fdZ ede fdZ ede fdZedefd Zedefd Zedefd Zedefd Zedefd ZedefdZedefdZedefdZedefdZ ede!fdZ"ede#fdZ$ddZ%ddZ&de'de'ddfdZ(ddZ)dS) AsyncClientz7Client for making asynchronous (non-blocking) requests.r/c||_t|j|_t|j|_t |j|_t|j|_t|j|_ t|j|_ t|j|_t|j|_t#|j|_d|_dSr1)r3r_modelsr_tuningsr _cachesr _batchesr_filesr_file_search_storesr_liver_tokensr _operations_nextgen_client_instancer4s r6r7zAsyncClient.__init__ns!Dt/00DL !122DMt/00DL !122DMT-..DK4T5EFFD4+,,DJt/00DL&t'788DKOD!!!r9r:c |j|jS|jj}|jrt jdt d|j}|?|j|j |j |j |j t jdt d|jj }|jjjpi}d|v}tr|st jdt d||j|j}n t"dz}t%|j|jj|j|j||jr |jdz nd|t1|j |_|j}|jjr)d |_|jj|_|jj|_|jS) N>extra_body properties are not supported in `.interactions` yetcategory stacklevel?Granular retry options are not supported in `.interactions` yet transportzCAsync interactions client cannot use aiohttp, fallingback to httpx.rbase_urlrO api_versiondefault_headers http_clienttimeout max_retriesclient_adapterT)rur3 _http_options extra_bodywarningswarn UserWarning retry_options initial_delay max_delayexp_basejitterhttp_status_codes_async_httpx_clientasync_client_argsr attemptsr#r"rrOrrUrr-r> _is_vertexrD_vertex_projectrI_vertex_location)r5 http_opts retry_optsrrhas_custom_transportrclients r6_nextgen_clientzAsyncClient._nextgen_client|s $0  ** .Im J (J ,   +   *   (  ' 3m K ,(6HNB&*;;/m O *"5"A'kk'!+k$?# ()!),5,=G !D((49$:JKK % % %D! *F  :f#/7f $ 0 9f  ((r9cdtsdatjdtd|jjS)NTEInteractions usage is experimental and may change in future versions.rry!_interactions_experimental_warnedrrrr interactionsr?s r6rzAsyncClient.interactionssB -*.'m Q   ,,r9c|jjSr1rwebhooksr?s r6rzAsyncClient.webhooks   ((r9cdtsdatjdtd|jjS)NT?Agents usage is experimental and may change in future versions.rry_agent_experimental_warnedrrrragentsr?s r6rzAsyncClient.agentssB &#' m K   &&r9c4t|do|jduSNruhasattrrur?s r6_has_nextgen_clientzAsyncClient._has_nextgen_client' 011 2 %T1r9c|jSr1rlr?s r6modelszAsyncClient.models <r9c|jSr1rmr?s r6tuningszAsyncClient.tunings =r9c|jSr1rnr?s r6cacheszAsyncClient.cachesrr9c|jSr1rqr?s r6file_search_storeszAsyncClient.file_search_stores  ##r9c|jSr1ror?s r6batcheszAsyncClient.batchesrr9c,t|jSN)modules)rrr?s r6chatszAsyncClient.chatss dk * * **r9c|jSr1rpr?s r6fileszAsyncClient.files ;r9c|jSr1)rrr?s r6livezAsyncClient.lives :r9c|jSr1rsr?s r6 auth_tokenszAsyncClient.auth_tokensrr9c|jSr1rtr?s r6 operationszAsyncClient.operations  r9NcK|jd{V|jr!|jd{VdSdS)aCloses the async client explicitly. However, it doesn't close the sync client, which can be closed using the Client.close() method or using the context manager. Usage: .. code-block:: python from google.genai import Client async_client = Client( vertexai=True, project='my-project-id', location='us-central1' ).aio response_1 = await async_client.models.generate_content( model='gemini-2.0-flash', contents='Hello World', ) response_2 = await async_client.models.generate_content( model='gemini-2.0-flash', contents='Hello World', ) # Close the client to release resources. await async_client.aclose() N)r3acloserrcloser?s r6rzAsyncClient.aclose sq2   ! ! # ######## )  & & ( (((((((((())r9c K|Sr1r_r?s r6 __aenter__zAsyncClient.__aenter__(s Kr9argskwargscBK~~|d{VdSr1)rr5rrs r6 __aexit__zAsyncClient.__aexit__+s1 f ++--r9c tj|dS#t$rYdSwxYwr1)asyncioget_running_loop create_taskr Exceptionr?s r6__del__zAsyncClient.__del__/sT    ,,T[[]];;;;;     dd s8< A  A r:N)r:rj)*rXrYrZr[rr7propertyr"r AsyncNextGenInteractionsResourcerr)rr+rr\rrrrrr rrrr rrrrrrrrrrrrrrrrr_r9r6rjrjks?? P P P P P @):@)@)@) 8@)D  -< - - - 8 - )-))) 8)  ') ' ' ' 8 ' 4 8  k 8 | 8 k 8 $"7$$$ 8$ | 8 +Z+++ 8+ Z 8 I 8 ; 8 / 8))))<3#$      r9rjceZdZUdZejdZeee d<ejdZ eee d<ejdZ eee d<d S) DebugConfigzGConfiguration options that change client network behavior when testing.c,tjddS)NGOOGLE_GENAI_CLIENT_MODEosgetenvr_r9r6zDebugConfig.:sbi(BDIIr9)default_factory client_modec,tjddS)NGOOGLE_GENAI_REPLAYS_DIRECTORYrr_r9r6rzDebugConfig.>sbi(H$OOr9replays_directoryc,tjddS)NGOOGLE_GENAI_REPLAY_IDrr_r9r6rzDebugConfig.Bsbi(@$GGr9 replay_idN) rXrYrZr[pydanticFieldrrr]__annotations__rrr_r9r6rr6sOO-x~II   +x}&4X^OO&&&Xc],X^GG)Xc]r9rceZdZdZddddddddddeedeedeedeejj j deed eed ee d ee e effd Ze d%deedeedeejj j deed eed ee d ee d efdZed efdZed efdZed efdZed efdZed efdZed efdZed efdZ ed e!fdZ"ed e#fdZ$ed e%fdZ&ed e'fdZ(ed e)fdZ*ed e+fdZ,ed e-fdZ.ed e/fdZ0ed efdZ1d&dZ2d'd Z3d!e4d"e4d dfd#Z5d&d$Z6dS)(Clienta Client for making synchronous requests. Use this client to make a request to the Gemini Developer API or Gemini Enterprise Agent Platform (previously Vertex AI API) and then wait for the response. To initialize the client, provide the required arguments either directly or by using environment variables. Gemini API users and Vertex AI users in `api_key="your-api-key"` or by defining `GOOGLE_API_KEY="your-api-key"` as an environment variable Gemini Enterprise Agent Platform API users can provide inputs argument as `enterprise=True, project="your-project-id", location="us-central1"` or by defining `GOOGLE_GENAI_USE_ENTERPRISE=true`, `GOOGLE_CLOUD_PROJECT` and `GOOGLE_CLOUD_LOCATION` environment variables. Attributes: api_key: The `API key `_ to use for authentication. Applies to the Gemini Developer API only. enterprise (bool): Indicates whether the client should use the Gemini Enterprise Agent Platform endpoints (previously Vertex AI API). Defaults to False (uses Gemini Developer API endpoints). When `enterprise` and `vertexai` are both set, and they have conflicting values, a `ValueError` will be raised. vertexai (bool): Legacy flag for `enterprise`. credentials: The credentials to use for authentication when calling the Gemini Enterprise Agent Platform APIs. Credentials can be obtained from environment variables and default credentials. For more information, see `Set up Application Default Credentials `_. Applies to the Vertex AI API only. project: The `Google Cloud project ID `_ to use for quota. Can be obtained from environment variables (for example, ``GOOGLE_CLOUD_PROJECT``). Applies to the Vertex AI API only. Find your `Google Cloud project ID `_. location: The `location `_ to send API requests to (for example, ``us-central1``). Can be obtained from environment variables. Applies to the Vertex AI API only. debug_config: Config settings that control network behavior of the client. This is typically used when running test code. http_options: Http options to use for the client. These options will be applied to all requests made by the client. Example usage: `client = genai.Client(http_options=types.HttpOptions(api_version='v1'))`. Usage for the Gemini Developer API: .. code-block:: python from google import genai client = genai.Client(api_key='my-api-key') Usage for the Gemini Enterprise Agent Platform API: .. code-block:: python from google import genai client = genai.Client( enterprise=True, project='my-project-id', location='us-central1' ) N) enterpriser>rO credentialsrDrI debug_config http_optionsrr>rOrrDrIrrc (|p t|_||||krtd||n|} t|tr t di|}t | pd|} | r|r| |_nt | }|| |||||j||_ t|j |_ t|j |_ t|j |_t!|j |_t%|j |_t)|j |_t-|j |_t1|j |_t5|j |_d|_dS)a@Initializes the client. Args: enterprise (bool): Indicates whether the client should use the Gemini Enterprise Agent Platform endpoints (previously Vertex AI API). Defaults to False (uses Gemini Developer API endpoints). When `enterprise` and `vertexai` are both set, and they have conflicting values, a `ValueError` will be raised. vertexai (bool): Legacy flag for `enterprise`. api_key (str): The `API key `_ to use for authentication. Applies to the Gemini Developer API only. credentials (google.auth.credentials.Credentials): The credentials to use for authentication when calling the Vertex AI APIs. Credentials can be obtained from environment variables and default credentials. For more information, see `Set up Application Default Credentials `_. Applies to the Vertex AI API only. project (str): The `Google Cloud project ID `_ to use for quota. Can be obtained from environment variables (for example, ``GOOGLE_CLOUD_PROJECT``). Applies to the Vertex AI API only. location (str): The `location `_ to send API requests to (for example, ``us-central1``). Can be obtained from environment variables. Applies to the Vertex AI API only. debug_config (DebugConfig): Config settings that control network behavior of the client. This is typically used when running test code. http_options (Union[HttpOptions, HttpOptionsDict]): Http options to use for the client. NzXenterprise and vertexai flags have conflicting values, please set enterprise value only.F)rr>rOrrDrIrrr_)r _debug_config ValueError isinstancer^rrr_get_api_clientr3rj_aiorrlrrmr rnrrqr rorrprrsrrtru) r5rr>rOrrDrIrrresolved_vertexairs r6r7zClient.__init__sX&6D("6:;Q;Q  $   '1&< (,%%1 00<00l-6 EEH6 6 ( "H555 ++"'!,DD,--DI$*++DLD,--DM$*++DL/0@AADD,--DM())DK$*++DL!$"233DFJD!!!r9r:c |r0|jdvr't|j|j|j|||||| St ||||||S)N)recordreplayauto) moderrr>rOrrDrIr)r>rOrrDrIr)rr rrrrs r6rzClient._get_api_clients 05 ' *(:!#     !    r9c |j|jS|jj}|jrt jdt d|j}|?|j|j |j |j |j t jdt d||j |j }n tdz}t|j|jj|j|j|jj|jr |jdz nd|t-|j|_|j}|jjr)d|_|jj|_|jj|_|jS) Nrwrxryr|rr~rT)rur3rrrrrrrrrrrrr#r$rrOrrU _httpx_clientrrar>rrDrrIr)r5rrrrs r6rzClient._nextgen_clientsx $0  ** .Im J (J ,   +   *   (  ' 3m K *"5"A'kk'!+k$:# ()!)$2,5,=G !D((44T5EFF % % %D! *F  :f#/7f $ 0 9f  ((r9cdtsdatjdtd|jjS)NTrryrr?s r6rzClient.interactions8sB -*.'mO   ,,r9c|jjSr1rr?s r6rzClient.webhooksDrr9cdtsdatjdtd|jjS)NTrrryrr?s r6rz Client.agentsHsB &#' mI   &&r9c4t|do|jduSrrr?s r6rzClient._has_nextgen_clientTrr9c,t|jSr)rrr?s r6rz Client.chats[s  % % %%r9c|jSr1)rr?s r6aioz Client.aio_s 9r9c|jSr1rr?s r6rz Client.modelscrr9c|jSr1rr?s r6rzClient.tuningsgrr9c|jSr1rr?s r6rz Client.cacheskrr9c|jSr1rr?s r6rzClient.file_search_storesorr9c|jSr1rr?s r6rzClient.batchessrr9c|jSr1rr?s r6rz Client.fileswrr9c|jSr1rr?s r6rzClient.auth_tokens{rr9c|jSr1rr?s r6rzClient.operationsrr9c|jjpdS)z6Returns whether the client is using the Vertex AI API.Fr=r?s r6r>zClient.vertexais   $ --r9c||j|jr|jdSdS)aCloses the synchronous client explicitly. However, it doesn't close the async client, which can be closed using the Client.aio.aclose() method or using the async context manager. Usage: .. code-block:: python from google.genai import Client client = Client( vertexai=True, project='my-project-id', location='us-central1' ) response_1 = client.models.generate_content( model='gemini-2.0-flash', contents='Hello World', ) response_2 = client.models.generate_content( model='gemini-2.0-flash', contents='Hello World', ) # Close the client to release resources. client.close() N)r3rrrr?s r6rz Client.closesI2  #   """""##r9c|Sr1r_r?s r6 __enter__zClient.__enter__s Kr9rrc2~~|dSr1)rrs r6__exit__zClient.__exit__s fJJLLLLLr9cR |dS#t$rYdSwxYwr1)rrr?s r6rzClient.__del__s:  jjlllll     dd s  &&)NNNNNNNr)r:r)7rXrYrZr[rr\r]googleauthr Credentialsrrrrr7 staticmethodrrrr$rNextGenInteractionsResourcerr(rr*rrrrrjrrrrrr rrrr rrrrrrrr>rr"rr$rr_r9r6rrFsAAL$(!%#CG# $,0DHSKSKSK4.SK SK } SK FK3?@ SK}SK SK[)SKU;#?@ASKSKSKSKj!%#CG# $,0,0!!!}!FK3?@!} !  ! [) ![)!!!!<!F 2)52)2)2) 82)h  -7 - - - 8 - )())) 8)  'n ' ' ' 8 ' 4 8  &U&&& 8& ; 8 f 8 w 8 f 8 $"2$$$ 8$ w 8 U 8 6 8 * 8 .... 8.####<C34      r9r)Jrrtypingrrr google.authr&rr3r _base_urlr_replay_api_clientr rr r rr r rrrrrrrrrrrrrrrrrtokensrrrrrtypesrrrrhttpxr r!r%r"r#r$_interactions.resourcesr&rr'r*r(r)r*r+rrr-rarj BaseModelrrr_r9r6r5sD  ''''''''''&&&&&&######//////********''''''''$$$$$$$$GGGGGGGG$$$$$$$$''''''''33333333''''''''********AAAAAAAAAA $$$$$$ccccccccccXXXXXXXXqqqqqqqqqqqq$)!")Y6M$O6H H H H H H H H V     ($    k k k k k k k k k k r9