&`iddlmZmZddlmZmZmZmZmZm Z ddl m Z GddeZ GddZ e ZdS) )Enumauto)AnyCallableDictListOptionalUnion) cli_loggerceZdZdZeZeZeZeZeZ eZ eZ eZ eZ eZeZdS)CreateClusterEventaEvents to track in ray.autoscaler.sdk.create_or_update_cluster. Attributes: up_started : Invoked at the beginning of create_or_update_cluster. ssh_keypair_downloaded : Invoked when the ssh keypair is downloaded. cluster_booting_started : Invoked when when the cluster booting starts. acquiring_new_head_node : Invoked before the head node is acquired. head_node_acquired : Invoked after the head node is acquired. ssh_control_acquired : Invoked when the node is being updated. run_initialization_cmd : Invoked before all initialization commands are called and again before each initialization command. run_setup_cmd : Invoked before all setup commands are called and again before each setup command. start_ray_runtime : Invoked before ray start commands are run. start_ray_runtime_completed : Invoked after ray start commands are run. cluster_booting_completed : Invoked after cluster booting is completed. N)__name__ __module__ __qualname____doc__r up_startedssh_keypair_downloadedcluster_booting_startedacquiring_new_head_nodehead_node_acquiredssh_control_acquiredrun_initialization_cmd run_setup_cmdstart_ray_runtimestart_ray_runtime_completedcluster_booting_completedx/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/autoscaler/_private/event_system.pyr r s(J!TVV"dff"dff466!TVVDFFM"&$&& $rr c eZdZdZdZdedeeegdfe eegdfffdZ d de de eee ffdZdefd ZdS) _EventSystemzEvent system that handles storing and calling callbacks for events. Attributes: callback_map (Dict[str, List[Callable]]) : Stores list of callbacks for events when registered. ci|_dSN callback_map)selfs r__init__z_EventSystem.__init__1sreventcallbackNc|tjvrtj|d|j|gt|tur|gn|dS)agStores callback handler for event. Args: event: Event that callback should be called on. See CreateClusterEvent for details on the events available to be registered against. callback (Callable[[Dict], None]): Callable object that is invoked when specified event occurs. zA is not currently tracked, and this callback will not be invoked.N) r __members__valuesr warningr% setdefaultextendtypelist)r&r(r)s radd_callback_handlerz!_EventSystem.add_callback_handler4s *6==?? ? ?  111    $$UB//66x..44XJJ(     r event_datacd|i}||d<||jvr|j|D]}||dSdS)a:Executes all callbacks for event. Args: event: Event that is invoked. See CreateClusterEvent for details on the available events. event_data (Dict[str, Any]): Argument that is passed to each callable object stored for this particular event. N event_namer$)r&r(r3r)s rexecute_callbackz_EventSystem.execute_callbackLsc  J#( < D% % % -e4 % %$$$$ & % % %rc,||jvr |j|=dSdS)zClears stored callable objects for event. Args: event: Event that has callable objects stored in map. See CreateClusterEvent for details on the available events. Nr$)r&r(s rclear_callbacks_for_eventz&_EventSystem.clear_callbacks_for_event_s+ D% % %!%((( & %rr#)rrrrr'strr rrrr2r r rr6r8rrrr!r!)s  $.Xtfdl5K0LLM    2QU%%'%5=d38n5M%%%%&)s))))))rr!N)enumrrtypingrrrrr r "ray.autoscaler._private.cli_loggerr r r!global_event_systemrrrr>s================999999''''''''D>)>)>)>)>)>)>)>)B#lnnr