&`iDddlZddlZddlZddlZddlZddlZddlmZmZm Z m Z m Z m Z m Z mZmZddlZddlmZmZddlmZddlmZmZddlmZddlmZedZejeZ d e ed e e!e e"ffd Z#d e e$e%ffd Z&d e e$effdZ'ded e%fdZ( ddeegefee e$efgeffde e e$efde de e$de!d egeff dZ)Gddej*Z+GddZ,dS)N) AnyCallableContextManagerDictListOptionalTupleTypeVarUnion)find_free_portis_ipv6) ActorHandle)StartTracebackStartTracebackWithWorkerRank) RayActorError) ObjectRefT remote_valuesreturnc|}t|dkrtj|\}}|D]} tj|#t $rC}||}td|dd|fcYd}~cSd}~wt$r+}||}t||d}~wwxYwt|dkdS)a$Check for actor failure when retrieving the remote values. Args: remote_values: List of object references from Ray actor methods. Returns: A tuple of (bool, Exception). The bool is True if evaluating all object references is successful, False otherwise. rzWorker z has failed.FN) worker_rank)TN) copylenraywaitgetrindexloggerinfo Exceptionr)r unfinishedfinished object_refexcfailed_actor_rankfailed_worker_ranks m/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/_internal/utils.pycheck_for_failurer("s5##%%J j//A  "x 33*#  J  ####  " " "$1$7$7 $C$C! E&7EEEFFFcz!!!!!!!!   %2%8%8%D%D"2 2  j//A  * :s*A C#6B!C! C.&CCctj}tt |r t jn t j}||fS)z4Returns the IP address and a free port on this node.)rutilget_node_ip_addressr r socketAF_INET6AF_INET)addrports r'get_address_and_portr1Hs? 8 ' ' ) )D WT]]N&// O OD :env_varscd|D}tj|dS)zyUpdates the environment variables on this worker process. Args: env_vars: Environment variables to set. c4i|]\}}|t|S)str).0kvs r' z#update_env_vars..Us$888tq!CFF888r2N)itemsosenvironupdate)r3 sanitizeds r'update_env_varsrAOs= 98x~~'7'7888IJi     r2fnctj|j}tjjtjjtjjhtfd|DS)aCounts the number of required parameters of a function. NOTE: *args counts as 1 required parameter. Examples -------- >>> def fn(a, b, /, c, *args, d=1, e=2, **kwargs): ... pass >>> count_required_parameters(fn) 4 >>> fn = lambda: 1 >>> count_required_parameters(fn) 0 >>> def fn(config, a, b=1, c=2): ... pass >>> from functools import partial >>> count_required_parameters(partial(fn, a=0)) 1 cZg|]'}|jtjjk|jv%|(Sr6)defaultinspect Parameteremptykind)r8ppositional_param_kindss r' z-count_required_parameters..xsC   yG-333BX8X8X 8X8X8Xr2) rF signature parametersvaluesrGPOSITIONAL_ONLYPOSITIONAL_OR_KEYWORDVAR_POSITIONALr)rBparamsrKs @r'count_required_parametersrTYs. r " " - 4 4 6 6F )/(          r2 train_funcFconfigtrain_func_context fn_arg_namediscard_returnscN t}|r"tjfd}| n |dkr|d|d}t||dkr(intj  fd}n tj  fd}|S)aValidates and constructs the training function to execute. Args: train_func: The training function to execute. This can either take in no arguments or a ``config`` dict. config (Optional[Dict]): Configurations to pass into ``train_func``. If None then an empty Dict will be created. train_func_context: Context manager for user's `train_func`, which executes backend-specific logic before and after the training function. fn_arg_name (Optional[str]): The name of training function to use for error messages. discard_returns: Whether to discard any returns from train_func or not. Returns: A valid training function. Raises: ValueError: if the input ``train_func`` is invalid. cN |i|dS#t$r }t|d}~wwxYwNr r)argskwargserUs r'discard_return_wrapperz4construct_train_func..discard_return_wrappersJ , D+F+++++ , , ,$!+ ,s $$z: should take in 0 or 1 required arguments, but it accepts z required arguments instead.Nc 5cdddS#1swxYwYdS#t$r }t|d}~wwxYwr\r])r`rVrWwrapped_train_funcs r'train_fnz&construct_train_func..train_fns ,''))66--f55666666666666666666 , , ,$!+ ,s, 2 % 2)2)2 A AA c 5cdddS#1swxYwYdS#t$r }t|d}~wwxYwr\r])r`rWrds r'rez&construct_train_func..train_fns ,''))00--//000000000000000000 , , ,$!+ ,s, 1 $ 1(1(1 AAA)rT functoolswraps ValueError) rUrVrWrXrYnum_required_paramsraerr_msgrerds ``` @r'construct_train_funcrls6.4J?? (  $ $ , , , , % $ , 4'Q A A" A A A !!!  ! !~6 + , , , , , , , , - , , , + , , , , , , , - , , Or2c&eZdZdZiZfdZxZS) SingletonzSingleton Abstract Base Class https://stackoverflow.com/questions/33364070/implementing -singleton-as-metaclass-but-for-abstract-classes c||jvr(tt|j|i||j|<|j|Sr\) _instancessuperrn__call__)clsr^r_ __class__s r'rrzSingleton.__call__sE cn $ $"@% 3"7"7"@$"Q&"Q"QCN3 ~c""r2)__name__ __module__ __qualname____doc__rprr __classcell__)rts@r'rnrnsI J#########r2rnc$eZdZdZdefdZdZdS) ActorWrapperzDWraps an actor to provide same API as using the base class directly.actorc||_dSr\)r|)selfr|s r'__init__zActorWrapper.__init__s  r2c8t|j|fdS)NcBtjj|i|Sr\)rrremote)r^r_ actor_methods r'z*ActorWrapper.__getattr__..s$sw/B|/BD/SF/S/S'T'Tr2)getattrr|)r~itemrs @r' __getattr__zActorWrapper.__getattr__s%tz400 TTTTTr2N)rurvrwrxrrrr6r2r'r{r{sINNkUUUUUr2r{)rUF)-abcrgrFloggingr=r,typingrrrrrrr r r rray._common.network_utilsr r ray.actorrray.air._internal.utilrrray.exceptionsr ray.typesrr getLoggerrurboolr r(r7intr1rArTrlABCMetarnr{r6r2r'rs                         ========!!!!!!)((((( GCLL  8 $ $# ?# 4)$ $%####LeCHo!d38n!!!!$($s$$$$V".! CChr1uoxc3h0@!0C'DDEC T#s(^ $C'C# C  C b!e_ CCCCL # # # # # # # # U U U U U U U U U Ur2