feat(custom-agent): add custom agent support #51

This commit introduces support for custom agents in the project. It includes the necessary changes to the `CustomAgentHandler` class to execute queries against a custom agent server. The `CustomAgentChatProcessor` and `AutoDevInputSection` classes have been modified to integrate with the new custom agent functionality. Additionally, the `CoUnitSettingService` has been updated to provide access to the custom agent settings.

Author: phodal

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

@@ -27,9 +27,9 @@ class CustomAgentChatProcessor(val project: Project) {
         val request = originPrompt.trim()
         val selectedAgent: CustomAgentConfig = ui.getSelectedCustomAgent()
 
-        val response = customAgentHandler.executeQuery(request, selectedAgent)
+        val response = customAgentHandler.execute(request, selectedAgent)
         if (response == null) {
-            logger.error("can not find intention for request: $request")
+            logger.error("error for custom agent: $selectedAgent with request: $request")
             return
         }
 

src/main/kotlin/cc/unitmesh/devti/counit/CustomAgentHandler.kt

@@ -1,11 +1,31 @@
 package cc.unitmesh.devti.counit
 
+import cc.unitmesh.devti.counit.configurable.customAgentSetting
 import com.intellij.openapi.components.Service
 import com.intellij.openapi.project.Project
+import okhttp3.MediaType.Companion.toMediaTypeOrNull
+import okhttp3.OkHttpClient
+import okhttp3.Request
+import okhttp3.RequestBody
+import java.time.Duration
 
 @Service(Service.Level.PROJECT)
 class CustomAgentHandler(val project: Project) {
-    fun executeQuery(input: String, selectedAgent: Any): String? {
-        return null
+    private var client = OkHttpClient()
+
+    fun execute(input: String, selectedAgent: Any): String? {
+        val serverAddress = project.customAgentSetting.serverAddress ?: return null
+        val body = RequestBody.create("application/json; charset=utf-8".toMediaTypeOrNull(), input)
+        val builder = Request.Builder()
+
+        client = client.newBuilder().build()
+        val call = client.newCall(builder.url(serverAddress).post(body).build())
+
+        call.execute().use { response ->
+            if (!response.isSuccessful) {
+                return null
+            }
+            return response.body?.string()
+        }
     }
 }

src/main/kotlin/cc/unitmesh/devti/counit/configurable/CoUnitSettingService.kt

@@ -3,7 +3,7 @@ package cc.unitmesh.devti.counit.configurable
 import com.intellij.openapi.components.*
 import com.intellij.openapi.project.Project
 
-val Project.customRagSettings: CoUnitProjectSettingsService
+val Project.customAgentSetting: CoUnitProjectSettingsService
     get() = service<CoUnitProjectSettingsService>()
 
 @Service(Service.Level.PROJECT)

src/main/kotlin/cc/unitmesh/devti/gui/chat/AutoDevInputSection.kt

@@ -2,7 +2,7 @@ package cc.unitmesh.devti.gui.chat
 
 import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.AutoDevIcons
-import cc.unitmesh.devti.counit.configurable.customRagSettings
+import cc.unitmesh.devti.counit.configurable.customAgentSetting
 import cc.unitmesh.devti.counit.model.CustomAgentConfig
 import cc.unitmesh.devti.llms.tokenizer.Tokenizer
 import cc.unitmesh.devti.llms.tokenizer.TokenizerImpl
@@ -33,7 +33,6 @@ import com.intellij.util.ui.JBEmptyBorder
 import com.intellij.util.ui.JBUI
 import com.intellij.util.ui.UIUtil
 import com.intellij.util.ui.components.BorderLayoutPanel
-import kotlinx.serialization.decodeFromString
 import kotlinx.serialization.json.Json
 import java.awt.Color
 import java.awt.Component
@@ -55,7 +54,7 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
     private val buttonPresentation: Presentation
     private val button: ActionButton
 
-    private val defaultRag: CustomAgentConfig = CustomAgentConfig("Normal", "Normal")
+    private val defaultRag: CustomAgentConfig = CustomAgentConfig("<Select Custom Agent>", "Normal")
     private var customRag: ComboBox<CustomAgentConfig> = ComboBox(MutableCollectionComboBoxModel(listOf()))
 
     private val logger = logger<AutoDevInputSection>()
@@ -118,7 +117,7 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         )
         layoutPanel.setOpaque(false)
 
-        if (project.customRagSettings.enableCustomRag) {
+        if (project.customAgentSetting.enableCustomRag) {
             customRag = ComboBox(MutableCollectionComboBoxModel(loadRagApps()))
             customRag.setRenderer(SimpleListCellRenderer.create { label: JBLabel, value: CustomAgentConfig?, _: Int ->
                 if (value != null) {
@@ -151,7 +150,7 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
 
 
     private fun loadRagApps(): List<CustomAgentConfig> {
-        val ragsJsonConfig = project.customRagSettings.ragsJsonConfig
+        val ragsJsonConfig = project.customAgentSetting.ragsJsonConfig
         if (ragsJsonConfig.isEmpty()) return listOf(defaultRag)
 
         val rags = try {
@@ -212,7 +211,7 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
     }
 
     fun hasSelectedAgent(): Boolean {
-        if (!project.customRagSettings.enableCustomRag) return false
+        if (!project.customAgentSetting.enableCustomRag) return false
         if (customRag.selectedItem == null) return false
         return customRag.selectedItem != defaultRag
     }

src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatCodingService.kt

@@ -4,7 +4,7 @@ import cc.unitmesh.cf.core.llms.LlmMsg
 import cc.unitmesh.devti.AutoDevBundle
 import cc.unitmesh.devti.util.LLMCoroutineScope
 import cc.unitmesh.devti.counit.CustomAgentChatProcessor
-import cc.unitmesh.devti.counit.configurable.customRagSettings
+import cc.unitmesh.devti.counit.configurable.customAgentSetting
 import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.util.parser.PostCodeProcessor
 import cc.unitmesh.devti.provider.ContextPrompter
@@ -34,7 +34,7 @@ class ChatCodingService(var actionType: ChatActionType, val project: Project) {
         val requestPrompt = prompter.requestPrompt()
         val displayPrompt = prompter.displayPrompt()
 
-        if (project.customRagSettings.enableCustomRag && ui.hasSelectedCustomAgent()) {
+        if (project.customAgentSetting.enableCustomRag && ui.hasSelectedCustomAgent()) {
             counitProcessor.handleChat(prompter, ui, context)
             return
         }