o `۷i@sddlZddlZddlmZmZmZddlmZddlm Z dZ dZ dee efdeee efe ffd d Zdee efdee effd d Zdee efdee effd dZde de fddZde de fddZdee efdee effddZdS)N)AnyDictTuple)get_ray_temp_dir) cli_loggerz6The field {} is not supported for on-premise clusters.zlocal.cluster.nodeconfigreturncCst|}dD]#}||r*|dkr t||ivr |dfSt|}t|qtiidi|d<t|d<d|dvrFt|}|dfSt |}|dfS) z\ Prepare local cluster config for ingestion by cluster launcher and autoscaler. ) head_node worker_nodesavailable_node_typesr F) node_config resourceshead_node_typecoordinator_addressproviderT) copydeepcopygetLOCAL_CLUSTER_NODE_TYPEunsupported_field_messageformatrabortprepare_coordinatorprepare_manual)rfielderr_msgrZ/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/autoscaler/_private/local/config.py prepare_local s"       rcCsHt|}d|vrtd|dt}|dd|d<|d|d<|S)N max_workersz[The field `max_workers` is required when using an automatically managed on-premise cluster.r min_workersr)rrrrrpop)r node_typerrrr+s   rc Cs t|}d|dvsd|dvrtdt|dd}|dt}|d||d|}|d}||krLtd|d |d |d ||d<n||d<||krltd |d |d |d ||d<||d<n||d<||krtd |d|d|d|S)ahValidates and sets defaults for configs of manually managed on-prem clusters. - Checks for presence of required `worker_ips` and `head_ips` fields. - Defaults min and max workers to the number of `worker_ips`. - Caps min and max workers at the number of `worker_ips`. - Writes min and max worker info into the single worker node type. worker_ipsrhead_ipzbPlease supply a `head_ip` and list of `worker_ips`. Alternatively, supply a `coordinator_address`.r rr z%The value of `min_workers` supplied (z6) is greater than the number of available worker ips (z). Setting `min_workers=z`.z%The value of `max_workers` supplied (z). Setting `max_workers=z3) is less than the number of available worker ips (z ). At most z. Ray worker nodes will connect to the cluster.) rrrrlenr setdefaultr!warning)rnum_ipsr"r rrrrr<sR       r cluster_namecCtjtd|S)Nzcluster-{}.lockospathjoinrrr)rrr get_lock_pathvr0cCr*)Nzcluster-{}.stater+r/rrrget_state_pathzr1r2cCs|S)Nr)rrrrbootstrap_local~sr3)rr,typingrrrray._common.utilsr"ray.autoscaler._private.cli_loggerrrrstrboolrrrr0r2r3rrrrs  *"":&