%`i2pddlZddlZddlmZddlmZddlmZdZej dddka tZ ej ddid Zejad t _d Zd efd ZdZdZdZedZedZdefdZdZdZdZdZdS)N)contextmanagerwraps) auto_init_ray__ray_client_mode_key__RAY_CLIENT_MODE01FTc\ttds dt_tjS)zGet's the value of `_client_hook_status_on_thread`. Since `_client_hook_status_on_thread` is a thread-local variable, we may need to add and set the 'status' attribute. statusT)hasattr_client_hook_status_on_threadr q/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/client_mode_hook.py!_get_client_hook_status_on_threadrs) 0( ; ;4/3%, ( //rvalc|t_dSN)rr rs r_set_client_hook_statusr's+.!(((rc:t}dt_|SNF)rrr )outs r_disable_client_hookr,s + - -C+0!( Jrc dadS)zkForce client mode to be enabled. NOTE: This should not be used in tests, use `enable_client_mode`. TNis_client_mode_enabledrrr_explicitly_enable_client_moder3s "rc dadSrrrrr_explicitly_disable_client_moder!;s"rc#vKt} dVt|dS#t|wxYwr)rrrs rdisable_client_hookr#@sH  C% $$$$$$$$$s'8c#rKt dVtdS#twxYwr)rr!rrrenable_client_moder%IsE"$$$* ')))))'))))s&6funccLddlmtfd}|S)zDecorator for whether to use the 'regular' ray version of a function, or the Ray Client version of that function. Args: func: This function. This is set when this function is used as a decorator. rrayctr-jdkstrtj|i|S|i|S)Ninit)client_mode_should_convert__name__!is_client_mode_enabled_by_defaultgetattr)argskwargsr&r)s rwrapperz!client_mode_hook..wrapper]s_ & ' ' D}&&*K&2wsDM22DCFCCCtT$V$$$r)ray.util.clientr)r)r&r2r)s` @rclient_mode_hookr4RsM$##### 4[[ % % % % %[ % Nrc:tpto tS)z;Determines if functions should be converted to client mode.)rr.rrrrr,r,ms C"C . + - -.rc<tfd}|S)aWraps a function called during client mode for execution as a remote task. Can be used to implement public features of ray client which do not belong in the main ray API (`ray.*`), yet require server-side execution. An example is the creation of placement groups: `ray.util.placement_group.placement_group()`. When called on the client side, this function is wrapped in a task to facilitate interaction with the GCS. cddlm}ttrA|d}|j|i|}||S|i|S)Nrr()num_cpus)r3r)rr,remoteget)r0r1r)frefr&s rr2z!client_mode_wrap..wrappers'''''' & ' ' & A &&t,,A!(D+F++C773<< tT$V$$$rr)r&r2s` rclient_mode_wrapr=ws5 4[[ % % % %[ % Nrc ddlm}t|td}|||s+||}t |t|||}|j||fi|S)aXRuns a preregistered ray RemoteFunction through the ray client. The common case for this is to transparently convert that RemoteFunction to a ClientRemoteFunction. This happens in circumstances where the RemoteFunction is declared early, in a library and only then is Ray used in client mode -- necessitating a conversion. rr(N) r3r)r/RAY_CLIENT_MODE_ATTR_converted_key_exists_convert_functionsetattr_get_converted_remote)func_clsin_args in_kwargsr1r)key client_funcs rclient_mode_convert_functionrJs$##### (0$ 7 7C S66s;; ##H--.444$$S))K ; w < r[s %%%%%%5555551(93??3F%;! $c*+++!0  1 1'+$000///// """### %%%***86...8===*>>>>>r