From e1f65022a4edba835381db20622e5458ef062b9e Mon Sep 17 00:00:00 2001
From: shaohuzhang1 <80892890+shaohuzhang1@users.noreply.github.com>
Date: Sun, 9 Feb 2025 17:35:31 +0800
Subject: [PATCH] fix: When the thinking process label is empty, do not process
 the thinking process (#2179)

---
 apps/application/flow/tools.py | 15 ++++++++++++---
 1 file changed, 12 insertions(+), 3 deletions(-)

diff --git a/apps/application/flow/tools.py b/apps/application/flow/tools.py
index 2d4ebbcfdbd..219f39e88ed 100644
--- a/apps/application/flow/tools.py
+++ b/apps/application/flow/tools.py
@@ -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):