
    -`i                     t    U d dl mZ d dlmZmZ d dlmZ erd dlmZ da	edz  e
d<   defdZdefd	ZddZdS )    )TYPE_CHECKING)ECConnectorBaseECConnectorRole)ECConnectorFactory)
VllmConfigN_EC_CONNECTOR_AGENTreturnc                  2    t           
J d            t           S )Nz)disaggregated EC cache is not initializedr        /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/distributed/ec_transfer/ec_transfer_state.pyget_ec_transferr      s    **,W***r   c                      t           d uS )Nr   r   r   r   has_ec_transferr      s    d**r   vllm_configr   c                     | j         dS | j         j        r)t          $t          j        | t
          j                  adS dS dS )z(
    Initialize EC cache connector.
    N)configrole)ec_transfer_configis_ec_transfer_instancer   r   create_connectorr   WORKER)r   s    r   ensure_ec_transfer_initializedr      s^     %- 	&>
'0A_%;
 
 
	
 
''r   )r   r   r	   N)typingr   .vllm.distributed.ec_transfer.ec_connector.baser   r   1vllm.distributed.ec_transfer.ec_connector.factoryr   vllm.configr   r   __annotations__r   boolr   r   r   r   r   <module>r!      s    !                    Q P P P P P '&&&&&&.2 _t+ 2 2 2    
+ + + + +
 
 
 
 
 
r   