feat(provider): enable DevIn agent responses #101

This commit enables the execution of DevIn agent responses within the custom agent chat processor by adding a new interface for DevIn response providers and implementing a custom agent response provider for DevIn language. The DevIn custom agent response provider executes the DevIn code and notifies the project of the result.

Author: phodal

example/custom_agent/server.py

@@ -93,13 +93,39 @@ def mock_devins(messages: Messages):
 Here are the patches for your
 
 ```devin
-/write:src/main/java/com/example/Controller.java#L1-L12
-\\`\\`\\`java
-public class Controller {
-    public void method() {
-        System.out.println("Hello, World!");
-    }
-}
+/patch
+
+\\`\\`\\`patch
+Index: src/main/java/cc/unitmesh/untitled/demo/controller/BlogCategoryController.java
+IDEA additional info:
+Subsystem: com.intellij.openapi.diff.impl.patch.CharsetEP
+<+>UTF-8
+===================================================================
+diff --git a/src/main/java/cc/unitmesh/untitled/demo/controller/BlogCategoryController.java b/src/main/java/cc/unitmesh/untitled/demo/controller/BlogCategoryController.java
+--- a/src/main/java/cc/unitmesh/untitled/demo/controller/BlogCategoryController.java	(revision b5985862c79fe42043697bc5d5f38b470020be3d)
++++ b/src/main/java/cc/unitmesh/untitled/demo/controller/BlogCategoryController.java	(date 1709616724534)
+@@ -4,7 +4,19 @@
+ 
+ @RestController
+ public class BlogCategoryController {
+-    // devti://story/github/1
++    public void addCategory(String categoryName) {
++        // ... add category logic here
++    }
++
++    public void deleteCategory(long categoryId) {
++        // ... delete category logic here
++    }
+ 
+-    // Close a bank account
++    public void updateCategory(long categoryId, String newCategoryName) {
++        // ... update category logic here
++    }
++
++    public void listCategories() {
++        // ... list all categories logic here
++    }
+ }
 \\`\\`\\`
 ```"""
 

exts/devins-lang/src/main/kotlin/cc/unitmesh/devti/language/provider/DevInsCustomAgentResponse.kt

@@ -0,0 +1,30 @@
+package cc.unitmesh.devti.language.provider
+
+import cc.unitmesh.devti.AutoDevNotifications
+import cc.unitmesh.devti.language.DevInLanguage
+import cc.unitmesh.devti.language.compiler.DevInsCompiler
+import cc.unitmesh.devti.language.psi.DevInFile
+import cc.unitmesh.devti.provider.custom.AgentResponseProvider
+import cc.unitmesh.devti.provider.custom.CustomAgentContext
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiFileFactory
+import java.util.*
+
+
+class DevInsCustomAgentResponse : AgentResponseProvider {
+    override val name: String = "DevIn"
+
+    override fun execute(project: Project, context: CustomAgentContext): String {
+        val filename = "DevIns-${UUID.randomUUID()}.devin"
+
+        val devInFile = PsiFileFactory.getInstance(project)
+            .createFileFromText(filename, DevInLanguage, context.response) as DevInFile
+
+        val devInsCompiler = DevInsCompiler(project, devInFile)
+        val result = devInsCompiler.compile()
+
+        AutoDevNotifications.notify(project, result.output)
+
+        return result.output
+    }
+}

exts/devins-lang/src/main/resources/cc.unitmesh.devti.language.xml

@@ -44,4 +44,8 @@
                 class="cc.unitmesh.devti.language.actions.DevInsRunFileAction"
                 use-shortcut-of="RunClass"/>
     </actions>
+
+    <extensions defaultExtensionNs="cc.unitmesh">
+        <customAgentResponse implementation="cc.unitmesh.devti.language.provider.DevInsCustomAgentResponse" />
+    </extensions>
 </idea-plugin>

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

@@ -8,7 +8,10 @@ import cc.unitmesh.devti.gui.chat.ChatCodingPanel
 import cc.unitmesh.devti.gui.chat.ChatRole
 import cc.unitmesh.devti.llms.LLMProvider
 import cc.unitmesh.devti.provider.ContextPrompter
+import cc.unitmesh.devti.provider.custom.AgentResponseProvider
+import cc.unitmesh.devti.provider.custom.CustomAgentContext
 import cc.unitmesh.devti.util.LLMCoroutineScope
+import cc.unitmesh.devti.util.parser.Code
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.components.service
 import com.intellij.openapi.diagnostic.logger
@@ -38,6 +41,8 @@ class CustomAgentChatProcessor(val project: Project) {
         }
 
         selectedAgent.state = CustomAgentState.FINISHED
+
+        var devInCode: String? = ""
         when (selectedAgent.responseAction) {
             CustomAgentResponseAction.Direct -> {
                 val message = ui.addMessage("loading", false, "")
@@ -50,6 +55,12 @@ class CustomAgentChatProcessor(val project: Project) {
                 val content = sb.toString().removeSurrounding("\"")
                 llmProvider.appendLocalMessage(content, ChatRole.Assistant)
                 message.reRenderAssistantOutput()
+
+                val code = Code.parse(content)
+                 if (code.language.displayName == "DevIn") {
+                     devInCode = code.text
+                 }
+
                 ui.hiddenProgressBar()
                 ui.updateUI()
             }
@@ -64,6 +75,11 @@ class CustomAgentChatProcessor(val project: Project) {
                 llmProvider.appendLocalMessage(msg, ChatRole.Assistant)
                 ui.hiddenProgressBar()
                 ui.updateUI()
+
+                val code = Code.parse(msg)
+                if (code.language.displayName == "DevIn") {
+                    devInCode = code.text
+                }
             }
 
             CustomAgentResponseAction.TextChunk -> {
@@ -101,5 +117,11 @@ class CustomAgentChatProcessor(val project: Project) {
                 ui.hiddenProgressBar()
             }
         }
+
+        if (!devInCode.isNullOrEmpty()) {
+            AgentResponseProvider.instance("DevIn").forEach {
+                it.execute(project, CustomAgentContext(selectedAgent, devInCode))
+            }
+        }
     }
 }

src/main/kotlin/cc/unitmesh/devti/provider/custom/AgentResponseProvider.kt

@@ -10,7 +10,7 @@ data class CustomAgentContext(
     val response: String
 )
 
-internal interface AgentResponseProvider {
+interface AgentResponseProvider {
     val name: String
 
     @RequiresBackgroundThread