Support real-time code formatting when LLM return results #252 (#254)

jiangyunpeng · web-flow · commit 6cdd753bc78a · 2025-01-04T08:05:04.000+08:00
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatCodingPanel.kt b/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/ChatCodingPanel.kt
@@ -260,7 +260,6 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp
             }
         }
 
-        messageView.reRenderAssistantOutput()
         return text
     }
 
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/MessageView.kt b/core/src/main/kotlin/cc/unitmesh/devti/gui/chat/MessageView.kt
@@ -1,7 +1,6 @@
 package cc.unitmesh.devti.gui.chat
 
 
-import cc.unitmesh.devti.gui.component.DisplayComponent
 import com.intellij.openapi.actionSystem.DataProvider
 import com.intellij.openapi.actionSystem.ex.ActionUtil
 import com.intellij.openapi.actionSystem.impl.ActionToolbarImpl
@@ -23,9 +22,9 @@ import kotlin.jvm.internal.Ref
 class MessageView(val message: String, val role: ChatRole, private val displayText: String) :
     JBPanel<MessageView>(), DataProvider {
     private val myNameLabel: Component
-    private val component: DisplayComponent = DisplayComponent(message)
     private var centerPanel: JPanel = JPanel(VerticalLayout(JBUI.scale(8)))
     private var messageView: SimpleMessage? = null
+    private var blocks: MutableList<MessageBlock> = ArrayList()
 
     init {
         isDoubleBuffered = true
@@ -55,17 +54,10 @@ class MessageView(val message: String, val role: ChatRole, private val displayTe
         centerPanel.add(myNameLabel)
         add(centerPanel, BorderLayout.CENTER)
 
-        if (role == ChatRole.User) {
-            ApplicationManager.getApplication().invokeLater {
-                val simpleMessage = SimpleMessage(displayText, message, role)
-                messageView = simpleMessage
-                renderInPartView(simpleMessage)
-            }
-        } else {
-            component.updateMessage(message)
-            component.revalidate()
-            component.repaint()
-            centerPanel.add(component)
+        ApplicationManager.getApplication().invokeLater {
+            val simpleMessage = SimpleMessage(displayText, message, role)
+            messageView = simpleMessage
+            renderInPartView(simpleMessage)
         }
     }
 
@@ -89,25 +81,58 @@ class MessageView(val message: String, val role: ChatRole, private val displayTe
     }
 
     private fun renderInPartView(message: SimpleMessage) {
-        val parts = layoutAll(message)
-        parts.forEach {
-            val blockView = when (it) {
-                is CodeBlock -> {
-                    val project = ProjectManager.getInstance().openProjects.firstOrNull()
-                    CodeBlockView(it, project!!) { }
+        val incrBlocks = layoutAll(message)
+        if (incrBlocks.isEmpty()) {
+            return
+        }
+        if (this.blocks.isEmpty()) {
+            blocks.addAll(incrBlocks)
+        } else {
+            var newBlock: MessageBlock
+            var i = 0
+            while (i < minOf(this.blocks.size, incrBlocks.size)) {
+                val oldBlock = this.blocks[i]
+                newBlock = incrBlocks[i]
+                if (oldBlock.getIdentifier() == newBlock.getIdentifier()) { // 如果新旧内容一样忽略
+                    i++
+                    continue
                 }
-
-                else -> TextBlockView(it)
+                oldBlock.setNeedUpdate(true)
+                oldBlock.addContent(newBlock.getTextContent())
+                i++
             }
+            for (j in i until incrBlocks.size) {
+                newBlock = incrBlocks[j]
+                newBlock.setNeedUpdate(true)
+                this.blocks.add(newBlock) // 添加到 blocks
+            }
+        }
+        renderComponent();
+    }
 
-            blockView.initialize()
-            val component = blockView.getComponent() ?: return@forEach
+    private fun renderComponent(){
+        blocks.forEach { block ->
+            if (!block.isNeedUpdate()) return@forEach
 
-            component.setForeground(JBUI.CurrentTheme.Label.foreground())
-            centerPanel.add(component)
+            if (block.getMessageBlockView() == null) {
+                val blockView = when (block) {
+                    is CodeBlock -> {
+                        val project = ProjectManager.getInstance().openProjects.first()
+                        CodeBlockView(block, project) { }
+                    }
+                    else -> TextBlockView(block)
+                }
+                blockView.initialize()
+                block.setMessageBlockView(blockView)
+
+                val component = blockView.getComponent() ?: return@forEach
+                component.setForeground(JBUI.CurrentTheme.Label.foreground())
+                centerPanel.add(component)
+            }
+            block.setNeedUpdate(false)
         }
-    }
 
+    }
     private var answer: String = ""
 
     fun updateContent(content: String) {
@@ -124,7 +149,7 @@ class MessageView(val message: String, val role: ChatRole, private val displayTe
 
     fun reRenderAssistantOutput() {
         ApplicationManager.getApplication().invokeLater {
-            centerPanel.remove(component)
+
             centerPanel.updateUI()
 
             centerPanel.add(myNameLabel)
@@ -148,8 +173,8 @@ class MessageView(val message: String, val role: ChatRole, private val displayTe
         override fun done() {
             try {
                 get()
-                component.updateMessage(message)
-                component.updateUI()
+                val simpleMessage = SimpleMessage(message, this@MessageView.message, role)
+                renderInPartView(simpleMessage)
             } catch (e: Exception) {
                 logger.error(message, e.message)
             }
@@ -246,7 +271,7 @@ class MessageView(val message: String, val role: ChatRole, private val displayTe
 
     override fun getData(dataId: String): CompletableMessage? {
         return if (CompletableMessage.key.`is`(dataId)) {
-           return this.messageView
+            return this.messageView
         } else {
             null
         }
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/llms/azure/AzureOpenAIProvider.kt b/core/src/main/kotlin/cc/unitmesh/devti/llms/azure/AzureOpenAIProvider.kt
@@ -199,4 +199,4 @@ class AzureOpenAIProvider(val project: Project) : LLMProvider {
             return customOpenAiHost
         }
     }
-}
+}
diff --git a/core/src/main/kotlin/com/intellij/temporary/gui/block/CodeBlockView.kt b/core/src/main/kotlin/com/intellij/temporary/gui/block/CodeBlockView.kt
@@ -8,6 +8,7 @@ import com.intellij.openapi.Disposable
 import com.intellij.openapi.actionSystem.ActionGroup
 import com.intellij.openapi.actionSystem.ActionManager
 import com.intellij.openapi.actionSystem.ActionPlaces
+import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.application.ReadAction
 import com.intellij.openapi.editor.Document
 import com.intellij.openapi.editor.Editor
@@ -80,7 +81,9 @@ class CodeBlockView(
             if (codePartEditorInfo!!.language == code.language) {
                 editorInfo!!.language = code.language
             }
-
+            ApplicationManager.getApplication().runWriteAction{
+                editorInfo!!.editor.document.setText(code.text)
+            }
             editorInfo!!.code.set(code.text)
         }
 
diff --git a/core/src/main/kotlin/com/intellij/temporary/gui/block/MessageBlock.kt b/core/src/main/kotlin/com/intellij/temporary/gui/block/MessageBlock.kt
@@ -3,11 +3,19 @@ package com.intellij.temporary.gui.block
 
 import cc.unitmesh.devti.util.parser.CodeFence
 import com.intellij.lang.Language
+import java.math.BigInteger
+import java.security.MessageDigest
+import java.security.NoSuchAlgorithmException
 
 interface MessageBlock {
     val type: MessageBlockType
     fun getTextContent(): String
     fun getMessage(): CompletableMessage
+    fun setNeedUpdate(needUpdate: Boolean)
+    fun isNeedUpdate(): Boolean
+    fun getIdentifier(): String
+    fun getMessageBlockView(): MessageBlockView?
+    fun setMessageBlockView(messageBlockView: MessageBlockView)
     fun addContent(addedContent: String)
     fun replaceContent(content: String)
     fun addTextListener(textListener: MessageBlockTextListener)
@@ -16,19 +24,58 @@ interface MessageBlock {
 
 abstract class AbstractMessageBlock(open val completableMessage: CompletableMessage) : MessageBlock {
     private val contentBuilder: StringBuilder = StringBuilder()
+    private var identifier: String = ""
+    private var needUpdate: Boolean = true
+    private var messageBlockView: MessageBlockView? = null
     private val textListeners: MutableList<MessageBlockTextListener> = mutableListOf()
 
+    companion object {
+        fun encode(bytes: ByteArray): String {
+            var md5 = ""
+            try {
+                val messageDigest = MessageDigest.getInstance("MD5")
+                messageDigest.update(bytes)
+                md5 = BigInteger(1, messageDigest.digest()).toString(16)
+            } catch (noSuchAlgorithmException: NoSuchAlgorithmException) {
+                // empty catch block
+            }
+            return md5
+        }
+    }
+    override fun isNeedUpdate(): Boolean {
+        return needUpdate
+    }
+
+    override fun setNeedUpdate(needUpdate: Boolean) {
+        this.needUpdate = needUpdate
+    }
+
+    override fun getMessageBlockView(): MessageBlockView? {
+        return messageBlockView;
+    }
+
+    override fun setMessageBlockView(messageBlockView: MessageBlockView) {
+        this.messageBlockView = messageBlockView;
+    }
+
+    override fun getIdentifier(): String {
+        return identifier;
+    }
+
     override fun addContent(addedContent: String) {
+        contentBuilder.clear()
         contentBuilder.append(addedContent)
         onContentAdded(addedContent)
         val content = contentBuilder.toString()
+        identifier = encode(content.toByteArray())
         onContentChanged(content)
         fireTextChanged(content)
     }
 
     override fun replaceContent(content: String) {
         contentBuilder.clear()
         contentBuilder.append(content)
+        identifier = encode(content.toByteArray())
         onContentChanged(content)
         fireTextChanged(content)
     }

Original file line number	Diff line number	Diff line change
`@@ -260,7 +260,6 @@ class ChatCodingPanel(private val chatCodingService: ChatCodingService, val disp`
`260`	`260`	`}`
`261`	`261`	`}`
`262`	`262`
`263`		`- messageView.reRenderAssistantOutput()`
`264`	`263`	`return text`
`265`	`264`	`}`
`266`	`265`
Original file line number	Diff line number	Diff line change
`@@ -199,4 +199,4 @@ class AzureOpenAIProvider(val project: Project) : LLMProvider {`
`199`	`199`	`return customOpenAiHost`
`200`	`200`	`}`
`201`	`201`	`}`
`202`		`-}`
	`202`	`+}`