From be067861c6ab4df53a45e332caf53c163330290d Mon Sep 17 00:00:00 2001 From: Chauncey Date: Sat, 11 Oct 2025 10:43:39 +0800 Subject: [PATCH] [Frontend] Improve the performance of `is_reasoning_end` (#25735) Signed-off-by: chaunceyjiang --- vllm/reasoning/basic_parsers.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vllm/reasoning/basic_parsers.py b/vllm/reasoning/basic_parsers.py index b4106a4f5..f47ffe621 100644 --- a/vllm/reasoning/basic_parsers.py +++ b/vllm/reasoning/basic_parsers.py @@ -59,7 +59,8 @@ class BaseThinkingReasoningParser(ReasoningParser): ) def is_reasoning_end(self, input_ids: list[int]) -> bool: - return self.end_token_id in input_ids + end_token_id = self.end_token_id + return any(input_id == end_token_id for input_id in reversed(input_ids)) def extract_content_ids(self, input_ids: list[int]) -> list[int]: """