Skip to content

Commit 548ac5a

Browse files
author
ochafik
committed
fix test-chat
1 parent e44a8eb commit 548ac5a

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

tests/test-chat.cpp

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -392,22 +392,16 @@ static void test_template_output_parsers() {
392392
inputs_tools_builtin.tools.push_back(python_tool);
393393

394394
{
395+
// Not supported yet
395396
const common_chat_template tmpl(read_file("models/templates/CohereForAI-c4ai-command-r-plus-tool_use.jinja"), "<s>", "</s>");
396-
assert_equals(COMMON_CHAT_FORMAT_GENERIC, common_chat_params_init(tmpl, inputs_no_tools).format);
397+
assert_equals(COMMON_CHAT_FORMAT_GENERIC, common_chat_params_init(tmpl, inputs_tools).format);
397398
}
398399
{
399400
const common_chat_template tmpl(read_file("models/templates/CohereForAI-c4ai-command-r7b-12-2024-tool_use.jinja"), "<s>", "</s>");
400401
std::vector<std::string> end_tokens{ "<|END_OF_TURN_TOKEN|>" };
401402

402403
assert_equals(COMMON_CHAT_FORMAT_CONTENT_ONLY, common_chat_params_init(tmpl, inputs_no_tools).format);
403404
assert_equals(COMMON_CHAT_FORMAT_COMMAND_R7B, common_chat_params_init(tmpl, inputs_tools).format);
404-
assert_equals(
405-
COMMON_CHAT_FORMAT_COMMAND_R7B,
406-
common_chat_params_init(
407-
common_chat_template(read_file("models/templates/CohereForAI-c4ai-command-r-plus-tool_use.jinja"),
408-
"<s>", "</s>"),
409-
inputs_tools)
410-
.format);
411405

412406
test_template(tmpl, end_tokens, tool_call_plan_message_with_idx, tools,
413407
"<|START_THINKING|>I'm not so sure<|END_THINKING|>"

0 commit comments

Comments
 (0)