From d4629dc43f92c189e90a3e2f2f8a52648aed4d9d Mon Sep 17 00:00:00 2001 From: lkchen Date: Tue, 17 Jun 2025 20:03:01 -0700 Subject: [PATCH] [Misc] Add __str__ for RequestStatus (#19780) Signed-off-by: Linkun Chen --- tests/v1/test_request.py | 15 +++++++++++++++ vllm/v1/request.py | 3 +++ 2 files changed, 18 insertions(+) create mode 100644 tests/v1/test_request.py diff --git a/tests/v1/test_request.py b/tests/v1/test_request.py new file mode 100644 index 000000000000..2dc90f83caba --- /dev/null +++ b/tests/v1/test_request.py @@ -0,0 +1,15 @@ +# SPDX-License-Identifier: Apache-2.0 +from vllm.v1.request import RequestStatus + + +def test_request_status_fmt_str(): + """Test that the string representation of RequestStatus is correct.""" + assert f"{RequestStatus.WAITING}" == "WAITING" + assert f"{RequestStatus.WAITING_FOR_FSM}" == "WAITING_FOR_FSM" + assert f"{RequestStatus.WAITING_FOR_REMOTE_KVS}" == "WAITING_FOR_REMOTE_KVS" + assert f"{RequestStatus.RUNNING}" == "RUNNING" + assert f"{RequestStatus.PREEMPTED}" == "PREEMPTED" + assert f"{RequestStatus.FINISHED_STOPPED}" == "FINISHED_STOPPED" + assert f"{RequestStatus.FINISHED_LENGTH_CAPPED}" == "FINISHED_LENGTH_CAPPED" + assert f"{RequestStatus.FINISHED_ABORTED}" == "FINISHED_ABORTED" + assert f"{RequestStatus.FINISHED_IGNORED}" == "FINISHED_IGNORED" diff --git a/vllm/v1/request.py b/vllm/v1/request.py index 53fd70fabecf..694e271e5ad7 100644 --- a/vllm/v1/request.py +++ b/vllm/v1/request.py @@ -171,6 +171,9 @@ class RequestStatus(enum.IntEnum): FINISHED_ABORTED = enum.auto() FINISHED_IGNORED = enum.auto() + def __str__(self): + return self.name + @staticmethod def is_finished(status: "RequestStatus") -> bool: return status > RequestStatus.PREEMPTED