o Zh&@sddlZddlmZmZddlZddlZddlmZm Z m Z ddl m Z ddl mZmZddlmZmZddlmZmZdd lmZmZdd lmZdd lmZmZdd lmZm Z dd l!m"Z"m#Z#GdddZ$Gdddej%Z&GdddZ'dS)N)OptionalUnion) BaseApiClient HttpOptionsHttpOptionsDict)ReplayApiClient) AsyncBatchesBatches) AsyncCachesCaches) AsyncChatsChats) AsyncFilesFiles) AsyncLive) AsyncModelsModels)AsyncOperations Operations) AsyncTuningsTuningsc@seZdZdZdefddZedefddZede fdd Z ede fd d Z ede fd d ZedefddZedefddZedefddZedefddZdS) AsyncClientz7Client for making asynchronous (non-blocking) requests. api_clientcCs^||_t|j|_t|j|_t|j|_t|j|_t |j|_ t |j|_ t |j|_dSN) _api_clientr_modelsr_tuningsr _cachesr _batchesr_filesr_liver _operations)selfrr$J/var/www/html/lang_env/lib/python3.10/site-packages/google/genai/client.py__init__%s      zAsyncClient.__init__returncC|jSrrr#r$r$r%models0zAsyncClient.modelscCr(rrr*r$r$r%tunings4r,zAsyncClient.tuningscCr(rrr*r$r$r%caches8r,zAsyncClient.cachescCr(rrr*r$r$r%batches<r,zAsyncClient.batchescC t|jdSN)modules)r r+r*r$r$r%chats@ zAsyncClient.chatscCr(rr r*r$r$r%filesDr,zAsyncClient.filescCr(r)r!r*r$r$r%liveHr,zAsyncClient.livecCr(rr"r*r$r$r% operationsLr,zAsyncClient.operationsN)__name__ __module__ __qualname____doc__rr&propertyrr+rr.r r0r r2r r6rr9rr:rr<r$r$r$r%r"s& rc@sfeZdZUdZejdddZeee d<ejdddZ eee d<ejdddZ eee d <d S) DebugConfigzGConfiguration options that change client network behavior when testing.cC tddS)NZGOOGLE_GENAI_CLIENT_MODEosgetenvr$r$r$r%T zDebugConfig.)default_factory client_modecCrC)NZGOOGLE_GENAI_REPLAYS_DIRECTORYrDr$r$r$r%rGXrHreplays_directorycCrC)NZGOOGLE_GENAI_REPLAY_IDrDr$r$r$r%rG\rH replay_idN) r=r>r?r@pydanticFieldrJrstr__annotations__rKrLr$r$r$r%rBPs rBc@speZdZdZdddddddddeedeedeejj j deedeed ee d ee e effd d Ze       d"deedeedeejj j deedeed ee d ee fd dZedefddZedefddZedefddZedefddZedefddZedefddZedefddZ ede!fddZ"edefd d!Z#dS)#Clienta` Client for making synchronous requests. Use this client to make a request to the Gemini Developer API or 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 express mode can provide API key by providing input argument `api_key="your-api-key"` or by defining `GOOGLE_API_KEY="your-api-key"` as an environment variable Vertex AI API users can provide inputs argument as `vertexai=True, project="your-project-id", location="us-central1"` or by defining `GOOGLE_GENAI_USE_VERTEXAI=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. vertexai: Indicates whether the client should use the Vertex AI API endpoints. Defaults to False (uses Gemini Developer API endpoints). Applies to the Vertex AI API only. 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: 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: 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 Vertex AI API: .. code-block:: python from google import genai client = genai.Client( vertexai=True, project='my-project-id', location='us-central1' ) Nvertexaiapi_key credentialsprojectlocation debug_config http_optionsrSrTrUrVrWrXrYc Cs|pt|_|j||||||j|d|_t|j|_t|j|_t|j|_ t |j|_ t |j|_ t|j|_t|j|_dS)a{Initializes the client. Args: vertexai (bool): Indicates whether the client should use the Vertex AI API endpoints. Defaults to False (uses Gemini Developer API endpoints). Applies to the Vertex AI API only. 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. rRN)rBZ _debug_config_get_api_clientrr_aiorrrrr rr rrr rr")r#rSrTrUrVrWrXrYr$r$r%r&s" (     zClient.__init__c CsB|r|jdvrt|j|j|j||||||d St||||||dS)N)recordZreplayauto) moderLrKrSrTrUrVrWrY)rSrTrUrVrWrY)rJrrLrKrrRr$r$r%rZs(  zClient._get_api_clientr'cCr3r4)rr+r*r$r$r%r6r7z Client.chatscCr(r)r[r*r$r$r%aior,z Client.aiocCr(rr)r*r$r$r%r+r,z Client.modelscCr(rr-r*r$r$r%r.r,zClient.tuningscCr(rr/r*r$r$r%r0 r,z Client.cachescCr(rr1r*r$r$r%r2r,zClient.batchescCr(rr8r*r$r$r%r9r,z Client.filescCr(rr;r*r$r$r%r<r,zClient.operationscCs |jjpdS)z6Returns whether the client is using the Vertex AI API.F)rrSr*r$r$r%rSs zClient.vertexai)NNNNNNN)$r=r>r?r@rboolrOgoogleauthrU CredentialsrBrrrr& staticmethodrZrArr6rr_rr+rr.r r0r r2rr9rr<rSr$r$r$r%rQ`s>  <  #rQ)(rEtypingrrZ google.authrarMrrrrZ_replay_api_clientrr2r r r0r r r6r rr9rrr:rr+rrr<rrr.rrrZ BaseModelrBrQr$r$r$r%s"  .