feat(agent): add display message handling for custom agents #379

Enhance custom agent execution by adding display message accumulation via StringBuilder. This allows for better message tracking and display in the UI. Also includes minor refactoring of prompt handling and adds a showLoading method to NormalChatCodingPanel.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/agent/custom/CustomAgentChatProcessor.kt

@@ -29,20 +29,23 @@ class CustomAgentChatProcessor(val project: Project) {
 
     fun handleChat(prompter: ContextPrompter, ui: NormalChatCodingPanel, llmProvider: LLMProvider): String? {
         val originPrompt = prompter.requestPrompt()
-        val displayMessage = originPrompt
+        val displayMessage = StringBuilder()
 
         val request = originPrompt.trim()
         val selectedAgent: CustomAgentConfig = ui.getSelectedCustomAgent()
 
         selectedAgent.state = CustomAgentState.HANDLING
 
-        val response: Flow<String>? = customAgentExecutor.execute(request, selectedAgent)
+        ui.showLoading()
+
+        val response: Flow<String>? = customAgentExecutor.execute(request, selectedAgent, displayMessage)
         if (response == null) {
             logger.error("error for custom agent: $selectedAgent with request: $request")
             return null
         }
 
-        ui.addMessage(originPrompt, true, displayMessage)
+        val message = displayMessage.toString()
+        ui.addMessage(message, true, message)
 
         var llmResponse = ""
         selectedAgent.state = CustomAgentState.FINISHED

core/src/main/kotlin/cc/unitmesh/devti/agent/custom/CustomAgentExecutor.kt

@@ -10,7 +10,6 @@ import cc.unitmesh.devti.llms.custom.Message
 import cc.unitmesh.devti.llms.custom.updateCustomFormat
 import cc.unitmesh.devti.provider.devins.CustomAgentContext
 import cc.unitmesh.devti.provider.devins.LanguageProcessor
-import cc.unitmesh.devti.util.readText
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
@@ -32,37 +31,37 @@ class CustomAgentExecutor(val project: Project) : CustomSSEProcessor(project) {
     override var requestFormat: String = ""
     override var responseFormat: String = ""
 
-    fun execute(promptText: String, agent: CustomAgentConfig): Flow<String>? {
-        var prompt = promptText
-
+    fun execute(promptText: String, agent: CustomAgentConfig, displayMessage: StringBuilder): Flow<String>? {
         if (agent.isFromDevIns) {
             val devin = LanguageProcessor.devin()!!
             val file = project.baseDir.findFileByRelativePath(agent.devinScriptPath)!!
-            prompt = runBlocking {
+            val prompt = runBlocking {
                 val context = CustomAgentContext(
                     agent, "", filePath = file,
                     initVariables = mapOf("input" to promptText)
                 )
                 devin.execute(project, context)
             }
 
+            displayMessage.append(prompt)
             messages.add(Message("user", prompt))
             return LlmFactory.create(project).stream(prompt, "")
         }
 
+        displayMessage.append(promptText)
         messages.add(Message("user", promptText))
 
         this.requestFormat = agent.connector?.requestFormat ?: this.requestFormat
         this.responseFormat = agent.connector?.responseFormat ?: this.responseFormat
 
-        val customRequest = CustomRequest(listOf(Message("user", prompt)))
+        val customRequest = CustomRequest(listOf(Message("user", promptText)))
         var request = if (requestFormat.isNotEmpty()) {
             customRequest.updateCustomFormat(requestFormat)
         } else {
             Json.encodeToString<CustomRequest>(customRequest)
         }
 
-        request = replacePlaceholders(request, prompt)
+        request = replacePlaceholders(request, promptText)
 
         val body = request.toRequestBody("application/json".toMediaTypeOrNull())
         val builder = Request.Builder()

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

@@ -190,6 +190,10 @@ class NormalChatCodingPanel(private val chatCodingService: ChatCodingService, va
         return messageView
     }
 
+    fun showLoading() {
+
+    }
+
     fun getHistoryMessages(): List<LlmMsg.ChatMessage> {
         val messages = mutableListOf<LlmMsg.ChatMessage>()
         for (i in 0 until myList.componentCount) {

core/src/main/kotlin/cc/unitmesh/devti/intentions/action/task/TestCodeGenTask.kt

@@ -220,7 +220,8 @@ class TestCodeGenTask(val request: TestCodeGenRequest, displayMessage: String) :
         val agent = loadTestRagConfig() ?: return ""
 
         val query = testPromptContext.sourceCode
-        val stringFlow: Flow<String> = CustomAgentExecutor(project).execute(query, agent) ?: return ""
+        val stringFlow: Flow<String> = CustomAgentExecutor(project)
+            .execute(query, agent, StringBuilder()) ?: return ""
 
         val responseBuilder = StringBuilder()
         runBlocking {

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/run/DevInsRunConfigurationProfileState.kt

@@ -40,7 +40,6 @@ import com.intellij.openapi.vfs.VirtualFileManager
 import com.intellij.ui.components.panels.NonOpaquePanel
 import kotlinx.coroutines.flow.*
 import kotlinx.coroutines.launch
-import kotlinx.coroutines.runBlocking
 import java.awt.BorderLayout
 import javax.swing.JComponent
 
@@ -221,7 +220,8 @@ open class DevInsRunConfigurationProfileState(
         agent: CustomAgentConfig
     ) {
         ApplicationManager.getApplication().invokeLater {
-            val stringFlow: Flow<String>? = CustomAgentExecutor(project = myProject).execute(output, agent)
+            val stringFlow: Flow<String>? = CustomAgentExecutor(project = myProject)
+                .execute(output, agent, StringBuilder())
             if (stringFlow != null) {
                 AutoDevCoroutineScope.scope(myProject).launch {
                     val llmResult = StringBuilder()

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/run/runner/CustomRemoteAgentLlmExecutor.kt

@@ -17,7 +17,8 @@ class CustomRemoteAgentLlmExecutor(
 ) : ShireLlmExecutor(context) {
     override fun execute(postFunction: PostFunction) {
         ApplicationManager.getApplication().invokeLater {
-            val stringFlow: Flow<String>? = CustomAgentExecutor(project = context.myProject).execute(context.prompt, agent)
+            val stringFlow: Flow<String>? = CustomAgentExecutor(project = context.myProject)
+                .execute(context.prompt, agent, StringBuilder())
 
             val console = context.console
             if (stringFlow == null) {