Unverified Commit 648e1cd2 authored by Ayush Agarwal's avatar Ayush Agarwal Committed by GitHub
Browse files

chore: configure nemotron nano parsers (#5396)


Signed-off-by: default avatarayushag <ayushag@nvidia.com>
parent c3dc3de4
......@@ -27,6 +27,7 @@ fn get_reasoning_parser_map() -> &'static HashMap<&'static str, ReasoningParserT
map.insert("step3", ReasoningParserType::Step3);
map.insert("mistral", ReasoningParserType::Mistral);
map.insert("granite", ReasoningParserType::Granite);
map.insert("nemotron_nano", ReasoningParserType::NemotronDeci); // nemotron nano is <think>...</think>
map
})
}
......@@ -213,6 +214,7 @@ mod tests {
"step3",
"mistral",
"granite",
"nemotron_nano",
];
for parser in available_parsers {
assert!(parsers.contains(&parser));
......
......@@ -43,6 +43,7 @@ pub fn get_tool_parser_map() -> &'static HashMap<&'static str, ToolCallConfig> {
map.insert("qwen3_coder", ToolCallConfig::qwen3_coder());
map.insert("jamba", ToolCallConfig::jamba());
map.insert("default", ToolCallConfig::default());
map.insert("nemotron_nano", ToolCallConfig::qwen3_coder()); // nemotron nano follows qwen3_coder format
map
})
}
......@@ -207,6 +208,7 @@ mod tests {
"deepseek_v3_2",
"qwen3_coder",
"jamba",
"nemotron_nano",
];
for parser in available_parsers {
assert!(parsers.contains(&parser));
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment