🚸 fix constant re-rendering of code vision in jetbrains

continuedev · Mar 15, 2024 · 7955821 · 7955821
1 parent f5305ab
commit 7955821
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/...tlin/com/github/continuedev/continueintellijextension/autocomplete/AutocompleteService.kt b/...tlin/com/github/continuedev/continueintellijextension/autocomplete/AutocompleteService.kt
@@ -105,7 +105,15 @@ class AutocompleteService(private val project: Project) {
             cancelCompletion(pendingCompletion!!)
             pendingCompletion = null
         }
-        editor.inlayModel.getInlineElementsInRange(0, editor.document.textLength).forEach { it.dispose() }
-        editor.inlayModel.getBlockElementsInRange(0, editor.document.textLength).forEach { it.dispose() }
+        editor.inlayModel.getInlineElementsInRange(0, editor.document.textLength).forEach {
+            if (it.renderer is ContinueCustomElementRenderer) {
+                it.dispose()
+            }
+        }
+        editor.inlayModel.getBlockElementsInRange(0, editor.document.textLength).forEach {
+            if (it.renderer is ContinueMultilineCustomElementRenderer) {
+                it.dispose()
+            }
+        }
     }
 }