diff --git a/.buildkite/test_areas/misc.yaml b/.buildkite/test_areas/misc.yaml index ce313060d8b7..e04016d6dcc2 100644 --- a/.buildkite/test_areas/misc.yaml +++ b/.buildkite/test_areas/misc.yaml @@ -281,6 +281,7 @@ steps: - vllm/model_executor/layers/quantization/quark/ - vllm/multimodal/ - vllm/outputs.py + - vllm/parser/ - vllm/platforms/ - vllm/pooling_params.py - vllm/ray/ diff --git a/tests/parser/test_parse.py b/tests/parser/test_parse.py index bac8b64bcbd6..ba8bc1427f2d 100644 --- a/tests/parser/test_parse.py +++ b/tests/parser/test_parse.py @@ -6,7 +6,7 @@ import pytest from vllm.entrypoints.openai.chat_completion.protocol import ChatCompletionRequest -from vllm.parser.abstract_parser import _WrappedParser +from vllm.parser.abstract_parser import DelegatingParser from vllm.reasoning.basic_parsers import BaseThinkingReasoningParser from vllm.tool_parsers.hermes_tool_parser import Hermes2ProToolParser @@ -65,9 +65,11 @@ def make_request(**overrides): def make_parser(tokenizer, reasoning=False, tool=False): - _WrappedParser.reasoning_parser_cls = ThinkReasoningParser if reasoning else None - _WrappedParser.tool_parser_cls = Hermes2ProToolParser if tool else None - return _WrappedParser(tokenizer) + class TestParser(DelegatingParser): + reasoning_parser_cls = ThinkReasoningParser if reasoning else None + tool_parser_cls = Hermes2ProToolParser if tool else None + + return TestParser(tokenizer) @pytest.mark.parametrize(