From 5f0af36af555a3813b9d30983bd29c384b84b647 Mon Sep 17 00:00:00 2001 From: bigmoyan Date: Sat, 12 Jul 2025 04:16:14 +0800 Subject: [PATCH] Update kimi-k2 tool calling docs, enable unit tests (#20821) Signed-off-by: wangzhengtao Co-authored-by: wangzhengtao Co-authored-by: wangzhengtao --- docs/features/tool_calling.md | 8 ++++++++ tests/tool_use/test_kimi_k2_tool_parser.py | 2 -- 2 files changed, 8 insertions(+), 2 deletions(-) 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"