o `۷i@sddlZddlmZddlmZddlmZmZddlm Z m Z m Z m Z m Z mZmZmZmZmZmZmZmZmZmZmZmZeeZgdZdZdZdZ d Z!d Z"gej#d e fd e fd e!fd e"fZ$dde%fddZ&GdddZ' dde%de(de%de)de%dee%f ddZ*dS)N)Optional) ray_constants) ErrorType JobConfig)ActorTableDataAvailableResourcesErrorTableDataGcsEntry GcsNodeInfo JobTableDataPlacementGroupTableData PubSubMessageResourceDemand ResourceLoad ResourcesDataResourceUsageBatchData TablePrefix TablePubsub TaskEventsTotalResourcesWorkerTableData)rr rrr rrrr rrrrrrrr rr i i`zgrpc.max_send_message_lengthzgrpc.max_receive_message_lengthzgrpc.keepalive_time_mszgrpc.keepalive_timeout_msFaddresscCsddlm}||t|dS)zReturns a GRPC channel to GCS. Args: address: GCS address string, e.g. ip:port aio: Whether using grpc.aio Returns: grpc.Channel or grpc.aio.Channel to GCS r)init_grpc_channel)options asynchronous)ray._private.grpc_utilsr _GRPC_OPTIONS)raiorrL/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/_private/gcs_utils.pycreate_gcs_channelIs r!c@s@eZdZd deedefddZeddZd d Z d d Z dS) GcsChannelNF gcs_addressrcCs||_||_dSN) _gcs_address_aio)selfr#rrrr __init__Xs zGcsChannel.__init__cC|jSr$)r%r'rrr r\szGcsChannel.addresscCst|j|j|_dSr$)r!r%r&_channelr*rrr connect`szGcsChannel.connectcCr)r$)r+r*rrr channelfszGcsChannel.channel)NF) __name__ __module__ __qualname__rstrboolr(propertyrr,r-rrrr r"Ws   r"hostportpassworduse_sslstorage_namespaceusernamecCsddlm}t|tstd|durd}t|tstdt|ts'td|dkr2td|t|ts;td t|tsDtd |||||||S) aThis function is used to cleanup the GCS storage in Redis. It supports Redis in cluster and non-cluster modes. Args: host: The Redis host address. port: The Redis port. username: The Redis username. password: The Redis password. use_ssl: Whether to encrypt the connection. storage_namespace: The namespace of the storage to be deleted. r)del_key_prefix_from_storagezHost must be a stringNzUsername must be a stringzPassword must be a stringzInvalid port: zuse_ssl must be a booleanz"storage namespace must be a string) ray._rayletr: isinstancer1 ValueErrorr2 TypeError)r4r5r6r7r8r9r:rrr cleanup_redis_storagejs$       r@)Fr$)+loggingtypingr ray._privaterray.core.generated.common_pb2rrray.core.generated.gcs_pb2rrrr r r r r rrrrrrrrr getLoggerr.logger__all__WORKERDRIVER_MAX_MESSAGE_LENGTH_GRPC_KEEPALIVE_TIME_MS_GRPC_KEEPALIVE_TIMEOUT_MSGLOBAL_GRPC_OPTIONSrr1r!r"intr2r@rrrr sN  L