Skip to content

Commit 4dc6528

Browse files
authored
Support MiniMax tool call format
1 parent 021c229 commit 4dc6528

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

include/minja/chat-template.hpp

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -198,12 +198,14 @@ class chat_template {
198198
dummy_user_msg,
199199
make_tool_calls_msg(json::array({make_tool_call("ipython", dummy_args_obj.dump())})),
200200
}), {}, false);
201-
auto tool_call_renders_str_arguments = contains(out, "<parameter=argument_needle>") || contains(out, "\"argument_needle\":") || contains(out, "'argument_needle':");
201+
auto tool_call_renders_str_arguments = contains(out, "<parameter=argument_needle>") || contains(out, "\"argument_needle\":")
202+
|| contains(out, "'argument_needle':") || contains(out, "<parameter name=\"argument_needle\">");
202203
out = try_raw_render(json::array({
203204
dummy_user_msg,
204205
make_tool_calls_msg(json::array({make_tool_call("ipython", dummy_args_obj)})),
205206
}), {}, false);
206-
auto tool_call_renders_obj_arguments = contains(out, "<parameter=argument_needle>") || contains(out, "\"argument_needle\":") || contains(out, "'argument_needle':");
207+
auto tool_call_renders_obj_arguments = contains(out, "<parameter=argument_needle>") || contains(out, "\"argument_needle\":")
208+
|| contains(out, "'argument_needle':") || contains(out, "<parameter name=\"argument_needle\">");
207209

208210
caps_.supports_tool_calls = tool_call_renders_str_arguments || tool_call_renders_obj_arguments;
209211
caps_.requires_object_arguments = !tool_call_renders_str_arguments && tool_call_renders_obj_arguments;

0 commit comments

Comments
 (0)