&`i4ddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl Z ddl m Z ddl mZmZmZmZmZmZmZddlZddlZddlZddlmcmcmZddlmcmcmZddlmZddlm Z m!Z!ddl"m#Z#ddl$m%Z%ddl&m'Z'dd l(m)Z)dd l*m+Z+dd l,m-Z-dd l.m/Z/dd l0m1Z1ddl2m3Z3m4Z4m5Z5m6Z6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?m@Z@ddlAmBZBejCeDZEe'ddZFdZGGddejHZIdZJdeKfdZLdejMdeeeKeffdZNd)deKdeOfd ZPdeKdeOfd!ZQd*d#ZRd)d$ZSd%eeKdee/fd&ZTd'ZUeDd(kr eUdSdS)+N) defaultdict)AnyCallableDictListOptionalSetUnion) cloudpickle) build_address is_localhost) ray_constants)disable_client_hook) env_integer) setup_logger)%canonicalize_bootstrap_address_or_die)add_port_to_grpc_server) GcsClient JobConfig)CLIENT_SERVER_MAX_THREADS GRPC_OPTIONSOBJECT_TRANSFER_CHUNK_SIZEClientServerHandle ResponseCache) DataServicer)LogstreamServicer) serve_proxier)dumps_from_serverloads_from_client)current_serverTIMEOUT_FOR_SPECIFIC_SERVER_ScFtjfd}|S)z Decorator for gRPC stubs. Before calling the real stubs, checks if there's an existing entry in the caches. If there is, then return the cached entry. Otherwise, call the real function and use the real cache c t| d}t fd|Dr  |||S d} d}t d}|j|}|||}|_t |trH|tj j | t|||S  |||} np#t$rc} |||| |tj j | t| d} ~ wwxYw|||| | S)N) client_id thread_idreq_idc3 K|]}|vV dSN).0imetadatas q/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/client/server/server.py z7_use_response_cache..wrapper..<s(77Qq 777777r&r'r()dictinvocation_metadataanyintresponse_caches check_cache isinstance Exceptionset_codegrpc StatusCodeFAILED_PRECONDITION set_detailsstr update_cache) selfrequestcontext expected_idsr&r'r(response_cache cached_entryresper.funcs @r/wrapperz$_use_response_cache..wrapper8s335566; 7777,777 7 7 04gw// /[) [) Xh'((-i8%11)VDD  #, 22 #  !DEEE##C $5$5666""  4gw//DD     ' ' 61 = = =   T_@ A A A   A ' ' '    ##Ivt<<< s8 D E3AE..E3) functoolswraps)rIrJs` r/_use_response_cacherM1s;_T"""""H Nr1c eZdZdefdZ d.dejdejfdZe d.dej fdZ d.dej fdZ e d.dejfd Zd.dejfd Zd.dejfd Z d.dejfd Zd.dejfd Zd.defdZdededefdZdZdZdZdZ e d.dZ! d.dej"dede#de$j%de&ej'f dZ(dej"fdZ)dej"defdZ* d.dej+dej,fdZ- d.de.ee/fdededefd Z0d.dej1fd!Z2 d.d"ej3d#e4e5d$e6ee5fdej7fd%Z8 d.d"ej3d#e4e5d$e6ee5fdej7fd&Z9 d.d"ej3d#e4e5d$e6ee5fdej7fd'Z: d.d"ej3d#e4e5d$e6ee5fdej7fd(Z; d.d"ej3dej7fd)Zj?fd+Z@deded*e&e6fd,ZAd-ZBdS)/RayletServicerray_connect_handlerc\tt|_tt|_i|_i|_tt |_i|_t |_ tj |_ ||_ tt|_dS)zuConstruct a raylet service Args: ray_connect_handler: Function to connect to ray cluster N)rr2 object_refsclient_side_ref_map function_refs actor_refsset actor_ownersregistered_actor_classes named_actors threadingLock state_lockrPrr6)rArPs r/__init__zRayletServicer.__init__asCNdBSBSBMdBSBS 8:3>s3C3C(*%EE#.**#6 9D]9S9Sr1NrBreturncL|jr!tj|j}d|_nd}d}t 5t jr0t jjj }|j }ntj|j pd} |j|fi|nV#t$rI}t dt%jdd|cYd}~cdddSd}~wwxYwdddn #1swxYwY|t%jdS|}|rt+|jjj}||jjjt+|jjj} | |jjj|| krCt7|dkr0t%jdd |jjd |jjSt%jdS) NTz{}zRunning Ray Init failed:Fz0Call to `ray.init()` on the server failed with: )okmsgr`rz.Runtime environment doesn't match request one z current one ) job_configpickleloads _client_jobrrayis_initialized_privateworker global_worker core_workerget_job_configjsonray_init_kwargsrPr9logger exceptionray_client_pb2 InitResponse_get_proto_job_configrVruntime_env_infourisworking_dir_uriupdatepy_modules_urislen) rArBrCrccurrent_job_configrj extra_kwargsrHjob_uriscurrent_job_uriss r/InitzRayletServicer.Initts   g&899J%)J " "J! " "  !## ,:%+%7%F%F%H%H""#z'*A*ITJJ ,D,ZHH<HHHH $$%?@@@)6 RqRR                         !.$777 7 5577   :6;KLLH OOJ7<L M M M""38H     # #"38H   +++H 0A0A%2N#-#>#CNN#5#F#KNN *d3333sBAD B*)D * C=42C8&C='D 8C==D  DDcF t5tjj|j|j|j|j}dddn #1swxYwYn)#t$r}t||d}Yd}~nd}~wwxYwtj |S)N) overwrite namespaceF)already_exists) rrg experimental internal_kv_internal_kv_putkeyvaluerrr9return_exception_in_contextrr KVPutResponse)rArBrCrrHs r/KVPutzRayletServicer.KVPuts #$&&  !$!1!=!N!NKM%/%/ "O""                # # # '7 3 3 3"NNNNNN #+>JJJJs:A%=A A%AA% A!A%% B /BB c. t5tjj|j|j}dddn #1swxYwYn)#t$r}t||d}Yd}~nd}~wwxYwtj |S)Nrr1)r) rrgrr_internal_kv_getrrr9rrr KVGetResponse)rArBrCrrHs r/KVGetzRayletServicer.KVGets $&&  (4EEK7+<F                   '7 3 3 3EEEEEE +%8888:A1A  A AAAA A?#A::A?c: t5tjj|j|j|j}dddn #1swxYwYn)#t$r}t||d}Yd}~nd}~wwxYwtj |S)N) del_by_prefixrr) deleted_num) rrgrr_internal_kv_delrrrr9rrr KVDelResponse)rArBrCrrHs r/KVDelzRayletServicer.KVDels $&&  !.:KKK")"7%/L                     '7 3 3 3KKKKKK + DDDDs:A7A AAAAA B)BBc. t5tjj|j|j}dddn #1swxYwYn)#t$r}t||g}Yd}~nd}~wwxYwtj |S)Nr)keys) rrgrr_internal_kv_listprefixrr9rrrKVListResponse)rArBrCrrHs r/KVListzRayletServicer.KVLists $&&  '3EENg.?F                   '7 3 3 3DDDDDD ,$7777rc. t5tjj|j|j}dddn #1swxYwYn)#t$r}t||d}Yd}~nd}~wwxYwtj |S)NrF)exists) rrgrr_internal_kv_existsrrr9rrrKVExistsResponse)rArBrCrrHs r/KVExistszRayletServicer.KVExistss $&&  )5IIK7+<J                   '7 3 3 3FFFFFF .f====rct5tj|j}dddn #1swxYwYt jtj|S)N)all_namespaces) actors_json) rrgutillist_named_actorsrrrClientListNamedActorsResponserndumps)rArBrCactorss r/ListNamedActorszRayletServicer.ListNamedActorss! " " W WX//w?U/VVF W W W W W W W W W W W W W W W; 6**    s&AAActj}|j|_|jtjjkrt 5t j}dddn #1swxYwYd|D}|j tj |n,|jtjj krt 5t j }dddn #1swxYwYd|D}|j tj |n|jtjjkr%tj}t 5t j}t jj||_t jj||_|j|_|j|_|j|_||_dddn #1swxYwY|j |nAt 5||||_ dddn #1swxYwY|S)Nc4i|]\}}|t|Sr+floatr,kvs r/ z.RayletServicer.ClusterInfo..$IIItq!q%((IIIr1)tablec4i|]\}}|t|Sr+rrs r/rz.RayletServicer.ClusterInfo..rr1)!rrClusterInfoResponsetypeClusterInfoTypeCLUSTER_RESOURCESrrgcluster_resourcesitemsresource_tableCopyFrom ResourceTableAVAILABLE_RESOURCESavailable_resourcesRUNTIME_CONTEXTRuntimeContextget_runtime_context_commonutils hex_to_binary get_job_idjob_id get_node_idnode_idr-should_capture_child_tasks_in_placement_groupcapture_client_tasks gcs_addressget_runtime_env_string runtime_envruntime_context_return_debug_cluster_inforn)rArBrCrG resourcesfloat_resourcesctxrtcs r/ ClusterInfozRayletServicer.ClusterInfos133L <>9K K K$&& 4 4133  4 4 4 4 4 4 4 4 4 4 4 4 4 4 4JIy7H7HIIIO   ( (2@@@WW    \^;O O O$&& 6 6355  6 6 6 6 6 6 6 6 6 6 6 6 6 6 6JIy7H7HIIIO   ( (2@@@WW    \^;K K K 4CCEEC$&& ? ?-// [.<O>OPP #  E(#&/"%"<"<">"> ? ? ? ? ? ? ? ? ? ? ? ? ? ? ?  ) )# . . . .$&& N N ;;GWMM  N N N N N N N N N N N N N N N sIA((A,/A,3DDD=CJJJK,,K03K0cd}|jtjjkrt j}n|jtjjkrt j}n|jtjjkrt j }nl|jtjj kri}nO|jtjj kr&dtj j i}ntdt!j|S)z9Handle ClusterInfo requests that only return a json blob.N dashboard_urlzUnsupported cluster info type)rrrrNODESrgnodesIS_INITIALIZEDrhTIMELINEtimelinePING DASHBOARD_URLrirjget_dashboard_url TypeErrorrnr)rArBrCdatas r/rz)RayletServicer._return_debug_cluster_infos <>9? ? ?9;;DD \^;J J J%''DD \^;D D D<>>DD \^;@ @ @DD \^;I I I#S\%8%J%J%L%LMDD;<< <z$r1r&idc|j5||jvr]||j|vrNtd|d||j||= ddddS||jvr||j|vrtd|d||j||||r7td||j|= ddddS ddddS#1swxYwYdS)NzReleasing object z for TzReleasing actor Deleting reference to actor F) r\rRrpdebughexrWremove_can_remove_actor_refrU)rAr&rs r/releasezRayletServicer.release(s _  D,,,))444LL!ORVVXX!O!OI!O!OPPP(3B7         D---*9555LL!NBFFHH!N!N9!N!NOOO%i077;;;11"550 %NBFFHH%N%NOOO OB/         !                  sAE/B7E3EEEc|j5||||dddn #1swxYwYtjdSr*)r\_release_objects_release_actorsgccollect)rAr&s r/ release_allzRayletServicer.release_all;s _ , ,  ! !) , , ,   + + + , , , , , , , , , , , , , , , s+?AActfd|jD }|o|jvS)Nc3 K|]}|vV dSr*r+)r, actor_listactor_id_bytess r/r0z7RayletServicer._can_remove_actor_ref..Fs9  -7Nj (      r1)r4rWvaluesrY)rArno_owners ` r/rz$RayletServicer._can_remove_actor_refEsa    ;?;L;S;S;U;U      CN$2CCCr1c||jvrtd|dSt|j|}|j|=||jvr|j|=||jvr|j|=td|d|dS)Nz%Releasing client with no references: Released all z objects for client )rRrprrzrSr6)rAr&counts r/rzRayletServicer._release_objectsKs D, , , LLLLL M M M FD$Y/00  Y ' 0 0 0(3 , , ,$Y/ KUKK KKLLLLLr1cz||jvrtd|dSd}|j|}|D]S}|dz }||r7td||j|=Ttd|d|dS)Nz!Releasing client with no actors: rrrz actors for client: )rWrprpoprrrU)rAr&ractors_to_removeid_bytess r/rzRayletServicer._release_actorsWs D- - - LLHYHH I I I F,00;;( . .H QJE))(33 . LHLLNNLLMMMOH- KUKK KKLLLLLr1c|ddkr |j|j|jj}t 5t j||jj|jj dddn #1swxYwYn#t$r}t||Yd}~nd}~wwxYw|ddkr |j |j j}t 5t j||j jdddn #1swxYwYn6#t$r}t||Yd}~nd}~wwxYwt!dt#jdS) Nterminate_type task_object)force recursiveactor) no_restartzEClient requested termination without providing a valid terminate_typeTrb) WhichOneofrRr&rrrrgcancelrrr9rrUrkillr RuntimeErrorrrTerminateResponse)rAreqrC object_refrH actor_refs r/ TerminatezRayletServicer.Terminatefs+ >>* + +} < < 8!-cm.send_get_responses/%6vy$%O%O %(__ )A~~~/W~~~'+y&)CC(( ).g.Dl(S(S99H$,/I$IE"% *X\=W,W##C(6'A&*%/c %:)1-9+5 (((H*8)D$,V***J),,Z888899 %///#1#="'{/@/E/E$$$ .:xPVWWW$((......... /sCC E$AD==E)rzids ValueErrorrRrrrr"r rrprrr _on_completedr9) rArBr&r(rrCridrefr+rHs ````` r/_async_get_objectz RayletServicer._async_get_objects w{  q OgOO k!ny)--c488 !-!'.3.."+.. * W LL3. / / /$&&% %  /c /d / / / / / / / / / /D!!"3444K% % % % % % % % % % % % % % % % % % N W W W!-EARSTAUAUVVV V V V V V V WsB+D *C>1 D >DD DD E'E<EEc#$Kt|}|d}|9tjdt jtdVdS|||Ed{VdS)Nr&Fz.client_id is not specified in request metadatar) r2r3rrrr"r rr- _get_object)rArBrCr.r&s r/ GetObjectzRayletServicer.GetObjects335566LL--   ,!'OPP     '';; ; ; ; ; ; ; ; ; ;r1c #Kg}|jD]z}|j||d}|r||;t jdt jtd|d|VdS t d|zt5tj||j }dddn #1swxYwYnB#t$r5}t jdt j|VYd}~dSd}~wwxYwt|||}t!|} | dks Jdt#j| t&z } t)|j| D]J} | t&z} t-| | d zt&z} t jd || | | | | VKdS) NFzClientObjectRef z is not found for client rzget: %s)timeoutrrrTr)r,rRrappendrrr"r rr-rprrrgr6r9rrzrrrrr r!)rArBr& objectrefsr/r0rrHr%rrrr&r's r/r3zRayletServicer._get_objects ;  C"9-11#t<> __ A~~~G~~~y.H!HII g4lCC  H99Ej8a<3M"MNNC ,c *!)%       s<+C'3C C'CC'"C#C'' D&1*D!!D&cR||j|jd|j|S)z#gRPC entrypoint for unary PutObject) _put_objectr client_ref_idowner_id)rArBrCs r/ PutObjectzRayletServicer.PutObjects/ L'/W5Ew   r1rr<r=c t||}|r|j|}nd}t5tj||}dddn #1swxYwYnZ#t $rM} t dtj ddtj | cYd} ~ Sd} ~ wwxYw||j || <t|dkr"| |j||<t d|ztj | d S) aPut an object in the cluster with ray.put() via gRPC. Args: data: Pickled data. Can either be bytearray if this is called from the dataservicer, or bytes if called from PutObject. client_ref_id: The id associated with this object on the client. client_id: The client who owns this data, for tracking when to delete this reference. owner_id: The owner id of the object. context: gRPC context. N)_ownerz Put failed:r1F)rrrrzput: %sT)rr)r rUrrgr$r9rprqrr PutResponser rrRbinaryrzrSr) rArr<r&r=rCobjowner objectrefrHs r/r;zRayletServicer._put_objects& #D$//C 1$&& 7 7GC666  7 7 7 7 7 7 7 7 7 7 7 7 7 7 7      ] + + +!-e;+ Y*+++)Y-=-=-?-?tLLLLsA0A!A A!AA!AA!! B8+AB3-B83B8cg}|jD]`}||j|jvrtdt |z||j|j|a|j}|j} t5tj |||dkr|nd\}}dddn #1swxYwYnI#t$r<} t d| tjdcYd} ~ Sd} ~ wwxYwtdt |dt |d |D} d |D} tjd | | S) Nz4Asking for a ref not associated with this client: %s) num_returnsr6z Exception F)rzwait:  c6g|]}|Sr+rB)r,ready_object_refs r/ z-RayletServicer.WaitObject..Cs2   *:  # # % %   r1c6g|]}|Sr+rK)r,remaining_object_refs r/rMz-RayletServicer.WaitObject..Fs4   $ ! ' ' ) )   r1T)rready_object_idsremaining_object_ids) object_idsrRr&r9r?r7rHr6rrgwaitrprrr WaitResponser) rArBrCrRr/rHr6ready_object_refsremaining_object_refsrHrPrQs r/ WaitObjectzRayletServicer.WaitObject+s" % I IC$*7+<===JSQTXXU   t/0AB3G H H H H) / <$&&  ;>8 +'."}}GG$<<<8!#8                 < < < LL)a)) * * *!.U;;; ; ; ; ; ; ; <   !23333S9N5O5O5O P     >O     (=   *-!5    sB:C#B7+ C7B;;C>B;?C D  1D>D D taskarglistkwargsctd|jdtjj|j t5|jtjj kr| ||||}n|jtjj kr| ||||}n|jtjj kr|||||}nj|jtjjkr|||}n9t#dtjj|jzd|_|cdddS#1swxYwYdS#t&$rN}tddtjdt+j|cYd}~Sd}~wwxYw) Nz schedule: rIz$Unimplemented Schedule task type: %sTzCaught schedule exception)exc_infoFr)rprnamerr ClientTaskRemoteExecTypeNamerrFUNCTION_schedule_functionACTOR_schedule_actorMETHOD_schedule_method NAMED_ACTOR_schedule_named_actorNotImplementedErrorrr9ClientTaskTicketr r)rArXrYrZrCrrHs r/SchedulezRayletServicer.SchedulePs*  yyy.3BGG RRR T    $&&  9 9 BBB!44T7FGTTFFY.";"AAA!11$QQFFY.";"BBB!224&'RRFFY.";"GGG!77gFFFF->(3BGG RRS $                        LL4tL D D D!2;#4Q#7#7       sDFD E5( F5E99F<E9=F G AGGGcL|j|j}|tdt ||j}t |j}| |jdi|}|j|i|}| ||j } tj | S)Nz7Can't run an actor the server doesn't have a handle for return_idsr+) rUr payload_idr9getattrr]decode_optionsoptionsremoteunify_and_track_outputsr&rrrj) rArXrYrZrC actor_handlemethodoptsoutputr,s r/rfzRayletServicer._schedule_methodrs**4?;;  UVV Vty11dl++  #V^++d++F2622**64>BB.#>>>>r1c*||j|jt|j}t|j}| |jdi|}t |5|j|i|}dddn #1swxYwY||j|j <|j |j |j tj|j gSNrmr+)lookup_or_register_actorror&rqbaseline_optionsrrr!rsrU _actor_idrBrWaddrrrj)rArXrYrZrC remote_classrwrs r/rdzRayletServicer._schedule_actorsC44 OT^^DBB.#>>>>rct|jdksJtj|j|jpd}|j}||jvr ||j|<|j |j  ||j |tj|jgS)Nrrm)rzrorg get_actorr]rr}rBrUrWr&r~rYrrrj)rArXrCr bin_actor_ids r/rhz$RayletServicer._schedule_named_actors4?##q(((( di)?4@@--// t . .,1DOL ) $.)--l;;; l+++.5?;Q;Q;S;S:TUUUUr1rrct5||jvr|j||}tj|}t j|std|t|dkrtj ||j|<n"tj di|||j|<dddn #1swxYwY|j|S)Nz6Attempting to register function that isn't a function.rr+) rrTrRrgrinspect isfunctionr9rzrs)rArr&rrfuncrefrIs r/rz&RayletServicer.lookup_or_register_funcs5! " " I I+++*95b9ww'')$//#P?c'lla&7&7-0Z-=-=D&r**-BSZ-B-B'-B-B4-H-HD&r* I I I I I I I I I I I I I I I!"%%sB(CC Cct5||jvr|j||}tj|}t j|std|t|dkrtj |}ntj di||}||j|<dddn #1swxYwY|j|S)Nz0Attempting to schedule actor that isn't a class.rr+) rrXrRrgrrisclassr9rzrs)rArr&rractor_class_ref actor_class reg_classs r/r{z'RayletServicer.lookup_or_register_actors! " " > >666"&"29"=b"A!go66 {33X#$VWWW?c'lla&7&7 # ; 7 7II 5 5 5W 5 5k B BI4=-b1 > > > > > > > > > > > > > > >,R00sB"B==CCc$|g}nt|tr|}n|g}|D]b}||j|vrtd|||j||<cd|DS)NzAlready saw object_ref c6g|]}|Sr+rK)r,outs r/rMz:RayletServicer.unify_and_track_outputs..s 000 000r1)r8listrBrRrpwarning)rArxr&outputsrs r/rtz&RayletServicer.unify_and_track_outputss >GG  % % GGhG < >>???8;D Y ' 5 5000000r1r*)C__name__ __module__ __qualname__rr]rr InitRequestrsrrMrrrrrrrrrrrrrrr?rbytesboolrrrrrr GetRequestr5queueQueuerr"r1r4r3 PutRequestrAr>r bytearrayr;rTrWr^rrrrjrkrfrdrbrhrgremote_functionRemoteFunctionrr{rtr+r1r/rOrO`sTHTTTT(<@2424%124  $24242424h K Kn.J K K K K 9 9n.J 9 9 9 9 E En.J E E E E 8 8~/L 8 8 8 8 > >1P > > > > $   5    %%N4V%%%%N  3    "%D&DDD M M M M M M9999> FWFW*FWFW FW k FW ., -FWFWFWFWP <!: < < < <$>#<$$$$$N;?  %0  #     &M&ME9$%&M&M &M  &M&M&M&MP# # >3N# # # # T  ' c S#X   (    N ??'?c?S#X ?  ( ????. VV'VcVS#X V  ( VVVV0 ??'?c?S#X ?  ( ????&8< V V"- V  ( V V V V&&$'&2:4.&  +&&&&"11$'12:4.1111" 1 1 1 1 1r1rOc|H|t||tjjdSdSr*)r>encode_exceptionr:r;r<ABORTED)errrCs r/rrsN,S11222 011111 r1r^cvtj|}tj|Sr*)r rbase64standard_b64encodedecode)rqrs r/rrs/  Y ' 'D  $T * * 1 1 3 33r1rrcx|jsdStj|j}t|tsJ|Sr*)pickled_optionsrdrer8r2)rrrws r/rqrqs?  "t </ 0 0D dD ! !!! ! Kr1hostportc* d dtdtttffd}ddlm}|p|}|t dtd}t|}t|}t}tj ||tj ||tj||t|st!|d |t!||d |t#|||| } || S) Nrcroct5tjstjdd|i|cdddS ddddS#1swxYwYdS)Nrcr+)rrgrhinit)rcros r/default_connect_handlerz&serve..default_connect_handlers! " " J J%'' JxII:III J J J J J J J J J J J J J J J J J J J J J J J J J J Js&AAAr)$create_grpc_server_with_interceptorsray_client_serverF) max_workersthread_name_prefixrr asynchronousz 127.0.0.1::) task_servicer data_servicer logs_servicer grpc_serverr*)rrr?rray._private.grpc_utilsrrrrOrrray_client_pb2_grpc"add_RayletDriverServicer_to_server(add_RayletDataStreamerServicer_to_server'add_RayletLogStreamerServicer_to_serverr rrr&) rrrPrrserverrrrcurrent_handles r/servers_ $JJJ9=c3hJJJJMLLLLL-H1H 1 1-. F ##677M //M%''M:=&QQQ@PVWWW? vVVV   =(;T(;(;<<<Ft$4$4d$4$4555'### N  LLNNN r1ct5tjidddn #1swxYwYdfd }t|||}|fS)NcTtjrStjd|iS)Nrc)rgrhr)rcroargsinforZs r/rPz+init_and_serve..ray_connect_handler&s6     DK8DCzCFCC Cr1)rPr*)rrgrr)rrrrZrP server_handlers `` @r/init_and_server!s   ))x((()))))))))))))))DDDDDDDD$:MNNNM 4  s 044Fc|dt5tj|ddddS#1swxYwYdS)Nr)stoprrgshutdown)r_exiting_interpreters r/shutdown_with_serverr2s KKNNN   ++ )***++++++++++++++++++sAA  A c,ddtffd }|S)Nrccr0rtjd|d|dStjd|d|dStjdd|i|dS)N)address_redis_username_redis_passwordrc)rrcrcr+)rgr)rcrorredis_passwordredis_usernames r/rPz/create_ray_handler..ray_connect_handler9s  ? T#$2$2)  & SZSS?SSSSS H > > >o > > > > >r1r*r)rrrrPs``` r/create_ray_handlerr8sA ? ? ? ? ? ? ? ? ? ? r1rc@t|}t|S)zr Try to create a gcs client based on the command line args or by autodetecting a running Ray cluster. r)rrrs r/try_create_gcs_clientrKs" 4G<TY 22H4yyH A##D$7@@ KKLLL(LLMMM yG I I L..&*&D    ty$)-@AA% = Y M $'3LLNNU!6t|!D!DJ$0HHTTT ,=='J}--+D>  $ $ $ Q QQ4<QQ  ########  $ JqMMMy---'3a77,I)))Q.)(A--+---)A-22-1NNNKK#8 W W WXXXA YD  As?N+.BK N+ L%AL N+ L%%BN++O O__main__r*)F)VrrKrrrnloggingrrdrrZr collectionsrtypingrrrrrr r r;rgray._private.state!ray.core.generated.ray_client_pb2core generatedrr&ray.core.generated.ray_client_pb2_grpcrr ray._common.network_utilsr r ray._privaterray._private.client_mode_hookrray._private.ray_constantsrray._private.ray_loggingrray._private.servicesrray._private.tls_utilsr ray._rayletrray.job_configrray.util.client.commonrrrrr#ray.util.client.server.dataservicerr"ray.util.client.server.logservicerrray.util.client.server.proxierr%ray.util.client.server.server_picklerrr #ray.util.client.server.server_stubsr! getLoggerrrpr"rMRayletDriverServicerrOrr?r TaskOptionsrqr5rrrrrrr+r1r/rs     ######BBBBBBBBBBBBBBBBBB ::::::::::::DDDDDDDDDDDDAAAAAAAA&&&&&&======222222111111GGGGGG::::::!!!!!!$$$$$$=<<<<<@@@@@@888888VVVVVVVV>>>>>>  8 $ $ + ,KR P P,,,^C 1C 1C 1C 1C 1(=C 1C 1C 1L222434444 N68DcN;S!!!3!!!!H!!C!!!!"++++ &&8C=&Xi5H&&&&aaaH zDFFFFFr1