feat(devins-language): add support for custom agent execution #101

Add support for executing custom agents for language compilation, improving interaction with the console.

Author: phodal

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

@@ -1,6 +1,8 @@
 package cc.unitmesh.devti.language.run
 
 // DONT REMOVE THIS IMPORT
+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.llms.LLMProvider
@@ -25,8 +27,8 @@ import com.intellij.openapi.project.Project
 import com.intellij.openapi.vfs.VirtualFileManager
 import com.intellij.psi.PsiManager
 import com.intellij.ui.components.panels.NonOpaquePanel
+import kotlinx.coroutines.flow.Flow
 import kotlinx.coroutines.launch
-import kotlinx.coroutines.flow.collect
 import kotlinx.coroutines.runBlocking
 import java.awt.BorderLayout
 import java.io.OutputStream
@@ -72,6 +74,57 @@ open class DevInsRunConfigurationProfileState(
 
         val output = compileResult.output
 
+        val agent = compileResult.workingAgent
+
+        if (agent != null) {
+            agentRun(output, console, processHandler, agent)
+        } else {
+            defaultRun(output, console, processHandler, compileResult.isLocalCommand)
+        }
+
+        return DefaultExecutionResult(console, processHandler)
+    }
+
+    private fun agentRun(
+        output: String,
+        console: ConsoleViewWrapperBase,
+        processHandler: ProcessHandler,
+        agent: CustomAgentConfig
+    ) {
+        output.split("\n").forEach {
+            if (it.contains("<DevInsError>")) {
+                console.print(it, ConsoleViewContentType.LOG_ERROR_OUTPUT)
+            } else {
+                console.print(it, ConsoleViewContentType.USER_INPUT)
+            }
+            console.print("\n", ConsoleViewContentType.NORMAL_OUTPUT)
+        }
+
+        console.print("\n--------------------\n", ConsoleViewContentType.NORMAL_OUTPUT)
+
+        ApplicationManager.getApplication().invokeLater {
+            val stringFlow: Flow<String>? = CustomAgentExecutor(project = myProject).execute(output, agent)
+            if (stringFlow != null) {
+                LLMCoroutineScope.scope(myProject).launch {
+                    runBlocking {
+                        stringFlow.collect {
+                            console.print(it, ConsoleViewContentType.NORMAL_OUTPUT)
+                        }
+                    }
+
+                    console.print("\nDone!", ConsoleViewContentType.SYSTEM_OUTPUT)
+                    processHandler.detachProcess()
+                }
+            }
+        }
+    }
+
+    private fun defaultRun(
+        output: String,
+        console: ConsoleViewWrapperBase,
+        processHandler: ProcessHandler,
+        isLocalMode: Boolean
+    ) {
         // contains <DevInsError> means error
         output.split("\n").forEach {
             if (it.contains("<DevInsError>")) {
@@ -85,7 +138,7 @@ open class DevInsRunConfigurationProfileState(
         console.print("\n--------------------\n", ConsoleViewContentType.NORMAL_OUTPUT)
 
         ApplicationManager.getApplication().invokeLater {
-            if (compileResult.isLocalCommand) {
+            if (isLocalMode) {
                 console.print("Local command detected, running in local mode", ConsoleViewContentType.SYSTEM_OUTPUT)
                 processHandler.detachProcess()
                 return@invokeLater
@@ -102,8 +155,6 @@ open class DevInsRunConfigurationProfileState(
                 processHandler.detachProcess()
             }
         }
-
-        return DefaultExecutionResult(console, processHandler)
     }
 
     @Throws(ExecutionException::class)