&`i,ddlZddlZddlmZddlmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZddlmZe dZe dZejeZeGd d ZeGd d ZGd dZede de ddGddZdS)N) dataclass)DictListOptionalTupleTypeTypeVar)get_ray_doc_version) ActorHandle) DeprecatedT ActorMetadatac(eZdZUdZeed<eed<dS) ActorWrapperz+Class containing an actor and its metadata.actormetadataN)__name__ __module__ __qualname____doc__r __annotations__rh/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/actor_group.pyrrs055 rrc^eZdZUeed<eed<eeeefed<eed<eed<dS) ActorConfignum_cpusnum_gpus resources init_args init_kwargsN) rrrfloatrrrstrrrrrrrsVOOOOOOS%Z())))rrc*eZdZdddefdZdZdZdS)ActorGroupMethod actor_group ActorGroup method_namecFtj||_||_dSN)weakrefrefr& _method_name)selfr&r(s r__init__zActorGroupMethod.__init__#s"";{33'rcBtd|jd|jd)NzIActorGroup methods cannot be called directly. Instead of running 'object.z()', try 'object.z .remote()'.) TypeErrorr-r.argskwargss r__call__zActorGroupMethod.__call__'s@ 6"&"3 6 6( 6 6 6   rcVfdjDS)NcZg|]'}t|jjji(Sr)getattrrr-remote).0ar3r4r.s r z+ActorGroupMethod.remote..0sJ    7GAGT. / / 6 G G G   r)r&actorsr2s```rr9zActorGroupMethod.remote/sI      %%''.    rN)rrrr#r/r5r9rrrr%r%"sT(L(s((((        rr%zcFor stateless/task processing, use ray.util.multiprocessing, see details in https://docs.ray.io/en/z/ray-more-libs/multiprocessing.html. For stateful/actor processing such as batch prediction, use Datasets.map_batches(compute=ActorPoolStrategy, ...), see details in https://docs.ray.io/en/z4/data/api/dataset.html#ray.data.Dataset.map_batches.T)messagewarningceZdZdZ ddedededed eee efd ee d eefd Z d Z dZ dZdZddefdZdeefdZdefdZedZdS)r'aGroup of Ray Actors that can execute arbitrary functions. ``ActorGroup`` launches Ray actors according to the given specification. It can then execute arbitrary Python functions in each of these actors. If not enough resources are available to launch the actors, the Ray cluster will automatically scale up if autoscaling is enabled. Args: actor_cls: The class to use as the remote actors. num_actors: The number of the provided Ray actors to launch. Defaults to 1. num_cpus_per_actor: The number of CPUs to reserve for each actor. Fractional values are allowed. Defaults to 1. num_gpus_per_actor: The number of GPUs to reserve for each actor. Fractional values are allowed. Defaults to 0. resources_per_actor (Optional[Dict[str, float]]): Dictionary specifying the resources that will be requested for each actor in addition to ``num_cpus_per_actor`` and ``num_gpus_per_actor``. init_args, init_kwargs: If ``actor_cls`` is provided, these args will be used for the actor initialization. rN actor_cls num_actorsnum_cpus_per_actornum_gpus_per_actorresources_per_actorr r!cddlm}|d|dkrtd|d|dks|dkrtd|d|dg|_||_t ||||pd |pi |_tj|jj |jj |jj ||_ | dS) Nr)record_library_usagezutil.ActorGroupzFThe provided `num_actors` must be greater than 0. Received num_actors=z instead.zXThe number of CPUs and GPUs per actor must not be negative. Received num_cpus_per_actor=z and num_gpus_per_actor=.r)rrrr r!)rrr)ray._common.usage.usage_librH ValueErrorr=rCr actor_configrayr9rrr _remote_clsstart) r.rBrCrDrErFr r!rHs rr/zActorGroup.__init__YsD EDDDDD./// ??/9   ! !%7!%;%;<&8<<'9<<<  $''')o2#)r     3:&/&/'1      rcpt|jdkrtdt||S)Nrz9This ActorGroup has been shutdown. Please start it again.)lenr= RuntimeErrorr%r.items r __getattr__zActorGroup.__getattr__s> t{  q K  d+++rc*t|jSr*)rQr=r.s r__len__zActorGroup.__len__s4;rc|j|Sr*r=rSs r __getitem__zActorGroup.__getitem__s{4  rc:|jr't|jdkrtdtd|jd||jtt|jddS)z*Starts all the actors in this actor group.rz_The actors have already been started. Please call `shutdown` first if you want to restart them.z Starting  actors.z" actors have successfully started.N)r=rQrRloggerdebugrC add_actorsrWs rrOzActorGroup.starts ; 3t{++a//    ::::;;; ((( DK((LLLMMMMMr patience_sctdt|jd|dkr$|jD]}t j|jnid|jD}t j||\}}|r=td|jD]}t j|jtdg|_dS) aqShutdown all the actors in this actor group. Args: patience_s: Attempt a graceful shutdown of the actors for this many seconds. Fallback to force kill if graceful shutdown is not complete after this time. If this is less than or equal to 0, immediately force kill all actors. zShutting down r]rcJg|] }|jj!Sr)r__ray_terminate__r9)r:ws rr<z'ActorGroup.shutdown..s)QQQ299;;QQQr)timeoutz8Graceful termination failed. Falling back to force kill.zShutdown successful.N)r^r_rQr=rMkillrwait)r.rbr done_refsdonenot_dones rshutdownzActorGroup.shutdowns  @c$+&6&6@@@AAA ?? & &%%%% &RQT[QQQI XiDDDND( * WXXX![**EHU[)))) +,,, r actor_indexescg}tt|jD]&}||vr ||j|'||_dS)zRemoves the actors with the specified indexes. Args: actor_indexes (List[int]): The indexes of the actors to remove. N)rangerQr=append)r.rn new_actorsis r remove_actorszActorGroup.remove_actorss[  s4;''(( 2 2A %%!!$+a.111  rc<g}g}t|D]y}|jj|jji|jj}||t|dr,||jztj |}t|dkrdgt|z}tt|D]7}|j t||||8dS)ztAdds ``num_actors`` to this ActorGroup. Args: num_actors: The number of actors to add. get_actor_metadatarN)rr)rprNr9rLr r!rqhasattrrvrMgetrQr=r)r.rCrrnew_actor_metadata_rrrss rr`zActorGroup.add_actorss3  z"" M MA+D$+",040A0ME   e $ $ $u233 M"))%*B*I*I*K*KLLL7-.. x==A  vJ/Hs:'' X XA K  |*Q-(ST+VVV W W W W X Xrc$d|jDS)Ncg|] }|j Sr)r)r:r;s rr<z-ActorGroup.actor_metadata..s000q 000rrZrWs ractor_metadatazActorGroup.actor_metadatas00DK0000r)rArArNNN)ra)rrrrrintr"rrr#rr/rUrXr[rOrmrrtr`propertyr}rrrr'r'6s_:$%$%:>%)&*...." . " . &d3:&67 .E?.d^....`,,,   !!! N N N56 !49 ! ! ! !XSXXXX211X111rr')loggingr+ dataclassesrtypingrrrrrr rMray._private.utilsr ray.actorr ray.util.annotationsr r r getLoggerrr^rrr%r'rrrrs!!!!!!================ 222222!!!!!!++++++ GCLL((  8 $ $            (  j!4!4!6!6 j j2133  j j j  h1h1h1h1h1h1h1h1h1h1r