@@ -23,14 +23,23 @@ def __init__(self, reasoning_content_start, reasoning_content_end):
2323 self .all_content = ""
2424 self .reasoning_content_start_tag = reasoning_content_start
2525 self .reasoning_content_end_tag = reasoning_content_end
26- self .reasoning_content_start_tag_len = len (reasoning_content_start )
27- self .reasoning_content_end_tag_len = len (reasoning_content_end )
28- self .reasoning_content_end_tag_prefix = reasoning_content_end [0 ]
26+ self .reasoning_content_start_tag_len = len (
27+ reasoning_content_start ) if reasoning_content_start is not None else 0
28+ self .reasoning_content_end_tag_len = len (reasoning_content_end ) if reasoning_content_end is not None else 0
29+ self .reasoning_content_end_tag_prefix = reasoning_content_end [
30+ 0 ] if self .reasoning_content_end_tag_len > 0 else ''
2931 self .reasoning_content_is_start = False
3032 self .reasoning_content_is_end = False
3133 self .reasoning_content_chunk = ""
3234
3335 def get_reasoning_content (self , chunk ):
36+ # 如果没有开始思考过程标签那么就全是结果
37+ if self .reasoning_content_start_tag is None or len (self .reasoning_content_start_tag ) == 0 :
38+ self .content += chunk .content
39+ return {'content' : chunk .content , 'reasoning_content' : '' }
40+ # 如果没有结束思考过程标签那么就全部是思考过程
41+ if self .reasoning_content_end_tag is None or len (self .reasoning_content_end_tag ) == 0 :
42+ return {'content' : '' , 'reasoning_content' : chunk .content }
3443 self .all_content += chunk .content
3544 if not self .reasoning_content_is_start and len (self .all_content ) >= self .reasoning_content_start_tag_len :
3645 if self .all_content .startswith (self .reasoning_content_start_tag ):
0 commit comments