o $i) @sddlmZmZmZddlmZddlmZdada ddZ dd Z d efd d Z ed dZ edddeeefdeeeefdefddZedddeeefdeeeefdefddZedededdfddZe d'dddeeefdeeefdedeeeefdef ddZeddd deeefd!edeeeefdefd"d#Zeddd$eeefdeeeefdeefd%d&ZdS)()ListOptionalUnion)client_mode_hook) GcsClientFNcCs dadadS)NFglobal_gcs_client _initializedr r Y/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/experimental/internal_kv.py_internal_kv_reset sr cCstSNrr 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) namespacekeyrreturncC>t|tr |}t|tr|}t|tsJt||S)z Fetch the value of a binary key.) isinstancestrencodebytesrinternal_kv_getrrr r r _internal_kv_get!   rcCr)zCheck key exists or not.)rrrrrinternal_kv_existsrr r r _internal_kv_exists/rr uri expiration_scCs t||S)z'Pin a runtime_env URI for expiration_s.)rpin_runtime_env_uri)r!r"r r r _pin_runtime_env_uri=s r$Tvalue overwritecCslt|tr |}t|tr|}t|tr|}t|tr*t|tr*t|ts,Jt||||dkS)zGlobally 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)rrrrboolrinternal_kv_put)rr%r&rr r r _internal_kv_putCs   r)) del_by_prefixrr*cCs@t|tr |}t|tr|}t|tsJt|||Sr )rrrrrinternal_kv_del)rr*rr r r _internal_kv_delas  r,prefixcCs0t|tr |}t|tr|}t||S)zBList all keys in the internal KV store that start with the prefix.)rrrrinternal_kv_keys)r-rr r r _internal_kv_listps   r/)T)typingrrrray._private.client_mode_hookr ray._rayletrr rr rrrrrrr'r intr$r)r,r/r r r r s