&`i$ddlmZddlmZmZmZddlmZmZddl m Z ddl m Z dZ e GddeeZe Gd d eeZe Gd d eeZe Gd deZe GddeZe GddeZe GddeZGddeZe GddeZe GddeZe GddeZe GddeZe Gdd eZe Gd!d"eZe Gd#d$eZd%S)&)Enum)DictListOptional) BaseModelField) JobDetails) DeveloperAPIiPc>eZdZdZdZdZdZdZdZdZ dZ d Z d e fd Z d S) RunStatusz5Enumeration of the possible statuses for a Train run. INITIALIZING SCHEDULINGRUNNING RESTARTINGRESIZINGFINISHEDERROREDABORTEDreturncL|tjtjtjfvSN)r rrrselfs w/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/v2/_internal/state/schema.py is_terminalzRunStatus.is_terminal#s *I,=y?PQQQN)__name__ __module__ __qualname____doc__r rrrrrrrboolrrrr r sh??"LJGJHHGGRTRRRRRRrr c2eZdZdZdZdZdZdZdZde fdZ d S) RunAttemptStatusz=Enumeration of the possible statuses for a Train run attempt.PENDINGrrrrrcL|tjtjtjfvSr)r$rrrrs rrzRunAttemptStatus.is_terminal9s&  %  $  $   rN) rrrr r%rrrrr!rr"rrr$r$'sRGGGGHGG T      rr$ceZdZdZdZdZdS) ActorStatusz5Enumeration of the statuses for a Train worker actor.ALIVEDEADN)rrrr r)r*r"rrr(r(As ?? E DDDrr(cFeZdZUdZedZeeefe d<dS)TrainResourcesz.Resources allocated for a Train worker or run.zVA dictionary specifying the types and amounts of resources allocated (e.g., CPU, GPU). description resourcesN) rrrr rr/rstrfloat__annotations__r"rrr,r,KsQ88"'%&###ItCJrr,ceZdZUdZedZeed<edZeed<edZ eed<ed Z e ed <ed Z e ed <ed Z e ed<edZeed<edZeeed<edZeeed<edZeed<edZee ed<dS) TrainWorkerz"Metadata about a Ray Train worker.z6The global rank of the worker in the training cluster.r- world_rankz)The local rank of the worker on its node. local_rankz-The rank of the worker's node in the cluster. node_rankz$The unique ID of the worker's actor.actor_idz6The unique ID of the node where the worker is running.node_idz7The IP address of the node where the worker is running.node_ipzThe process ID of the worker.pidz*A list of GPU IDs allocated to the worker.gpu_idsz'The current status of the worker actor.statusz-The resources allocated to this Train worker.r/z.The path to the log file for the Train worker. log_file_pathN)rrrr rr5intr2r6r7r8r0r9r:r;r<rr=rr(r/r,r>r"rrr4r4Us,,eLJe(STTTJTTTU'VWWWIsWWWE&LMMMHcMMM5LGS5MGSu!@AAACAAA+WXXXGT#YXXX$)E=%%%FH[ !!&C!!!I~$)5D$$$M8C=rr4ceZdZUdZedZeed<edZeed<edZ e eed<ed Z e eed <d S) MemoryInfoz'Memory usage information for a process.z2The resident set size (RSS) memory usage in bytes.r-rssz-The virtual memory size (VMS) usage in bytes.vmszThe number of page faults.pfaultszThe number of page-ins.pageinsN) rrrr rrBr?r2rCrDrrEr"rrrArArs11u!UVVVCVVVu!PQQQCQQQ"U/KLLLGXc]LLL"U/HIIIGXc]IIIIIrrAceZdZUdZedZeed<edZe e e ed<edZ e ed<d S) ProcessStatsz(CPU and memory statistics for a process.zThe percentage of CPU usage.r- cpuPercentzOMemory statistics, including total memory, free memory, and memory usage ratio.memz"Detailed memory usage information. memoryInfoN)rrrr rrHr1r2rIrrr?rJrAr"rrrGrG|s22*HIIIJIII$u"   C$s) #U/STTTJ TTTTTrrGcXeZdZUdZedZeed<edZeed<dS)ProcessGPUUsagez#GPU usage statistics for a process.zThe process ID.r-r;zThe GPU memory usage in bytes.gpuMemoryUsageN) rrrr rr;r?r2rMr"rrrLrLsT--u!2333C333%,LMMMNCMMMMMrrLceZdZUdZedZeed<edZe ed<edZ eed<ed Z e e ed <ed Ze ed <ed Ze ed<edZeed<dS)GPUStatszStatistics for a GPU.z!The unique identifier of the GPU.r-uuidzThe index of the GPU.indexzThe name of the GPU.namez&The percentage utilization of the GPU.utilizationGpuz'The amount of GPU memory used in bytes. memoryUsedz(The total amount of GPU memory in bytes. memoryTotalz0GPU usage statistics for the associated process. processInfoN)rrrr rrPr0r2rQr?rRrSrr1rTrUrVrLr"rrrOrOs "EFFFD#FFF#:;;;E3;;;"8999D#999&+e<'''NHUO*STTTJTTT+UVVVKVVV#(5F$$$KrrOcteZdZUdZeddZeeed<ee dZ e e ed<dS) DecoratedTrainWorkerzJDetailed metadata for a Ray Train worker, including process and GPU stats.Nz1CPU and memory statistics for the worker process.r- processStatszDA list of GPUs used by the worker process, with detailed statistics.)default_factoryr.gpus) rrrr rrYrrGr2listr[rrOr"rrrXrXsTT+05 M,,,L(<(!5%D$x.rrXcTeZdZUdZedZeed<edZeed<edZ e ed<ed Z e eed <ed Z eed <ed Ze eed<edZeeed<edZeeed<dS)TrainRunAttemptz:Metadata for an individual attempt to execute a Train run.z+Unique identifier for the parent Train run.r-run_idz6Unique identifier for this specific Train run attempt. attempt_idz6The current execution status of the Train run attempt.r=zLAdditional details about the status, including error messages if applicable. status_detailzGThe UNIX timestamp (in nanoseconds) when the Train run attempt started. start_time_nszlThe UNIX timestamp (in nanoseconds) when the Train run attempt ended. If null, the attempt is still ongoing. end_time_nszBThe resources (e.g., CPU, GPU) allocated to the Train run attempt.r/zLList of Train workers participating in this attempt, sorted by global ranks.workersN)rrrr rr_r0r2r`r=r$rarrbr?rcr/rr,rdr4r"rrr^r^sDD%$QRRRFCRRReLJ %uL   F $)53$$$M8C=/M3"'1"""K# ',eX'''ItN#"'""""GT+ rr^cBeZdZUdZedZeeed<dS)DecoratedTrainRunAttemptzKDetailed metadata for a Train run attempt, including decorated worker data.zIA list of Train workers with detailed statistics, sorted by global ranks.r-rdN) rrrr rrdrrXr2r"rrrfrfsNUU*/%"+++GT& 'rrfcjeZdZUdZedZeed<edZeed<edZ eed<ed Z eed <ed Z e ed <ed Z eeed<edZeed<edZeeed<edZeeed<dS)TrainRunz?Metadata for a Ray Train run, including its details and status.z$Unique identifier for the Train run.r-idz.Human-readable name assigned to the Train run.rRz.The Ray Job ID associated with this Train run.job_idz.Unique ID of the actor managing the Train run.controller_actor_idz.The current execution status of the Train run.r=zTAdditional details about the current status, including error messages if applicable.raz?The UNIX timestamp (in nanoseconds) when the Train run started.rbzdThe UNIX timestamp (in nanoseconds) when the Train run ended. If null, the run is still in progress.rcz6The path to the log file for the Train run controller.controller_log_file_pathN)rrrr rrir0r2rRrjrkr=r rarrbr?rcrlr"rrrhrhsIIe FGGGBGGG"RSSSD#SSS%$TUUUFCUUU$uD   DFI$)52$$$M8C=UM3"'1"""K#/4eL///hsmrrhcreZdZUdZedZeeed<eddZ e e ed<dS)DecoratedTrainRunzJDetailed metadata for a Ray Train run, including attempts and job details.z1A list of attempts made to execute the Train run.r-attemptsNzADetailed information about the job that initiated this Train run. job_details) rrrr rrorrfr2rprr r"rrrnrnsTT/4uG000Hd+,). W)))K*%rrncBeZdZUdZedZeeed<dS)TrainRunsResponsez3Response containing a list of decorated Train runs.z,A list of Train runs with detailed metadata.r- train_runsN) rrrr rrsrrnr2r"rrrrrrsL==*/%B+++J&'rrrN)enumrtypingrrrray._common.pydantic_compatrr)ray.dashboard.modules.job.pydantic_modelsr ray.util.annotationsr MAX_ERROR_STACK_TRACE_LENGTHr0r r$r(r,r4rArGrLrOrXr^rfrhrnrrr"rrrzs''''''''''88888888@@@@@@------$RRRRRTRRR6     sD   2#tY)8JJJJJJJJUUUUU9UUUNNNNNiNNN     y         ;   i@y8         r