&`iMXddlZddlZddlmZmZmZmZddlmZm Z GddZ dS)N)DictListOptionalSet)InstanceInstanceUpdateEventc (eZdZUdZdZeededfed<e d"de de de j d e d e f d Z e d e fd Ze d e j d efdZe d e j d efdZe d e j d efdZd e j d efdZe d"de de j d e d efdZe de de j d e fdZe de ded efdZe d ededffdZe d#de dedd edfdZe d#de dedd edfdZe d#de dedd eefdZed e j d edfdZe de ded e fd Z ed!Z!dS)$ InstanceUtilzt A helper class to group updates and operations on an Instance object defined in instance_manager.proto NzInstance.InstanceStatus_reachable_from instance_id instance_typestatusdetailsreturnct}d|_||_||_||_t ||||S)a Returns a new instance with the given status. Args: instance_id: The instance id. instance_type: The instance type. status: The status of the new instance. details: The details of the status transition. r)rversionr rrr _record_status_transition)r rrrinstances }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/autoscaler/v2/instance_manager/common.py new_instancezInstanceUtil.new_instancesI ::*!. ..xIIIcBttjS)z/ Returns a random instance id. )struuiduuid4rrrandom_instance_idzInstanceUtil.random_instance_id-s 4:<<   rinstance_statusc  |tjksJ|tjtjtjtjtjtjtjtj tj tj h vS)z Returns True if the instance is in a status where there could exist a cloud instance allocated by the cloud provider. ) rUNKNOWN ALLOCATEDRAY_INSTALLING RAY_RUNNING RAY_STOPPINGRAY_STOP_REQUESTED RAY_STOPPED TERMINATINGRAY_INSTALL_FAILEDTERMINATION_FAILEDALLOCATION_TIMEOUTrs ris_cloud_instance_allocatedz(InstanceUtil.is_cloud_instance_allocated4sf ("22222    #   !  '    '  '  ' #  rc|tjksJ|ttjvrdS|ttjvrdSdS)z Returns True if the instance is in a status where the ray process is running on the cloud instance. i.e. RAY_RUNNING, RAY_STOP_REQUESTED, RAY_STOPPING FT)rr!r get_reachable_statusesr%r$r,s ris_ray_runningzInstanceUtil.is_ray_runningHsj("22222 lAA  !    5 lAA(BVWW W W4urc|tjksJtjt|vrdS|ttjvrdSdS)z Returns True if the instance is in a status where the ray process is pending to be started on the cloud instance. FT)rr!r$r r/r,s ris_ray_pendingzInstanceUtil.is_ray_pending[sl("22222  |'J'J ( (   5 lAA(BVWW W W5trcNtjt|vS)zt Returns True if the instance is in a status where it may transition to RAY_RUNNING status. )rr$r r/r,s ris_ray_running_reachablez%InstanceUtil.is_ray_running_reachableos) #|'J'J ( (   rrnew_instance_statusc|t|jvrdS||_t|||dS)aBTransitions the instance to the new state. Args: instance: The instance to update. new_instance_status: The new status to transition to. details: The details of the transition. Returns: True if the status transition is successful, False otherwise. FT)r get_valid_transitionsrr)rr5rs r set_statuszInstanceUtil.set_statusxsU" 5577H I I5-..x9LgVVVtrctj}|jt j|||dS)zRecords the status transition. Args: instance: The instance to update. status: The new status to transition to. )r timestamp_nsrN)timetime_nsstatus_historyappendr StatusHistory)rrrnow_nss rrz&InstanceUtil._record_status_transitionsR&&  " &#        r timeout_sc `|j}t||}t|dksCJd|jdt|dt j|dt|d}tj |z |dzkrd Sd S) al Returns True if the instance has been in the current status for more than the timeout_seconds. Args: instance: The instance to check. timeout_seconds: The timeout in seconds. Returns: True if the instance has been in the current status for more than the timeout_s seconds. )select_instance_statusz instance z has  z statusgeAFT) rr get_status_transition_times_nslenr rInstanceStatusNamesortedr;r<)rrA cur_statusstatus_times_nsstatus_time_nss r has_timeoutzInstanceUtil.has_timeouts_ &EE ZF  ?##q((( A, A A33G3G A A&++J77 A A A)(( 004 <>>N *y3 ? ?5trctjtjhtjtjtjtjhtjtjtjtjtjtj tj tj htjtjtj tj tj tj htjtj tjtj tj tj htj tjtj tjtj htjtj htjtj tj tj htj tj tj htj tj tjhtjtj htj ttjttj tj tj htjtiSN)rQUEUED REQUESTEDr"ALLOCATION_FAILEDr#r$r+r%r'r( TERMINATEDr)r&r*setr!rrrr7z"InstanceUtil.get_valid_transitionss O"  "* !   '$+%$$#)!2  #$+ $$##&*  +%$$# #  '%$$# *  '$*  !$$# $  $# #  #+ #  '$*    &  '$# *  ceeu[ [ rrCzInstance.StatusHistorycdg}|jD]%}|r |j|kr||&|S)aH Returns the status history of the instance. Args: instance: The instance. select_instance_status: The go-to status to search for, i.e. select only status history when the instance transitions into the status. If None, returns all status updates. )r=rr>)rrChistory status_updates rget_status_transitionsz#InstanceUtil.get_status_transitionsbsO%4 * *M& !15KKK NN= ) ) ) )rc~t||}|d|r|dSdS)z Returns the last status transition of the instance. Args: instance: The instance. instance_status: The status to search for. If None, returns the last status update. c|jSrQr:)xs rz9InstanceUtil.get_last_status_transition..s1>r)keyrFN)r rZsort)rrCrXs rget_last_status_transitionz'InstanceUtil.get_last_status_transitionzsI55h@VWW 11 222  2; trcLdt||DS)aL Returns a list of timestamps of the instance status update. Args: instance: The instance. instance_status: The status to search for. If None, returns all status updates timestamps. Returns: The list of timestamps of the instance status updates. cg|] }|j Srr]).0es r z?InstanceUtil.get_status_transition_times_ns..s*    N   r)r rZ)rrCs rrGz+InstanceUtil.get_status_transition_times_nss9   !880    rcR|j||j|S)at Returns the set of instance status that is reachable from the given instance status following the status transitions. This method is memoized. Args: instance_status: The instance status to start from. Returns: The set of instance status that is reachable from the given instance status. )r _compute_reachable)clsrs rr/z#InstanceUtil.get_reachable_statusess-   &  " " $ $ $"?33rupdatec|jrOdtj|jd|jd|jd|jd|jd|j Sdtj|j dtj|jd|jd|jd|jd|jd|j S) z3Returns a log string for the given instance update.z New instance z (id=z, type=z, cloud_instance_id=z , ray_id=z): zUpdate instance z->) upsertrrIrJr5r rcloud_instance_idnode_idrr)rrks rget_log_str_for_updatez#InstanceUtil.get_log_str_for_updates? = @*//0JKK@@'@@080F@@&.%?@@#* @@06~ @@  <&++HO<< < <&++F,FGG < <# < <-5,B < <"*!;  < < &  < < ,2>  < < rc|}fdi|_tjD]%}t }||||j|<&dS)zo Computes and memorize the from status sets for each status machine with a DFS search. cj||D](}||vr"|||||)|S)z^ Regular DFS algorithm to find all reachable nodes from a given node. )add)graphstartvisited next_nodedfss rrxz,InstanceUtil._compute_reachable..dfssQ#5\ 3 3 G++KK ***Cy'222NrN)r7r rrIvaluesrV)rjvalid_transitionsrrvrxs @rrizInstanceUtil._compute_reachables  5577     !-4466 R RFeeG*-#.?*Q*QC  ' ' R Rr)r rQ)"__name__ __module__ __qualname____doc__r rrr__annotations__ staticmethodrrrIrrboolr-r0r2r4r8rintrOr7rrZrbrG classmethodr/rrprirrrr r s X &,E(F FG     '   \.!!!!\!  X5L QU   \ &(?D\$(?D\& (2I d    %4  \0  $,$; FI   \ $h34\8^ 4!3'@#AA$^ ^ ^ \^ @GK ()B C & '\.GK ()B C * +\$GK   ()B C  c   \ ,4!04 & '444[4$  ;N SV   \ &RR[RRRrr ) r;rtypingrrrr'ray.core.generated.instance_manager_pb2rrr rrrrs ,,,,,,,,,,,,QQQQQQQQ\R\R\R\R\R\R\R\R\R\Rr