
    &`i^                     X    d dl mZ d dlmZ erd dlmZmZ d dlmZ  G d d          Z	dS )    )SimpleNamespace)TYPE_CHECKING)JobIDNodeIDRuntimeContextc                       e Zd ZdZd ZddZd Zed             Zedd	            Z	edd            Z
edefd            Zedefd            Zedefd            ZdefdZedefd            ZdS )_ClientWorkerPropertyAPIzHEmulates the properties of the ray._private.worker object for the clientc                     |J || _         d S N)worker)selfr   s     s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/client/runtime_context.py__init__z!_ClientWorkerPropertyAPI.__init__   s    !!!    returnr   c                 $    ddl m}  ||           S )z=Creates a RuntimeContext backed by the properites of this APIr   r   )ray.runtime_contextr   )r   r   s     r   build_runtime_contextz._ClientWorkerPropertyAPI.build_runtime_context   s&     	766666~d###r   c                 b    dd l mc mc m} | j                            |j        j                  S )Nr   )!ray.core.generated.ray_client_pb2core	generatedray_client_pb2r   get_cluster_infoClusterInfoTypeRUNTIME_CONTEXT)r   r   s     r   _fetch_runtime_contextz/_ClientWorkerPropertyAPI._fetch_runtime_context   sI    BBBBBBBBBBBB{++*:
 
 	
r   c                     ddl m} |S )Nr   )SCRIPT_MODE)ray._private.workerr    )r   r    s     r   modez_ClientWorkerPropertyAPI.mode   s    333333r   r   c                 R    ddl m}  ||                                 j                  S )Nr   )r   )rayr   r   job_id)r   r   s     r   current_job_idz'_ClientWorkerPropertyAPI.current_job_id$   s3    uT00229:::r   r   c                 R    ddl m}  ||                                 j                  S )Nr   )r   )r$   r   r   node_id)r   r   s     r   current_node_idz(_ClientWorkerPropertyAPI.current_node_id*   s3    vd1133;<<<r   c                 4    |                                  j        S r   )r   	namespacer   s    r   r+   z"_ClientWorkerPropertyAPI.namespace0   s    **,,66r   c                 4    |                                  j        S r   )r   capture_client_tasksr,   s    r   -should_capture_child_tasks_in_placement_groupzF_ClientWorkerPropertyAPI.should_capture_child_tasks_in_placement_group4   s    **,,AAr   c                 4    |                                  j        S r   )r   runtime_envr,   s    r   r1   z$_ClientWorkerPropertyAPI.runtime_env8   s    **,,88r   c                 4    | j                                         S r   )r   ping_serverr,   s    r   check_connectedz(_ClientWorkerPropertyAPI.check_connected<   s    {&&(((r   c                 P    t          |                                 j                  S )N)address)r   r   gcs_addressr,   s    r   
gcs_clientz#_ClientWorkerPropertyAPI.gcs_client?   s"    t'B'B'D'D'PQQQQr   N)r   r   )r   r   )r   r   )__name__
__module____qualname____doc__r   r   r   propertyr"   r&   r)   strr+   boolr/   r1   r4   r8    r   r   r
   r
   	   sj       RR  $ $ $ $
 
 
   X
 ; ; ; X;
 = = = X=
 73 7 7 7 X7 Bt B B B XB 9S 9 9 9 X9) ) ) ) ) RC R R R XR R Rr   r
   N)
typesr   typingr   r$   r   r   r   r   r
   r@   r   r   <module>rC      s    ! ! ! ! ! !             3!!!!!!!!2222228R 8R 8R 8R 8R 8R 8R 8R 8R 8Rr   