
    &`i#              	       6   d dl Z d dlZd dlZd dlZd dlmc mZ d dlmZ d dl	m
Z
 d dlmZ d dlmZ d dlmZ d dlmZ d Z e             d dlZd dlZd d	lmZ d d
lmZ edk    r e j        d          Ze                    dded           e                    ddedd           e                    dded           e                    dded           e                    ddedd           e                    ddd ej        ej         ej!                   e                    ddeej"        ej#                   e                    d deej$        d!%                    ej$                             e                    d"ded#           e                    d$ded%           e                    d&dedd'           e                    d(dedd)           e                    d*ded+d,           e                    d-ded+d.           e&                                Z'ej(        d/k    re'j)        nd Z*ej(        d/k    re'j+        nd0Z, e-e'j.        e'j/        e'j0        e'j1        e*e,1          Z2 e
j3        e'j4        e'j5        e*e,            ee'j6        e'j7        2          Z8 ee'j9        e2e8e'j:        e'j;        e'j<        3          Z=d4ej>        d5ej?        fd6Z@d4ej>        d5ej?        fd7ZAd4ej>        d5ej?        fd8ZB ejC         ee          g9          ZDeDjE        F                    d:e@           eDjE        F                    d;eA           eDjE        F                    d<eB            e            ZGdZHej(        d=vrd> ZI ee'j0        e8eIeG          ZH	  ejJ        eDe'j;        e'j<        eG?           dS # eK$ rSZLe=jM        N                    d@eL            eHeHO                                  ejP        eLjQ                   Y dZL[LdS dZL[Lww xY wdS )A    N)get_or_create_event_loop)logging_utils)get_token_auth_middleware)create_check_raylet_task)	GcsClient)runtime_env_agent_pb2c                  D   t           j                            t           j                            t                              } t
          j                            dt           j                            | d                     t
          j                            d|            d S )Nr   thirdparty_files)ospathabspathdirname__file__sysinsertjoin)my_dirs    w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/runtime_env/agent/main.pyimport_libsr      sf    W__RW__X6677FHOOArw||F,>??@@@HOOAv    )web)RuntimeEnvAgent__main__zRuntime env agent.)descriptionz--node-ip-addressTzthe IP address of this node.)requiredtypehelpz--runtime-env-agent-portzCThe port on which the runtime env agent will receive HTTP requests.)r   r   defaultr   z--gcs-addresszThe address (ip:port) of GCS.z--cluster-id-hexzThe cluster id in hex.z--runtime-env-dirz?Specify the path of the resource directory used by runtime_env.z--logging-levelFc                 N    t          j        |                                           S )N)logginggetLevelNameupper)ss    r   <lambda>r$   C   s    w+AGGII66 r   )r   r   r   choicesr   z--logging-formatz--logging-filenamezJSpecify the name of log file, log to stdout if set empty, default is "{}".z--logging-rotate-bytesz+Specify the max bytes for rotating log filez--logging-rotate-backup-countz,Specify the backup count of rotated log filez	--log-dirz"Specify the path of log directory.z
--temp-dirz?Specify the path of the temporary directory use by Ray process.z--stdout-filepath z.The filepath to dump runtime env agent stdout.z--stderr-filepathz.The filepath to dump runtime env agent stderr.win32   )logging_levellogging_formatlog_dirfilename	max_bytesbackup_count)address
cluster_id)runtime_env_dirlogging_params
gcs_clienttemp_dirr/   runtime_env_agent_portrequestreturnc                   K   |                                   d {V }t          j                    } |                     |           t                              |            d {V }t          j        |                                d          S Nzapplication/octet-stream)bodycontent_type)	readr   GetOrCreateRuntimeEnvRequestParseFromStringagentGetOrCreateRuntimeEnvr   ResponseSerializeToStringr6   datareplys      r   get_or_create_runtime_envrF      s      \\^^######'DFF%%%11'::::::::|((**9S
 
 
 	
r   c                   K   |                                   d {V }t          j                    } |                     |           t                              |            d {V }t          j        |                                d          S r9   )	r<   r   !DeleteRuntimeEnvIfPossibleRequestr>   r?   DeleteRuntimeEnvIfPossibler   rA   rB   rC   s      r   delete_runtime_env_if_possiblerJ      s      \\^^######'IKK%%%66w????????|((**9S
 
 
 	
r   c                   K   |                                   d {V }t          j                    } |                     |           t                              |            d {V }t          j        |                                d          S r9   )	r<   r   GetRuntimeEnvsInfoRequestr>   r?   GetRuntimeEnvsInfor   rA   rB   rC   s      r   get_runtime_envs_inforN      s      \\^^######'ACC%%%..w77777777|((**9S
 
 
 	
r   )middlewaresz/get_or_create_runtime_envz/delete_runtime_env_if_possiblez/get_runtime_envs_info)r'   cygwinc                 ~    t           j                            dt          j         dt          j         d|             d S )Nz1Raylet is dead! Exiting Runtime Env Agent. addr: z, port: 
)r?   _loggerinfoargsnode_ip_addressr5   )msgs    r   parent_dead_callbackrX      sZ    M- 4       r   )hostportloopzSystemExit! )Rargparser    r   r   ray._private.ray_constants_privateray_constantsray._common.utilsr   ray._privater   5ray._private.authentication.http_token_authenticationr   ray._private.process_watcherr   ray._rayletr   ray.core.generatedr   r   aiohttpruntime_env_constsr   runtime_env_agentr   __name__ArgumentParserparseradd_argumentstrintLOGGER_LEVELLOGGER_LEVEL_CHOICESLOGGER_LEVEL_HELPLOGGER_FORMATLOGGER_FORMAT_HELPRUNTIME_ENV_AGENT_LOG_FILENAMEformat
parse_argsrU   platformlogging_rotate_byteslogging_rotation_byteslogging_rotate_backup_countlogging_rotation_backup_countdictr)   r*   r+   logging_filenamer2    redirect_stdout_stderr_if_neededstdout_filepathstderr_filepathgcs_addresscluster_id_hexr3   r1   r4   rV   r5   r?   RequestrA   rF   rJ   rN   Applicationapprouteradd_postr[   check_raylet_taskrX   run_app
SystemExiterS   rT   	exceptionexitcode r   r   <module>r      s     				 



 2 2 2 2 2 2 2 2 2      ' & & & & &      B A A A A A ! ! ! ! ! !     
               - - - - - -z$X$1EFFFF
+	     "R     $S7V     T:R     N     66*2,     +-     "A77=v=8
 8
  	 	 	  :	     ';	     1     N     =     =     D ;>,':Q:QT66WX,/LG,C,C(( " T(*&(2  N 3M2%	   4#3@STTTJO,%$#:  E
 
 
 
 
 

ck 
cl 
 
 
 

S[ 
S\ 
 
 
 
 #/'@'@'I'I&J
K
K
KCJ46OPPPJ)+I   J02GHHH##%%D
|...	 	 	 54L*&:D
 
%,		
 	
 	
 	
 	
 	
    -!--... ('')))M s   N> >PAPP