fix(chat-coding-service): handle custom RAG requests #51

Previously, the chat coding service was not correctly identifying and handling custom RAG (Reusable Arbitrary Grammar) requests. This commit modifies the service to properly recognize custom RAG prompts and invoke the appropriate processing logic. The import of `kotlinx.serialization.json.Json` has been retained, and a new method `isCustomRag(input: String)` has been added to the `CoUnitPreProcessor` class to handle these requests. Additionally, the `AutoDevInputSection` and `ChatCodingPanel` have been updated to include a `ComboBox` for selecting custom RAGs and to expose a method `usedCustomRag()` to indicate whether a custom RAG has been selected.

Author: phodal

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

@@ -12,8 +12,6 @@ import com.intellij.openapi.components.Service
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import kotlinx.coroutines.launch
-// keep this import
-import kotlinx.serialization.json.Json
 
 const val CO_UNIT = "/counit"
 
@@ -22,10 +20,10 @@ class CoUnitPreProcessor(val project: Project) {
     private val llmFactory = LlmFactory()
 
     private val coUnitPromptGenerator = CoUnitPromptGenerator(project)
-    private val json = Json { ignoreUnknownKeys = true }
     private val llmProvider = llmFactory.create(project)
 
-    fun isCoUnit(input: String): Boolean {
+    fun isCustomRag(input: String): Boolean {
+        // handle current category
         return project.customRagSettings.enableCustomRag && input.startsWith(CO_UNIT)
     }
 

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

@@ -47,6 +47,8 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
     private val documentListener: DocumentListener
     private val buttonPresentation: Presentation
     private val button: ActionButton
+    private val customRag: ComboBox<String>
+
     val editorListeners: EventDispatcher<AutoDevInputListener> =
         EventDispatcher.create(AutoDevInputListener::class.java)
     private var tokenizer: Tokenizer? = null
@@ -107,11 +109,11 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
             JBColor(3684930, 3750720)
         )
         layoutPanel.setOpaque(false)
-        val customRags = ComboBox(arrayOf("Normal")).also {
+        customRag = ComboBox(arrayOf("Normal")).also {
 //             todo: load from json config
         }
 
-        layoutPanel.addToLeft(customRags)
+        layoutPanel.addToLeft(customRag)
         layoutPanel.addToCenter(horizontalGlue)
         layoutPanel.addToRight(button)
         addToBottom(layoutPanel)
@@ -187,6 +189,10 @@ class AutoDevInputSection(private val project: Project, val disposable: Disposab
         return ValidationInfo(errorMessage, this as JComponent).asWarning()
     }
 
+    fun usedCustomRag(): Boolean {
+        return customRag.selectedItem != "Normal"
+    }
+
     private val maxHeight: Int
         get() {
             val decorator: InternalDecorator = UIUtil.getParentOfType(

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

@@ -244,4 +244,8 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
         myList.removeAll()
         updateUI()
     }
+
+    fun usedCustomRag(): Boolean {
+        return inputSection.usedCustomRag()
+    }
 }

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

@@ -33,11 +33,9 @@ class ChatCodingService(var actionType: ChatActionType, val project: Project) {
         val requestPrompt = prompter.requestPrompt()
         val displayPrompt = prompter.displayPrompt()
 
-        counitProcessor.isCoUnit(requestPrompt).let {
-            if (it) {
-                counitProcessor.handleChat(prompter, ui, context)
-                return
-            }
+        if (counitProcessor.isCustomRag(requestPrompt) && ui.usedCustomRag()) {
+            counitProcessor.handleChat(prompter, ui, context)
+            return
         }
 
         ui.addMessage(requestPrompt, true, displayPrompt)