%`iMddlZddlZddlmZmZmZmZddlZddl m Z ddl m Z ddl mZddlmZddlmZddlmZmZejeZeGd d eZdaejZee d efd ZdS) N)AnyDictListOptional)client_mode_hook)actors)(parse_pg_formatted_resources_to_original)TaskID) RuntimeEnv) Deprecated PublicAPIcteZdZdZdZedddeeeffdZ e eddd Z defd Z e ed dd Z defd ZdefdZe edddZdeefdZdefdZdeefdZdeefdZe edddZdeefdZdeefdZe dZe dZe edddZdeefdZe dZdZd Z e d!Z!e d"Z"e d#Z#ed$ddeee$effd%Z%deee$effd&Z&deee$effd'Z'd(S))RuntimeContextz)A class used for getting runtime context.c|J||_dSN)workerselfrs g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/runtime_context.py__init__zRuntimeContext.__init__s!!! z4Use get_xxx_id() methods to get relevant ids insteadT)messagewarningreturnc|j|j|jd}|jjt jjjkr"|j |j|d<|j |j |d<|S)zxGet a dictionary of the current context. Returns: dict: Dictionary of the current context. )job_idnode_id namespaceNtask_idactor_id) rrrrmoderay_private WORKER_MODErr )rcontexts rgetzRuntimeContext.getsfk|   ; s|2> > >|'%)\ "}(&*m #rzUse get_job_id() insteadcJ|jj}|rJ|S)aGet current job ID for this worker or driver. Job ID is the id of your Ray drivers that create tasks or actors. Returns: If called by a driver, this returns the job ID. If called in a task, return the job ID of the associated driver. )rcurrent_job_idis_nilrrs rrzRuntimeContext.job_id.s'+==??""" rc|tjs Jd|jj}|S)aGet current job ID for this worker or driver. Job ID is the id of your Ray drivers that create tasks or actors. Returns: If called by a driver, this returns the job ID. If called in a task, return the job ID of the associated driver. The job ID will be hex format. Raises: AssertionError: If not called in a driver or worker. Generally, this means that ray.init() was not called. z=Job ID is not available because Ray has not been initialized.)r"is_initializedrr(hexr*s r get_job_idzRuntimeContext.get_job_id>sF    K K J K K +zz||rzUse get_node_id() insteadcJ|jj}|rJ|S)aRGet the ID for the node that this process is running on. This can be called from within a driver, task, or actor. When called from a driver that is connected to a remote Ray cluster using Ray Client, this returns the ID of the head node. Returns: A node id for this worker or driver. )rcurrent_node_idr)rrs rrzRuntimeContext.node_idRs)+->>#####rc|tjs Jd|jj}|S)aGet the ID for the node that this process is running on. This can be called from within a driver, task, or actor. When called from a driver that is connected to a remote Ray cluster using Ray Client, this returns the ID of the head node. Returns: A node id in hex format for this worker or driver. Raises: AssertionError: If not called in a driver or worker. Generally, this means that ray.init() was not called. z>Node ID is not available because Ray has not been initialized.)r"r,rr0r-r1s r get_node_idzRuntimeContext.get_node_idbsF    L L K L L +-{{}}rcxtjs Jd|jjS)zGet current worker ID for this worker or driver process. Returns: A worker id in hex format for this worker or driver process. z@Worker ID is not available because Ray has not been initialized.)r"r,r worker_idr-rs r get_worker_idzRuntimeContext.get_worker_idvsE    N N M N N {$((***rzUse get_task_id() insteadc|jjtjjjksJd|jj|}|s|ndS)aZGet current task ID for this worker. Task ID is the id of a Ray task. This shouldn't be used in a driver process. Example: .. testcode:: import ray @ray.remote class Actor: def ready(self): return True @ray.remote def f(): return True # All the below code generates different task ids. # Task ids are available for actor creation. a = Actor.remote() # Task ids are available for actor tasks. a.ready.remote() # Task ids are available for normal tasks. f.remote() Returns: The current worker's task id. None if there's no task id. ZThis method is only available when the process is a worker. Current mode: N)rr!r"r#r$_get_current_task_idr)rrs rrzRuntimeContext.task_idsoH K  3 ? ? ? ? ;(, (8 ; ; @ ? ?++--%nn..8wwD8rc|jjtjjjkr)t d|jjdS|}|s| ndS)aGet current task ID for this worker. Task ID is the id of a Ray task. The ID will be in hex format. This shouldn't be used in a driver process. Example: .. testcode:: import ray @ray.remote class Actor: def get_task_id(self): return ray.get_runtime_context().get_task_id() @ray.remote def get_task_id(): return ray.get_runtime_context().get_task_id() # All the below code generates different task ids. a = Actor.remote() # Task ids are available for actor tasks. print(ray.get(a.get_task_id.remote())) # Task ids are available for normal tasks. print(ray.get(get_task_id.remote())) .. testoutput:: :options: +MOCK 16310a0f0a45af5c2746a0e6efb235c0962896a201000000 c2668a65bda616c1ffffffffffffffffffffffff01000000 Returns: The current worker's task id in hex. None if there's no task id. JThis method is only available when the process is a worker. Current mode: N) rr!r"r#r$loggerrr:r)r-r;s r get_task_idzRuntimeContext.get_task_idsL ; s|2> > > NN<)-)9<<   4++--$+NN$4$4>w{{}}}$>rc|jjSr)rcurrent_task_idr6s rr:z#RuntimeContext._get_current_task_ids {**rc|jjtjjjkr)t d|jjdS|jjS)aBGet current task name for this worker. Task name by default is the task's funciton call string. It can also be specified in options when triggering a task. Example: .. testcode:: import ray @ray.remote class Actor: def get_task_name(self): return ray.get_runtime_context().get_task_name() @ray.remote class AsyncActor: async def get_task_name(self): return ray.get_runtime_context().get_task_name() @ray.remote def get_task_name(): return ray.get_runtime_context().get_task_name() a = Actor.remote() b = AsyncActor.remote() # Task names are available for actor tasks. print(ray.get(a.get_task_name.remote())) # Task names are avaiable for async actor tasks. print(ray.get(b.get_task_name.remote())) # Task names are available for normal tasks. # Get default task name print(ray.get(get_task_name.remote())) # Get specified task name print(ray.get(get_task_name.options(name="task_name").remote())) .. testoutput:: :options: +MOCK Actor.get_task_name AsyncActor.get_task_name get_task_name task_nams Returns: The current worker's task name r=N)rr!r"r#r$r>rcurrent_task_namer6s r get_task_namezRuntimeContext.get_task_names\d ; s|2> > > NN<)-)9<<   4{,,rc|jjtjjjkr)t d|jjdS|jjS)aGet current task function name string for this worker. Example: .. testcode:: import ray @ray.remote class Actor: def get_task_function_name(self): return ray.get_runtime_context().get_task_function_name() @ray.remote class AsyncActor: async def get_task_function_name(self): return ray.get_runtime_context().get_task_function_name() @ray.remote def get_task_function_name(): return ray.get_runtime_context().get_task_function_name() a = Actor.remote() b = AsyncActor.remote() # Task functions are available for actor tasks. print(ray.get(a.get_task_function_name.remote())) # Task functions are available for async actor tasks. print(ray.get(b.get_task_function_name.remote())) # Task functions are available for normal tasks. print(ray.get(get_task_function_name.remote())) .. testoutput:: :options: +MOCK [python modual name].Actor.get_task_function_name [python modual name].AsyncActor.get_task_function_name [python modual name].get_task_function_name Returns: The current worker's task function call string r=N)rr!r"r#r$r>rcurrent_task_function_namer6s rget_task_function_namez%RuntimeContext.get_task_function_names\V ; s|2> > > NN<)-)9<<   4{55rzUse get_actor_id() insteadc|jjtjjjksJd|jj|jj}|s|ndS)zGet the current actor ID in this worker. ID of the actor of the current process. This shouldn't be used in a driver process. Returns: The current actor id in this worker. None if there's no actor id. r9N)rr!r"r#r$r r)rr s rr zRuntimeContext.actor_idKsg K  3 ? ? ? ? ;(, (8 ; ; @ ? ?;''00:xxd:rc|jjtjjjkr)t d|jjdS|jj}|s| ndS)a6Get the current actor ID in this worker. ID of the actor of the current process. This shouldn't be used in a driver process. The ID will be in hex format. Returns: The current actor id in hex format in this worker. None if there's no actor id. r=N) rr!r"r#r$r>debugr r)r-rIs r get_actor_idzRuntimeContext.get_actor_id^s{ ; s|2> > > LL<)-)9<<   4;'%-__%6%6@x||~~~D@rc|jjtjjjkr)t d|jjdS|jj}|s |jj ndS)aGet the current actor name of this worker. This shouldn't be used in a driver process. The name is in string format. Returns: The current actor name of this worker. If a current worker is an actor, and if actor name doesn't exist, it returns an empty string. If a current worker is not an actor, it returns None. r=N) rr!r"r#r$r>rr r) actor_namerIs rget_actor_namezRuntimeContext.get_actor_namessy ; s|2> > > NN<)-)9<<   4;'-5__->->Ht{%%DHrc|jjS)zvGet the current namespace of this worker. Returns: The current namespace of this worker. )rrr6s rrzRuntimeContext.namespaces{$$rc|jr Jdt|j}|o |ddkS)zCheck whether this actor has been restarted. Returns: Whether this actor has been ever restarted. z0This method should't be called inside Ray tasks.)r NumRestartsr)r r)rr-)r actor_infos rwas_current_actor_reconstructedz.RuntimeContext.was_current_actor_reconstructedsb $$&& > > = > > &T]%6%6%8%8999 zCRuntimeContext.get_assigned_resources...s&//VQS//////r)sum)raresmappings r z9RuntimeContext.get_assigned_resources..sE   W //w/////   r) rr!r"r#r$check_connected core_worker resource_idsitemsr )rresource_id_map resource_mapresults rget_assigned_resourcesz%RuntimeContext.get_assigned_resourcess K  3 ? ? ? ? ;(, (8 ; ; @ ? ? ##%%%+1>>@@  / 5 5 7 7   :,GG rc|jjS)zGet the runtime env string used for the current driver or worker. Returns: The runtime env string currently using by this worker. )r runtime_envr6s rget_runtime_env_stringz%RuntimeContext.get_runtime_env_strings {&&rcNtj|S)zGet the runtime env used for the current driver or worker. Returns: The runtime env currently using by this worker. The type of return value is ray.runtime_env.RuntimeEnv. )r deserializersr6s rrrzRuntimeContext.runtime_envs!%d&A&A&C&CDDDrc|j}||j}|rt d|j|S)zvGet the current actor handle of this actor itself. Returns: The handle of current actor. z*This method is only available in an actor.)rrir r) RuntimeErrorrjget_actor_handle)rrr s r current_actorzRuntimeContext.current_actors]   ? ??   MKLL L!228<<z6RuntimeContext.get_accelerator_ids..s2U2U2Ur3r772U2U2Ur)rrir"r# accelerators"get_all_accelerator_resource_names,get_accelerator_ids_for_accelerator_resource)rrids_dictaccelerator_resource_nameaccelerator_idss rrz"RuntimeContext.get_accelerator_idss   )+\ & I I K K V V %$QQ)C-CCCO3V2U_2U2U2UH. / /rcF|j}||jS)z Get the node labels of the current worker. Returns: A dictionary of label key-value pairs. )rricurrent_node_labelsrs rget_node_labelszRuntimeContext.get_node_labelss&   ))rN)(__name__ __module__ __qualname____doc__rr rrrr&propertyrr.rr3r7rrr?r r:rDrGr rLrOrrTrWrZr\rprsrrryr}rrrrr`rrrrs33ZFPTT#s(^&Z2DAAA  BAX C(Z3TBBB  CBX S( +s + + + +Z3TBBB'9'9CBX'9R-?Xc]-?-?-?-?^+f++++8-x}8-8-8-8-t16 16161616fZ4dCCC;;DCX;"AhsmAAAA*I IIII,%%X% = =X =Z>MMM..NMX.; ;;;; I IX I2'''EEXE = =X =..X.Z;TJJJ*$sDI~"6***KJ*T#tCy.%9, *c49n!5 * * * * * *rrrct5t#ttjjjatcdddS#1swxYwYdS)aGet the runtime context of the current driver/worker. The obtained runtime context can be used to get the metadata of the current driver, task, or actor. Example: .. testcode:: import ray # Get the job id. ray.get_runtime_context().get_job_id() # Get the actor id. ray.get_runtime_context().get_actor_id() # Get the task id. ray.get_runtime_context().get_task_id() N)_runtime_context_lock_runtime_contextrr"r#r global_workerr`rrget_runtime_contextr.s*    #-cl.A.OPP                    s1AA  A )logging threadingtypingrrrrray._private.workerr"ray._private.client_mode_hookrray._private.staterray._private.utilsr ray._rayletr ray.runtime_envr ray.util.annotationsr r getLoggerrr>objectrrLockrrr`rrrsa,,,,,,,,,,,,::::::%%%%%%GGGGGG&&&&&&66666666  8 $ $ V*V*V*V*V*VV*V* V*r& ((  ^       r