refactor(custom-agent): improve response action handling and add removeLastMessage function #51

Author: phodal

src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentChatProcessor.kt

@@ -1,17 +1,14 @@
 package cc.unitmesh.devti.counit
 
-import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.counit.model.CustomAgentConfig
 import cc.unitmesh.devti.counit.model.CustomAgentState
 import cc.unitmesh.devti.counit.model.ResponseAction
 import cc.unitmesh.devti.gui.chat.ChatCodingPanel
 import cc.unitmesh.devti.gui.chat.ChatContext
 import cc.unitmesh.devti.provider.ContextPrompter
-import cc.unitmesh.devti.util.LLMCoroutineScope
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
-import kotlinx.coroutines.launch
 import kotlinx.coroutines.runBlocking
 
 @Service(Service.Level.PROJECT)
@@ -45,13 +42,6 @@ class CustomAgentChatProcessor(val project: Project) {
                 ui.updateUI()
             }
 
-            ResponseAction.Stream -> {
-                ui.addMessage(AutoDevBundle.message("autodev.loading"))
-                LLMCoroutineScope.scope(project).launch {
-//                    ui.updateMessage(response)
-                }
-            }
-
             ResponseAction.TextChunk -> {
                 val sb = StringBuilder()
                 runBlocking {

src/main/kotlin/cc/unitmesh/devti/counit/model/AgentResponseAction.kt

@@ -13,11 +13,6 @@ enum class ResponseAction {
      */
     Direct,
 
-    /**
-     * Stream response
-     */
-    Stream,
-
     /**
      * Text splitting result
      */

src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatCodingPanel.kt

@@ -255,6 +255,11 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         progressBar.isVisible = false
     }
 
+    /**
+     * Remove the last message from the list and update the UI.
+     *
+     * @return void
+     */
     fun removeLastMessage() {
         if (myList.componentCount > 0) {
             myList.remove(myList.componentCount - 1)