diff --git a/docs/features/tool_calling.md b/docs/features/tool_calling.md index d3caeaba65f74..35e01861c5d5f 100644 --- a/docs/features/tool_calling.md +++ b/docs/features/tool_calling.md @@ -282,6 +282,14 @@ Supported models: Flags: `--tool-call-parser deepseek_v3 --chat-template {see_above}` +### Kimi-K2 Models (`kimi_k2`) + +Supported models: + +* `moonshotai/Kimi-K2-Instruct` + +Flags: `--tool-call-parser kimi_k2` + ### Models with Pythonic Tool Calls (`pythonic`) A growing number of models output a python list to represent tool calls instead of using JSON. This has the advantage of inherently supporting parallel tool calls and removing ambiguity around the JSON schema required for tool calls. The `pythonic` tool parser can support such models. diff --git a/tests/tool_use/test_kimi_k2_tool_parser.py b/tests/tool_use/test_kimi_k2_tool_parser.py index 8768203a7119a..bd030632f167b 100644 --- a/tests/tool_use/test_kimi_k2_tool_parser.py +++ b/tests/tool_use/test_kimi_k2_tool_parser.py @@ -10,8 +10,6 @@ from vllm.entrypoints.openai.protocol import FunctionCall, ToolCall from vllm.entrypoints.openai.tool_parsers import KimiK2ToolParser from vllm.transformers_utils.tokenizer import get_tokenizer -pytest.skip("skip kimi_k2 parser test", allow_module_level=True) - # Use a common model that is likely to be available MODEL = "moonshotai/Kimi-K2-Instruct"