[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:
2918c1b49c

---------

Signed-off-by: pz1116 <zpbzpb123123@gmail.com>
This commit is contained in:
pz1116
2025-11-22 18:52:48 +08:00
committed by GitHub
parent 9b3a484b46
commit ea3372fb0c

View File

@@ -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"))