Skip to content

Commit

Permalink
fix: When the thinking process label is empty, do not process the thi…
Browse files Browse the repository at this point in the history
…nking process (#2179)
  • Loading branch information
shaohuzhang1 authored Feb 9, 2025
1 parent d9abe8d commit e1f6502
Showing 1 changed file with 12 additions and 3 deletions.
15 changes: 12 additions & 3 deletions apps/application/flow/tools.py
Original file line number Diff line number Diff line change
Expand Up @@ -23,14 +23,23 @@ def __init__(self, reasoning_content_start, reasoning_content_end):
self.all_content = ""
self.reasoning_content_start_tag = reasoning_content_start
self.reasoning_content_end_tag = reasoning_content_end
self.reasoning_content_start_tag_len = len(reasoning_content_start)
self.reasoning_content_end_tag_len = len(reasoning_content_end)
self.reasoning_content_end_tag_prefix = reasoning_content_end[0]
self.reasoning_content_start_tag_len = len(
reasoning_content_start) if reasoning_content_start is not None else 0
self.reasoning_content_end_tag_len = len(reasoning_content_end) if reasoning_content_end is not None else 0
self.reasoning_content_end_tag_prefix = reasoning_content_end[
0] if self.reasoning_content_end_tag_len > 0 else ''
self.reasoning_content_is_start = False
self.reasoning_content_is_end = False
self.reasoning_content_chunk = ""

def get_reasoning_content(self, chunk):
# 如果没有开始思考过程标签那么就全是结果
if self.reasoning_content_start_tag is None or len(self.reasoning_content_start_tag) == 0:
self.content += chunk.content
return {'content': chunk.content, 'reasoning_content': ''}
# 如果没有结束思考过程标签那么就全部是思考过程
if self.reasoning_content_end_tag is None or len(self.reasoning_content_end_tag) == 0:
return {'content': '', 'reasoning_content': chunk.content}
self.all_content += chunk.content
if not self.reasoning_content_is_start and len(self.all_content) >= self.reasoning_content_start_tag_len:
if self.all_content.startswith(self.reasoning_content_start_tag):
Expand Down

0 comments on commit e1f6502

Please sign in to comment.