feat(devins-lang): improve conversation service and compiler #100

- Refactored createConversation method to accept result object instead of input string and script path, and updated related methods to reflect the change. Added tryReRun method to handle re-running conversations.

This commit improves the conversation service and compiler by refactoring the createConversation method to accept a result object instead of an input string and script path. It also adds a tryReRun method to handle re-running conversations.

Author: phodal

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

@@ -4,6 +4,7 @@ import cc.unitmesh.devti.agent.CustomAgentExecutor
 import cc.unitmesh.devti.agent.model.CustomAgentConfig
 import cc.unitmesh.devti.language.compiler.DevInsCompiler
 import cc.unitmesh.devti.language.psi.DevInFile
+import cc.unitmesh.devti.language.run.flow.DevInsConversationService
 import cc.unitmesh.devti.language.status.DevInsRunListener
 import cc.unitmesh.devti.llms.LLMProvider
 import cc.unitmesh.devti.llms.LlmFactory
@@ -25,6 +26,7 @@ import com.intellij.execution.ui.ConsoleViewContentType
 import com.intellij.openapi.actionSystem.ActionManager
 import com.intellij.openapi.actionSystem.DefaultActionGroup
 import com.intellij.openapi.application.ApplicationManager
+import com.intellij.openapi.components.service
 import com.intellij.openapi.project.Project
 import com.intellij.ui.components.panels.NonOpaquePanel
 import kotlinx.coroutines.flow.*
@@ -89,6 +91,8 @@ open class DevInsRunConfigurationProfileState(
         val compiler = DevInsCompiler(myProject, file)
         val compileResult = compiler.compile()
 
+        myProject.service<DevInsConversationService>().createConversation(configuration.getScriptPath(), compileResult)
+
         val output = compileResult.output
         val agent = compileResult.workingAgent
 
@@ -122,13 +126,17 @@ open class DevInsRunConfigurationProfileState(
             val stringFlow: Flow<String>? = CustomAgentExecutor(project = myProject).execute(output, agent)
             if (stringFlow != null) {
                 LLMCoroutineScope.scope(myProject).launch {
+                    val llmResult = StringBuilder()
                     runBlocking {
                         stringFlow.collect {
+                            llmResult.append(it)
                             console.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
                         }
                     }
 
                     console.print("\nDone!", ConsoleViewContentType.SYSTEM_OUTPUT)
+                    myProject.service<DevInsConversationService>()
+                        .updateLlmResponse(configuration.getScriptPath(), llmResult.toString())
                     processHandler.detachProcess()
                 }
             }
@@ -161,13 +169,17 @@ open class DevInsRunConfigurationProfileState(
             }
 
             LLMCoroutineScope.scope(myProject).launch {
+                val llmResult = StringBuilder()
                 runBlocking {
                     llm.stream(output, "").collect {
+                        llmResult.append(it)
                         console.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
                     }
                 }
 
                 console.print("\nDone!", ConsoleViewContentType.SYSTEM_OUTPUT)
+                myProject.service<DevInsConversationService>()
+                    .updateLlmResponse(configuration.getScriptPath(), llmResult.toString())
                 processHandler.detachProcess()
             }
         }

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

@@ -11,8 +11,8 @@ class DevInsConversationService(val project: Project) {
      */
     private val cachedConversations: MutableMap<String, DevInsConversation> = mutableMapOf()
 
-    fun createConversation(input: String, scriptPath: String): DevInsConversation {
-        val conversation = DevInsConversation(input, scriptPath, DevInsCompiledResult(), "", "")
+    fun createConversation(scriptPath: String, result: DevInsCompiledResult): DevInsConversation {
+        val conversation = DevInsConversation(scriptPath, result, "", "")
         cachedConversations[scriptPath] = conversation
         return conversation
     }
@@ -21,15 +21,9 @@ class DevInsConversationService(val project: Project) {
         return cachedConversations[scriptPath]
     }
 
-    fun updateCompiledResult(scriptPath: String, result: DevInsCompiledResult) {
+    fun updateLlmResponse(scriptPath: String, llmResponse: String) {
         cachedConversations[scriptPath]?.let {
-            cachedConversations[scriptPath] = it.copy(result = result)
-        }
-    }
-
-    fun updateApiResponse(scriptPath: String, apiResponse: String) {
-        cachedConversations[scriptPath]?.let {
-            cachedConversations[scriptPath] = it.copy(apiResponse = apiResponse)
+            cachedConversations[scriptPath] = it.copy(llmResponse = llmResponse)
         }
     }
 
@@ -38,13 +32,31 @@ class DevInsConversationService(val project: Project) {
             cachedConversations[path] = it.copy(ideOutput = ideOutput)
         }
     }
+
+    fun tryReRun(scriptPath: String) {
+        if (cachedConversations.isEmpty()) {
+            return
+        }
+
+        val conversation = cachedConversations[scriptPath] ?: return
+        if (conversation.hadReRun) {
+            return
+        }
+
+        conversation.hadReRun = true
+
+        // call llm again to re-run
+    }
 }
 
 
 data class DevInsConversation(
-    val input: String,
     val scriptPath: String,
     val result: DevInsCompiledResult,
-    val apiResponse: String,
-    val ideOutput: String
-)
+    val llmResponse: String,
+    val ideOutput: String,
+    val messages: MutableList<cc.unitmesh.devti.llms.custom.Message> = mutableListOf(),
+    var hadReRun: Boolean = false
+) {
+    // update messages when has Error or Warning
+}

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

@@ -5,6 +5,7 @@ import cc.unitmesh.devti.language.psi.DevInVisitor
 import com.intellij.execution.process.ProcessEvent
 import com.intellij.openapi.application.runReadAction
 import com.intellij.openapi.components.Service
+import com.intellij.openapi.components.service
 import com.intellij.openapi.project.Project
 import com.intellij.psi.PsiComment
 import com.intellij.psi.PsiElement
@@ -30,12 +31,12 @@ class DevInsFlowProcessor(val project: Project) {
      */
     fun process(output: String, event: ProcessEvent, scriptPath: String) {
         val devInFile: DevInFile? = runReadAction { DevInFile.lookup(project, scriptPath) }
+        project.service<DevInsConversationService>().updateIdeOutput(scriptPath, output)
         if (event.exitCode == 0) {
             // continue
         }
         if (event.exitCode != 0) {
-            // stop
-            // call compiler to fix issue
+            project.service<DevInsConversationService>().tryReRun(scriptPath)
         }
     }
 

src/main/kotlin/cc/unitmesh/devti/gui/chat/message/AutoDevRateMessageAction.kt

@@ -18,7 +18,6 @@ enum class ChatMessageRating {
 
 abstract class AutoDevRateMessageAction : DumbAwareToggleAction() {
     override fun getActionUpdateThread(): ActionUpdateThread = ActionUpdateThread.EDT
-
     abstract fun getReaction(): ChatMessageRating
     abstract fun getReactionIcon(): Icon
     abstract fun getReactionIconSelected(): Icon