[router] add kimi-k2 tool parser (#9702)

Co-authored-by: Chang Su <chang.s.su@oracle.com>
This commit is contained in:
Simo Lin
2025-08-27 11:04:55 -07:00
committed by GitHub
co-authored by Chang Su
parent 6f6beca49d
commit 5c06dcb75a
5 changed files with 440 additions and 4 deletions
+6 -1
View File
@@ -1,5 +1,6 @@
use crate::tool_parser::parsers::{
DeepSeekParser, JsonParser, LlamaParser, MistralParser, PythonicParser, QwenParser, Step3Parser,
DeepSeekParser, JsonParser, KimiK2Parser, LlamaParser, MistralParser, PythonicParser,
QwenParser, Step3Parser,
};
use crate::tool_parser::traits::ToolParser;
use std::collections::HashMap;
@@ -113,8 +114,12 @@ impl ParserRegistry {
// DeepSeek V3 parser - Unicode tokens with JSON blocks
self.register_parser("deepseek", Arc::new(DeepSeekParser::new()));
// Step3 parser - StepTML XML format
self.register_parser("step3", Arc::new(Step3Parser::new()));
// Kimi K2 parser - Token-based with indexed functions
self.register_parser("kimik2", Arc::new(KimiK2Parser::new()));
}
/// Register default model mappings