&`i;dZddlZddlZddlmZddlmZmZmZm Z m Z m Z ddl m Z ddlmZerddlmZddlmZdd lmZdd lmZmZmZejeZd ZGd d ZdS)z^This file defines the interface between the ray client worker and the overall ray module API. N)Future) TYPE_CHECKINGAnyCallableListOptionalUnion)ray_option_utils)_ClientWorkerPropertyAPI) ActorClass) DataResponse)RemoteFunction)ClientActorHandleClientObjectRef ClientStubcZt|tr|dS|S)Nzutf-8) isinstancestrencode)values g/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/util/client/api.py _as_bytesrs+%%||G$$$ Lc reZdZdZdFdZdddZdZdZdZd d d e e fd Z d e d dfdZ d e d dfdZdGdZ dFdedeed dfdZdHded e efdZdddIdZddddJdZd efd Zd!Zd"Zd#Zd$Zd%Zd efd&ZdFd'eed ee efd(Z d efd)Z!dd*d+e"ee fdee"ee fd efd,Z#dd*d+e"ee fdee"ee fd e fd-Z$ dKdd*d+e"ee fd.e"ee fd/edee"ee fd ef d0Z%ddd1d+e"ee fd2edee"ee fd e&fd3Z'dd*d4e"ee fdee"ee fd e e fd5Z(d6ed7e&d dfd8Z)dd9d efd:Z*d;dZ,d+ed efd?Z-d+efd@Z.dAddBe/dCgdfd dfdDZ0d efdEZ1dS)L _ClientAPIzThe Client-side methods corresponding to the ray API. Delegates to the Client Worker that contains the connection to the ClientServer. Nc||_dSN)worker)selfrs r__init__z_ClientAPI.__init__ s  rtimeoutc:|j||S)zget is the hook stub passed on to replace `ray.get` Args: vals: [Client]ObjectRef or list of these refs to retrieve. timeout: Optional timeout in milliseconds r!)rget)rvalsr"s rr$z_ClientAPI.get#s{tW555rc&|jj|i|S)zput is the hook stub passed on to replace `ray.put` Args: val: The value to `put`. args: opaque arguments kwargs: opaque keyword arguments )rputrargskwargss rr'z_ClientAPI.put,st{////rc&|jj|i|S)zwait is the hook stub passed on to replace `ray.wait` Args: args: opaque arguments kwargs: opaque keyword arguments )rwaitr(s rr,z_ClientAPI.wait6s t{0000rcFddlm}t|dkrCt|dkr0t|dr|d|dSt|dkrt|dksJtj||S)a remote is the hook stub passed on to replace `ray.remote`. This sets up remote functions or actors, as the decorator, but does not execute them. Args: args: opaque arguments kwargs: opaque keyword arguments r)remote_decoratorN)options)ray.util.client.commonr.lencallabler remote_args_error_string)rr)r*r.s rremotez_ClientAPI.remote?s <;;;;; t99>>c&kkQ..8DG3D3D.1##D111$q':: : IINNs6{{Q  4 / .////rinstancerreturnc.|jj|g|Ri|S)a2call_remote is called by stub objects to execute them remotely. This is used by stub objects in situations where they're called with .remote, eg, `f.remote()` or `actor_cls.remote()`. This allows the client stub objects to delegate execution to be implemented in the most effective way whether it's in the client, clientserver, or raylet worker. Args: instance: The Client-side stub reference to a remote object args: opaque arguments kwargs: opaque keyword arguments )r call_remote)rr6r)r*s rr9z_ClientAPI.call_remoteVs)'t{&xA$AAA&AAAridc6|j|S)apAttempts to release an object reference. When client references are destructed, they release their reference, which can opportunistically send a notification through the datachannel to release the reference being held for that object on the server. Args: id: The id of the reference to release on the server side. )r call_releaserr:s rr<z_ClientAPI.call_releasefs{''+++rc6|j|S)aAttempts to retain a client object reference. Increments the reference count on the client side, to prevent the client worker from attempting to release the server reference. Args: id: The id of the reference to retain on the client side. )r call_retainr=s rr?z_ClientAPI.call_retainrs{&&r***rc4|jS)zsclose cleans up an API connection by closing any channels or shutting down any servers gracefully. )rclosers rrAz_ClientAPI.close}s{  """rname namespacerc8|j||S)zReturns a handle to an actor by name. Args: name: The name passed to this actor by Actor.options(name="name").remote() )r get_actor)rrCrDs rrFz_ClientAPI.get_actors{$$T9555rFall_namespacesc6|j|S)aList all named actors in the system. Actors must have been created with Actor.options(name="name").remote(). This works for both detached & non-detached actors. By default, only actors in the current namespace will be returned and the returned entries will simply be their name. If `all_namespaces` is set to True, all actors in the cluster will be returned regardless of namespace, and the retunred entries will be of the form '/'. )rlist_named_actors)rrGs rrIz_ClientAPI.list_named_actorss{,,^<<.annotate_methods9&&-3M-B*"f,,39:M3N0Mr)r2)rr)r* valid_kwargs error_stringkeykey_error_stringrhs ` rrgz_ClientAPI.methods&':;  ,)5 , , ,  4yyA~~#f++///   rcNt|jS)zReturn a Ray RuntimeContext describing the state on the server Returns: A RuntimeContext wrapping a client making get_cluster_info calls. )r rbuild_runtime_contextrBs rget_runtime_contextz_ClientAPI.get_runtime_contexts ( 44JJLLLrcgSrrBs r get_gpu_idsz_ClientAPI.get_gpu_ids!s rfilenamectdddlmcmcm}|j|jj }|@t|d5}tj ||ddddS#1swxYwYdS|S)NzBTimeline will include events from other clients using this server.rw) loggerwarningrXrYrZr[rr\r]TIMELINEopenjsondump)rryr[ all_eventsoutfiles rtimelinez_ClientAPI.timeline$s P    CBBBBBBBBBBB[11  * 3    h$$ / *g... / / / / / / / / / / / / / / / / / / sBBBcdS)z.Hook for internal_kv._internal_kv_initialized.TrwrBs r_internal_kv_initializedz#_ClientAPI._internal_kv_initialized4s trrDrkcn|jt|t|S)z)Hook for internal_kv._internal_kv_exists.r)rinternal_kv_existsrrrkrDs r_internal_kv_existsz_ClientAPI._internal_kv_exists;s6{-- cNNi &:&:.   rcn|jt|t|S)z&Hook for internal_kv._internal_kv_get.r)rinternal_kv_getrrs r_internal_kv_getz_ClientAPI._internal_kv_getCs6{** cNNi &:&:+   rr overwritec|jt|t||t|S)z&Hook for internal_kv._internal_kv_put.r)rinternal_kv_putr)rrkrrrDs r_internal_kv_putz_ClientAPI._internal_kv_putKsB{** cNNIe,,i9YCWCW+   r del_by_prefixrDrcp|jt||t|S)z&Hook for internal_kv._internal_kv_del.r)rinternal_kv_delr)rrkrrDs r_internal_kv_delz_ClientAPI._internal_kv_delXs8{** cNN-9YCWCW+   rprefixcn|jt|t|S)z'Hook for internal_kv._internal_kv_list.r)rinternal_kv_listr)rrrDs r_internal_kv_listz_ClientAPI._internal_kv_listds8{++ f  9)=)=,   ruri expiration_sc8|j||S)z*Hook for internal_kv._pin_runtime_env_uri.)rpin_runtime_env_uri)rrrs r_pin_runtime_env_uriz_ClientAPI._pin_runtime_env_urios{..sLAAArr c6|j|S)z@Register a ClientActorClass for the ActorClass and return a UUID)r_convert_actor)rrKs rrz_ClientAPI._convert_actorss{))%000rfuncrc6|j|S)z@Register a ClientRemoteFunc for the ActorClass and return a UUID)r_convert_function)rrs rrz_ClientAPI._convert_functionws{,,T222rc6|j|S)z)Given a UUID, return the converted object)r_get_convertedrrks rrz_ClientAPI._get_converted{s{))#...rc6|j|S)zACheck if a key UUID is present in the store of converted objects.)r_converted_key_existsrs rrz _ClientAPI._converted_key_existss{00555rc|ds"td|||S)N_zNot available in Ray client: `ray.{}`. This method is only available within Ray remote functions and is not yet implemented in the client API.) startswithNotImplementedErrorformat__getattribute__rs r __getattr__z_ClientAPI.__getattr__sN~~c"" %117  $$S)))rrefcallbackr c<|j||dSr)rregister_callback)rrrs r_register_callbackz_ClientAPI._register_callbacks" %%c844444rcddlmcmcm}|j|jjddS)Nr dashboard_url) rXrYrZr[rr\r] DASHBOARD_URLr$r_s r_get_dashboard_urlz_ClientAPI._get_dashboard_urlsXBBBBBBBBBBBB{++  * 8  #or " " #rr)r7N)F)rKr)rQr)T)2__name__ __module__ __qualname____doc__r r$r'r,r5rrr9bytesr<r?rArrrFboolrIrNrTrVr`rgrorrrulistrxrrrr rrrintrrrrrrrrrrrrwrrrrs$(66666000111000.BLBd6lBBBB ,u , , , , , +e + + + + +####59 6 6 6$,SM 6  6 6 6 6 = = =c = = = =>B>>>>>>7rsC %%%%%%FFFFFFFFFFFFFFFF((((((DDDDDDV$$$$$$>>>>>>222222UUUUUUUUUU  8 $ $ {#{#{#{#{#{#{#{#{#{#r