[Feature] Add Command Tool Parser

Signed-off-by: Doil Kim <[email protected]>
Co-authored-by: 김종곤 <[email protected]>

Author: Ubuntu

vllm/entrypoints/openai/tool_parsers/__init__.py

@@ -2,7 +2,7 @@
 # SPDX-FileCopyrightText: Copyright contributors to the vLLM project
 
 from .abstract_tool_parser import ToolParser, ToolParserManager
-from .command_tool_parser import CommandJsonToolParser
+from .command_tool_parser import CommandToolParser
 from .deepseekv3_tool_parser import DeepSeekV3ToolParser
 from .granite_20b_fc_tool_parser import Granite20bFCToolParser
 from .granite_tool_parser import GraniteToolParser
@@ -23,5 +23,5 @@
     "Internlm2ToolParser", "Llama3JsonToolParser", "JambaToolParser",
     "Llama4PythonicToolParser", "PythonicToolParser", "Phi4MiniJsonToolParser",
     "DeepSeekV3ToolParser", "xLAMToolParser", "MinimaxToolParser",
-    "CommandJsonToolParser"
+    "CommandToolParser"
 ]

vllm/entrypoints/openai/tool_parsers/command_tool_parser.py

@@ -23,8 +23,8 @@
 logger = init_logger(__name__)
 
 
-@ToolParserManager.register_module("command_json")
-class CommandJsonToolParser(ToolParser):
+@ToolParserManager.register_module("command")
+class CommandToolParser(ToolParser):
 
     def __init__(self, tokenizer: PreTrainedTokenizerBase):
         super().__init__(tokenizer)
@@ -47,7 +47,7 @@ def __init__(self, tokenizer: PreTrainedTokenizerBase):
         if (self.tool_call_start_token_id is None
                 or self.tool_call_end_token_id is None):
             raise RuntimeError(
-                "CommandJsonToolParser cannot find start/end tokens in vocab")
+                "CommandToolParser cannot find start/end tokens in vocab")
 
     def extract_tool_calls(
             self, model_output: str,