@ghyper9023/pi-dev-workflow 0.3.3 → 0.4.1

package/extensions/dev-prompts.ts

@@ -26,16 +26,20 @@
 import type { ExtensionAPI, ExtensionCommandContext } from "@earendil-works/pi-coding-agent";
 import { runGrillPhase, runPRDPhase, saveAnswerFile, recoverFromBackup, type GrillOptions } from "./grill-me-agent";
 import { discoverAgents } from "./sub-agents";
+import { runWorkflow, loadCheckpointFromFile, type WorkflowStepDef } from "./workflow-engine";
+import { uiSelect, uiConfirm, uiInput, BACK_MARKER } from "./ui-helpers";
 
 // ── Helpers ──────────────────────────────────────────────────
 
-/** Ask a single question. Returns `undefined` on cancel (Esc). */
+/** Ask a single question with proper wrapping. Returns `undefined` on cancel (Esc), or BACK_MARKER for back. */
 async function ask(
 	ctx: ExtensionCommandContext,
 	label: string,
 	placeholder: string,
+	backable = false,
+	initialValue = "",
 ): Promise<string | undefined> {
-	return ctx.ui.input(label, { placeholder, required: false });
+	return uiInput(ctx, label, placeholder, false, backable, initialValue);
 }
 
 /** Check if a field value is empty or explicitly "无". */
@@ -360,12 +364,276 @@ const _refactorGrillAgent = discoverAgents().find(a => a.name === "dev-refactor-
 const _testGrillAgent = discoverAgents().find(a => a.name === "dev-test-grill-agent")!;
 const _perfGrillAgent = discoverAgents().find(a => a.name === "dev-perf-grill-agent")!;
 
+// ── Workflow configurations ──────────────────────────────────
+
+const FEAT_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 生成实施计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🔧 实施代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+	{
+		id: "trimmer-reviewer",
+		label: "✂️ 精简代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "trimmer",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 300_000,
+	},
+	{
+		id: "docWriter",
+		label: "📝 更新文档",
+		type: "confirm",
+		agentName: "docWriter",
+		timeoutMs: 300_000,
+	},
+];
+
+const FIX_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析根因并制定修复计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🔧 修复代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+	{
+		id: "docWriter",
+		label: "📝 更新文档",
+		type: "confirm",
+		agentName: "docWriter",
+		timeoutMs: 300_000,
+	},
+];
+
+const REFACTOR_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析重构计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🔧 重构代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+	{
+		id: "trimmer-reviewer",
+		label: "✂️ 精简代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "trimmer",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 300_000,
+	},
+];
+
+const PERF_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析性能问题并制定优化计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "⚡ 优化代码 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+];
+
+const TEST_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析测试计划",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "worker-reviewer",
+		label: "🧪 编写测试 → 审查",
+		type: "loop-group",
+		loopAgentName: "worker",
+		reviewAgentName: "reviewer",
+		maxLoops: 3,
+		timeoutMs: 900_000,
+	},
+];
+
+const DOC_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "planner",
+		label: "📋 分析文档需求",
+		type: "auto",
+		agentName: "planner",
+		timeoutMs: 900_000,
+	},
+	{
+		id: "docWriter",
+		label: "📝 撰写文档",
+		type: "auto",
+		agentName: "docWriter",
+		timeoutMs: 300_000,
+	},
+];
+
+const STYLE_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "trimmer-reviewer",
+		label: "✂️ 风格调整 → 审查",
+		type: "loop-group",
+		loopAgentName: "trimmer",
+		reviewAgentName: "reviewer",
+		maxLoops: 2,
+		timeoutMs: 300_000,
+	},
+];
+
+const SECURITY_WORKFLOW_STEPS: WorkflowStepDef[] = [
+	{
+		id: "reviewer",
+		label: "🔒 安全审查",
+		type: "auto",
+		agentName: "reviewer",
+		timeoutMs: 300_000,
+	},
+];
+
 // ── Command runner ───────────────────────────────────────────
 
+/** Format workflow steps into a readable list. */
+function formatWorkflowSteps(steps: WorkflowStepDef[]): string {
+	return steps.map((s, i) => `${i + 1}. ${s.label}`).join("\n");
+}
+
 /**
- * Run a wizard with an optional Grill phase:
- * Wizard → Assemble → (Grill?) → Send
+ * Prompt user to choose workflow mode and optionally customize sub-agent chain.
+ *
+ * Returns true if workflow was started and handled (caller should return immediately).
+ * Returns false if caller should fall through to direct prompt sending.
  */
+async function promptWorkflowDecision(
+	ctx: ExtensionCommandContext,
+	pi: ExtensionAPI,
+	finalPrompt: string,
+	defaultSteps: WorkflowStepDef[],
+): Promise<boolean> {
+	if (!defaultSteps || defaultSteps.length === 0) return false;
+
+	const choice = await uiSelect(
+		ctx,
+		"🚀 选择工作流模式",
+		[
+			"1. 使用默认链式子代理（推荐）",
+			"2. 自定义链式子代理",
+			"3. 退出工作流（直接发送 prompt 给主代理）",
+		],
+	);
+
+	if (!choice || choice.startsWith("3")) {
+		return false;
+	}
+
+	if (choice.startsWith("1")) {
+		saveAnswerFile(ctx.cwd, finalPrompt);
+		await runWorkflow(ctx, pi, finalPrompt, { steps: defaultSteps }, "快速链式");
+		return true;
+	}
+
+	// ── Custom mode: let user pick steps and set timeouts (with back support) ──
+	const customSteps: WorkflowStepDef[] = [];
+	let stepIdx = 0;
+	while (stepIdx >= 0 && stepIdx < defaultSteps.length) {
+		const step = defaultSteps[stepIdx]!;
+
+		// If this step was already added, show current config, else show default
+		const existingStep = customSteps.find(cs => cs.id === step.id);
+		const include = await uiConfirm(
+			ctx,
+			`📌 ${step.label}${existingStep ? ' (已添加)' : ''}`,
+			`类型: ${step.type}\n默认超时: ${(step.timeoutMs / 60000).toFixed(0)} 分钟`,
+			stepIdx > 0, // backable only after first step
+		);
+		if (include === "back") {
+			// Go back: remove the last added step
+			if (customSteps.length > 0) {
+				customSteps.pop();
+				stepIdx--;
+				continue;
+			}
+			return false; // nothing to go back to
+		}
+		if (include === undefined) return false; // Esc
+
+		if (!include) {
+			stepIdx++;
+			continue; // skip this step
+		}
+
+		const existingTimeout = existingStep?.timeoutMs;
+		const timeoutStr = await uiInput(
+			ctx,
+			`⏱️ ${step.label} - 超时时间(分钟)`,
+			existingTimeout
+				? `当前: ${(existingTimeout / 60000).toFixed(0)} 分钟，留空保持`
+				: `留空保持默认 (${(step.timeoutMs / 60000).toFixed(0)} 分钟)`,
+		);
+		if (timeoutStr === undefined) { stepIdx++; continue; } // Esc on input = skip
+
+		if (existingStep) {
+			// Update existing step's timeout
+			existingStep.timeoutMs = timeoutStr ? parseInt(timeoutStr, 10) * 60 * 1000 || step.timeoutMs : step.timeoutMs;
+		} else {
+			customSteps.push({
+				...step,
+				timeoutMs: timeoutStr ? parseInt(timeoutStr, 10) * 60 * 1000 || step.timeoutMs : step.timeoutMs,
+			});
+		}
+		stepIdx++;
+	}
+
+	if (customSteps.length === 0) {
+		return false;
+	}
+
+	saveAnswerFile(ctx.cwd, finalPrompt);
+	await runWorkflow(ctx, pi, finalPrompt, { steps: customSteps }, "自定义");
+	return true;
+}
+
 async function runWizardWithGrill(
 	ctx: ExtensionCommandContext,
 	pi: ExtensionAPI,
@@ -374,17 +642,30 @@ async function runWizardWithGrill(
 	questions: Array<{ label: string; placeholder: string; key: string }>,
 	assembler: (answers: Record<string, string>) => string,
 	grillOptions?: GrillOptions,
+	workflowConfig?: { steps: WorkflowStepDef[] },
 ): Promise<void> {
-	ctx.ui.notify(`📋 /dev-${type} — ${label}，请逐项填写以下信息（留空跳过对应段落，Esc 取消）`, "info");
-
 	const answers: Record<string, string> = {};
-	for (const q of questions) {
-		const val = await ask(ctx, q.label, q.placeholder);
+	let idx = 0;
+
+	while (idx >= 0 && idx < questions.length) {
+		const q = questions[idx]!;
+		const existingVal = answers[q.key];
+		const placeholder = existingVal
+			? `(之前: ${existingVal.slice(0, 60)}) ${q.placeholder}`
+			: q.placeholder;
+		const val = await ask(ctx, q.label, placeholder, true, existingVal || "");
 		if (val === undefined) {
-			ctx.ui.notify("❌ 已取消", "warning");
+			return;
+		}
+		if (val === BACK_MARKER) {
+			if (idx > 0) {
+				idx--;
+				continue;
+			}
 			return;
 		}
 		answers[q.key] = val;
+		idx++;
 	}
 
 	const basePrompt = assembler(answers);
@@ -400,39 +681,32 @@ async function runWizardWithGrill(
 			loaderLabel: grillOptions.loaderLabel,
 		});
 		if (grillResult.cancelled) {
-			ctx.ui.notify("❌ 操作已取消", "warning");
 			return;
 		}
 		finalPrompt = grillResult.enhancedPrompt;
 	}
 
-	// ── Guard & persist before sending ───────────────────────
+	// ── Workflow phase ───────────────────────────
+	if (workflowConfig && workflowConfig.steps.length > 0) {
+		const handled = await promptWorkflowDecision(ctx, pi, finalPrompt, workflowConfig.steps);
+		if (handled) return;
+	}
+
+	// ── Guard & persist before sending ──────────────────────
 	if (!finalPrompt) {
-		ctx.ui.notify("⚠️ 最终提示词为空，正在尝试从备份文件恢复...", "warning");
 		const recovered = recoverFromBackup(ctx.cwd);
 		if (recovered) {
 			finalPrompt = recovered;
-			ctx.ui.notify("✅ 已从备份文件恢复提示词内容", "success");
 		} else {
-			ctx.ui.notify("❌ 错误：最终提示词为空且无可用备份，无法发送", "error");
 			return;
 		}
 	}
 	const answerPath = saveAnswerFile(ctx.cwd, finalPrompt);
-
-	ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
 	pi.sendUserMessage(finalPrompt, { deliverAs: "followUp" });
-	ctx.ui.notify(`📝 /dev-${type} 提示词已投递，主代理正在处理。备份: ${answerPath}`, "info");
 }
 
 /**
  * Run a wizard: ask questions, assemble prompt, send to agent.
- * @param ctx     Command context
- * @param pi      Extension API (for sendUserMessage)
- * @param type    Display name of the prompt type (e.g. "feat")
- * @param label   Friendly label (e.g. "新功能/创意生成")
- * @param questions  Array of [label, placeholder] pairs
- * @param assembler  Function that takes answers and returns the prompt string
  */
 async function runWizard(
 	ctx: ExtensionCommandContext,
@@ -441,28 +715,48 @@ async function runWizard(
 	label: string,
 	questions: Array<{ label: string; placeholder: string; key: string }>,
 	assembler: (answers: Record<string, string>) => string,
+	workflowConfig?: { steps: WorkflowStepDef[] },
 ): Promise<void> {
-	ctx.ui.notify(`📋 /dev-${type} — ${label}，请逐项填写以下信息（留空跳过对应段落，Esc 取消）`, "info");
-
 	const answers: Record<string, string> = {};
-
-	for (const q of questions) {
-		const val = await ask(ctx, q.label, q.placeholder);
+	let idx = 0;
+
+	while (idx >= 0 && idx < questions.length) {
+		const q = questions[idx]!;
+		const existingVal = answers[q.key];
+		const placeholder = existingVal
+			? `(之前: ${existingVal.slice(0, 60)}) ${q.placeholder}`
+			: q.placeholder;
+		const val = await ask(ctx, q.label, placeholder, true, existingVal || "");
 		if (val === undefined) {
-			ctx.ui.notify("❌ 已取消", "warning");
+			// Esc → cancel whole wizard
+			return;
+		}
+		if (val === BACK_MARKER) {
+			// Go back to previous question
+			if (idx > 0) {
+				idx--;
+				continue;
+			}
+			// Already at first question → cancel
 			return;
 		}
 		answers[q.key] = val;
+		idx++;
 	}
 
 	const prompt = assembler(answers);
-	ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
+
+	// ── Workflow phase ───────────────────────────
+	if (workflowConfig && workflowConfig.steps.length > 0) {
+		const handled = await promptWorkflowDecision(ctx, pi, prompt, workflowConfig.steps);
+		if (handled) return;
+	}
+
+	// Persist prompt before sending
+	saveAnswerFile(ctx.cwd, prompt);
 
 	// Send the assembled prompt to the main agent
 	pi.sendUserMessage(prompt, { deliverAs: "followUp" });
-
-	// Also show the prompt in a notification for reference
-	ctx.ui.notify(`📝 /dev-${type} 提示词已投递，主代理正在处理`, "info");
 }
 
 // ── Questions for each command ────────────────────────────────
@@ -555,66 +849,55 @@ const COMPARE_QUESTIONS = [
 export default function (pi: ExtensionAPI) {
 	// ── /dev-feat ──────────────────────────────────────────────
 	pi.registerCommand("dev-feat", {
-		description: "(prompt wizard) 新功能/创意生成 — 支持设计评审 (Grill) + PRD 生成",
+		description: "(prompt wizard) 新功能/创意生成 — 支持设计评审 (Grill) + 自动化工作流",
 		handler: async (_args, ctx) => {
-			// ── Phase 1: Wizard ──────────────────────────────────
-			ctx.ui.notify("📋 /dev-feat — 新功能/创意生成，请逐项填写以下信息（留空跳过对应段落，Esc 取消）", "info");
-
 			const answers: Record<string, string> = {};
-			for (const q of FEAT_QUESTIONS) {
-				const val = await ask(ctx, q.label, q.placeholder);
+			let featIdx = 0;
+			while (featIdx >= 0 && featIdx < FEAT_QUESTIONS.length) {
+				const q = FEAT_QUESTIONS[featIdx]!;
+				const existingVal = answers[q.key];
+				const placeholder = existingVal
+					? `(之前: ${existingVal.slice(0, 60)}) ${q.placeholder}`
+					: q.placeholder;
+				const val = await ask(ctx, q.label, placeholder, true, existingVal || "");
 				if (val === undefined) {
-					ctx.ui.notify("❌ 已取消", "warning");
+					return;
+				}
+				if (val === BACK_MARKER) {
+					if (featIdx > 0) {
+						featIdx--;
+						continue;
+					}
 					return;
 				}
 				answers[q.key] = val;
+				featIdx++;
 			}
 
-			// ── Phase 2: Assemble base prompt ───────────────────
 			const basePrompt = assembleFeatPrompt(answers as FeatFields);
-			ctx.ui.notify(`✅ 基本信息已收集，共 ${FEAT_QUESTIONS.length} 项`, "success");
 
-			// ── Phase 3: Grill (设计评审) ───────────────────────
 			const grillResult = await runGrillPhase(basePrompt, ctx);
 			if (grillResult.cancelled) {
-				ctx.ui.notify("❌ 操作已取消", "warning");
 				return;
 			}
 			const finalPrompt = grillResult.enhancedPrompt;
 
-			// ── Phase 4: Send to main agent ─────────────────────
+			if (FEAT_WORKFLOW_STEPS.length > 0) {
+				const handled = await promptWorkflowDecision(ctx, pi, finalPrompt, FEAT_WORKFLOW_STEPS);
+				if (handled) return;
+			}
+
 			if (!finalPrompt) {
-				ctx.ui.notify("⚠️ 最终提示词为空，正在尝试从备份文件恢复...", "warning");
 				const recovered = recoverFromBackup(ctx.cwd);
 				if (recovered) {
-					ctx.ui.notify("✅ 已从备份文件恢复提示词内容", "success");
-					const recoveredPath = saveAnswerFile(ctx.cwd, recovered);
-					ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
+					const answerPath = saveAnswerFile(ctx.cwd, recovered);
 					pi.sendUserMessage(recovered, { deliverAs: "followUp" });
-					ctx.ui.notify(`📝 /dev-feat 提示词已投递（恢复自备份），主代理正在处理。备份: ${recoveredPath}`, "info");
-					// ── Phase 5: Wait for agent to finish ───────────────
-					await ctx.waitForIdle();
-					// ── Phase 6: PRD generation ─────────────────────────
-					const moduleHint = (answers as FeatFields).module || "feature";
-					await runPRDPhase(recovered, moduleHint, pi, ctx);
-					return;
-				} else {
-					ctx.ui.notify("❌ 错误：最终提示词为空且无可用备份，无法发送", "error");
 					return;
 				}
+				return;
 			}
 			const answerPath = saveAnswerFile(ctx.cwd, finalPrompt);
-			ctx.ui.notify(`✅ 提示词已组装完成，正在发送给主代理...`, "success");
 			pi.sendUserMessage(finalPrompt, { deliverAs: "followUp" });
-			ctx.ui.notify(`📝 /dev-feat 提示词已投递，主代理正在处理。备份: ${answerPath}`, "info");
-
-			// ── Phase 5: Wait for agent to finish ───────────────
-			await ctx.waitForIdle();
-
-			// ── Phase 6: PRD generation ─────────────────────────
-			const moduleHint = (answers as FeatFields).module || "feature";
-			const grillContext = finalPrompt;
-			await runPRDPhase(grillContext, moduleHint, pi, ctx);
 		},
 	});
 
@@ -628,10 +911,11 @@ export default function (pi: ExtensionAPI) {
 				{
 					agentDef: _fixGrillAgent,
 					title: "🐛 Bug 根因分析评审",
-					description: "是否进入 Bug 根因分析评审 (Grill) 模式？\nAI 会从复现条件、根因推理、修复方案、回归风险等维度挑战你的理解。",
+					description: "AI 会从复现条件、根因推理、修复方案、回归风险等维度挑战你的理解。",
 					questionTitle: "Bug 根因分析",
 					loaderLabel: "🧠 AI 正在分析代码并生成根因评审问题...",
 				},
+				{ steps: FIX_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -646,10 +930,11 @@ export default function (pi: ExtensionAPI) {
 				{
 					agentDef: _docGrillAgent,
 					title: "📄 文档大纲评审",
-					description: "是否进入文档大纲评审 (Grill) 模式？\nAI 会从受众定位、结构安排、示例选择等维度审视你的文档计划。",
+					description: "AI 会从受众定位、结构安排、示例选择等维度审视你的文档计划。",
 					questionTitle: "文档大纲评审",
 					loaderLabel: "🧠 AI 正在分析并生成文档大纲评审问题...",
 				},
+				{ steps: DOC_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -664,10 +949,11 @@ export default function (pi: ExtensionAPI) {
 				{
 					agentDef: _refactorGrillAgent,
 					title: "🔧 重构方案评审",
-					description: "是否进入重构方案评审 (Grill) 模式？\nAI 会从模块边界、API 兼容性、测试策略、迁移风险等维度审视你的重构计划。",
+					description: "AI 会从模块边界、API 兼容性、测试策略、迁移风险等维度审视你的重构计划。",
 					questionTitle: "重构方案评审",
 					loaderLabel: "🧠 AI 正在分析代码并生成重构评审问题...",
 				},
+				{ steps: REFACTOR_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -682,10 +968,11 @@ export default function (pi: ExtensionAPI) {
 				{
 					agentDef: _testGrillAgent,
 					title: "🧪 测试计划评审",
-					description: "是否进入测试计划评审 (Grill) 模式？\nAI 会从覆盖维度、边界条件、模拟策略等角度审视你的测试方案。",
+					description: "AI 会从覆盖维度、边界条件、模拟策略等角度审视你的测试方案。",
 					questionTitle: "测试计划评审",
 					loaderLabel: "🧠 AI 正在分析并生成测试评审问题...",
 				},
+				{ steps: TEST_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -708,10 +995,11 @@ export default function (pi: ExtensionAPI) {
 				{
 					agentDef: _perfGrillAgent,
 					title: "⚡ 性能优化方案评审",
-					description: "是否进入性能优化方案评审 (Grill) 模式？\nAI 会从基准测试方法、优化方向、回归风险等维度审视你的方案。",
+					description: "AI 会从基准测试方法、优化方向、回归风险等维度审视你的方案。",
 					questionTitle: "性能优化方案评审",
 					loaderLabel: "🧠 AI 正在分析并生成性能优化评审问题...",
 				},
+				{ steps: PERF_WORKFLOW_STEPS },
 			);
 		},
 	});
@@ -720,7 +1008,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerCommand("dev-style", {
 		description: "(prompt wizard) 风格/格式调整 — 交互填写后发送优化提示词给主代理",
 		handler: async (_args, ctx) => {
-			await runWizard(ctx, pi, "style", "风格/格式调整", STYLE_QUESTIONS, assembleStylePrompt);
+			await runWizard(ctx, pi, "style", "风格/格式调整", STYLE_QUESTIONS, assembleStylePrompt, { steps: STYLE_WORKFLOW_STEPS });
 		},
 	});
 
@@ -728,7 +1016,7 @@ export default function (pi: ExtensionAPI) {
 	pi.registerCommand("dev-security", {
 		description: "(prompt wizard) 安全审查 — 交互填写后发送优化提示词给主代理",
 		handler: async (_args, ctx) => {
-			await runWizard(ctx, pi, "security", "安全审查", SECURITY_QUESTIONS, assembleSecurityPrompt);
+			await runWizard(ctx, pi, "security", "安全审查", SECURITY_QUESTIONS, assembleSecurityPrompt, { steps: SECURITY_WORKFLOW_STEPS });
 		},
 	});
 
@@ -747,4 +1035,16 @@ export default function (pi: ExtensionAPI) {
 			await runWizard(ctx, pi, "compare", "对比评估", COMPARE_QUESTIONS, assembleComparePrompt);
 		},
 	});
+
+	// ── /dev-workflow-continue — 恢复中断的工作流 ─────────────
+	pi.registerCommand("dev-workflow-continue", {
+		description: "恢复上次中断的自动化工作流（从 checkpoint 继续）",
+		handler: async (_args, ctx) => {
+			const cp = loadCheckpointFromFile(ctx.cwd);
+			if (!cp) {
+				return;
+			}
+			await runWorkflow(ctx, pi, cp.prompt, { steps: FEAT_WORKFLOW_STEPS }, "恢复");
+		},
+	});
 }

package/extensions/git-commands.ts

@@ -13,6 +13,7 @@
 
 import type { ExtensionAPI } from "@earendil-works/pi-coding-agent";
 import { discoverAgents, spawnSubagent, extractFinalOutput, type AgentDef } from "./sub-agents";
+import { uiInput } from "./ui-helpers";
 
 // ── Helpers ──────────────────────────────────────────────────
 
@@ -165,18 +166,13 @@ export default function (pi: ExtensionAPI) {
 
 			let message = args.trim();
 			if (!message) {
-				const input = await ctx.ui.input("Commit message", {
-					placeholder: "直接回车让 AI 自动生成，或输入信息后提交...",
-					required: false,
-				});
+				const input = await uiInput(ctx, "Commit message", "直接回车让 AI 自动生成，或输入信息后提交...");
 				if (input === undefined) {
-					ctx.ui.notify("Commit cancelled", "warning");
 					return;
 				}
 				message = input.trim();
 			}
 
-			ctx.ui.notify("🤖 正在委派 git-sub-agent 处理...", "info");
 
 			const task = message
 				? `Stage all changes with git add -A, then commit with message: "${message}". Do NOT ask for confirmation.`
@@ -193,7 +189,6 @@ export default function (pi: ExtensionAPI) {
 			const agent = getAgent(ctx, gitAgent, "git-agent");
 			if (!agent) return;
 
-			ctx.ui.notify("🤖 正在委派 git-sub-agent 处理...", "info");
 			await runSubAgent(
 				agent,
 				"Push commits to remote with git push. Do NOT ask for confirmation.",
@@ -211,18 +206,13 @@ export default function (pi: ExtensionAPI) {
 
 			let message = args.trim();
 			if (!message) {
-				const input = await ctx.ui.input("Commit message", {
-					placeholder: "直接回车让 AI 自动生成，或输入信息后提交并推送...",
-					required: false,
-				});
+				const input = await uiInput(ctx, "Commit message", "直接回车让 AI 自动生成，或输入信息后提交并推送...");
 				if (input === undefined) {
-					ctx.ui.notify("Commit & push cancelled", "warning");
 					return;
 				}
 				message = input.trim();
 			}
 
-			ctx.ui.notify("🤖 正在委派 git-sub-agent 处理...", "info");
 
 			const task = message
 				? `Stage all changes with git add -A, commit with message: "${message}", then push. Do NOT ask for confirmation.`