&`it ddlZddlZddlZddlZddlmZmZmZmZm Z ddl Z ddl m cm Z ddlmZddlmZddlmZejeZdZdZdedefd Zdedefd Zd e eeefefd ed dde eeefeffdZd e eeefefdede eeefeffdZd e eeefefd dde eeefeffdZ dedeefdZ!dedeefdZ"dedeefdZ#dS)N)AnyCallableDictOptionalUnion) load_class) RuntimeEnvzray_runtime_env_func::cpttjtjdS)N60)intosenvironget ray_constants2RAY_WORKER_PROCESS_SETUP_HOOK_LOAD_TIMEOUT_ENV_VARw/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/runtime_env/setup_hook.pyget_import_export_timeoutrs/    Ld    rkeyreturnc^ttj|zSN)RUNTIME_ENV_FUNC_IDENTIFIERbase64 b64encodedecoders r_decode_function_keyrs$ &)9#)>)>)E)E)G)G GGrc|tsJtj|t tdSr) startswithrr b64decodelenrs r_encode_function_keyr$s@ >>5 6 666 6  C$? @ @ B BC D DDr runtime_env setup_funcworkerz ray.Workerct|tsJ |j|t }n2#t $r%}t jd|d}~wwxYw| di}tj |vsJdtj dt||tj <||d<|j |d<|S)Ntimeoutz$Failed to export the setup function.env_vars The env var, ?, is not permitted because it is reserved for the internal use.worker_process_setup_hook) isinstancerfunction_actor_managerexport_setup_funcr Exceptionray exceptionsRuntimeEnvSetupErrorrr!WORKER_PROCESS_SETUP_HOOK_ENV_VARr__name__)r%r&r'rer+s rexport_setup_func_callabler9$s j( + +++ ++==  9 ; ;>   n11 2   z2..H  :( J J J H G H H H K J JAU AAH] <='K  0:/BK+, s(A A1 A,,A1setup_func_modulect|tsJ|di}tj|vsJdtjd||tj<||d<|S)Nr+r,r-)r/strrrr6)r%r:r+s rexport_setup_func_moduler=Cs ' - --- -z2..H  :( J J J H G H H H K J JARH] <=&K  rc |d}||St|trt|||St|trt ||St dt|d)aUploads the worker_process_setup_hook to GCS with a key. runtime_env["worker_process_setup_hook"] is converted to a decoded key that can load the worker setup hook function from GCS. i.e., you can use internalKV.Get(runtime_env["worker_process_setup_hook]) to access the worker setup hook from GCS. Args: runtime_env: The runtime_env. The value will be modified when returned. worker: ray.worker instance. decoder: GCS requires the function key to be bytes. However, we cannot json serialize (which is required to serialize runtime env) the bytes. So the key should be decoded to a string. The given decoder is used to decode the function key. r.Nz2worker_process_setup_hook must be a function, got .)rr/rr9r<r= TypeErrortype)r%r'r&s r*upload_worker_process_setup_hook_if_neededrBRs*!<==J*h'' )+z6JJJ J $ $ ' Z@@@ WDDTDT W W W   rworker_process_setup_hook_keycz|J|tst|St|S)zLoad the setup hook from a given key and execute. Args: worker_process_setup_hook_key: The key to import the setup hook from GCS. Returns: An error message if it fails. None if it succeeds. )r!r"load_and_execute_setup_hook_module load_and_execute_setup_hook_func)rCs rload_and_execute_setup_hookrGvsF ) 4 4 4 ( 3 34O P PO12OPPP/0MNNNrc t|}|dS#t$r!d|d|dtj}|cYSwxYw)Nz)Failed to execute the setup hook method, z; from ``ray.init(runtime_env={'worker_process_setup_hook': z})``. Please make sure the given module exists and is available from ray workers. For more details, see the error trace below. )rr2 traceback format_exc)rCr& error_messages rrErEs =>>  t     (, ( (/L ( ( #%%  ( (  s(AAc"tjjj}|jsJ|j} |t|t}n7#t$r*dtdtj }|cYSwxYw tj |j}n(#t$rdtj }|cYSwxYw |n0#t$r#d|jdtj }|cYSwxYwdS)Nr)z#Failed to import setup hook within z seconds. z-Failed to deserialize the setup hook method. z7Failed to execute the setup hook method. Function name: )r3_privater' global_worker connectedr0fetch_registered_methodr$rr2rIrJpickleloadsfunction function_name)rCr' func_managerworker_setup_func_inforKr&s rrFrFs\ .F   0L !-!E!E !> ? ?-//"F" "   ((** ( (#%% ( (   \"8"ABB  X @T@V@V X X     (%3 ( (#%% ( (    4s50A1B  B B++"CC C*D  D )$rloggingr rItypingrrrrrr3ray._private.ray_constantsrNrray.cloudpickle cloudpicklerRray._common.utilsrray.runtime_envr getLoggerr7loggerrrbytesr<rr$r9r=rBrGrErFrrrrbs  77777777777777 222222222 ((((((&&&&&&  8 $ $6HeHHHHH EcEeEEEE tCH~z12  4S>: %& > tCH~z12   4S>: %&    ! tCH~z12! !  4S>: %&! ! ! ! HO#&O c]OOOO$#& c](%#&% c]%%%%%%r