%`i8ddlZddlZddlZddlZddlZddlZddlZddlmZddl m Z m Z m Z m Z ddlZddlmZmZmZddlmZmZddlmZmZddlmZddlmZmZej e!Z"d Z#eeGd d eZ$eGd d Z%Gdde%Z&de'de e'e'ffdZ(de e'de%fdZ)e dde e'de*de%fdZ+dS)N) dataclass)AnyDictOptionalTuple) RAY_ADDRESS_ENVIRONMENT_VARIABLE"RAY_NAMESPACE_ENVIRONMENT_VARIABLE$RAY_RUNTIME_ENV_ENVIRONMENT_VARIABLE)get_ray_client_dependency_error split_address) BaseContextinit) JobConfig) Deprecated PublicAPIzYhttps://docs.ray.io/en/latest/cluster/running-applications/job-submission/ray-client.htmlceZdZUdZeeed<eed<eed<eed<eed<eej j j ed<dd Z dd Z dd Zd Zdedd fdZd S) ClientContextz@ Basic context manager for a ClientBuilder connection. dashboard_urlpython_version ray_version ray_commit _num_clients_context_to_restorereturnc.||SN) _swap_contextselfs f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/client_builder.py __enter__zClientContext.__enter__-s  NcX|d|dS)NF)_disconnect_with_contextr)rexcs r __exit__zClientContext.__exit__1s. %%e,,, r"c||d|dSNT)rr$rs r disconnectzClientContext.disconnect5s@  %%d+++ r"c~|j5tjjj|j|_dSdSr)rrayutilclient set_contextrs r rzClientContext._swap_context:s>  # /'*x':'F'F(((D $ $ $ 0 /r"force_disconnectctjjjrQtjjjs|r%tjjdSdStjjj j dStjjj j rt ddStjdS)z Disconnect Ray. If it's a ray client and created with `allow_multiple`, it will do nothing. For other cases this either disconnects from the remote Client Server or shuts the current driver down. NzqThe current Ray Cluster is scoped to this process. Disconnecting is not possible as it will shutdown the cluster.)r+r,r- is_connected is_defaultclient_connectr)_privateworker global_workernodeis_headloggerdebugshutdown)rr/s r r$z&ClientContext._disconnect_with_context@s 8?  + + - - x"--// 53C 5'2244444 5 5\ . 3 ; F \ . 3 ; ; = =  LL      LNNNNNr")rrrN)__name__ __module__ __qualname____doc__rstr__annotations__intr+r,r- RayAPIStubr!r&r)rboolr$r"r rrs C=   OOO!#(/"<====  $r"rceZdZdZdeeddfdZdeeefddfdZ deddfd Z de fd Z d Z dd Zdd ZdS) ClientBuilderz Builder for a Ray Client connection. This class can be subclassed by custom builder classes to modify connection behavior to include additional features or altered semantics. One example is the ``_LocalClientBuilder``. addressrNcttd||_t|_i|_d|_d|_d|_d|_ dS)NzRay Client requires pip package `ray[client]`. If you installed the minimal Ray (e.g. `pip install ray`), please reinstall by executing `pip install ray[client]`.FT) r ValueErrorrIr _job_config_remote_init_kwargs_allow_multiple_connections _credentials _metadata_deprecation_warn_enabled)rrIs r __init__zClientBuilder.__init__`sl * , , 8K   $;;#% ,1( *.&&&r"envc:|j||S)a Set an environment for the session. Args: env (Dict[st, Any]): A runtime environment to use for this connection. See :ref:`runtime-environments` for what values are accepted in this dict. )rLset_runtime_env)rrSs r rSzClientBuilder.envss  ((--- r" namespacec:|j||S)zl Sets the namespace for the session. Args: namespace: Namespace to use. )rLset_ray_namespace)rrVs r rVzClientBuilder.namespace~s **9555 r"c t|jr||tjjj}tjjdk}|js|s|rtdd}|jr)tjjj d}tjj |j |j|j|j|j}tjjj}t'||d|d|d|dtjjj }|jr)tjjj ||S) a Begin a connection to the address passed in via ray.client(...). Returns: ClientInfo: Dataclass with information about the setting. This includes the server's version of Python & Ray as well as the dashboard_url. rzsThe client has already connected to the cluster with allow_multiple=True. Please set allow_multiple=True to proceedN) job_configrOray_init_kwargsmetadatarrr num_clientsrrrrrr)rQ_client_deprecation_warn_fill_defaults_from_envr+r,r-r1num_connected_contextsrNrKr.r3connectrIrLrOrMrP_get_dashboard_urlr get_context)rdefault_cli_connectedhas_cli_connected old_ray_cxtclient_info_dictrcxts r rbzClientBuilder.connects  ) ,  ) ) + + + $$&&& !$ 3 @ @ B BHOBBDDqH0 ) "      + @(/-99$??K82:: L'* 4^ ;  +>>@@ '+,<=(7' 5)-8 # 3 ? ? A A      + 9 HO  + +K 8 8 8 r"cNtjt}|r!|jj||tjt}|r5|jj+| tj |dSdSdSr) osenvirongetr rL ray_namespacerVr runtime_envrSjsonloads)rnamespace_env_varruntime_env_vars r r`z%ClientBuilder._fill_defaults_from_envsJNN+MNN  .!1!?!G NN, - - -*..)MNN  2t/;C HHTZ00 1 1 1 1 1 2 2CCr"c z|d||d|d=|d||d|d=|ddur d|_|d=d|vr|d|_|d=d|vr|d|_|d=|rtjt}t| |j }t|dkr5td d |||_d |}t$d ||S) aa When a client builder is constructed through ray.init, for example `ray.init(ray://..., namespace=...)`, all of the arguments passed into ray.init with non-default values are passed again into this method. Custom client builders can override this method to do their own handling/validation of arguments. rVNroallow_multipleTrOrPrzGot unexpected kwargs: {}, z:Passing the following kwargs to ray.init() on the server: )rmrVrSrNkeysrOrPinspect signatureray_driver_initset difference parameterslen RuntimeErrorformatjoinrMr9info)rkwargs expected_sig extra_argsunknowns r _init_argszClientBuilder._init_argss ::k " " . NN6+. / / /{# ::m $ $ 0 HHVM* + + +}% ::& ' '4 / //3D ,'( V[[]] * * &~ 6D ~& &++-- ' '#K0DN{#  ",_==LV[[]]++66|7N7S7S7U7UVVJ:"""/66tyy7L7LMM(.D $ii''G KK,"),,    r"c|jj}|jj}g}|jrRt |t r|d|jdn|d|jd|r|d|d|r|dd|}d|d}tj dtd |d td d S)z Generates a warning for user's if this ClientBuilder instance was created directly or through ray.client, instead of relying on internal methods (ray.init, or auto init) "z"ray://z namespace="zruntime_env=rvz ray.init()zStarting a connection through `ray.client` will be deprecated in future ray versions in favor of `ray.init`. See the docs for more details: zR. You can replace your call to `ray.client().connect()` with the following:  ) stacklevelN) rLrnrorI isinstance_LocalClientBuilderappendrwarningswarnCLIENT_DOCS_URLDeprecationWarning)rrVroreplacement_argsargs_strreplacement_calls r r_z&ClientBuilder._client_deprecation_warnsG $2 &2  < C$ 344 C !''(;DL(;(;(;<<<< ''(A$,(A(A(ABBB  @  # #$>)$>$>$> ? ? ?  F  # #$D E E E99-..2x222   *, * *&  * * *       r")rrHr<)r=r>r?r@rrArRrrrSrVrrbr`rr_rFr"r rHrHXs . .$....& tCH~ /    3?99999v222))))V' ' ' ' ' ' r"rHceZdZdefdZdS)rrc |jr||tj|j|j}t|ddtj dtj dtj dtj tj ddS) zQ Begin a connection to the address passed in via ray.client(...) )rIrZ webui_urlz{}.{}.{}rNr^) rQr_r`r+rrIrLrrsys version_info __version__ __commit__)rconnection_dicts r rbz_LocalClientBuilder.connect"s  ) ,  ) ) + + + $$&&&(4r?rrbrFr"r rr!s/       r"rrIrc2d|vrd|z}t|S)z Splits address into a module string (scheme) and an inner_address. If the scheme is not present, then "ray://" is prepended to the address. z://zray://)r )rIs r _split_addressr:s'  GW$  ! !!r"c|dkrtdS|3tjj|}t|St |\}} t j|}n(#t$r}td|d||d}~wwxYwdt|vsJd|d| |S)NlocalzModule: z6 does not exist. This module was parsed from Address: rHz does not have ClientBuilder.) rr+r4servicescanonicalize_bootstrap_addressr importlib import_module ExceptionrdirrH)rI module_string inner_addressmodulees r _get_builder_from_addressrEs'"7+++,'FFwOO"7+++#1'#:#: M=(77  >} > >4; > >    c   >->>>       . ..sA33 B=BBTrQctjt}|r*|(td|dtd|}t |}||_|S)a[ Creates a ClientBuilder based on the provided address. The address can be of the following forms: * None: Connects to or creates a local cluster and connects to it. * ``"local"``: Creates a new cluster locally and connects to it. * ``"IP:Port"``: Connects to a Ray Client Server at the given address. * ``"module://inner_address"``: load module.ClientBuilder & pass inner_address The _deprecation_warn_enabled flag enables deprecation warnings, and is for internal use only. Set it to False to suppress client deprecation warnings. NzUsing address (z$) instead of auto-detection because z is set.)rkrlrmrr9r:rrQ)rIrQ env_addressbuilders r r-r-[s$*..!ABBKw  Bk B B7 B B B   '00G(AG% Nr"r(),rrxrploggingrkrr dataclassesrtypingrrrrray.util.client_connectr+ray._private.ray_constantsrr r ray._private.utilsr r ray._private.workerr rrzray.job_configrray.util.annotationsrr getLoggerr=r9rrrHrrArrrEr-rFr"r rsz  !!!!!!------------ NMMMMMMMDDDDDDDD$$$$$$66666666  8 $ $%   55555K55  5p E E E E E E E  E P     -   2"C"E#s(O""""/x}/////, EI c]>B r"