Skip to content

Commit 5f0af36

Browse files
MoyanZittowangzhengtaowangzhengtao
authored
Update kimi-k2 tool calling docs, enable unit tests (#20821)
Signed-off-by: wangzhengtao <wangzhengtao@moonshot.cn> Co-authored-by: wangzhengtao <wangzhengtao@moonshot.cn> Co-authored-by: wangzhengtao <wangzhengtao@msh.team>
1 parent 0d21b26 commit 5f0af36

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

docs/features/tool_calling.md

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -282,6 +282,14 @@ Supported models:
282282

283283
Flags: `--tool-call-parser deepseek_v3 --chat-template {see_above}`
284284

285+
### Kimi-K2 Models (`kimi_k2`)
286+
287+
Supported models:
288+
289+
* `moonshotai/Kimi-K2-Instruct`
290+
291+
Flags: `--tool-call-parser kimi_k2`
292+
285293
### Models with Pythonic Tool Calls (`pythonic`)
286294

287295
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.

tests/tool_use/test_kimi_k2_tool_parser.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -10,8 +10,6 @@
1010
from vllm.entrypoints.openai.tool_parsers import KimiK2ToolParser
1111
from vllm.transformers_utils.tokenizer import get_tokenizer
1212

13-
pytest.skip("skip kimi_k2 parser test", allow_module_level=True)
14-
1513
# Use a common model that is likely to be available
1614
MODEL = "moonshotai/Kimi-K2-Instruct"
1715

0 commit comments

Comments
 (0)