&`i!ddlZddlmZmZmZddlZddlmZmZm Z m Z ddl m Z deede fdZe d Gd d Ze d Gd d ZdS)N)AnyListOptional)GetTimeoutErrorRayChannelErrorRayChannelTimeoutError RayTaskError) PublicAPI return_valsreturn_single_outputct|tr||D]+}t|tr|,|rt |dksJ|dS|S)a! Process return values for return to the DAG caller. Any exceptions found in return_vals will be raised. If return_single_output=True, it indicates that the original DAG did not have a MultiOutputNode, so the DAG caller expects a single return value instead of a list. r) isinstance Exceptionr as_instanceof_causelen)r r vals u/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/experimental/compiled_dag_ref.py_process_return_valsrs+y)),, c< ( ( ,))++ + ,;1$$$$1~ alpha) stabilitycneZdZdZ ddddedeefdZdZd Zd Z d Z d Z dd ee fdZ dS)CompiledDAGRefa A reference to a compiled DAG execution result. This is a subclass of ObjectRef and resembles ObjectRef. For example, similar to ObjectRef, ray.get() can be called on it to retrieve the result. However, there are several major differences: 1. ray.get() can only be called once per CompiledDAGRef. 2. ray.wait() is not supported. 3. CompiledDAGRef cannot be copied, deep copied, or pickled. 4. CompiledDAGRef cannot be passed as an argument to another task. Ndagray.experimental.CompiledDAGexecution_index channel_indexcV||_||_||_d|_|j|_dS)a Args: dag: The compiled DAG that generated this CompiledDAGRef. execution_index: The index of the execution for the DAG. A DAG can be executed multiple times, and execution index indicates which execution this CompiledDAGRef corresponds to. actor_execution_loop_refs: The actor execution loop refs that are used to execute the DAG. This can be used internally to check the task execution errors in case of exceptions. channel_index: The index of the DAG's output channel to fetch the result from. A DAG can have multiple output channels, and channel index indicates which channel this CompiledDAGRef corresponds to. If channel index is not provided, this CompiledDAGRef wraps the results from all output channels. FN)_dag_execution_index_channel_index_ray_get_calleddag_output_channels_dag_output_channels)selfrrrs r__init__zCompiledDAGRef.__init__2s4, /+$$'$;!!!rc\d|jd|jd|jdS)NzCompiledDAGRef(, execution_index=, channel_index=)r get_idr!r"r&s r__str__zCompiledDAGRef.__str__OsK 4di..00 4 4#4 4 4!0 4 4 4 rc td)Nz CompiledDAGRef cannot be copied. ValueErrorr.s r__copy__zCompiledDAGRef.__copy__Vs;<<>>>$## % % %J%% 0 0 0&4/ 0      #K666sHA A++D?+C5+CC5 C2 C C2*C--C22C55Dr;)__name__ __module__ __qualname____doc__intrr'r/r3r7r9r>floatrHrrrr$s   (, << +<< } <<<<:   ===BBB>>> X X X,7,78E?,7,7,7,7,7,7rrc ^eZdZdZ ddddedddeefd Zd Zd Zd Z d Z dZ dZ dS)CompiledDAGFuturea A reference to a compiled DAG execution result, when executed with asyncio. This differs from CompiledDAGRef in that `await` must be called on the future to get the result, instead of `ray.get()`. This resembles async usage of ObjectRefs. For example, similar to ObjectRef, `await` can be called directly on the CompiledDAGFuture to retrieve the result. However, there are several major differences: 1. `await` can only be called once per CompiledDAGFuture. 2. ray.wait() is not supported. 3. CompiledDAGFuture cannot be copied, deep copied, or pickled. 4. CompiledDAGFuture cannot be passed as an argument to another task. Nrrrfutzasyncio.Futurerc>||_||_||_||_dSr;)r r!_futr")r&rrrVrs rr'zCompiledDAGFuture.__init__s( / +rc\d|jd|jd|jdS)NzCompiledDAGFuture(r)r*r+r,r.s rr/zCompiledDAGFuture.__str__sK 4!1!1!3!3 4 4#4 4 4!0 4 4 4 rc td)Nz#CompiledDAGFuture cannot be copied.r1r.s rr3zCompiledDAGFuture.__copy__s>???rc td)Nz(CompiledDAGFuture cannot be deep copied.r1r5s rr7zCompiledDAGFuture.__deepcopy__sCDDDrc td)Nz$CompiledDAGFuture cannot be pickled.r1r.s rr9zCompiledDAGFuture.__reduce__s?@@@rc#K|jtd|j}d|_|j|jsO|Ed{V}|jxjdz c_|j|j||j|j|j }t|dS)NzVCompiledDAGFuture can only be awaited upon once, and it has already been awaited upon.rT) rXr2r _has_execution_resultsr! __await___max_finished_execution_index_cache_execution_resultsrCr"r)r&rVresultr s rr_zCompiledDAGFuture.__await__s 9 - i y//0EFF N # //////F I 3 3q 8 3 3 I . .t/Df M M Mi66  !4#6  $K666rc~|jjrdS|jdS|j|j|jdSr;)r r<rXr=r!r"r.s rr>zCompiledDAGFuture.__del__sE 9   F 9  F ++D,A4CVWWWWWrr;) rMrNrOrPrQrr'r/r3r7r9r_r>rSrrrUrUs  &(, , , + , , ,  } , , , ,   @@@EEEAAA777.XXXXXrrU)asynciotypingrrrrGray.exceptionsrrrr ray.util.annotationsr boolrrrUrSrrrisV&&&&&&&&&&  +*****d3it, Wq7q7q7q7q7q7q7q7h WJXJXJXJXJXJXJXJXJXJXr