fix(completion): improve chatbot response handling with JSON parsing enhancements

This commit enhances the chatbot response handling by improving the JSON parsing logic. It includes changes to remove the leading and trailing ```json` tags from the string to extract the JSON response correctly. Additionally, the logger has been refactored to use a lambda expression instead of a reflection-based approach for improved performance and readability.

Author: phodal

src/main/kotlin/cc/unitmesh/devti/actions/chat/CodeCompleteChatAction.kt

@@ -22,6 +22,8 @@ class CodeCompleteChatAction : AnAction() {
         return ActionUpdateThread.BGT
     }
 
+    private val logger = logger<ChatCodingService>()
+
     override fun actionPerformed(e: AnActionEvent) {
         val project = e.project ?: return
         val document = e.getData(CommonDataKeys.EDITOR)?.document
@@ -52,7 +54,7 @@ class CodeCompleteChatAction : AnAction() {
                 val chatCodingService = ChatCodingService(actionType, project)
                 val toolWindowManager =
                     ToolWindowManager.getInstance(project).getToolWindow(AutoDevToolWindowFactory.Util.id) ?: run {
-                        logger<ChatCodingService>().warn("Tool window not found")
+                        logger.warn("Tool window not found")
                         return@runReadAction
                     }
 
@@ -65,10 +67,8 @@ class CodeCompleteChatAction : AnAction() {
                 contentManager.addContent(content)
 
                 toolWindowManager.activate {
-                    val chatContext = ChatContext(
-                        null, prefixText, suffixText
-                    )
-                    chatCodingService.handlePromptAndResponse(contentPanel, prompter, chatContext, false)
+                    val chatContext = ChatContext(null, prefixText, suffixText)
+                    chatCodingService.handlePromptAndResponse(contentPanel, prompter, chatContext, true)
                 }
             } catch (ignore: IndexNotReadyException) {
                 return@runReadAction

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

@@ -14,7 +14,6 @@ import com.intellij.openapi.components.Service
 import com.intellij.openapi.diagnostic.logger
 import com.intellij.openapi.project.Project
 import kotlinx.coroutines.launch
-import kotlinx.serialization.decodeFromString
 import kotlinx.serialization.json.Json
 
 const val CO_UNIT = "/counit"
@@ -56,8 +55,7 @@ class CoUnitPreProcessor(val project: Project) {
             llmProvider.appendLocalMessage(result, ChatRole.Assistant)
 
             val explain = try {
-                val fixedResult = fix(result)
-                json.decodeFromString<ExplainQuery>(fixedResult)
+                json.decodeFromString<ExplainQuery>(extractJsonResponse(result))
             } catch (e: Exception) {
                 LOG.error("parse result error: $e")
                 return@launch
@@ -111,8 +109,15 @@ class CoUnitPreProcessor(val project: Project) {
         return sb.toString()
     }
 
-    private fun fix(result: String): String {
-        // remove start and end ```json
+    /**
+     * This method is used to extract JSON response from a given string.
+     * It removes the leading and trailing ````json` tags from the string,
+     * which are used to denote JSON code blocks in markdown files.
+     *
+     * @param result The string containing the JSON response surrounded by ````json` tags.
+     * @return The extracted JSON response string without the ````json` tags.
+     */
+    private fun extractJsonResponse(result: String): String {
         return result
             .removePrefix("```json")
             .removeSuffix("```")

src/main/kotlin/cc/unitmesh/devti/llms/custom/CustomLLMProvider.kt

@@ -65,12 +65,9 @@ class CustomLLMProvider(val project: Project) : LLMProvider {
     override fun prompt(promptText: String): String = this.prompt(promptText, "")
 
     override fun stream(promptText: String, systemPrompt: String, keepHistory: Boolean): Flow<String> {
-        if (!keepHistory) {
+        if (!keepHistory || project.coderSetting.state.noChatHistory) {
             clearMessage()
         }
-        if (project.coderSetting.state.noChatHistory) {
-            messages.clear()
-        }
 
         messages += Message("user", promptText)