biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[Bugfix] deepseek_r1_reasoning_parser put reason content in wrong field in certain edge case (#13097)

Browse Source
This commit is contained in:
LikeSundayLikeRain
2025-02-13 02:11:26 -05:00
committed by GitHub
parent 60c68df6d1
commit 04f50ad9d1
2 changed files with 6 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
vllm/entrypoints/openai/reasoning_parsers/deepseek_r1_reasoning_parser.py
View File

@@ -128,7 +128,7 @@ class DeepSeekR1ReasoningParser(ReasoningParser):
# Thus we assume the reasoning content is always at the start.
# Ref https://huggingface.co/deepseek-ai/DeepSeek-R1/commit/8a58a132790c9935686eb97f042afa8013451c9f
if self.think_end_token not in model_output:
return None, model_output
return model_output, None
else:
# Add a start token if it's missing to keep compatibility.
if self.think_start_token not in model_output: