o ۾i@stUddlmZddlmZmZddlmZerddlmZda edBe d<defdd Z de fd d Z dddZdS)) TYPE_CHECKING)ECConnectorBaseECConnectorRole)ECConnectorFactory) VllmConfigN_EC_CONNECTOR_AGENTreturncCstdusJdtS)Nz)disaggregated EC cache is not initializedrr r b/home/ubuntu/.local/lib/python3.10/site-packages/vllm/distributed/ec_transfer/ec_transfer_state.pyget_ec_transfersr cCstduS)Nr r r r r has_ec_transfersr vllm_configrcCs:|jdurdS|jjrtdurtj|tjdadSdSdS)z( Initialize EC cache connector. N)configrole)ec_transfer_configis_ec_transfer_instancerrcreate_connectorrWORKER)rr r r ensure_ec_transfer_initializeds  r)rrrN)typingr.vllm.distributed.ec_transfer.ec_connector.baserr1vllm.distributed.ec_transfer.ec_connector.factoryr vllm.configrr__annotations__r boolr rr r r r s