feat: Enhance completion provider with error handling

- Add ApplicationManager.invokeLater for UI updates
- Implement exception handling for cancellations and errors
- Improve logging for completion process and errors

Author: yangbobo2021

src/main/kotlin/ai/devchat/plugin/completion/editor/CompletionProvider.kt

@@ -10,6 +10,8 @@ import kotlinx.coroutines.flow.MutableStateFlow
 import kotlinx.coroutines.flow.asStateFlow
 import java.util.concurrent.atomic.AtomicInteger
 import java.util.concurrent.atomic.AtomicReference
+import com.intellij.openapi.application.ApplicationManager
+
 
 @Service
 class CompletionProvider {
@@ -23,34 +25,46 @@ class CompletionProvider {
 
     private val currentContext = AtomicReference<CompletionContext?>(null)
 
-    fun provideCompletion(editor: Editor, offset: Int, manually: Boolean = false) {
-        val currentSequence = completionSequence.incrementAndGet()
-        val agentService = service<AgentService>()
-        val inlineCompletionService = service<InlineCompletionService>()
+fun provideCompletion(editor: Editor, offset: Int, manually: Boolean = false) {
+    logger.info("start provideCompletion")
+    val currentSequence = completionSequence.incrementAndGet()
+    val agentService = service<AgentService>()
+    val inlineCompletionService = service<InlineCompletionService>()
 
-        val oldContext = currentContext.getAndSet(null)
-        oldContext?.job?.cancel()
-        inlineCompletionService.dismiss()
+    val oldContext = currentContext.getAndSet(null)
+    oldContext?.job?.cancel()
+    inlineCompletionService.dismiss()
 
-        val job = agentService.scope.launch {
+    val job = agentService.scope.launch {
+        try {
             logger.info("Trigger completion at $offset")
-            agentService.provideCompletion(editor, offset, manually).let {
-                if (isActive) {  // Check if the job is still active before updating
-                    if (it != null && completionSequence.get() == currentSequence) {
-                        logger.info("Show completion at $offset: $it")
-                        inlineCompletionService.show(editor, offset, it)
-                    }
-                    currentContext.set(null)
-                    ongoingCompletionFlow.value = null
+            val result = agentService.provideCompletion(editor, offset, manually)
+            if (isActive && result != null && completionSequence.get() == currentSequence) {
+                logger.info("Show completion at $offset: $result")
+                ApplicationManager.getApplication().invokeLater {
+                    inlineCompletionService.show(editor, offset, result)
                 }
             }
-        }
+        } catch (e: CancellationException) {
+            // 方案1：以较低的日志级别记录
+            logger.info("Completion was cancelled: ${e.message}")
+            // 或者方案2：完全忽略
+            // // 不做任何处理
 
-        val newContext = CompletionContext(editor, offset, job)
-        currentContext.set(newContext)
-        ongoingCompletionFlow.value = newContext
+            null
+        } catch (e: Exception) {
+            logger.error("Error in completion job", e)
+        } finally {
+            currentContext.set(null)
+            ongoingCompletionFlow.value = null
+        }
     }
 
+    val newContext = CompletionContext(editor, offset, job)
+    currentContext.set(newContext)
+    ongoingCompletionFlow.value = newContext
+}
+
     fun clear() {
         val inlineCompletionService = service<InlineCompletionService>()
         inlineCompletionService.dismiss()