refactor(chat): add support for custom RAG settings in chat coding service #51

This commit introduces the ability to handle custom RAG settings in the chat coding service. It modifies the `ChatCodingService` to check if custom RAG is enabled and if the user has selected a custom RAG. If both conditions are met, the chat is processed using the custom RAG settings.

The `CoUnitPreProcessor` is modified to include a new method `isCustomRag` which checks if the input string starts with `CO_UNIT` and if custom RAG is enabled. The `CustomRagSettings` class is also introduced to manage the custom RAG configuration.

The `AutoDevInputSection` and `ChatCodingPanel` are updated to expose methods `usedCustomRag` and `selectedCustomRag` to check if the user has selected a custom RAG. These methods are used in the `ChatCodingService` to determine if custom RAG processing should be applied.

Author: phodal

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

@@ -22,11 +22,6 @@ class CoUnitPreProcessor(val project: Project) {
     private val coUnitPromptGenerator = CoUnitPromptGenerator(project)
     private val llmProvider = llmFactory.create(project)
 
-    fun isCustomRag(input: String): Boolean {
-        // handle current category
-        return project.customRagSettings.enableCustomRag && input.startsWith(CO_UNIT)
-    }
-
     fun handleChat(prompter: ContextPrompter, ui: ChatCodingPanel, context: ChatContext?) {
         val originRequest = prompter.requestPrompt()
         ui.addMessage(originRequest, true, originRequest)

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

@@ -22,7 +22,6 @@ import com.intellij.openapi.wm.impl.InternalDecorator
 import com.intellij.temporary.gui.block.AutoDevCoolBorder
 import com.intellij.ui.JBColor
 import com.intellij.ui.content.ContentManager
-import com.intellij.ui.popup.list.ComboBoxPopup
 import com.intellij.util.EventDispatcher
 import com.intellij.util.ui.JBEmptyBorder
 import com.intellij.util.ui.JBUI
@@ -188,7 +187,7 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         return ValidationInfo(errorMessage, this as JComponent).asWarning()
     }
 
-    fun usedCustomRag(): Boolean {
+    fun selectedCustomRag(): Boolean {
         return customRag.selectedItem != "Normal"
     }
 

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

@@ -245,7 +245,7 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         updateUI()
     }
 
-    fun usedCustomRag(): Boolean {
-        return inputSection.usedCustomRag()
+    fun selectedCustomRag(): Boolean {
+        return inputSection.selectedCustomRag()
     }
 }

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

@@ -4,6 +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.CoUnitPreProcessor
+import cc.unitmesh.devti.counit.configurable.customRagSettings
 import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.util.parser.PostCodeProcessor
 import cc.unitmesh.devti.provider.ContextPrompter
@@ -33,7 +34,7 @@ class ChatCodingService(var actionType: ChatActionType, val project: Project) {
         val requestPrompt = prompter.requestPrompt()
         val displayPrompt = prompter.displayPrompt()
 
-        if (counitProcessor.isCustomRag(requestPrompt) && ui.usedCustomRag()) {
+        if (project.customRagSettings.enableCustomRag && ui.selectedCustomRag()) {
             counitProcessor.handleChat(prompter, ui, context)
             return
         }