o
-i6 @ s8 d dl mZ d dlmZ d dlmZ G dd deZdS ) )Sequence) DeltaMessage)BaseThinkingReasoningParserc sr e Zd ZdZedefddZedefddZdeded ed
ee d ee d ee de
d
B f fdd Z Z S )DeepSeekR1ReasoningParserz
Reasoning parser for DeepSeek R1 model.
The DeepSeek R1 model uses ... tokens to denote reasoning
text. This parser extracts the reasoning content from the model output.
returnc C dS )z(The token that starts reasoning content.z selfr r h/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/vllm/reasoning/deepseek_r1_reasoning_parser.py start_token z%DeepSeekR1ReasoningParser.start_tokenc C r )z&The token that ends reasoning content.zr r r r r end_token r
z#DeepSeekR1ReasoningParser.end_token
previous_text current_text
delta_textprevious_token_idscurrent_token_idsdelta_token_idsNc s t ||||||}|d urP| j|vrP| j|vrP| j|v rA|| j}|d | } ||t| j d }
t| |
r=|
dS d dS | j|v rKt|dS t|dS |S )N) reasoningcontent)r )r )superextract_reasoning_streamingstart_token_id end_token_idfindr lenr ) r
r r r r r r ret end_indexr r __class__r r r s2
z5DeepSeekR1ReasoningParser.extract_reasoning_streaming)
__name__
__module__ __qualname____doc__propertystrr r r intr r
__classcell__r r r r r
s* r N)collections.abcr 'vllm.entrypoints.openai.engine.protocolr vllm.reasoning.basic_parsersr r r r r r s