&`i1ddlZddlmZmZddlmZddlmZmZm Z m Z ddl m Z ddl mZmZddlmZdZeZeGd d ZeGd d ZeGd dZeGddZeGddZeGddZeGddeZeGddeZeGddeZeGddZeGddZeGdd Z eGd!d"Z!dS)#N) dataclassfield)Enum)DictListOptionalTuple) InstanceUtil) NodeState NodeStatus)InstanceNodeTerminatedc:eZdZUdZeed<dZeed<dZeed<dS) ResourceUsage resource_namegtotalusedN) __name__ __module__ __qualname__rstr__annotations__rfloatrl/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/autoscaler/v2/schema.pyrrsDM3E5D%rrc0eZdZUeeed<eed<dS) NodeUsageusage idle_time_msN)rrrrrrintrrrrrs1  rrc^eZdZUeed<eed<eed<eed<dZeeed<dZeeed<dZee ed<dZ eeed <dZ eeed <dZ ee eed <dZeeeefed <d eeeffdZd eeeffdZd eeeffdZdS)NodeInfoinstance_type_nameray_node_type_name instance_id ip_addressN node_statusnode_idresource_usagefailure_detaildetails node_activitylabelsreturnc@|jiSd|jjDS)Nc(i|]}|j|jSrrr.0rs r z,NodeInfo.total_resources..CsLLLQLLLrr+r selfs rtotal_resourceszNodeInfo.total_resources@s*   &ILL$2E2KLLLLrc@|jiSd|jjDS)Nc8i|]}|j|j|jz Srrrrr4s rr7z0NodeInfo.available_resources..Hs%UUUa16!1UUUrr8r9s ravailable_resourceszNodeInfo.available_resourcesEs*   &IUU4;N;TUUUUrc@|jiSd|jjDS)Nc(i|]}|j|jSr)rrr4s rr7z+NodeInfo.used_resources..MsKKKAKKKrr8r9s rused_resourceszNodeInfo.used_resourcesJs*   &IKK1D1JKKKKr)rrrrrr)rr*r+rr,r-r.rr/rrr;r?rBrrrr$r$'s]OOO!%K#%%%!GXc]!!!*.NHY'...$(NHSM(((!GXc]!!!)-M8DI&---'+FHT#s(^ $+++Mc5j!1MMMM VT#u*%5VVVV LS%Z 0LLLLLLrr$ceZdZUGddeZeed<eed<eed<eed<eed<dZe eed <dZ e eed <dS) LaunchRequestceZdZdZdZdS)LaunchRequest.StatusFAILEDPENDINGN)rrrrGrHrrrStatusrFRsrrIr%r&countstate request_ts_sN failed_ts_sr-) rrrrrIrrr"rMrr-rrrrDrDPs  JJJ MMM!%K#%%%!GXc]!!!!!rrDc@eZdZUeeefed<eed<defdZdS)ResourceRequestByCountbundlerJr0c(d|jd|jdS)N[ ])rJrPr9s r__str__zResourceRequestByCount.__str__ms.4:.. ....rN) rrrrrrrr"rUrrrrOrOfsQ e  JJJ///////rrOc&eZdZUeeed<dS)ResourceDemandbundles_by_countN)rrrrrOrrrrrWrWqs&1222222rrWcneZdZUeed<dZeeed<dZeeed<dZeeed<dZ dS)PlacementGroupResourceDemandr-Npg_idstrategyrKc|jsdSd}tj||j}|sdS|jd\}}|d\}}||_||_||_dS)Nz ^.*:.*\|.*$:|)r-rematchsplitr[r\rK)r:patternrar[r-r\rKs r __post_init__z*PlacementGroupResourceDemand.__post_init__s|  F!$,//  F++C00w!--,,%    r) rrrrrr[rr\rKrdrrrrZrZwslLLLE8C="Hhsm"""E8C=rrZceZdZdS)RayTaskActorDemandNrrrrrrrfrfDrrfceZdZdS)ClusterConstraintDemandNrgrrrrjrjrhrrjceZdZUeeZeeed<eeZ ee ed<eeZ ee ed<dS)ResourceDemandSummarydefault_factoryplacement_group_demandray_task_actor_demandcluster_constraint_demandN) rrrrlistrorrZrrprfrqrjrrrrlrlsBGBBBD!=>7.sNNNQNNNrrr9s rr;zClusterStatus.total_resourcessNN$2MNNNNrc$d|jDS)Nc8i|]}|j|j|jz Srr>r4s rr7z5ClusterStatus.available_resources..s%WWWa16!1WWWrrr9s rr?z!ClusterStatus.available_resourcessWW4;VWWWWrN)rrrrrrr|rr$rr}r~rDrrrrrrlrrtrrrrr;r?rrrr{r{s$)5#>#>#>L$x.>>>!&t!???#(5#>#>#>L$x.>>>27%2M2M2MD/MMM.3e-///+5///E5///Oc5j!1OOOOXT#u*%5XXXXXXrr{ceZdZUdZdZeeed<dZee ed<dZ ee ed<de e effdZde fdZde fd ZdS) AutoscalerInstanceaw AutoscalerInstance represents an instance that's managed by the autoscaler. This includes two states: 1. the instance manager state: information of the underlying cloud instance. 2. the ray node state, e.g. resources, ray node status. The two states are linked by the cloud instance id, which should be set when the ray node is started. Ncloud_instance_idray_node im_instancer0c ddddddddd}||jdu|jdu|jduf\}}|s||fS|jO|jH|j!tj|jjrdSn tj|jjsd S|j|j|jjkrd S|j|j|j|jjkrd SdS) zValidate the autoscaler instance state. Returns: A tuple of (valid, error_msg) where: - valid is whether the state is valid - error_msg is the error message for the validation results. )Fz Not possible)Tr)TzKThere's a ray node w/o cloud instance id, must be started not by autoscaler)Fz7Not possible - no way to link im instance with ray node)FzKNot possible since cloud instance id is either part of im state or ray node))TTT)TTF)TFT)TFF)FTT)FTF)FFT)FFFN)FzEinstance should be in a status where cloud instance is not allocated.)FzAinstance should be in a status where cloud instance is allocated.)Fz cloud instance id doesn't match.)rrrr is_cloud_instance_allocatedstatusr')r:state_combinationsvalid error_msgs rvalidatezAutoscalerInstance.validates:!8!+" #" #-",#-)  ..&$. % D(  y $)# #   'DM,A%-;Drsw ((((((((............BBBBBBCCCCCCCC<<<<<< 0      %L%L%L%L%L%L%L %LP """"""" "* /////// / 3333333 3  > 8                n               ''''''' '" XXXXXXX XB @ @ @ @ @ @ @  @ @ @ r