From d8beca5024890898b53c0498d727aa3fec9b9db9 Mon Sep 17 00:00:00 2001 From: Martin Hickey Date: Wed, 27 Aug 2025 18:21:16 +0100 Subject: [PATCH] Remove v1 engine proxy as no longer required As V1 is now default engine, there is not longer a need to set class proxy for V1 to V0 engine. Signed-off-by: Martin Hickey --- vllm/engine/async_llm_engine.py | 7 ------- vllm/engine/llm_engine.py | 5 ----- 2 files changed, 12 deletions(-) diff --git a/vllm/engine/async_llm_engine.py b/vllm/engine/async_llm_engine.py index 4fb028627a8c..bab67489b980 100644 --- a/vllm/engine/async_llm_engine.py +++ b/vllm/engine/async_llm_engine.py @@ -1117,10 +1117,3 @@ async def collective_rpc(self, """ return await self.engine.collective_rpc_async(method, timeout, args, kwargs) - - -# TODO(v1): Remove this class proxy when V1 goes default. -if envs.is_set("VLLM_USE_V1") and envs.VLLM_USE_V1: - from vllm.v1.engine.async_llm import AsyncLLM - - AsyncLLMEngine = AsyncLLM # type: ignore diff --git a/vllm/engine/llm_engine.py b/vllm/engine/llm_engine.py index 03c2f0375da4..5e8e241d23de 100644 --- a/vllm/engine/llm_engine.py +++ b/vllm/engine/llm_engine.py @@ -1893,8 +1893,3 @@ def collective_rpc(self, kwargs: Optional[dict[str, Any]] = None) -> list[_R]: return self.model_executor.collective_rpc(method, timeout, args, kwargs) - - -if envs.is_set("VLLM_USE_V1") and envs.VLLM_USE_V1: - from vllm.v1.engine.llm_engine import LLMEngine as V1LLMEngine - LLMEngine = V1LLMEngine # type: ignore