&`iz ddlmZmZmZddlZdedeefdZ d dedeeejjej j fdeeejjej j feffdZ d deeejjej j fdeeejjej j feffd Z dS) )DictOptionalUnionNnode_ipreturncntjD] }|dr|d|kr |dcS!dS)z?Returns the node ID for the first alive node with the input IP.AliveNodeManagerAddressNodeIDN)raynodes)rnodes e/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/air/util/node.py_get_node_id_from_node_iprsI "" = "T"677BB> ! ! ! 4node_idremote_func_or_actor_classcxtjj|d}d|i}||S|jdi|S)aSchedule a remote function or actor class on a given node. Args: node_id: The node to schedule on. remote_func_or_actor_class: A Ray remote function or actor class to schedule on the input node. If None, this function will directly return the options dict to pass to another remote function or actor class as remote options. Returns: The provided remote function or actor class, but with options modified to force placement on the input node. If remote_func_or_actor_class is None, the options dict to pass to another remote function or actor class as remote options kwargs. F)rsoftscheduling_strategyN)r utilscheduling_strategiesNodeAffinitySchedulingStrategyoptions)rrrrs r_force_on_noders\*(8WWeX%&9:G!) - % - 8 8 8 88rcltj}t||S)aSchedule a remote function or actor class on the current node. If using Ray Client, the current node is the client server node. Args: remote_func_or_actor_class: A Ray remote function or actor class to schedule on the current node. If None, this function will directly return the options dict to pass to another remote function or actor class as remote options. Returns: The provided remote function or actor class, but with options modified to force placement on the current node. If remote_func_or_actor_class is None, the options dict to pass to another remote function or actor class as remote options kwargs. )r get_runtime_context get_node_idr)rcurrent_node_ids r_force_on_current_noder!0s/(-//;;==O /+E F FFr)N) typingrrrr strrremote_functionRemoteFunctionactor ActorClassrr!rrrr(sF(((((((((( sx} 99 9 ( c!0#)2FFG!9  5$3SY5II JD PQ 9999H GG ( c!0#)2FFG!G 5$3SY5II JD PQ GGGGGGr