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