o c²“i) ã@sžddlmZmZmZddlmZddlmZdada dd„Z dd „Z d efd d „Z ed d„ƒZ eddœdeeefdeeeefdefdd„ƒZeddœdeeefdeeeefdefdd„ƒZedededdfdd„ƒZe d'ddœdeeefdeeefdedeeeefdef dd„ƒZeddd œdeeefd!edeeeefdefd"d#„ƒZeddœd$eeefdeeeefdeefd%d&„ƒZdS)(é)ÚListÚOptionalÚUnion)Úclient_mode_hook)Ú GcsClientFNcCs dadadS)NF©Úglobal_gcs_clientÚ _initialized©r r úP/home/ubuntu/.local/lib/python3.10/site-packages/ray/experimental/internal_kv.pyÚ_internal_kv_reset sr cCstS©N©rr r r r Úinternal_kv_get_gcs_clientsrÚ gcs_clientcCs|dusJ‚|adadS)z;Initialize the internal KV for use in other function calls.NTr)rr r r Ú_initialize_internal_kvs rcCstduSr rr r r r Ú_internal_kv_initializedsr)Ú namespaceÚkeyrÚreturncCó>t|tƒr | ¡}t|tƒr| ¡}t|tƒsJ‚t ||¡S)z Fetch the value of a binary key.)Ú isinstanceÚstrÚencodeÚbytesrÚinternal_kv_get©rrr r r Ú_internal_kv_get!ó   rcCr)zCheck key exists or not.)rrrrrÚinternal_kv_existsrr r r Ú_internal_kv_exists/rr ÚuriÚ expiration_scCs t ||¡S)z'Pin a runtime_env URI for expiration_s.)rÚpin_runtime_env_uri)r!r"r r r Ú_pin_runtime_env_uri=s r$TÚvalueÚ overwritecCslt|tƒr | ¡}t|tƒr| ¡}t|tƒr| ¡}t|tƒr*t|tƒr*t|tƒs,J‚t ||||¡dkS)zÈGlobally associates a value with a given binary key. This only has an effect if the key does not already have a value. Returns: already_exists: whether the value already exists. r)rrrrÚboolrÚinternal_kv_put)rr%r&rr r r Ú_internal_kv_putCs   ÿþýr))Ú del_by_prefixrr*cCs@t|tƒr | ¡}t|tƒr| ¡}t|tƒsJ‚t |||¡Sr )rrrrrÚinternal_kv_del)rr*rr r r Ú_internal_kv_delas  r,ÚprefixcCs0t|tƒr | ¡}t|tƒr| ¡}t ||¡S)zBList all keys in the internal KV store that start with the prefix.)rrrrÚinternal_kv_keys)r-rr r r Ú_internal_kv_listps   r/)T)ÚtypingrrrÚray._private.client_mode_hookrÚ ray._rayletrr rr rrrrrrr'r Úintr$r)r,r/r r r r Ús„   ÿ ÿÿ þ ÿ ÿÿ þ ýû ÿ þýû úü ÿýü ûÿ ÿÿþ