From 478340fe115b414bb828b54fa1769265b175341a Mon Sep 17 00:00:00 2001
From: Carl-Robert Linnupuu <carlrobertoh@gmail.com>
Date: Fri, 1 Nov 2024 14:27:18 +0000
Subject: [PATCH] fix: code completion state when session contains remining
 completion

---
 .../codecompletions/DebouncedCodeCompletionProvider.kt   | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/DebouncedCodeCompletionProvider.kt b/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/DebouncedCodeCompletionProvider.kt
index 635bec2d..b9489a4c 100644
--- a/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/DebouncedCodeCompletionProvider.kt
+++ b/src/main/kotlin/ee/carlrobert/codegpt/codecompletions/DebouncedCodeCompletionProvider.kt
@@ -101,7 +101,14 @@ class DebouncedCodeCompletionProvider : DebouncedInlineCompletionProvider() {
             null -> false
         }
 
-        return event is InlineCompletionEvent.DocumentChange && codeCompletionsEnabled
+        if (!codeCompletionsEnabled) {
+            return false
+        }
+
+        val containsActiveCompletion =
+            REMAINING_EDITOR_COMPLETION.get(event.toRequest()?.editor)?.isNotEmpty() ?: false
+
+        return event is InlineCompletionEvent.DocumentChange || containsActiveCompletion
     }
 
     private fun ProducerScope<InlineCompletionElement>.getEventListener(