feat(agent): add support for DevIns language in code blocks #144

unit-mesh · Apr 11, 2024 · ddd7565 · ddd7565
1 parent df04902
commit ddd7565
Show file tree

Hide file tree

Showing 5 changed files with 34 additions and 6 deletions.
diff --git a/docs/agent/agent-action.md b/docs/agent/agent-action.md
@@ -49,4 +49,12 @@ enum class ResponseAction {
 
 <img src="https://unitmesh.cc/auto-dev/custom-agent-webview.png" alt="Custom AI Agent Dropdown" width="600px"/>
 
+### DevIns
 
+> The DevIns response action will handle the response in the DevIns language.
+
+just like the following example:
+
+```
+/write:xxx.java#L1-L12
+```
diff --git a/src/main/kotlin/cc/unitmesh/devti/agent/CustomAgentChatProcessor.kt b/src/main/kotlin/cc/unitmesh/devti/agent/CustomAgentChatProcessor.kt
@@ -57,9 +57,9 @@ class CustomAgentChatProcessor(val project: Project) {
                 message.reRenderAssistantOutput()
 
                 val code = Code.parse(content)
-                 if (code.language.displayName == "DevIn") {
-                     devInCode = code.text
-                 }
+                if (code.language.displayName == "DevIn") {
+                    devInCode = code.text
+                }
 
                 ui.hiddenProgressBar()
                 ui.updateUI()
@@ -116,6 +116,20 @@ class CustomAgentChatProcessor(val project: Project) {
                 ui.appendWebView(content, project)
                 ui.hiddenProgressBar()
             }
+
+            CustomAgentResponseAction.DevIns -> {
+                ui.addMessage(AutoDevBundle.message("autodev.loading"))
+                var msg = ""
+                LLMCoroutineScope.scope(project).launch {
+                    msg = ui.updateMessage(response)
+                }
+
+                llmProvider.appendLocalMessage(msg, ChatRole.Assistant)
+                ui.hiddenProgressBar()
+                ui.updateUI()
+
+                devInCode = msg
+            }
         }
 
         if (!devInCode.isNullOrEmpty()) {

diff --git a/src/main/kotlin/cc/unitmesh/devti/agent/model/CustomAgentResponseAction.kt b/src/main/kotlin/cc/unitmesh/devti/agent/model/CustomAgentResponseAction.kt
@@ -34,5 +34,10 @@ enum class CustomAgentResponseAction {
     /**
      * Display result in WebView
      */
-    WebView
+    WebView,
+
+    /**
+     * Handle with DevIns language in code block
+     */
+    DevIns
 }
diff --git a/src/main/kotlin/cc/unitmesh/devti/util/parser/CodeUtil.kt b/src/main/kotlin/cc/unitmesh/devti/util/parser/CodeUtil.kt
@@ -57,7 +57,7 @@ class Code(val language: Language, val text: String, val isComplete: Boolean) {
                 return Code(findLanguage("markdown"), content.replace("\\n", "\n"), codeClosed)
             }
 
-            if (languageId == "devin") {
+            if (languageId == "devin" || languageId == "devins") {
                 trimmedCode = trimmedCode.replace("\\`\\`\\`", "```")
             }
 

diff --git a/src/main/resources/cc/unitmesh/devti/custom/schema/autodev-custom-agent.json b/src/main/resources/cc/unitmesh/devti/custom/schema/autodev-custom-agent.json
@@ -50,7 +50,8 @@
         "enum": [
           "Direct",
           "TextChunk",
-          "WebView"
+          "WebView",
+          "DevIns"
         ]
       }
     },