[core] Add tags parameter to wake_up() (#15500)

Signed-off-by: Eric <erictang000@gmail.com>
2025-04-02 01:59:27 -07:00
parent 90969fb39a
commit ddb94c2605
18 changed files with 144 additions and 71 deletions
--- a/vllm/engine/async_llm_engine.py
+++ b/vllm/engine/async_llm_engine.py
@@ -1225,8 +1225,8 @@ class AsyncLLMEngine(EngineClient):
    async def sleep(self, level: int = 1) -> None:
        self.engine.sleep(level)

-    async def wake_up(self) -> None:
-        self.engine.wake_up()
+    async def wake_up(self, tags: Optional[list[str]] = None) -> None:
+        self.engine.wake_up(tags)

    async def is_sleeping(self) -> bool:
        return self.engine.is_sleeping()