&`i+ddlmZmZmZmZmZddlmZddlm Z ddl m Z ddl m Z dZe GddeZe Gd d eZe Gd d Zd S))AnyDictListOptionalUnion)DAGNodeget_dag_node_str)type_to_string) DeveloperAPI__in_context_manager__c 2eZdZdZddddeeeeeee feffffdZ de e dee e fdee e fd ee e ffd Z d Zd efd Zde de fdZd e fdZde fdZdeee fd e fdZdZdZd e fdZxZS) InputNodeaoRay dag node used in DAG building API to mark entrypoints of a DAG. Should only be function or class method. A DAG can have multiple entrypoints, but only one instance of InputNode exists per DAG, shared among all DAGNodes. Example: .. code-block:: m1.forward / \ dag_input ensemble -> dag_output \ / m2.forward In this pipeline, each user input is broadcasted to both m1.forward and m2.forward as first stop of the DAG, and authored like .. code-block:: python import ray @ray.remote class Model: def __init__(self, val): self.val = val def forward(self, input): return self.val * input @ray.remote def combine(a, b): return a + b with InputNode() as dag_input: m1 = Model.bind(1) m2 = Model.bind(2) m1_output = m1.forward.bind(dag_input[0]) m2_output = m2.forward.bind(dag_input.x) ray_dag = combine.bind(m1_output, m2_output) # Pass mix of args and kwargs as input. ray_dag.execute(1, x=2) # 1 sent to m1, 2 sent to m2 # Alternatively user can also pass single data object, list or dict # and access them via list index, object attribute or dict key str. ray_dag.execute(UserDataObject(m1=1, m2=2)) # dag_input.m1, dag_input.m2 ray_dag.execute([1, 2]) # dag_input[0], dag_input[1] ray_dag.execute({"m1": 1, "m2": 2}) # dag_input["m1"], dag_input["m2"] N) input_type_other_args_to_resolverc2t|dkst|dkrtdi|_||_|+t |t r|i}t ||d<tgii|dS)aInputNode should only take attributes of validating and converting input data rather than the input data itself. User input should be provided via `ray_dag.execute(user_input)`. Args: input_type: Describes the data type of inputs user will be giving. - if given through singular InputNode: type of InputNode - if given through InputAttributeNodes: map of key -> type Used when deciding what Gradio block to represent the input nodes with. _other_args_to_resolve: Internal only to keep InputNode's execution context throughput pickling, replacement and serialization. User should not use or pass this field. rz-InputNode should not take any args or kwargs.Nresult_type_string)other_args_to_resolve) len ValueErrorinput_attribute_nodesr isinstancetyper super__init__)selfrrargskwargs __class__s f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/dag/input_node.pyrzInputNode.__init__Cs( t99>>S[[A--LMM M%'"$  !jT&B&B !%-)+&;I*;U;U "#7 8 R;QRRRRRnew_args new_kwargs new_optionsnew_other_args_to_resolvec"t|S)N)r)rrr"r#r$r%s r _copy_implzInputNode._copy_implds0IJJJJr!c|s Jdt|dkrt|dkr|dSt|i|S)zExecutor of InputNode.zInputNode is a singleton instance that should be only used in context manager for dag building and execution. See the docstring of class InputNode for examples.r)_in_context_managerr DAGInputDatarrrs r _execute_implzInputNode._execute_implmsl''))   /  ) t99>>c&kkQ..7NT,V,,,r!returncT|jrt|jvrdS|jtS)z2Return if InputNode is created in context manager.F)_bound_other_args_to_resolveIN_CONTEXT_MANAGERrs r r+zInputNode._in_context_manager{s21 I!)JJJ545GH Hr!keyvalc||j|<dS)zqSet field in parent DAGNode attribute that can be resolved in both pickle and JSON serialization Nr1)rr4r5s r set_contextzInputNode.set_contexts25)#...r!c"t|dS)N __InputNode__r r3s r __str__zInputNode.__str__so666r!ct|ts Jd||jvrt||d|j|<|j|S)Nz5Please only access dag input attributes with str key. __getattr__)rstrrInputAttributeNoderr4s r r=zInputNode.__getattr__sr    C C B C C  d0 0 0.@c=//D &s +)#..r!ct|ttfs Jdd}|j#||jvrt |j|}||jvrt ||d||j|<|j|S)NRPlease only use int index or str as first-level key to access fields of dag input. __getitem__)rr>intrr rr?)rr4rs r rCzInputNode.__getitem__s#Sz**   *  *  ? &3$/+A+A'(<==J d0 0 0.@c=*//D &s +)#..r!c<|td|S)NT)r8r2r3s r __enter__zInputNode.__enter__s +T222 r!cdSN)rrs r __exit__zInputNode.__exit__s r!c2d|jvr |jdSdSzvGet type of the output of this DAGNode. Generated by ray.experimental.gradio_utils.type_to_string(). rNr7r3s r get_result_typezInputNode.get_result_type* 4#D D D45IJ J E Dr!)__name__ __module__ __qualname____doc__rrrrrDr>rrrr(r.boolr+r8r;r=rCrFrJrM __classcell__rs@r rr s44rJN# SSSU4eCHot.C)D#DEFSSSSSSBKs)KcNK#s(^ K $(S> KKKK - - -ITIIII5s55555 77777/s/////uS#X/3////    KKKKKKKKKr!rc eZdZdZ ddedeeefdedeffd Zde e d e ee fd e ee fd e ee ffd Z d Z defdZdefdZedeeeffdZxZS)r?a5Represents partial access of user input based on an index (int), object attribute or dict key (str). Examples: .. code-block:: python with InputNode() as dag_input: a = dag_input[0] b = dag_input.x ray_dag = add.bind(a, b) # This makes a = 1 and b = 2 ray_dag.execute(1, x=2) with InputNode() as dag_input: a = dag_input[0] b = dag_input[1] ray_dag = add.bind(a, b) # This makes a = 2 and b = 3 ray_dag.execute(2, 3) # Alternatively, you can input a single object # and the inputs are automatically indexed from the object: # This makes a = 2 and b = 3 ray_dag.execute([2, 3]) Ndag_input_noder4accessor_methodrc ||_||_||_tgii||||ddS)N)rWr4rXr)_dag_input_node_key_accessor_methodrr)rrWr4rXrrs r rzInputAttributeNode.__init__sc . /    "0#2'1   r!r"r#r$r%cVt|d|d|d|dS)NrWr4rXr)r?r's r r(zInputAttributeNode._copy_impls9" %&6 7 %e , %&7 8 %&: ;    r!c|t|jtr|j|jS|j}t|jtr:|jdkr ||jS|jdkrt ||jSdSt|jtr ||jStd)a0Executor of InputAttributeNode. 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. rCr=rBN) rrZr,r[r>r\getattrrDr)rrruser_input_python_objects r r.z InputAttributeNode._execute_impls d*L 9 9 ' 2 2 (,'; $$)S)) (M993DI>>*m;;"#;TYGGG<;DIs++ / :: 2r!r/c4t|d|jdS)Nz["z"])r r[r3s r r;zInputAttributeNode.__str__s &849&8&8&8999r!c2d|jvr |jdSdSrLr7r3s r rMz"InputAttributeNode.get_result_typerNr!c|jSrH)r[r3s r r4zInputAttributeNode.key&s yr!rH)rOrPrQrRrrrDr>rrrrr(r.r;rMpropertyr4rTrUs@r r?r?sOD   ! 38_          2  s)  cN  #s(^  $(S>     ::::::KKKKKU38_Xr!r?c8eZdZdZdZdeeefdefdZ dS)r,zIf user passed multiple args and kwargs directly to dag.execute(), we generate this wrapper for all user inputs as one object, accessible via list index or object attribute key. c<t||_||_dSrH)list_args_kwargsr-s r rzDAGInputData.__init__2s$ZZ  r!r4r/ct|tr |j|St|tr |j|St d)NrB)rrDrhr>rirr@s r rCzDAGInputData.__getitem__6sV c3   :c? " S ! ! <$ $. r!N) rOrPrQrRrrrDr>rrCrIr!r r,r,+sW  uS#X 3      r!r,N)typingrrrrrray.dagrray.dag.format_utilsr ray.experimental.gradio_utilsr ray.util.annotationsr r2rr?r,rIr!r rpsA33333333333333111111888888-------iKiKiKiKiKiKiKiKXoooooooodr!