o
    پi;                     @   sZ   d Z ddlZddlmZ ddlmZ ddlmZmZm	Z	m
Z
 ddlmZ defdd	ZdS )
z'Start bootstrap/kv-store-related server    N)Type)BaseKVBootstrapServer)DisaggregationModeKVClassTypeTransferBackendget_kv_class)
ServerArgsserver_argsc           
   
   C   s   t | j}t| j}|t jkrUt|tj}|| j| j	d}| j
dko&|tjk}|rSzddlm} td}|| W |S  tyR } zd}	|	d| 7 }	|	d }~ww |S d S )N)hostportr   )create_config_storeASCEND_MF_STORE_URLz+Failed create mf store, invalid ascend_url.z With exception )r   disaggregation_moder   disaggregation_transfer_backendPREFILLr   r   BOOTSTRAP_SERVERr
   disaggregation_bootstrap_port	node_rankASCENDmemfabric_hybridr   osgetenv	Exception)
r	   disagg_modetransfer_backendkv_bootstrap_server_classbootstrap_serveris_create_storer   
ascend_urleerror_message r!   V/home/ubuntu/.local/lib/python3.10/site-packages/sglang/srt/managers/disagg_service.pystart_disagg_service   s4   




r#   )__doc__r   typingr   sglang.srt.disaggregation.baser   sglang.srt.disaggregation.utilsr   r   r   r   sglang.srt.server_argsr   r#   r!   r!   r!   r"   <module>   s    