diff --git a/pkg/provider/runners/difysvapi.py b/pkg/provider/runners/difysvapi.py
index 3b072bc2..1a443a16 100644
--- a/pkg/provider/runners/difysvapi.py
+++ b/pkg/provider/runners/difysvapi.py
@@ -406,7 +406,7 @@ class DifyServiceAPIRunner(runner.RequestRunner):
             if chunk['event'] == 'message':
                 message_idx += 1
                 if remove_think:
-                    if message_idx == 1:
+                    if '<think>' in chunk['answer'] and not think_start:
                         think_start = True
                         continue
                     if '</think>' in chunk['answer'] and not think_end: