fix setup snippet_id

smallcloudai · Jan 12, 2024 · 7b5b80e · 7b5b80e
1 parent 6f8a687
commit 7b5b80e
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/main/kotlin/com/smallcloud/refactai/modes/completion/CompletionMode.kt b/src/main/kotlin/com/smallcloud/refactai/modes/completion/CompletionMode.kt
@@ -224,10 +224,6 @@ class CompletionMode(
                 return@streamedInferenceFetch
             }
 
-            if ((!completionInProgress)
-                || (choice.delta.isEmpty() && !choice.finishReason.isNullOrEmpty())) {
-                return@streamedInferenceFetch
-            }
             val completion: Completion = if (completionLayout?.lastCompletionData == null ||
                 completionLayout?.lastCompletionData?.createdTs != prediction.created) {
                 Completion(request.body.inputs.sources.values.toList().first(),
@@ -243,6 +239,10 @@ class CompletionMode(
             if (completion.snippetTelemetryId == null) {
                 completion.snippetTelemetryId = prediction.snippetTelemetryId
             }
+            if ((!completionInProgress)
+                || (choice.delta.isEmpty() && !choice.finishReason.isNullOrEmpty())) {
+                return@streamedInferenceFetch
+            }
             completion.updateCompletion(choice.delta)
             synchronized(this) {
                 renderCompletion(