Skip to content

Commit e1f6502

Browse files
authored
fix: When the thinking process label is empty, do not process the thinking process (#2179)
1 parent d9abe8d commit e1f6502

File tree

1 file changed

+12
-3
lines changed

1 file changed

+12
-3
lines changed

apps/application/flow/tools.py

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)