.`i6 F d dl mZ d dlmZ d dlmZ G d de ZdS ) )Sequence) DeltaMessage)BaseThinkingReasoningParserc e Zd ZdZedefd Zedefd Zdedededee d ee d
ee de
d z f fd Z xZ 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 dS )z(The token that starts reasoning content.z selfs /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/vllm/reasoning/deepseek_r1_reasoning_parser.py start_tokenz%DeepSeekR1ReasoningParser.start_token s y c dS )z&The token that ends reasoning content.zr
r s r
end_tokenz#DeepSeekR1ReasoningParser.end_token s zr
previous_text current_text
delta_textprevious_token_idscurrent_token_idsdelta_token_idsNc t |||||| }|| j |vr| j |vr| j |v rX| | j }|d | } ||t
| j z d }
t | |
r|
nd S | j |v rt | S t | S |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__s r
r z5DeepSeekR1ReasoningParser.extract_reasoning_streaming s gg11
O#+===#?:: O33 'OODN ; ; &z z 2 $YT^1D1D%D%F%FG# ''. 8GGD "&888#J7777 $j9999
r )
__name__
__module__ __qualname____doc__propertystrr r r intr r
__classcell__)r# s @r
r r
s S X 3 X'' ' '
%SM ' $C=
' "# '
' ' ' ' ' ' ' ' ' 'r r N)collections.abcr 'vllm.entrypoints.openai.engine.protocolr vllm.reasoning.basic_parsersr r r
r r
r/ sr % $ $ $ $ $ @ @ @ @ @ @ D D D D D D9 9 9 9 9 ; 9 9 9 9 9r