From ea3372fb0c7ff56d3057d2cab73b7e3948a2a70a Mon Sep 17 00:00:00 2001 From: pz1116 <47019764+Pz1116@users.noreply.github.com> Date: Sat, 22 Nov 2025 18:52:48 +0800 Subject: [PATCH] [Bugfix][KV Pool]fix get_ip import in mooncake_store (#4355) ### What this PR does / why we need it? fix import error for get_ip() in vllm main branch ### Does this PR introduce _any_ user-facing change? N ### How was this patch tested? - vLLM version: v0.11.0 - vLLM main: https://github.com/vllm-project/vllm/commit/2918c1b49c88c29783c86f78d2c4221cb9622379 --------- Signed-off-by: pz1116 --- vllm_ascend/distributed/mooncake/mooncake_store.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/vllm_ascend/distributed/mooncake/mooncake_store.py b/vllm_ascend/distributed/mooncake/mooncake_store.py index cee07c6a..ec885c59 100644 --- a/vllm_ascend/distributed/mooncake/mooncake_store.py +++ b/vllm_ascend/distributed/mooncake/mooncake_store.py @@ -5,13 +5,19 @@ import os from mooncake.store import ReplicateConfig # type: ignore from vllm.config import ParallelConfig from vllm.distributed.parallel_state import get_tensor_model_parallel_rank -from vllm.utils import get_ip, logger +from vllm.utils import logger from vllm_ascend.distributed.mooncake.config_data import MooncakeEngineKey from vllm_ascend.distributed.mooncake.transfer_engine import get_global_te +from vllm_ascend.utils import vllm_version_is from .config_data import MooncakeStoreConfig +if vllm_version_is("0.11.0"): + from vllm.utils import get_ip +else: + from vllm.utils.network_utils import get_ip + METADATA_BYTES_LEN = 24 BASE_PORT = int(os.getenv("VLLM_BASE_PORT", "8790"))