From 790e34e375feef2274b303e6a88c8fe0798687a7 Mon Sep 17 00:00:00 2001 From: NATURE Date: Fri, 30 Jan 2026 13:26:29 +0800 Subject: [PATCH] [Bugfix] Remove Mooncake/Yuanrong connector import warning (#1091) Signed-off-by: natureofnature --- .../omni_connectors/connectors/mooncake_connector.py | 8 +++++--- .../omni_connectors/connectors/yuanrong_connector.py | 10 +++++++--- 2 files changed, 12 insertions(+), 6 deletions(-) diff --git a/vllm_omni/distributed/omni_connectors/connectors/mooncake_connector.py b/vllm_omni/distributed/omni_connectors/connectors/mooncake_connector.py index 2f969b9bf..e28486481 100644 --- a/vllm_omni/distributed/omni_connectors/connectors/mooncake_connector.py +++ b/vllm_omni/distributed/omni_connectors/connectors/mooncake_connector.py @@ -15,7 +15,6 @@ except ImportError: try: from mooncake import MooncakeDistributedStore, ReplicateConfig except ImportError: - logger.warning("Mooncake not available, MooncakeOmniConnector will not work") MooncakeDistributedStore = None ReplicateConfig = None @@ -24,8 +23,11 @@ class MooncakeConnector(OmniConnectorBase): """Mooncake-based distributed connector for OmniConnector.""" def __init__(self, config: dict[str, Any]): - if MooncakeDistributedStore is None: - raise ImportError("Mooncake not available") + if MooncakeDistributedStore is None or ReplicateConfig is None: + raise ImportError( + "Mooncake components (MooncakeDistributedStore/ReplicateConfig) are not available. " + "Please ensure the 'mooncake' package is installed in your environment." + ) self.config = config self.host = config.get("host", "127.0.0.1") diff --git a/vllm_omni/distributed/omni_connectors/connectors/yuanrong_connector.py b/vllm_omni/distributed/omni_connectors/connectors/yuanrong_connector.py index c147d00fd..b2e4c85d5 100644 --- a/vllm_omni/distributed/omni_connectors/connectors/yuanrong_connector.py +++ b/vllm_omni/distributed/omni_connectors/connectors/yuanrong_connector.py @@ -11,16 +11,20 @@ logger = get_connector_logger(__name__) try: from datasystem.kv_client import KVClient, SetParam, WriteMode except ImportError: - logger.warning("Datasystem not available, YuanrongConnector will not work") KVClient = None + SetParam = None + WriteMode = None class YuanrongConnector(OmniConnectorBase): """Datasystem-based distributed connector for OmniConnector.""" def __init__(self, config: dict[str, Any]): - if KVClient is None: - raise ImportError("Datasystem not available") + if KVClient is None or SetParam is None or WriteMode is None: + raise ImportError( + "Datasystem components (KVClient/SetParam/WriteMode) are not available. " + "Please ensure the 'datasystem' package is installed in your environment." + ) self.config = config self.client = None