&`i,ddlmZmZmZmZmZmZddlmZddl Z ddl m Z m Z m Z mZddlmZddlmZddlmZddlmZeGd d eZGd d eZGd dZeGddeZdS))AnyDictListOptionalTupleUnion) ReferenceTypeN)BIND_INDEX_KEYIS_CLASS_METHOD_OUTPUT_KEYPARENT_CLASS_NODE_KEYPREV_CLASS_METHOD_CALL_KEY)DAGNode)get_dag_node_str) InputNode) DeveloperAPIc eZdZdZ dfd ZdeedeeefdeeefdeeeffdZ d Z d e fd Z d efd Z d efdZxZS) ClassNodez/Represents an actor creation in a Ray task DAG.Nc||_d|_t|||||rt ddS)Nother_args_to_resolvezInputNode handles user dynamic input the DAG, and cannot be used as args, kwargs, or other_args_to_resolve in ClassNode constructor because it is not available at class construction or binding time.)_body _last_callsuper__init___contains_input_node ValueError)selfclscls_args cls_kwargs cls_optionsr __class__s f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/dag/class_node.pyrzClassNode.__init__sw 7;    "7      $ $ & & 6   new_args new_kwargs new_optionsnew_other_args_to_resolvec4t|j||||SNr)rrrr%r&r'r(s r# _copy_implzClassNode._copy_impl.s, J   ";     r$c~tj|jjdi|jj|ji|jS)a6Executor of ClassNode by ray.remote() Args and kwargs are to match base class signature, but not in the implementation. All args and kwargs should be resolved and replaced with value in bound_args and bound_kwargs via bottom-up recursion when current node is executed. )rayremoteroptions_bound_options _bound_args _bound_kwargs)rargskwargss r# _execute_implzClassNode._execute_impl=sY  CJtz " "  , ,* , , T% =)-); = = r$returnch|}|D]}t|trdSdS)z_Check if InputNode is used in children DAGNodes with current node as the root. TF)_get_all_child_nodes isinstancer)rchildren_dag_nodeschilds r#rzClassNode._contains_input_nodeKsH"6688'  E%++ tt ur$ method_namec|dkr6dt|jvr tdt|dt |j|t ||i}|S)Nbindz .bind() cannot be used again on  )dirrAttributeErrortypegetattr_UnboundClassMethodNode)rr> call_nodes r# __getattr__zClassNode.__getattr__Ush & V3tz??%B%B !QDJJ!Q!Q!QRR R K(((+D+rBB r$cFt|t|jSN)rstrrrs r#__str__zClassNode.__str__^sc$*oo666r$rJ)__name__ __module__ __qualname____doc__rrrrrKr,r7boolrrHrM __classcell__r"s@r#rrs99# 2  s)  cN  #s(^  $(S>         ds777777777r$rc:eZdZdededefdZdZdefdZdZ d S) rFactorr>r1c0||_||_||_dSrJ)_actor _method_name_options)rrVr>r1s r#rz _UnboundClassMethodNode.__init__cs ' r$ct|jt|jji}t |j|||j|}||j_|Sr*)r rXr rClassMethodNoderYrZ)rr5r6rnodes r#r@z_UnboundClassMethodNode.bindksV !4; & (>!       M"7    "&  r$attrcV|dkrtd||S)Nr0zd.remote() cannot be used on ClassMethodNodes. Use .bind() instead to express an symbolic actor call.)rC__getattribute__)rr^s r#rHz#_UnboundClassMethodNode.__getattr__{s: 8   5  ((.. .r$c ||_|SrJ)rZ)rr1s r#r1z_UnboundClassMethodNode.optionss  r$N) rNrOrPrrKdictrr@rHr1r.r$r#rFrFbsv i c D     /////r$rFcVeZdZdZdddefdZed dZedefdZd S) _ClassMethodOutputz7Represents a class method output in a Ray function DAG.class_method_callr\ output_idxc"||_||_dSrJ)_class_method_call _output_idx)rrerfs r#rz_ClassMethodOutput.__init__s"3%r$r8c|jSrJ)rhrLs r#rez$_ClassMethodOutput.class_method_calls &&r$c|jSrJ)rirLs r#rfz_ClassMethodOutput.output_idxs r$N)r8r\) rNrOrPrQintrpropertyrerfr.r$r#rdrdsAA&*;&&&&& '''X' C   X   r$rdc eZdZdZdedeedeeefdeeefdeeeff fd Zde ed eeefd eeefd eeeffd Z d Z defdZ defdZ defdZdefdZdZdedfdZedefdZedefdZedefdZededfdZedeefdZxZS)r\z method_args method_kwargsmethod_optionsrc|pg|_|pi|_|pi|_||_|t |_|td|_|td|_ |td|_ d|_ |j r!t|d|d|_ t||||dS)NFrr)r3r4r2rYgetr _parent_class_noder _prev_class_method_callr _bind_indexr _is_class_method_output_class_method_outputrdrr)rr>rorprqrr"s r#rzClassMethodNode.__init__s',"*0b,2!, " % %&; < <  " % %&@$ G G $+@*C*C D+ +  .C-F-F &. . $ CG!  ' );A A))D %    "7      r$r%r&r'r(c4t|j||||Sr*)r\rYr+s r#r,zClassMethodNode._copy_impls-     ";     r$c|jrAt|j|j}|jdi|jj|ji|jS|j J|jd|j j S)a<Executor of ClassMethodNode by ray.remote() Args and kwargs are to match base class signature, but not in the implementation. All args and kwargs should be resolved and replaced with value in bound_args and bound_kwargs via bottom-up recursion when current node is executed. Nrr.) is_class_method_callrErurYr1r2r0r3r4ryrf)rr5r6 method_bodys r#r7zClassMethodNode._execute_impls  $ M!$"94;LMMKD&;&==)<==D!$  ,888#A&t'@'KL Lr$r8c2t||jdS)Nz())rrYrLs r#rMzClassMethodNode.__str__s):&>&>&>???r$c*|SrJ)rMrLs r#__repr__zClassMethodNode.__repr__s||~~r$c|jSrJ)rYrLs r#get_method_namezClassMethodNode.get_method_names   r$c|jSrJ)rwrLs r#_get_bind_indexzClassMethodNode._get_bind_indexs r$c0t|j|}|SrJ)rEru)rr>r}s r#_get_remote_methodz"ClassMethodNode._get_remote_methodsd5{CC r$zray.actor.ActorHandlec\t|jtjjsdS|jSrJ)r;rur/rV ActorHandlerLs r#_get_actor_handlez!ClassMethodNode._get_actor_handles*$1393HII 4&&r$c|jrS|jdd}|4||j}|d}|S|jJ|jjjS)z Return the number of return values from the class method call. If the node is a class method output, return the number of return values from the upstream class method call. num_returnsN) r|r2rtrrY __getstate__ryrer)rrmethods r#rzClassMethodNode.num_returns s~  $ K-11-FFK"001BCC$1133MB  ,888,>J Jr$c|j S)zu Return True if the node is a class method call, False if the node is a class method output. rxrLs r#r|z$ClassMethodNode.is_class_method_calls ///r$c|jS)zu Return True if the node is a class method output, False if the node is a class method call. rrLs r#is_class_method_outputz&ClassMethodNode.is_class_method_output#s ++r$c,|jdS|jjS)z Return the upstream class method call that returns multiple values. If the node is a class method output, return None. N)ryrerLs r#rez!ClassMethodNode.class_method_call+s  $ ,4(::r$c,|jdS|jjS)z Return the output index of the return value from the upstream class method call that returns multiple values. If the node is a class method call, return None. N)ryrfrLs r#rfzClassMethodNode.output_idx6s  $ ,4(33r$)rNrOrPrQrKrrrrrr,r7rMrrrlrrrrrmrrRr|rrerfrSrTs@r#r\r\skFF1 1 3Z1 CH~ 1 S#X 1 $CH~ 1 1 1 1 1 1 f  s)  cN  #s(^  $(S>     MMM&@@@@@#!!!!!     '8,C#D'''' KSKKKXK"0d000X0,,,,X,;8,=#>;;;X; 4HSM 4 4 4X 4 4 4 4 4r$r\)typingrrrrrrweakrefr r/ray.dag.constantsr r r r ray.dag.dag_noderray.dag.format_utilsrray.dag.input_noderray.util.annotationsrrobjectrFrdr\r.r$r#rs::::::::::::::::!!!!!!  %$$$$$111111((((((------M7M7M7M7M7M7M7M7`$$$$$f$$$N        $d4d4d4d4d4gd4d4d4d4d4r$