feat(plan): add plan review functionality and scrollable UI

phodal · phodal · commit 0d1eae1e05dd · 2025-03-17T15:06:23.000+08:00
#259 - Add plan review functionality with LLM integration for plan evaluation. - Introduce scrollable UI for plan display in PlanSketch. - Update plan template file extensions from `.devin` to `.vm`. - Add new plan reviewer templates in both English and Chinese.
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/observer/agent/AgentStateService.kt b/core/src/main/kotlin/cc/unitmesh/devti/observer/agent/AgentStateService.kt
@@ -4,6 +4,7 @@ import cc.unitmesh.devti.agent.tool.AgentTool
 import cc.unitmesh.devti.devin.dataprovider.BuiltinCommand
 import cc.unitmesh.devti.llms.custom.Message
 import cc.unitmesh.devti.observer.plan.AgentTaskEntry
+import cc.unitmesh.devti.observer.plan.MarkdownPlanParser
 import cc.unitmesh.devti.observer.plan.PlanUpdateListener
 import com.intellij.openapi.application.ApplicationManager
 import com.intellij.openapi.components.Service
@@ -58,6 +59,11 @@ class AgentStateService {
             .onPlanUpdate(items)
     }
 
+    fun updatePlan(content: String) {
+        val planItems = MarkdownPlanParser.parse(content)
+        updatePlan(planItems.toMutableList())
+    }
+
     fun resetState() {
         state = AgentState()
         ApplicationManager.getApplication().messageBus
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/observer/plan/PlanReviewAction.kt b/core/src/main/kotlin/cc/unitmesh/devti/observer/plan/PlanReviewAction.kt
@@ -1,11 +1,21 @@
 // filepath: /Volumes/source/ai/autocrud/core/src/main/kotlin/cc/unitmesh/devti/observer/plan/PlanReviewAction.kt
 package cc.unitmesh.devti.observer.plan
 
+import cc.unitmesh.devti.AutoDevNotifications
+import cc.unitmesh.devti.llms.LlmFactory
 import cc.unitmesh.devti.observer.agent.AgentStateService
+import cc.unitmesh.devti.statusbar.AutoDevStatus
+import cc.unitmesh.devti.statusbar.AutoDevStatusService
+import cc.unitmesh.devti.template.GENIUS_CODE
+import cc.unitmesh.devti.template.TemplateRender
+import cc.unitmesh.devti.util.AutoDevCoroutineScope
+import cc.unitmesh.devti.util.parser.CodeFence
+import com.intellij.execution.ui.ConsoleViewContentType
 import com.intellij.openapi.actionSystem.ActionUpdateThread
 import com.intellij.openapi.actionSystem.AnAction
 import com.intellij.openapi.actionSystem.AnActionEvent
-import org.intellij.markdown.IElementType
+import kotlinx.coroutines.launch
+import kotlinx.coroutines.runBlocking
 import org.intellij.markdown.MarkdownElementTypes
 import org.intellij.markdown.ast.ASTNode
 import org.intellij.markdown.ast.accept
@@ -22,12 +32,40 @@ class PlanReviewAction : AnAction() {
         val agentStateService = project.getService(AgentStateService::class.java)
 
         val currentPlan = agentStateService.getPlan()
+        val issue = agentStateService.buildOriginIntention() ?: ""
         val plan = MarkdownPlanParser.formatPlanToMarkdown(currentPlan)
 
-        /// call llm to evaluate the plan
         val allMessages = agentStateService.getAllMessages()
-        val withOutCode = allMessages.map {
-            removeAllMarkdownCode(it.content)
+        val withoutCodeMsgs = allMessages.map {
+            it.copy(role = it.role, content = removeAllMarkdownCode(it.content))
+        }
+
+        val templateRender = TemplateRender(GENIUS_CODE)
+        val systemPrompt = templateRender.getTemplate("plan-reviewer.vm")
+        val history = withoutCodeMsgs.joinToString {
+            "# Role ${it.role}\nMessage:\n${it.content}"
+        }
+
+        val stream = LlmFactory.create(project).stream(history, systemPrompt)
+        AutoDevCoroutineScope.scope(project).launch {
+            val llmResult = StringBuilder()
+            AutoDevStatusService.notifyApplication(AutoDevStatus.InProgress, "review the plan")
+            runBlocking {
+                stream.collect {
+                    llmResult.append(it)
+                }
+            }
+
+            val result = llmResult.toString()
+            AutoDevStatusService.notifyApplication(AutoDevStatus.Done, "review the plan")
+            AutoDevNotifications.notify(project, result)
+            val plan = CodeFence.parseAll(result).firstOrNull {
+                it.originLanguage == "plan"
+            }
+
+            if (plan !== null) {
+                agentStateService.updatePlan(result)
+            }
         }
     }
 }
@@ -68,7 +106,7 @@ fun removeAllMarkdownCode(markdownContent: String): String {
 private fun extractCodeFenceLanguage(node: ASTNode, markdownContent: String): String {
     val nodeText = node.getTextInNode(markdownContent).toString()
     val firstLine = nodeText.lines().firstOrNull() ?: ""
-    
+
     val languageMatch = Regex("^```(.*)$").find(firstLine.trim())
     return languageMatch?.groupValues?.getOrNull(1)?.trim() ?: ""
 }
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/sketch/SketchInputListener.kt b/core/src/main/kotlin/cc/unitmesh/devti/sketch/SketchInputListener.kt
@@ -36,7 +36,7 @@ open class SketchInputListener(
     override val templateRender: TemplateRender get() = TemplateRender(GENIUS_CODE)
     open var systemPrompt = ""
     open var planPrompt = ""
-    val planTemplate = templateRender.getTemplate("plan.devin")
+    val planTemplate = templateRender.getTemplate("plan.vm")
 
     open suspend fun setup() {
         systemPrompt = templateRender.renderTemplate(template, SketchRunContext.create(project, null, ""))
diff --git a/core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/plan/PlanSketch.kt b/core/src/main/kotlin/cc/unitmesh/devti/sketch/ui/plan/PlanSketch.kt
@@ -8,6 +8,7 @@ import cc.unitmesh.devti.sketch.ui.ExtensionLangSketch
 import com.intellij.lang.Language
 import com.intellij.openapi.project.Project
 import com.intellij.ui.components.JBPanel
+import com.intellij.ui.components.JBScrollPane
 import com.intellij.util.ui.JBEmptyBorder
 import com.intellij.util.ui.JBUI
 import java.awt.BorderLayout
@@ -16,6 +17,7 @@ import javax.swing.Box
 import javax.swing.BoxLayout
 import javax.swing.JComponent
 import javax.swing.JPanel
+import javax.swing.ScrollPaneConstants
 
 /**
  * Controller class for managing the plan data and UI updates
@@ -96,7 +98,14 @@ class PlanSketch(
         }
         
         planController.renderPlan()
-        add(contentPanel, BorderLayout.CENTER)
+        
+        val scrollPane = JBScrollPane(contentPanel).apply {
+            verticalScrollBarPolicy = ScrollPaneConstants.VERTICAL_SCROLLBAR_AS_NEEDED
+            horizontalScrollBarPolicy = ScrollPaneConstants.HORIZONTAL_SCROLLBAR_NEVER
+            border = null
+        }
+        
+        add(scrollPane, BorderLayout.CENTER)
 
         minimumSize = Dimension(200, 0)
         background = JBUI.CurrentTheme.ToolWindow.background()
@@ -130,3 +139,4 @@ class PlanSketch(
     
     override fun dispose() {}
 }
+
diff --git a/core/src/main/resources/genius/en/code/plan-reviewer.vm b/core/src/main/resources/genius/en/code/plan-reviewer.vm
@@ -0,0 +1,57 @@
+你是一位 Plan 审核者（Peer Reviewer）、纠正者，正在评估一名自动编程程序员的自动化过程的计划与实施情况，该程序员的目标是解决给定存储库中的特定问题。
+该程序员有可能更新计划不及时，所以你需要纠正计划。
+
+你不关心代码实现细节，而是关注整体架构、技术决策和计划的可行性。你的目标是确保方案既符合技术规范，又能满足业务需求。
+
+**输入数据：**
+- 程序员与环境交互的历史记录（可能是部分的，也可能是完整的）。
+- 计划的描述，包括技术方案、实施步骤、目标等。
+- 可能的业务背景或相关约束条件。
+- 更新后的计划，包含每个步骤的进度指示符：
+- `[✓]`：步骤完成，或正在进行该步骤。
+- `[!]`：步骤失败。
+- `[*]`：步骤正在进行中。
+
+**你的评审标准：**
+1. **路线合理性**：
+- 该计划是否清晰可行？
+- 是否遵循合理的技术架构和最佳实践？
+- 是否符合已有的技术规范（如异常管理、API 认证鉴权、微服务架构等）？
+
+2. **业务适配性**：
+- 该计划是否真正满足业务需求？
+- 是否有更简洁或更高效的方案？
+
+3. **可扩展性与长期维护**：
+- 该方案是否便于后续扩展？
+- 是否可能带来技术债务？
+
+4. **风险评估**：
+- 该计划是否存在明显的技术风险（如性能瓶颈、安全漏洞、可用性问题等）？
+- 是否考虑了异常情况和回退机制？
+
+5. **步骤进度评估**：
+- 你需要根据计划中的步骤进度指示符评估该计划的当前状态。
+- `[✓]`：已完成或当前正在进行的步骤，是否符合预期？是否已充分完成？
+- `[!]`：失败的步骤，是否存在重大问题或阻碍？应给出修正建议。
+- `[*]`：正在进行中的步骤，是否存在瓶颈或风险点？是否有足够的支持来完成该步骤？
+
+6. **你的输出格式**：
+- 优化后的方案。使用 `plan` 语言的 markdown 代码库
+- `plan` 中包含步骤的状态（`[✓]`, `[!]`, `[*]`）给出分析与评估；
+
+你永远只返回 markdown plan 代码块，返回示例：
+
+```plan
+1. 领域模型重构
+    - 创建聚合根：建立 Blog 聚合根，包含 Post、Comment 等子实体
+    - 充血模型改造：将业务逻辑从 Service 迁移到领域对象
+    - 值对象创建：构建 Slug、Content、Author 等值对象
+2. 分层架构调整
+3. 关键重构步骤：
+    - 分离领域模型与持久化实体
+    - 重构 BlogService 为领域服务+应用服务
+    - 创建工厂方法处理复杂对象创建
+    - 实现领域事件机制
+    - 添加业务约束校验逻辑
+```
diff --git a/core/src/main/resources/genius/en/code/plan.vm b/core/src/main/resources/genius/en/code/plan.vm
diff --git a/core/src/main/resources/genius/zh/code/plan-reviewer.vm b/core/src/main/resources/genius/zh/code/plan-reviewer.vm
@@ -0,0 +1,57 @@
+你是一位 Plan 审核者（Peer Reviewer）、纠正者，正在评估一名自动编程程序员的自动化过程的计划与实施情况，该程序员的目标是解决给定存储库中的特定问题。
+该程序员有可能更新计划不及时，所以你需要纠正计划。
+
+你不关心代码实现细节，而是关注整体架构、技术决策和计划的可行性。你的目标是确保方案既符合技术规范，又能满足业务需求。
+
+**输入数据：**
+- 程序员与环境交互的历史记录（可能是部分的，也可能是完整的）。
+- 计划的描述，包括技术方案、实施步骤、目标等。
+- 可能的业务背景或相关约束条件。
+- 更新后的计划，包含每个步骤的进度指示符：
+- `[✓]`：步骤完成，或正在进行该步骤。
+- `[!]`：步骤失败。
+- `[*]`：步骤正在进行中。
+
+**你的评审标准：**
+1. **路线合理性**：
+- 该计划是否清晰可行？
+- 是否遵循合理的技术架构和最佳实践？
+- 是否符合已有的技术规范（如异常管理、API 认证鉴权、微服务架构等）？
+
+2. **业务适配性**：
+- 该计划是否真正满足业务需求？
+- 是否有更简洁或更高效的方案？
+
+3. **可扩展性与长期维护**：
+- 该方案是否便于后续扩展？
+- 是否可能带来技术债务？
+
+4. **风险评估**：
+- 该计划是否存在明显的技术风险（如性能瓶颈、安全漏洞、可用性问题等）？
+- 是否考虑了异常情况和回退机制？
+
+5. **步骤进度评估**：
+- 你需要根据计划中的步骤进度指示符评估该计划的当前状态。
+- `[✓]`：已完成或当前正在进行的步骤，是否符合预期？是否已充分完成？
+- `[!]`：失败的步骤，是否存在重大问题或阻碍？应给出修正建议。
+- `[*]`：正在进行中的步骤，是否存在瓶颈或风险点？是否有足够的支持来完成该步骤？
+
+6. **你的输出格式**：
+- 优化后的方案。使用 `plan` 语言的 markdown 代码库
+- `plan` 中包含步骤的状态（`[✓]`, `[!]`, `[*]`）给出分析与评估；
+
+你永远只返回 markdown plan 代码块，返回示例：
+
+```plan
+1. 领域模型重构
+    - 创建聚合根：建立 Blog 聚合根，包含 Post、Comment 等子实体
+    - 充血模型改造：将业务逻辑从 Service 迁移到领域对象
+    - 值对象创建：构建 Slug、Content、Author 等值对象
+2. 分层架构调整
+3. 关键重构步骤：
+    - 分离领域模型与持久化实体
+    - 重构 BlogService 为领域服务+应用服务
+    - 创建工厂方法处理复杂对象创建
+    - 实现领域事件机制
+    - 添加业务约束校验逻辑
+```
diff --git a/core/src/main/resources/genius/zh/code/plan.vm b/core/src/main/resources/genius/zh/code/plan.vm
