feat(diff-repair): include user intention in diff repair context

Add user's original intention to the diff repair context for better context awareness during code repair. This involves modifying the `DiffRepairContext` data class, updating template files, and adding a method in `AgentStateService` to retrieve the intention.

Author: phodal

core/src/main/kotlin/cc/unitmesh/devti/observer/agent/AgentStateService.kt

@@ -35,6 +35,18 @@ class AgentStateService {
         return messages
     }
 
+    fun buildOriginIntention(): String? {
+        val intention = state.messages
+            .firstOrNull { it.role.lowercase() == "user" }
+            ?.content
+
+        if (intention != null) {
+            state.originIntention = intention
+        }
+
+        return intention
+    }
+
     fun resolveIssue() {
         // todo resolve issue
     }

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/patch/DiffLangSketch.kt

@@ -28,7 +28,6 @@ import com.intellij.ui.components.panels.VerticalLayout
 import com.intellij.util.containers.MultiMap
 import com.intellij.util.ui.JBUI
 import java.awt.BorderLayout
-import javax.swing.Box
 import javax.swing.BoxLayout
 import javax.swing.JButton
 import javax.swing.JComponent

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/patch/DiffRepair.kt

@@ -2,6 +2,7 @@ package cc.unitmesh.devti.sketch.ui.patch
 
 import cc.unitmesh.devti.llm2.model.ModelType
 import cc.unitmesh.devti.llms.LlmFactory
+import cc.unitmesh.devti.observer.agent.AgentStateService
 import cc.unitmesh.devti.template.GENIUS_CODE
 import cc.unitmesh.devti.template.TemplateRender
 import cc.unitmesh.devti.util.AutoDevCoroutineScope
@@ -19,7 +20,9 @@ fun applyDiffRepairSuggestion(project: Project, editor: Editor, oldCode: String,
     val templateRender = TemplateRender(GENIUS_CODE)
     val template = templateRender.getTemplate("repair-diff.vm")
 
-    templateRender.context = DiffRepairContext(oldCode, patchedCode)
+    val intention = project.getService(AgentStateService::class.java).buildOriginIntention()
+
+    templateRender.context = DiffRepairContext(intention, patchedCode, oldCode)
     val prompt = templateRender.renderTemplate(template)
 
     val flow: Flow<String> = LlmFactory.create(project, ModelType.FastApply).stream(prompt, "", false)

core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/patch/SingleFileDiffSketch.kt

@@ -226,8 +226,9 @@ class SingleFileDiffSketch(
 }
 
 data class DiffRepairContext(
-    val oldCode: String,
+    val intention: String?,
     val patchedCode: String,
+    val oldCode: String,
 ) : TemplateContext
 
 fun VirtualFile.readText(): String {

core/src/main/resources/genius/en/code/repair-diff.vm

@@ -1,7 +1,9 @@
 Please according the diff to repair the code, and return the repaired final code.
 
+User origin intention: ${context.intention}
+
 Here is the original code: ${context.oldCode}
 
-Here is the patched code:
+Here is the failed patched context:
 
 ${context.patchedCode}

core/src/main/resources/genius/zh/code/repair-diff.vm

@@ -1,10 +1,12 @@
 请根据我的原始代码和失败的 patch，生成最终的完整代码。
 
+用户的原始意图是: ${context.intention}
+
 Here is the original code:
 
 ${context.oldCode}
 
-Here is the patched code:
+Here is the failed patched context:
 
 ${context.patchedCode}