Skip to content

Commit 150199c

Browse files
Merge PR ggml-org#16932 (xml_toolcall, commit e5529dd) into testing-branch16 — unified XML tool-call parser + streaming reasoning + GLM4.5/4.6 + MiniMax-M2
1 parent 120b01e commit 150199c

File tree

11 files changed

+1446
-105
lines changed

11 files changed

+1446
-105
lines changed

common/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -50,6 +50,8 @@ add_library(${TARGET} STATIC
5050
base64.hpp
5151
chat-parser.cpp
5252
chat-parser.h
53+
chat-parser-xml-toolcall.h
54+
chat-parser-xml-toolcall.cpp
5355
chat.cpp
5456
chat.h
5557
common.cpp

common/chat-parser-xml-toolcall.cpp

Lines changed: 710 additions & 0 deletions
Large diffs are not rendered by default.

common/chat-parser-xml-toolcall.h

Lines changed: 37 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,37 @@
1+
#pragma once
2+
3+
#include "chat.h"
4+
5+
#include <nlohmann/json.hpp>
6+
7+
#include <optional>
8+
#include <string>
9+
#include <vector>
10+
11+
12+
// Sample config:
13+
// MiniMax-M2 (left): <minimax:tool_call>\n<invoke name="tool-name">\n<parameter name="key">value</parameter>\n...</invoke>\n...</minimax:tool_call>
14+
// GLM 4.5 (right): <tool_call>function_name\n<arg_key>key</arg_key>\n<arg_value>value</arg_value>\n</tool_call>
15+
struct xml_tool_call_format {
16+
std::string scope_start; // <minimax:tool_call>\n // \n // can be empty
17+
std::string tool_start; // <invoke name=\" // <tool_call>
18+
std::string tool_sep; // \">\n // \n // can be empty only for parse_xml_tool_calls
19+
std::string key_start; // <parameter name=\" // <arg_key>
20+
std::string key_val_sep; // \"> // </arg_key>\n<arg_value>
21+
std::string val_end; // </parameter>\n // </arg_value>\n
22+
std::string tool_end; // </invoke>\n // </tool_call>\n
23+
std::string scope_end; // </minimax:tool_call> // // can be empty
24+
// Set this if there can be dynamic spaces inside key_val_sep.
25+
// e.g. key_val_sep=</arg_key> key_val_sep2=<arg_value> for GLM4.5
26+
std::optional<std::string> key_val_sep2 = std::nullopt;
27+
bool allow_toolcall_in_think = false; // TODO: UNTESTED!!!
28+
};
29+
30+
// make a GBNF that accept any strings except those containing any of the forbidden strings.
31+
std::string make_gbnf_excluding(std::vector<std::string> forbids);
32+
33+
/**
34+
* Build grammar for xml-style tool call
35+
* form.scope_start and form.scope_end can be empty.
36+
*/
37+
void build_grammar_xml_tool_call(common_chat_params & data, const nlohmann::ordered_json & tools, const struct xml_tool_call_format & form);

common/chat-parser.h

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
#pragma once
22

33
#include "chat.h"
4+
#include "chat-parser-xml-toolcall.h"
45
#include "json-partial.h"
56
#include "regex-partial.h"
67

@@ -119,5 +120,14 @@ class common_chat_msg_parser {
119120
const std::vector<std::vector<std::string>> & content_paths = {}
120121
);
121122

123+
/**
124+
* Parse XML-Style tool call for given xml_tool_call_format. Return false for invalid syntax and get the position untouched.
125+
* form.scope_start, form.tool_sep and form.scope_end can be empty.
126+
*/
127+
bool try_consume_xml_tool_calls(const struct xml_tool_call_format & form);
128+
129+
// Parse content uses reasoning and XML-Style tool call
130+
void consume_reasoning_with_xml_tool_calls(const struct xml_tool_call_format & form, const std::string & start_think = "<think>", const std::string & end_think = "</think>");
131+
122132
void clear_tools();
123133
};

0 commit comments

Comments
 (0)