Skip to content

Commit 389a97c

Browse files
committed
Fail request if FSM fails to advance
1 parent e0cbad4 commit 389a97c

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

vllm/v1/core/sched/scheduler.py

Lines changed: 9 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -779,8 +779,15 @@ def update_from_output(
779779
# NOTE: structured_output_request
780780
# should not be None if use_structured_output, we have
781781
# check above, so safe to ignore type warning
782-
request.structured_output_request.grammar.accept_tokens( # type: ignore[union-attr]
783-
req_id, new_token_ids)
782+
if not request.structured_output_request.grammar.accept_tokens( # type: ignore[union-attr]
783+
req_id, new_token_ids):
784+
# Grammar FSM failed to advance - mark request as finished with error
785+
logger.error(
786+
"Structured output FSM failed to advance for request %s. "
787+
"Terminating request.", req_id)
788+
request.status = RequestStatus.FINISHED_ABORTED
789+
stopped = True
790+
self._free_request(request)
784791

785792
# Add newly generated spec token ids to the request.
786793
if spec_token_ids is not None:

0 commit comments

Comments
 (0)