%`i9UdZddlZddlmZddlmZmZmZmZm Z m Z ddl Z ddl m Z ddlmZmZddlmZddlmZdd lmZmZmZeGd d ZdKd ededefdZdZdLd edefdZdeeeefdeefdZ idee!e"dfddee#e"dfddeee"dfdedd eee"dfdeddeddeddd ee"deefd!"d#ee$d$"d%eee"dfdee!e"dfd&d'ee!e"dfd(ee"deeeefd)eed "d*ee!e"dfZ%d+ed,e"defd-Z&ed.dd"ed/e j'"edd0"ee$ee"dfd1d"ee$e"dfd2eee#e(fd3d"ee$e"dfd4d5Z)ee#e!e"dfeee"dfd"eee"dfd6ed7ded8d"ed9d"ed:d$"eee"dfeed"eee"dfd; Z*ie)e*e%Z+eeefe,d<<ie%e)Z-eeefe,d=<ie%e*Z.eeefe,d><d?e#e+/d@Z0dAeeeffdBZ1dAeeefdCe$fdDZ2dAeeefdEefdFZ3dAeeefdEefdGZ4dHeeefdIeeefdeeeffdJZ5dS)MzKManage, parse and validate options for Ray tasks, actors and actor methods.N) dataclass)AnyCallableDictOptionalTupleUnion) ray_constants)validate_fallback_strategyvalidate_label_selector)get_ray_doc_version)PlacementGroup)NodeAffinitySchedulingStrategyNodeLabelSchedulingStrategy PlacementGroupSchedulingStrategyceZdZUdZeeeeefed<dZ ee e gee fed<dZ e ed<de de fdZdS)OptionNtype_constraintvalue_constraint default_valuekeywordvaluec |j?t||js*td|d|jdt||j&||}|rt |dSdS)zValidate the option.NzThe type of keyword 'z ' must be z, but received type )r isinstance TypeErrortyper ValueError)selfrrpossible_error_messages p/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/_common/ray_option_utils.pyvalidatezOption.validate s   +eT%9:: 7G77t?S77)-e77  ,%)%:%:5%A%A "% 9 !7888 - , 9 9)__name__ __module__ __qualname__rrr rr__annotations__rrrstrrr!r"r rrs;?OXeD%+$567>>>BFhxx}(<=>EEEM3 9 9C 9 9 9 9 9 9r"rTnameinfiniterc|r)tttdffd|Stttdffd|S)aThis is used for positive and discrete options. Args: name: The name of the option keyword. infinite: If True, user could use -1 to represent infinity. default_value: The default value for this option. Returns: An Option object. Nc$||dkrdnddS)N The keyword 'zO' only accepts None, 0, -1 or a positive integer, where -1 represents infinity.r(xr)s r z"_counting_option..<s6 Q"WWdDDDDr"rc$||dkrdnddS)Nrr.z-' only accepts None, 0 or a positive integer.r(r/s r r1z"_counting_option..Ds* Ia$ PT P P Pr")rintr)r)r*rs` r _counting_optionr5.s  $t**  D D D D(       d4jj Q Q Q Q$    r"c|dkrd|dSt|tr1|dkr+t|tjjzdkrd|dS|dkrdn|}|tjjvr>tjj | |\}}|s|SdS) NrzThe quantity of resource z cannot be negativegz5The precision of the fractional quantity of resource z cannot go beyond 0.0001num_gpusGPU) rfloatr4ray_rayletRESOURCE_UNIT_SCALING_private accelerators"get_all_accelerator_resource_names$get_accelerator_manager_for_resource"validate_resource_request_quantity)r)quantity resource_namevalid error_messages r _validate_resource_quantityrFKs!||D4DDDD8U## OO 3;<< = = B B 'D ' ' ' "Z//EETM 1TTVVVV L % J J   , ,       ! 4r"cbttttdffd|S)z*This is used for resource related options.Nc,|dnt|SN)rFr/s r r1z"_resource_option..js19$$+FtQ+O+Or"r2)rr9r4r)r)rs` r _resource_optionrJfs9  T$ZZ OOOO#   r" resourcesreturnc|dSd|vsd|vr dS|D]\}}t||}|r|cSdS)NCPUr8z[Use the 'num_cpus' and 'num_gpus' keyword instead of 'CPU' and 'GPU' in 'resources' keyword)itemsrF)rKr)rBrs r _validate_resourcesrPost Ui// %  $//++**h!z..s,CC2I66CCCCCCr"zAretry_exceptions must be either a boolean or a list of exceptions)rboollisttupleallrSs r r1r1s_ q$  Q1tUm,, Q CCCCCCC Q$$Qr"c|dkrdndS)Nrz_generator_backpressure_num_objects=0 is not allowed. Use a value > 0. If the value is equal to 1, the behavior is identical to Python generator (generator 1 object whenever `next` is called). Use -1 to disable this feature. r(rSs r r1r1s 66$ Kr")rhrirX num_returnsrYretry_exceptions#_generator_backpressure_num_objectsc|dvrdndS)N)Ndetached non_detachedzOactor `lifetime` argument must be one of 'detached', 'non_detached' and 'None'.r(rSs r r1r1s 2 2 2$%r"max_concurrency max_restartsmax_task_retriesmax_pending_calls) concurrency_groupsenable_tensor_transportlifetimerrrr namespace get_if_existsallow_out_of_order_execution valid_options task_options actor_optionszThe @ray.remote decorator must be applied either with no arguments and no parentheses, for example '@ray.remote', or it must be applied using some of the arguments in the list zL, for example '@ray.remote(num_returns=2, resources={"CustomResource": 1})'.optionsc|dd}|d}|dvr|tddSdS)z2Check if deprecated placement group option exists.rZr[r`)r[NNzrPlacement groups should be specified via the scheduling_strategy option. The placement_group option is deprecated.)getr)rrZr`s r _check_deprecate_placement_groupr sbkk"3Y??O!++&;<< 0007J7V 8    107V7Vr"caller_stacklevelcp|d}|d}|d}|dkr/tjdtdt|dz|d kr/tjd tdt|dz|r1tjd tdt|dzdSdS) NrZr\r]r[zplacement_group parameter is deprecated. Use scheduling_strategy=PlacementGroupSchedulingStrategy(...) instead, see the usage at https://docs.ray.io/en/z&/ray-core/package-ref.html#ray-remote.rj stacklevelr-zplacement_group_bundle_index parameter is deprecated. Use scheduling_strategy=PlacementGroupSchedulingStrategy(...) instead, see the usage at https://docs.ray.io/en/zplacement_group_capture_child_tasks parameter is deprecated. Use scheduling_strategy=PlacementGroupSchedulingStrategy(...) instead, see the usage at https://docs.ray.io/en/)warningswarnr DeprecationWarning)rrrZr\r]s r )_warn_if_using_deprecated_placement_grouprs8/0O#*+I#J *12W*X')##  d':&;&; d d d (1,     $r))  d':&;&; d d d (1,     +   d':&;&; d d d (1,         r" in_optionsc &|D]W\}}|tvr(td|dttdt|||X|rd|vrtdt |dS)zOptions check for Ray tasks. Args: options: Options for Ray tasks. in_options: If True, we are checking the options under the context of ".options()". Invalid option keyword z& for remote functions. Valid ones are .rhz5Setting 'max_calls' is not supported in '.options()'.N)rOrrrvr!rrrkvs r validate_task_optionsr?s ''1 L 8!88"&|"4"4888  Q  A&&&&RkW,,PQQQ$W-----r"c |D]W\}}|tvr(td|dttdt|||X|rd|vrtd|dr$|dstdd |vr,t jd td td t|dS)zOptions check for Ray actors. Args: options: Options for Ray actors. in_options: If True, we are checking the options under the context of ".options()". rz for actors. Valid ones are rrz>Setting 'concurrency_groups' is not supported in '.options()'.rr)z8The actor name must be specified to use `get_if_exists`.rYzSetting 'object_store_memory' for actors is deprecated since it doesn't actually reserve the required object store memory. Use object spilling that's enabled by default (https://docs.ray.io/en/zb/ray-core/objects/object-spilling.html) instead to bypass the object store memory size limitation.rjrN) rOrrrvr!rrrr rrrs r validate_actor_optionsrSsP ((1 M ! !9!99"&}"5"5999  a!!!Q'''' *g55 L   {{?##UGKK,?,?USTTT''  IWjVkVk I I I     %W-----r"original_options new_optionsci||S)zUpdate original options with new options and return. The returned updated options contain shallow copy of original options. r()rrs r update_optionsrys / .+ ..r")TNrI)6__doc__r dataclassesrtypingrrrrrr r: ray._privater ray._private.label_utilsr r ray._private.utilsr ray.util.placement_grouprray.util.scheduling_strategiesrrrrr'rur5rFrJr9rPdictrrvr4_common_optionsrgDEFAULT_TASK_MAX_RETRIESrw_task_only_options_actor_only_optionsrr&rrkeysremote_args_error_stringrrrrrr(r"r rsQQQ!!!!!!>>>>>>>>>>>>>>>> &&&&&&322222333333 9999999 903$c:63s8De,<#=(3-$ ffdDDJJ/1U1UVV  ttDzzCC  TT$ZZ011   x((  FFCd$ % %   ,,   ,, ++,A5II vv dS.) "#FF3b$A$A$A# $*664d2D+E+E% &ttDzz*,L,LMM' (664d,--) *66 DJJ  , * '  + <&&T:::= >vvtTT$ZZ())? FDT"!+uAFFF$#]%K! 1===6 c44::     "6 dd4jj J J  tU Q Q   ,26 dd4jj   , ,U55p!&$dd4jj!9::%vtTT$ZZ&8MMM dd4jj % %('(95AA$$^1EEE(();1MMM))*=RPPPdd4jj)**VD666$*FD$$t**+=$>$>!0$$$$ tCK  #L_"K8J"K d3;KKK#Mo#M9L#M tCK MMMG!%m&8&8&:&:!;!;GGG  d38n      #s(^  03     F.4S>.t....(#.DcN#.#.#.#.#.L/38n/37S>/ #s(^//////r"