diff --git a/core/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomSSEProcessor.kt b/core/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomSSEProcessor.kt
index b6bd860f3..028bbd447 100644
--- a/core/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomSSEProcessor.kt
+++ b/core/src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomSSEProcessor.kt
@@ -114,6 +114,8 @@ open class CustomSSEProcessor(private val project: Project) {
 
                                 val completion = result.choices[0].message
                                 if (completion != null && completion.content != null) {
+                                    hasSuccessRequest = true
+
                                     output += completion.content
                                     trySend(completion.content)
                                 }