Skip to content

GitLab

Unverified Commit d82b0050 authored by MatejKosec's avatar MatejKosec Committed by GitHub
Browse files

feat: interleaved thinking support in reasoning parser (#6422) 


Signed-off-by: default avatarMatej Kosec <mkosec@nvidia.com>
parent 7409bd3a
Expand all Show whitespace changes
Inline Side-by-side
Showing with 662 additions and 105 deletions
lib/parsers/src/reasoning/base_parser.rs
View file @ d82b0050
This diff is collapsed.
lib/parsers/src/reasoning/mod.rs
View file @ d82b0050
...@@ -30,6 +30,7 @@ fn get_reasoning_parser_map() -> &'static HashMap<&'static str, ReasoningParserT ...@@ -30,6 +30,7 @@ fn get_reasoning_parser_map() -> &'static HashMap<&'static str, ReasoningParserT
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.insert("nemotron_nano", ReasoningParserType::NemotronDeci); // nemotron nano is <think>...</think>
map.insert("glm45", ReasoningParserType::NemotronDeci); // GLM-4.5/5 is <think>...</think>, no force_reasoning
map.insert( map.insert(
"minimax_append_think", "minimax_append_think",
ReasoningParserType::MiniMaxAppendThink, ReasoningParserType::MiniMaxAppendThink,
...@@ -225,6 +226,7 @@ mod tests { ...@@ -225,6 +226,7 @@ mod tests {
"mistral", "mistral",
"granite", "granite",
"nemotron_nano", "nemotron_nano",
"glm45",
"minimax_append_think", "minimax_append_think",
]; ];
for parser in available_parsers { for parser in available_parsers {
......
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