dylanyunlong/project_6_89d52222
1
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

[fix] </think> to none

Browse Source
This commit is contained in:
root
2026-09-02 06:43:19 +00:00
parent 2a85e77c6f
commit d83ab51908
2 changed files with 16 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
qwen3_6_scripts/reasoning/qwen3_reasoning_parser.py
View File

@@ -46,10 +46,14 @@ class Qwen3ReasoningParser(BaseThinkingReasoningParser):
if self.end_token not in model_output:
# Thinking enabled but output truncated before </think>.
return model_output, None
return model_output or None, None
reasoning, _, content = model_output.partition(self.end_token)
return reasoning, content or None
# Normalize empty strings to None. When the model skips thinking
# (outputs </think> immediately), reasoning is "" — returning None
# keeps reasoning_content out of the JSON response so the test
# framework sees "no reasoning" rather than "empty reasoning".
return reasoning or None, content or None
def count_reasoning_tokens(self, token_ids: Sequence[int]) -> int:
token_ids = list(token_ids)
@@ -109,4 +113,4 @@ class Qwen3ReasoningParser(BaseThinkingReasoningParser):
# Register immediately when this module is imported.
ReasoningParserManager.register_module("qwen3", Qwen3ReasoningParser)
ReasoningParserManager.register_module("qwen3", Qwen3ReasoningParser)