minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 13 Packages Projects Releases Wiki Activity

fix KimiK2Detector regex patterns with re.DOTALL (#19120)

Browse Source 
Signed-off-by: Xinyuan Tong <xinyuantong.cs@gmail.com>
This commit is contained in:
Xinyuan Tong
2026-02-21 22:13:08 +08:00
committed by GitHub
parent 4a362a0e04
commit 677b66af80
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/srt/function_call/kimik2_detector.py
+5 -3
View File
@@ -40,11 +40,13 @@ class KimiK2Detector(BaseFormatDetector):
self.tool_call_end_token: str = "<|tool_call_end|>" self.tool_call_end_token: str = "<|tool_call_end|>"
self.tool_call_regex = re.compile( self.tool_call_regex = re.compile(
r"<\|tool_call_begin\|>\s*(?P<tool_call_id>[\w\.]+:\d+)\s*<\|tool_call_argument_begin\|>\s*(?P<function_arguments>\{.*?\})\s*<\|tool_call_end\|>" r"<\|tool_call_begin\|>\s*(?P<tool_call_id>[\w\.]+:\d+)\s*<\|tool_call_argument_begin\|>\s*(?P<function_arguments>\{.*?\})\s*<\|tool_call_end\|>",
re.DOTALL,
) )
self.stream_tool_call_portion_regex = re.compile( self.stream_tool_call_portion_regex = re.compile(
r"<\|tool_call_begin\|>\s*(?P<tool_call_id>[\w\.]+:\d+)\s*<\|tool_call_argument_begin\|>\s*(?P<function_arguments>\{.*)" r"<\|tool_call_begin\|>\s*(?P<tool_call_id>[\w\.]+:\d+)\s*<\|tool_call_argument_begin\|>\s*(?P<function_arguments>\{.*)",
re.DOTALL,
) )
self._last_arguments = "" self._last_arguments = ""
@@ -87,7 +89,7 @@ class KimiK2Detector(BaseFormatDetector):
function_name = m.group("name") function_name = m.group("name")
function_idx = int(m.group("index")) function_idx = int(m.group("index"))
logger.info(f"function_name {function_name}") logger.debug(f"function_name {function_name}")
tool_calls.append( tool_calls.append(
ToolCallItem( ToolCallItem(