🐛 When generating prompts using deep thinking models, it is necessary to remove the deep thinking content. #1229

Author: jiangpeiling

test/backend/services/test_prompt_service.py

@@ -26,7 +26,7 @@
                 get_enabled_sub_agent_description_for_generate_prompt,
                 generate_system_prompt,
                 join_info_for_generate_system_prompt,
-                process_thinking_tokens
+                _process_thinking_tokens
             )
 
 
@@ -493,7 +493,7 @@ def test_process_thinking_tokens_normal_token(self):
         def mock_callback(text):
             callback_calls.append(text)
 
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "Hello", False, token_join, mock_callback)
 
         self.assertFalse(is_thinking)
@@ -508,7 +508,7 @@ def test_process_thinking_tokens_start_thinking(self):
         def mock_callback(text):
             callback_calls.append(text)
 
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "<think>", False, token_join, mock_callback)
 
         self.assertTrue(is_thinking)
@@ -523,7 +523,7 @@ def test_process_thinking_tokens_content_while_thinking(self):
         def mock_callback(text):
             callback_calls.append(text)
 
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "thinking content", True, token_join, mock_callback)
 
         self.assertTrue(is_thinking)
@@ -538,7 +538,7 @@ def test_process_thinking_tokens_end_thinking(self):
         def mock_callback(text):
             callback_calls.append(text)
 
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "</think>", True, token_join, mock_callback)
 
         self.assertFalse(is_thinking)
@@ -553,7 +553,7 @@ def test_process_thinking_tokens_content_after_thinking(self):
         def mock_callback(text):
             callback_calls.append(text)
 
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "World", False, token_join, mock_callback)
 
         self.assertFalse(is_thinking)
@@ -569,32 +569,32 @@ def mock_callback(text):
             callback_calls.append(text)
 
         # Start with normal content
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "Start ", False, token_join, mock_callback)
         self.assertFalse(is_thinking)
 
         # Enter thinking mode
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "<think>", False, token_join, mock_callback)
         self.assertTrue(is_thinking)
 
         # Thinking content (ignored)
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "thinking", True, token_join, mock_callback)
         self.assertTrue(is_thinking)
 
         # More thinking content (ignored)
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             " more", True, token_join, mock_callback)
         self.assertTrue(is_thinking)
 
         # End thinking
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "</think>", True, token_join, mock_callback)
         self.assertFalse(is_thinking)
 
         # Continue with normal content
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             " End", False, token_join, mock_callback)
         self.assertFalse(is_thinking)
 
@@ -606,7 +606,7 @@ def test_process_thinking_tokens_no_callback(self):
         """Test process_thinking_tokens without callback function"""
         token_join = []
 
-        is_thinking = process_thinking_tokens("Hello", False, token_join, None)
+        is_thinking = _process_thinking_tokens("Hello", False, token_join, None)
 
         self.assertFalse(is_thinking)
         self.assertEqual(token_join, ["Hello"])
@@ -619,7 +619,7 @@ def test_process_thinking_tokens_empty_token(self):
         def mock_callback(text):
             callback_calls.append(text)
 
-        is_thinking = process_thinking_tokens(
+        is_thinking = _process_thinking_tokens(
             "", False, token_join, mock_callback)
 
         self.assertFalse(is_thinking)