ci3(ddlZddlZddlmZmZmZddlmZmZddl Z ddl m Z ddlmZddlmZddlmZddlmZd d lmZer$dd lmZdd lmZdd lmZddlmZddlmZdd lmZ dZ!de"e#e#fddfdZ$e$e_%GddZ&GddZ'y)N)OptionalUnion TYPE_CHECKING) TracebackType ModuleType)version) _api_client)_common)client)types)live) agent_engines)datasets)evals)prompt_optimizer)promptszvertex-genai-modulesheadersreturnctdtj}|jdd}||vr|d|j |d<|jdd}||vr|d|j |d<yy)zJAppends Vertex Gen AI modules tracking information to the request headers./z user-agent zx-goog-api-clientN)_GENAI_MODULES_TELEMETRY_HEADER aip_version __version__getstrip)rtracking_label user_agent api_clients =/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/client.py_add_tracking_headersr#-s88+:Q:Q9RSN\2.JZ'#-,a/? @ F F H 0"5JZ'*4Q~6F'G'M'M'O#$(c2eZdZdZdej fdZeejdddZ eddZ eddZ edd Z edd Zeejd dd ZddZddZdeedeedeedd fdZddZy ) AsyncClientz'Async Gen AI Client for the Vertex SDK.r!c||_tj|j|_d|_d|_d|_d|_d|_yN) r r AsyncLive_live_evals_agent_engines_prompt_optimizer_prompts _datasets)selfr!s r"__init__zAsyncClient.__init__AsE%^^D$4$45 ,0 487;.2 /3r$zTThe Vertex SDK GenAI live module is experimental, and may change in future versions.rc|jSr()r*r0s r"rzAsyncClient.liveJs zzr$c|j tjdt|_|jj |j S#t$r}t d|d}~wwxYw)N.evalszzThe 'evals' module requires 'pandas' and 'tqdm'. Please install them using pip install google-cloud-aiplatform[evaluation])r+ importlib import_module __package__ ImportError AsyncEvalsr r0es r"rzAsyncClient.evalsRsm ;;  (55h L {{%%d&6&677  !:  A A, A''A,c|jtjdt|_|jj |j SNz.prompt_optimizer)r-r6r7r8AsyncPromptOptimizerr r3s r"rzAsyncClient.prompt_optimizerasE  ! ! )%.%<%<#[&D "%%::4;K;KLLr$c|j tjdt|_|jj |j S#t$r}t d|d}~wwxYwNz.agent_engineszThe 'agent_engines' module requires 'additional packages'. Please install them using pip install google-cloud-aiplatform[agent_engines])r,r6r7r8r9AsyncAgentEnginesr r;s r"rzAsyncClient.agent_enginesisv    & '0&=&=$'#""44T5E5EFF  !=  r=c|jtjdt|_|jj |j SNz.prompts)r.r6r7r8 AsyncPromptsr r3s r"rzAsyncClient.prompts{sA == %33DM}}))$*:*:;;r$z^The Vertex SDK GenAI async datasets module is experimental, and may change in future versions.c|jtjdt|_|jj |j SNz .datasets)r/r6r7r8 AsyncDatasetsr r3s r"rzAsyncClient.datasetssC >> !&44DN~~++D,<,<==r$NcTK|jjd{y7w)aCloses the async client explicitly. Example usage: from vertexai import Client async_client = vertexai.Client( project='my-project-id', location='us-central1' ).aio prompt_1 = await async_client.prompts.create(...) prompt_2 = await async_client.prompts.create(...) # Close the client to release resources. await async_client.aclose() N)r acloser3s r"rKzAsyncClient.acloses %%'''s (&(cK|Swr(r3s r" __aenter__zAsyncClient.__aenter__s  sexc_type exc_value tracebackc@K|jd{y7wr()rK)r0rOrPrQs r" __aexit__zAsyncClient.__aexit__s kkms c tjj|jy#t$rYywxYwr()asyncioget_running_loop create_taskrK Exceptionr3s r"__del__zAsyncClient.__del__s6   $ $ & 2 24;;= A   s14 AA)rzlive_module.AsyncLive)rzevals_module.AsyncEvals)rz,prompt_optimizer_module.AsyncPromptOptimizer)rz&agent_engines_module.AsyncAgentEngines)rzprompts_module.AsyncPrompts)rzdatasets_module.AsyncDatasets)rNrr&)__name__ __module__ __qualname____doc__ genai_client BaseApiClientr1propertyr experimental_warningrrrrrrrKrNrrXrrSrYrMr$r"r&r&>s14<#=#=4!W!!    8 8MMGG"<<!W!! ->  >("9%I&M*   r$r&cZeZdZdZddddddddeedeejjjdeedeedee jd ee e je jff d Zedd Zedd ZeddZe ddeedeejjjdeedeedee jd ee jd ee j,fdZeddZeddZeej6dddZy)ClientzmGen AI Client for the Vertex SDK. Use this client to interact with Vertex-specific Gemini features. Napi_key credentialsprojectlocation debug_config http_optionsrfrgrhrirjrkc l|xstj|_t|trt j di|}tjjd|||||j||_ t|j|_ d|_ d|_ d|_d|_d|_y)aInitializes the client. Args: api_key (str): The `API key `_ to use for authentication. Applies to Vertex AI in express mode 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 `_. project (str): The `Google Cloud project ID `_ to use for quota. Can be obtained from environment variables (for example, ``GOOGLE_CLOUD_PROJECT``). location (str): The `location `_ to send API requests to (for example, ``us-central1``). Can be obtained from environment variables. 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. T)vertexairfrgrhrirjrkNrM)r_ DebugConfig _debug_config isinstancedictr HttpOptionsrd_get_api_clientr r&_aior+r-r,r.r/)r0rfrgrhrirjrks r"r1zClient.__init__sF*G\-E-E-G lD ) ,,<|>#++%?   0 01 ,0 7;48.2 /3r$rc|j tjdt|_|jj |j S#t$r}t d|d}~wwxYw)Nr5z~The 'evals' module requires additional dependencies. Please install them using pip install google-cloud-aiplatform[evaluation])r+r6r7r8r9Evalsr r;s r"rz Client.evalssm ;;  (55h L {{  !1!122  !:  r=c|jtjdt|_|jj |j Sr?)r-r6r7r8PromptOptimizerr r3s r"rzClient.prompt_optimizersE  ! ! )%.%<%<#[&D "%%55d6F6FGGr$c|jSr()rtr3s r"aioz Client.aios yyr$c |rJ|jdvr> !&44DN~~&&t'7'788r$)rzevals_module.Evals)rz'prompt_optimizer_module.PromptOptimizerrZ)NNNNNN)rz!agent_engines_module.AgentEngines)rzprompts_module.Prompts)rzdatasets_module.Datasets)r[r\r]r^rstrgoogleauthrg Credentialsr_rnrr rrHttpOptionsDictr1rarrrz staticmethodr`rsrrr rbrrMr$r"rdrds"&EI!%"&;?RV54#54fkk55AAB 54 # 54 3- 54|77854uU%6%68M8M%MNO54n 3 3HH!%EI!%"&;?48 #fkk55AAB#3-  |778  u001  ,,, -4BB"77!W!! -9  9r$rd)(rUr6typingrrrr rr google.authrgoogle.cloud.aiplatformrr google.genair r r r_rrvertexai._genairagent_engines_modulerdatasets_moduler evals_modulerprompt_optimizer_modulerprompts_module live_modulerrqrr#append_library_version_headersr&rdrMr$r"rs 11+:$ /<5:3#9 P4S> Pd P.C *ssl[9[9r$