
    &`i                     V   d dl Z d dlZd dlZd dlmc mZ ej         G d d                      Ze	dk    r e j
                    Ze                    ddedd	           e                    d
dedd	           e                                \  ZZ ej                     e                    ej        ej                  Zej                                        Z e ej        eg                      ej        d          d          Z ee d ej        e                      dS dS )    Nc                   (    e Zd ZddZd Zd Zd ZdS )		StepActor      c                     || _         d| _        d| _        || _        t          j        j        j        }|j        	                                }t          j        d| | j        d           d S )NFr   JOB:T	overwrite)
interval_sstoppedcurrent_steptotal_stepsray_privateworkerglobal_workercore_workerget_actor_idray_kv_internal_kv_put)selfr   r   r   	worker_ids        /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/experimental/job/example_job/demo_script.py__init__zStepActor.__init__   sl    $&$2&3355	 2y 2 2D4EQUVVVVVV    c                    t           j        j        j        }|j                                        }| j        | j        k    r| j        sht          d| j
         d| j                    t          j        | j
                   | xj        dz  c_        t          j        d| | j        d           nt          d           n| j        | j        k    d| _        t          j        d| dd           dS )	Nz	Sleeping z secs to executing step r   r   Tr	   z"Stop called or reached final step.DONE)r   r   r   r   r   r   r   r   r   printr   timesleepr   r   )r   r   r   s      r   runzStepActor.run   s"   $2&3355	4#333< 0 0 0 -0 0   
4?+++!!Q&!!'&9&&(9T     :;;; 4#333  2y 2 2FdKKKKvr   c                     | j         S )N)r   r   s    r   get_stepzStepActor.get_step-   s      r   c                     d| _         d S )NT)r   r#   s    r   stopzStepActor.stop0   s    r   N)r   r   )__name__
__module____qualname__r   r!   r$   r&    r   r   r   r   
   sY        W W W W  .! ! !    r   r   __main__z--interval-sFr   z Address to use to connect to Ray)requiredtypedefaulthelpz--total-stepsr   z Password for connecting to Redis)r   r   r   z, )argparser   r   ray.experimental.internal_kvexperimentalinternal_kvr   remoter   r'   ArgumentParserparseradd_argumentintparse_known_argsargs_initr   r   
step_actorr!   refr   get_internal_kv_listjob_key_internal_kv_getr*   r   r   <module>rC      s     



 - - - - - - - - - & & & & & & & &R z$X$&&F
/     /     %%''GD!CHJJJ!!?0@ "  J .


!
!C	E'#'3%..&f&v..q1G	EW
:
://88
:
:;;;;;3 r   