
    ci                         d Z ddlZddlZddlmZmZ ddlmZ ddlm	Z	 ddlm
Z
 ddlmZ  ej                  d      Zerdd	lmZ  G d
 de	j$                        Zy)z[Preview] Live API client.    N)OptionalTYPE_CHECKING)
ModuleType)_api_module)_common)BaseApiClientzgoogle_genai.live)live_agent_enginesc                   d     e Zd ZdZdef fdZe ej                  d      dd              Z	 xZ
S )	AsyncLivez[Preview] AsyncLive.
api_clientc                 2    t         |   |       d | _        y )N)super__init___agent_engines)selfr   	__class__s     ;/tmp/pip-target-z3e9_cxr/lib/python/vertexai/_genai/live.pyr   zAsyncLive.__init__'   s    $48    z]The Vertex SDK GenAI agent engines module is experimental, and may change in future versions.c                     | j                    	 t        j                  dt              | _         | j                   j                  | j                        S # t        $ r}t	        d      |d }~ww xY w)Nz.live_agent_engineszThe 'agent_engines' module requires 'additional packages'. Please install them using pip install google-cloud-aiplatform[agent_engines])r   	importlibimport_module__package__ImportErrorAsyncLiveAgentEngines_api_client)r   es     r   agent_engineszAsyncLive.agent_engines+   sx     & '0&=&=)'# ""889I9IJJ  != 	s   A 	A,A''A,)returnz/live_agent_engines_module.AsyncLiveAgentEngines)__name__
__module____qualname____doc__r   r   propertyr   experimental_warningr   __classcell__)r   s   @r   r   r   $   sB    9= 9 !W!!	-K	 
Kr   r   )r"   r   loggingtypingr   r   typesr   google.genair   r   google.genai._api_clientr   	getLoggerloggervertexai._genair	   live_agent_engines_module
BaseModuler    r   r   <module>r1      sN     !   *  $   2			.	/
K&& Kr   