%`i#<ddlZddlmZmZddlZddlmcmZddlmcm Z ddl Zddl m Z ddl mZddlmZddlmZddgZejZd Zdd Z ddZddZ ddZdZddedefdZdeeej e!ffdZ"dS)N)ListTuple) build_address) GlobalState)GcsClientOptions) common_pb2free global_gccdtjjj}|jdS)z3Trigger gc.collect() on all workers in the cluster.N)ray_privateworker global_worker core_workerr rs m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_private/internal_api.pyr r s,\ .F   """""ctj|}t}tj|ddd}|||S)NTF)allow_cluster_id_nilfetch_cluster_id_if_nil)services%canonicalize_bootstrap_address_or_dierrcreate_initialize_global_state)addressstateoptionss rget_state_from_addressrsV x)UK  E"" ##rcddlm}ddlm}m}||Nd}|D]}|dr|}n| Jdt |d|d}nt ||}||d tfd tfg } || } | | d d} | S)zReturns global memory info.rinit_grpc_channelnode_manager_pb2node_manager_pb2_grpcNAlivezEvery raylet is deadNodeManagerAddressNodeManagerPortgrpc.max_send_message_lengthgrpc.max_receive_message_lengthrFinclude_memory_infogN@timeout) ray._private.grpc_utilsr0ray.core.generatedr2r3 node_tablerMAX_MESSAGE_LENGTHNodeManagerServiceStubFormatGlobalMemoryInfoFormatGlobalMemoryInfoRequest) rnode_manager_addressnode_manager_portr0r2r3rayletnoderaylet_addresschannelstubr-s rr%r%8sA:99999JJJJJJJJ#'8'@$$&&  DG}  !!#9!!!& ' (&1B*C  '';=NOO +-? @ .0B C G ! 7 7 @ @D  ' '6656QQ (  E Lrcddlm}ddlm}m}||Jt ||}||dt fdt fg}||}|| |d } | S) z@Returns NodeStats object describing memory usage in the cluster.rr/r1Nr7r8r9r:g>@r<) r>r0r?r2r3rrArB GetNodeStatsGetNodeStatsRequest) rErFr;r0r2r3rIrJrK node_statss rrOrO^s :99999JJJJJJJJ +0A0M0M M"#79JKKN +-? @ .0B C G ! 7 7 @ @D"",,AT,UU#J rc d}|dt|jjdz |jjt d|jjz|jjz dt d|jjz|jjz dz }|jjdkr2|dt|jjdz z }|jj dkre|dt|jj dz |jj t|jj dz |jj z z }|jj dkre|d t|jj dz |jjt|jj dz |jj z z }|jjr|d z }|S) zDReturns formatted string describing object store stats in all nodes.z6--- Aggregate object store stats across all nodes --- z=Plasma memory usage {} MiB, {} objects, {}% full, {}% needed idrz%Plasma filesystem mmap usage: {} MiB z:Spilled {} MiB, {} objects, avg write throughput {} MiB/s z:Restored {} MiB, {} objects, avg read throughput {} MiB/s z4Object fetches queued, waiting for available memory.)formatint store_statsobject_store_bytes_usednum_local_objectsroundobject_store_bytes_availobject_store_bytes_primary_copyobject_store_bytes_fallbackspill_time_total_sspilled_bytes_totalspilled_objects_totalrestore_time_total_srestored_bytes_totalrestored_objects_totalobject_pulls_queued)r- store_summarys rr&r&ys MM 6 !9[I J J   / #;<#<=    #CD#<=     M& 4q88AHH !=M N N    +a// I P PE%9[IJJ!7%9"$':;    -11 I P PE%:kJKK!8%:"$'<=    ,POO r object_refs local_onlyctjdttjjj}t|tjr|g}t|ts/td t||D]>}t|tjs"td |?| tjd5t!|dkr ddddS|j||ddddS#1swxYwYdS)a DeprecationWarning: `free` is a deprecated API and will be removed in a future version of Ray. If you have a use case for this API, please open an issue on GitHub. Free a list of IDs from the in-process and plasma object stores. This function is a low-level API which should be used in restricted scenarios. If local_only is false, the request will be send to all object stores. This method will not return any value to indicate whether the deletion is successful or not. This function is an instruction to the object store. If some of the objects are in use, the object stores will delete them later when the ref count is down to 0. Examples: .. testcode:: import ray @ray.remote def f(): return 0 obj_ref = f.remote() ray.get(obj_ref) # wait for object to be created first free([obj_ref]) # unpin & delete object globally Args: object_refs (List[ObjectRef]): List of object refs to delete. local_only: Whether only deleting the list of objects in local object store or all object stores. z`free` is a deprecated API and will be removed in a future version of Ray. If you have a use case for this API, please open an issue on GitHub.z*free() expects a list of ObjectRef, got {}zIAttempting to call `free` on the value {}, which is not an ray.ObjectRef.zray.freerN)warningswarnDeprecationWarningr r rr isinstance ObjectReflist TypeErrorrStypecheck_connected profilingprofilelenr free_objects)rdrer object_refs rr r sJ M O \ .F+s}--$"m k4 ( (  8 ? ?[@Q@Q R R   " *cm44 117 1C1C      : & &AA {  q AAAAAAAA '' Z@@@ AAAAAAAAAAAAAAAAAAs;EEE  E returnctjjj}||jS)aReturn the locally submitted ongoing retry tasks triggered by lineage reconstruction. NOTE: for the lineage reconstruction task status, this method only returns the status known to the submitter (i.e. it returns SUBMITTED_TO_WORKER instead of RUNNING). The return type is a list of pairs where pair.first is the lineage reconstruction task info and pair.second is the number of ongoing lineage reconstruction tasks of this type. )r r rrror.get_local_ongoing_lineage_reconstruction_tasksrs rrwrws8\ .F    L L N NNr)N)Nrr r!TFN)NN)NNT)F)#rgtypingrrr ray._private.profilingr rpray._private.servicesrray._private.workerray._common.network_utilsrray._private.stater ray._rayletrr?r__all___configmax_grpc_message_sizerAr rr#r%rOr&rlboolr LineageReconstructionTaskrTrwrrrs *********(((((((((333333******(((((()))))) ; [6688###    ####*####NLP6777tAAAAdAAAAAAAAAAHO * . 349OOOOOOr