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 ...@@ -27,6 +27,7 @@ fn get_reasoning_parser_map() -> &'static HashMap<&'static str, ReasoningParserT
map.insert("step3", ReasoningParserType::Step3); map.insert("step3", ReasoningParserType::Step3);
map.insert("mistral", ReasoningParserType::Mistral); map.insert("mistral", ReasoningParserType::Mistral);
map.insert("granite", ReasoningParserType::Granite); map.insert("granite", ReasoningParserType::Granite);
map.insert("nemotron_nano", ReasoningParserType::NemotronDeci); // nemotron nano is <think>...</think>
map map
}) })
} }
...@@ -213,6 +214,7 @@ mod tests { ...@@ -213,6 +214,7 @@ mod tests {
"step3", "step3",
"mistral", "mistral",
"granite", "granite",
"nemotron_nano",
]; ];
for parser in available_parsers { for parser in available_parsers {
assert!(parsers.contains(&parser)); assert!(parsers.contains(&parser));
......
...@@ -43,6 +43,7 @@ pub fn get_tool_parser_map() -> &'static HashMap<&'static str, ToolCallConfig> { ...@@ -43,6 +43,7 @@ pub fn get_tool_parser_map() -> &'static HashMap<&'static str, ToolCallConfig> {
map.insert("qwen3_coder", ToolCallConfig::qwen3_coder()); map.insert("qwen3_coder", ToolCallConfig::qwen3_coder());
map.insert("jamba", ToolCallConfig::jamba()); map.insert("jamba", ToolCallConfig::jamba());
map.insert("default", ToolCallConfig::default()); map.insert("default", ToolCallConfig::default());
map.insert("nemotron_nano", ToolCallConfig::qwen3_coder()); // nemotron nano follows qwen3_coder format
map map
}) })
} }
...@@ -207,6 +208,7 @@ mod tests { ...@@ -207,6 +208,7 @@ mod tests {
"deepseek_v3_2", "deepseek_v3_2",
"qwen3_coder", "qwen3_coder",
"jamba", "jamba",
"nemotron_nano",
]; ];
for parser in available_parsers { for parser in available_parsers {
assert!(parsers.contains(&parser)); 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