&`iEddlZddlmZmZmZmZmZmZmZm Z ddl Z eegdfZ ee gdfZ GddZdS)N)AnyCallableDictIterableOptionalSetTupleUnioncLeZdZdZddefdZ ddejdee dee fd Z dd e ejdee dee fd Z dejfd Zd eejfdZed efdZdejfdZ ddeeeefdeefdZdS)RayEventManageraOEvent manager for Ray futures. The event manager can be used to track futures and invoke callbacks when they resolve. Futures are tracked with :meth:`track_future`. Future can then be awaited with :meth:`wait`. When futures successfully resolve, they trigger an optional ``on_result`` callback that can be passed to :meth:`track_future`. If they fail, they trigger an optional ``on_error`` callback. Args: shuffle_futures: If True, futures will be shuffled before awaited. This will avoid implicit prioritization of futures within Ray. Tshuffle_futuresc"||_i|_dSN)_shuffle_futures_tracked_futures)selfr s }/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/air/execution/_internal/event_manager.py__init__zRayEventManager.__init__s /  Nfuture on_resulton_errorc||f|j|<dS)aTrack a single future and invoke callbacks on resolution. Control has to be yielded to the event manager for the callbacks to be invoked, either via :meth:`wait` or via :meth:`resolve_future`. Args: future: Ray future to await. on_result: Callback to invoke when the future resolves successfully. on_error: Callback to invoke when the future fails. N)r)rrrrs r track_futurezRayEventManager.track_future"s"*3H(=f%%%rfuturesc@|D]}||||dS)aTrack multiple futures and invoke callbacks on resolution. Control has to be yielded to the event manager for the callbacks to be invoked, either via :meth:`wait` or via :meth:`resolve_future`. Args: futures: Ray futures to await. on_result: Callback to invoke when the future resolves successfully. on_error: Callback to invoke when the future fails. )rrN)r)rrrrrs r track_futureszRayEventManager.track_futures5s?" N NF   f H  M M M M N Nrc<|j|ddS)zRemove future from tracking. The future will not be awaited anymore, and it will not trigger any callbacks. Args: future: Ray futures to discard. N)rpop)rrs rdiscard_futurezRayEventManager.discard_futureIs# !!&$/////rreturnc*t|jS)z)Get futures tracked by the event manager.)setrrs r get_futureszRayEventManager.get_futuresS4()))rc*t|jSr)lenrr$s r num_futureszRayEventManager.num_futuresWr&rc( |j|\}}n&#t$r}td|d|d}~wwxYw t j|}|r ||dSdS#t $r}|r ||n|Yd}~dSd}~wwxYw)aResolve a single future. This method will block until the future is available. It will then trigger the callback associated to the future and the event (success or error), if specified. Args: future: Ray future to resolve. zFuture z' is not tracked by this RayEventManagerN)rrKeyError ValueErrorrayget Exception)rrrreresults rresolve_futurezRayEventManager.resolve_future[s "&"7";";F"C"C Ixx   I&III   "WV__F " &!!!!! " "        s* A>AA,, B6B  Btimeout num_resultsc"t|}|jrtj||pt |}t jt|||\}}|D]}||dS)a8Wait up to ``timeout`` seconds for ``num_results`` futures to resolve. If ``timeout=None``, this method will block until all `num_results`` futures resolve. If ``num_results=None``, this method will await all tracked futures. For every future that resolves, the respective associated callbacks will be invoked. Args: timeout: Timeout in second to wait for futures to resolve. num_results: Number of futures to await. If ``None``, will wait for all tracked futures to resolve. )r4 num_returnsN) listr%rrandomshuffler(r-waitr2)rr4r5rready_rs rr;zRayEventManager.waitxs&t''))**   $ N7 # # #!1S\\ 8DMM7 TTTq ( (F    ' ' ' ' ( (r)T)NN)Nr3)__name__ __module__ __qualname____doc__boolrr- ObjectRefr_ResultCallback_ErrorCallbackrrrr rr%propertyintr)r2r floatr;rrr r s  04-1 >> >O,>>* >>>>,04-1 NN#-(NO,N>* NNNN(0S]0000*S/*****S***X*"S]"""">04%&((%s +,(c]((((((rr )r9typingrrrrrrr r r-rDr/rEr rIrrrKs MMMMMMMMMMMMMMMMMMMM C5$;'9+t+,J(J(J(J(J(J(J(J(J(J(r