feat(shire): implement chat completion task and editor interaction for code suggestions #379

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/gui/AutoDevToolWindowFactory.kt

@@ -71,7 +71,6 @@ class AutoDevToolWindowFactory : ToolWindowFactory, DumbAware {
     }
 
     companion object {
-
         fun labelNormalChat(
             toolWindowManager: ToolWindow,
             chatCodingService: ChatCodingService

core/src/main/kotlin/cc/unitmesh/devti/gui/chat/NormalChatCodingPanel.kt

@@ -18,7 +18,6 @@ import cc.unitmesh.devti.gui.toolbar.NewChatAction
 import cc.unitmesh.devti.provider.TextContextPrompter
 import cc.unitmesh.devti.provider.devins.LanguageProcessor
 import cc.unitmesh.devti.settings.AutoDevSettingsState
-import cc.unitmesh.devti.settings.locale.LanguageChangedCallback.componentStateChanged
 import cc.unitmesh.devti.sketch.createActionButton
 import cc.unitmesh.devti.sketch.ui.code.HtmlHighlightSketch
 import cc.unitmesh.devti.util.whenDisposed

core/src/main/resources/messages/AutoDevBundle_en.properties

@@ -270,3 +270,4 @@ shire.toolchain.function.not.found=Toolchain Not Found
 shire.run.local.mode=Run Local Model
 devins.llm.notfound=LLM not found
 devins.llm.done=Done
+intentions.step.prepare-context=Prepare Context

core/src/main/resources/messages/AutoDevBundle_zh.properties

@@ -260,3 +260,4 @@ shire.toolchain.function.not.found=Toolchain Not Found
 shire.run.local.mode=Run Local Model
 devins.llm.notfound=LLM not found
 devins.llm.done=Done
+intentions.step.prepare-context=Prepare Context

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/actions/DevInsRunFileAction.kt

@@ -6,6 +6,7 @@ import cc.unitmesh.devti.language.psi.DevInFile
 import cc.unitmesh.devti.language.run.DevInsConfiguration
 import cc.unitmesh.devti.language.run.DevInsConfigurationType
 import cc.unitmesh.devti.language.run.DevInsRunConfigurationProducer
+import cc.unitmesh.devti.language.run.runner.ShireConsoleView
 import cc.unitmesh.devti.language.status.DevInsRunListener
 import com.intellij.execution.ExecutionManager
 import com.intellij.execution.RunManager
@@ -105,23 +106,16 @@ class DevInsRunFileAction : DumbAwareAction() {
             val hintDisposable = Disposer.newDisposable()
             val connection = ApplicationManager.getApplication().messageBus.connect(hintDisposable)
             connection.subscribe(DevInsRunListener.TOPIC, object : DevInsRunListener {
-//                override fun runFinish(
-//                    allOutput: String,
-//                    llmOutput: String,
-//                    event: ProcessEvent,
-//                    scriptPath: String,
-//                    consoleView: DevInConsoleView?,
-//                ) {
-//                    future.complete(llmOutput)
-//                    connection.disconnect()
-//                    Disposer.dispose(hintDisposable)
-//                }
                 override fun runFinish(
-                    string: String,
+                    allOutput: String,
+                    llmOutput: String,
                     event: ProcessEvent,
-                    scriptPath: String
+                    scriptPath: String,
+                    consoleView: ShireConsoleView?,
                 ) {
-
+                    future.complete(llmOutput)
+                    connection.disconnect()
+                    Disposer.dispose(hintDisposable)
                 }
             })
 

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/compiler/service/DevInRunService.kt

@@ -1,6 +1,7 @@
 package cc.unitmesh.devti.language.compiler.service
 
 import cc.unitmesh.devti.language.run.DevInsConfiguration
+import cc.unitmesh.devti.language.run.runner.ShireConsoleView
 import cc.unitmesh.devti.language.status.DevInsRunListener
 import cc.unitmesh.devti.provider.RunService
 import com.intellij.execution.ExecutionManager
@@ -63,8 +64,10 @@ class DevInRunService : RunService {
         connection.subscribe(DevInsRunListener.TOPIC, object : DevInsRunListener {
             override fun runFinish(
                 string: String,
+                llmOutput: String,
                 event: ProcessEvent,
-                scriptPath: String
+                scriptPath: String,
+                consoleView: ShireConsoleView?
             ) {
                 future.complete(string)
                 /// append to input box

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/config/ChatCompletionTask.kt

@@ -0,0 +1,111 @@
+package cc.unitmesh.devti.language.config
+
+import cc.unitmesh.devti.AutoDevBundle
+import cc.unitmesh.devti.llms.LlmFactory
+import cc.unitmesh.devti.llms.cancelHandler
+import cc.unitmesh.devti.util.AutoDevCoroutineScope
+import cc.unitmesh.devti.util.parser.CodeFence
+import com.intellij.openapi.util.TextRange
+
+import com.intellij.openapi.actionSystem.CustomShortcutSet
+import com.intellij.openapi.actionSystem.KeyboardShortcut
+import com.intellij.openapi.application.invokeLater
+import com.intellij.openapi.diagnostic.logger
+import com.intellij.openapi.progress.ProgressIndicator
+import com.intellij.openapi.project.DumbAwareAction
+import kotlinx.coroutines.cancel
+import kotlinx.coroutines.flow.*
+import kotlinx.coroutines.launch
+import java.awt.event.KeyEvent
+import javax.swing.KeyStroke
+
+open class ChatCompletionTask(private val request: ShireCodeCompletionRequest) :
+    ShireInteractionTask(request.project, AutoDevBundle.message("intentions.chat.code.complete.name"), request.postExecute) {
+    private val logger = logger<ChatCompletionTask>()
+
+    private var isCanceled: Boolean = false
+
+    private var cancelCallback: ((String) -> Unit)? = null
+
+    override fun run(indicator: ProgressIndicator) {
+        indicator.isIndeterminate = false
+        indicator.fraction = 0.1
+        indicator.text = AutoDevBundle.message("intentions.step.prepare-context")
+
+        val flow: Flow<String> = LlmFactory.create(request.project)!!.stream(request.userPrompt, "", false)
+        logger.info("Prompt: ${request.userPrompt}")
+
+        val shortcutAction = DumbAwareAction.create {
+            isCanceled = true
+        }.apply {
+            registerCustomShortcutSet(
+                CustomShortcutSet(
+                    KeyboardShortcut(KeyStroke.getKeyStroke(KeyEvent.VK_ESCAPE, 0), null),
+                ),
+                request.editor.component
+            )
+        }
+
+        val editor = request.editor
+        val project = request.project
+
+        var currentOffset = request.startOffset
+        val modifyStart = request.startOffset
+
+        indicator.isIndeterminate = false
+        indicator.fraction = 0.5
+        indicator.text = AutoDevBundle.message("intentions.request.background.process.title")
+
+        AutoDevCoroutineScope.scope(request.project).launch {
+            val suggestion = StringBuilder()
+
+            flow.cancelHandler { cancelCallback = it }.cancellable().collect { char ->
+                if (isCanceled) {
+                    cancel()
+                    return@collect
+                }
+
+                suggestion.append(char)
+
+                invokeLater {
+                    if (!isCanceled && !request.isReplacement) {
+                        if (request.isInsertBefore) {
+                            InsertUtil.insertStreamingToDoc(project, char, editor, currentOffset)
+                            currentOffset += char.length
+                        } else {
+                            InsertUtil.insertStreamingToDoc(project, char, editor, currentOffset)
+                            currentOffset += char.length
+                        }
+                    }
+                }
+            }
+
+            val modifyEnd = currentOffset
+
+            if (request.isReplacement) {
+                val parsedContent = CodeFence.parse(suggestion.toString()).text
+                InsertUtil.replaceText(project, editor, parsedContent)
+            }
+
+            indicator.fraction = 0.8
+            logger.info("Suggestion: $suggestion")
+
+            val textRange = TextRange(modifyStart, modifyEnd)
+
+            request.postExecute.invoke(suggestion.toString(), textRange)
+            indicator.fraction = 1.0
+        }.invokeOnCompletion {
+            shortcutAction.unregisterCustomShortcutSet(editor.component)
+        }
+    }
+
+    override fun onThrowable(error: Throwable) {
+        super.onThrowable(error)
+    }
+
+    override fun onCancel() {
+        this.isCanceled = true
+        this.cancelCallback?.invoke("This job is canceled")
+        super.onCancel()
+    }
+}