refactor(SSEProcessor): set success flag on valid completion content

unit-mesh · Oct 23, 2024 · 9893e7a · 9893e7a
1 parent 3ef2312
commit 9893e7a
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -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)
                                 }