[Doc] Fix griffe warnings in HYV4 tool parser (#54412)
Signed-off-by: Harry Mellor <19981378+hmellor@users.noreply.github.com> Co-authored-by: Claude Opus 5 (1M context) <noreply@anthropic.com>
Harry Mellor committed
Aug 30, 2026 at 06:21 UTC
488b6da105222f4f8130b3aae4a0e67cfc61f522
1 file changed
+12
-8
vllm/tool_parsers/hy_v4_tool_parser.py
+12
-8
@@ -5,7 +5,8 @@ from __future__ import annotations
5
6
import ast
7
import json
8
-from typing import Any, TypedDict
8
+from collections.abc import Sequence
9
+from typing import TYPE_CHECKING, Any, TypedDict
10
11
import regex as re
12
@@ -26,6 +27,9 @@ from vllm.tokenizers import TokenizerLike
27
from vllm.tool_parsers.abstract_tool_parser import ToolParser
28
from vllm.tool_parsers.utils import partial_tag_overlap
29
30
+if TYPE_CHECKING:
31
+ from xgrammar import StructuralTag
32
+
33
logger = init_logger(__name__)
34
35
# =============================================================================
@@ -576,9 +580,9 @@ class HYV4ToolExtractor:
580
previous_text: str,
581
current_text: str,
582
delta_text: str,
579
- previous_token_ids,
580
- current_token_ids,
581
- delta_token_ids,
583
+ previous_token_ids: Sequence[int],
584
+ current_token_ids: Sequence[int],
585
+ delta_token_ids: Sequence[int],
586
tools: list[ToolSchema] | None,
587
*,
588
guided: bool = False,
@@ -989,7 +993,7 @@ class HYV4ToolParser(ToolParser):
993
request: ChatCompletionRequest | ResponsesRequest,
994
*,
995
reasoning: bool = False,
992
- ):
996
+ ) -> StructuralTag | None:
997
"""Build a structural tag matching HYV4's tool tokens.
998
999
Overridden only to pass the per-checkpoint ``token_suffix`` through to
@@ -1106,9 +1110,9 @@ class HYV4ToolParser(ToolParser):
1110
previous_text: str,
1111
current_text: str,
1112
delta_text: str,
1109
- previous_token_ids,
1110
- current_token_ids,
1111
- delta_token_ids,
1113
+ previous_token_ids: Sequence[int],
1114
+ current_token_ids: Sequence[int],
1115
+ delta_token_ids: Sequence[int],
1116
request: ChatCompletionRequest,
1117
) -> DeltaMessage | None:
1118
delta = self._extractor.extract_tool_calls_streaming(