Coverage for aceflow_mcp_server/prompts.py: 84%
55 statements
« prev ^ index » next coverage.py v7.10.1, created at 2025-08-02 16:56 +0800
1"""AceFlow MCP Prompts implementation."""
3try:
4 from fastmcp.prompts import prompt
5except ImportError:
6 # Mock implementation for testing
7 def prompt(func):
8 return func
9from typing import Dict, Any, Optional
10import json
11from pathlib import Path
14class AceFlowPrompts:
15 """AceFlow MCP Prompts collection."""
17 def __init__(self):
18 """Initialize prompts."""
19 pass
21 def _get_current_project_state(self) -> Dict[str, Any]:
22 """Get current project state for context."""
23 try:
24 # Look for project state in current directory
25 current_dir = Path.cwd()
27 # Search for AceFlow project root
28 project_root = current_dir
29 while project_root != project_root.parent:
30 if (project_root / ".aceflow").exists():
31 break
32 project_root = project_root.parent
33 else:
34 # Not found, use current directory
35 project_root = current_dir
37 state_file = project_root / ".aceflow" / "current_state.json"
38 if state_file.exists():
39 with open(state_file, 'r', encoding='utf-8') as f:
40 return json.load(f)
41 else:
42 return {
43 "project": {"name": "Unknown", "mode": "Unknown"},
44 "flow": {"current_stage": "unknown", "progress_percentage": 0}
45 }
46 except Exception:
47 return {
48 "project": {"name": "Unknown", "mode": "Unknown"},
49 "flow": {"current_stage": "unknown", "progress_percentage": 0}
50 }
52 @prompt
53 def workflow_assistant(
54 self,
55 task: Optional[str] = None,
56 context: Optional[str] = None
57 ) -> str:
58 """Generate workflow assistance prompt."""
59 try:
60 current_state = self._get_current_project_state()
61 current_stage = current_state.get("flow", {}).get("current_stage", "unknown")
62 project_name = current_state.get("project", {}).get("name", "Unknown")
63 mode = current_state.get("project", {}).get("mode", "Unknown")
64 progress = current_state.get("flow", {}).get("progress_percentage", 0)
66 base_prompt = f"""# AceFlow 工作流助手
68你是一个专业的AceFlow工作流助手，帮助用户管理结构化的软件开发项目。
70## 当前项目状态
71- **项目名称**: {project_name}
72- **工作流模式**: {mode}
73- **当前阶段**: {current_stage}
74- **完成进度**: {progress}%
76## 可用工具
77- **aceflow_init**: 初始化新项目 (支持 minimal/standard/complete/smart 模式)
78- **aceflow_stage**: 管理项目阶段 (status/next/list/reset 操作)
79- **aceflow_validate**: 验证项目合规性 (basic/complete 验证模式)
80- **aceflow_template**: 管理工作流模板 (list/apply/validate 操作)
82## 可用资源
83- **aceflow://project/state**: 获取详细项目状态
84- **aceflow://workflow/config**: 获取工作流配置
85- **aceflow://stage/guide/{{stage}}**: 获取特定阶段指导
87## 工作指导原则
89### 1. 项目初始化
90- 根据项目复杂度选择合适的工作流模式
91- minimal: 快速原型和概念验证
92- standard: 标准软件开发流程
93- complete: 企业级完整开发流程
94- smart: AI增强的自适应流程
96### 2. 阶段管理
97- 严格按照定义的工作流阶段顺序执行
98- 每个阶段完成后使用 aceflow_stage 工具更新状态
99- 确保每个阶段的交付物符合质量标准
101### 3. 质量保证
102- 定期使用 aceflow_validate 工具检查项目合规性
103- 所有项目文档和代码必须输出到 aceflow_result/ 目录
104- 保持跨对话的工作记忆和上下文连续性
106### 4. 文档规范
107- 使用标准的Markdown格式
108- 保持文档结构清晰，内容完整
109- 包含必要的代码示例和使用说明
111### 5. 协作支持
112- 提供清晰的阶段指导和最佳实践建议
113- 识别潜在风险并提供缓解策略
114- 为下一阶段转换做好准备工作
116## 响应方式
117- 始终基于当前项目状态提供建议
118- 提供具体、可操作的指导步骤
119- 主动识别和解决常见问题
120- 保持专业、友好的沟通风格
121- 优先使用中文进行交流
123## 特殊注意事项
124- 如果项目尚未初始化，优先指导用户进行项目初始化
125- 在执行任何操作前，建议先检查当前项目状态
126- 对于复杂任务，提供分步骤的执行计划
127- 及时更新项目状态，确保工作流程的连续性"""
129 if task:
130 base_prompt += f"""
132## 当前任务
133{task}
135请根据当前项目状态和任务要求，提供具体的执行指导和建议。"""
137 if context:
138 base_prompt += f"""
140## 额外上下文
141{context}
143请考虑这些额外信息来调整你的建议和指导。"""
145 # Add stage-specific guidance if we know the current stage
146 if current_stage != "unknown":
147 base_prompt += f"""
149## 当前阶段指导
150当前正在进行 '{current_stage}' 阶段。你可以使用资源 aceflow://stage/guide/{current_stage} 获取详细的阶段指导信息。
152请重点关注该阶段的：
153- 主要任务和目标
154- 输出要求和质量标准
155- 下一阶段的准备工作"""
157 return base_prompt
159 except Exception as e:
160 return f"""# AceFlow 工作流助手 (错误模式)
162抱歉，在生成工作流助手提示时发生错误: {str(e)}
164## 基础功能
165即使在错误模式下，你仍然可以：
166- 使用 aceflow_init 工具初始化新项目
167- 使用 aceflow_stage 工具管理项目阶段
168- 使用 aceflow_validate 工具验证项目状态
169- 使用 aceflow_template 工具管理模板
171## 建议
1721. 检查是否在正确的项目目录中
1732. 确认项目已正确初始化
1743. 重试操作或联系技术支持
176请继续使用可用的工具来协助项目开发。"""
178 @prompt
179 def stage_guide(self, stage: str) -> str:
180 """Generate stage-specific guidance prompt."""
181 try:
182 current_state = self._get_current_project_state()
183 project_name = current_state.get("project", {}).get("name", "Unknown")
184 mode = current_state.get("project", {}).get("mode", "Unknown")
186 prompt = f"""# {stage.upper()} 阶段指导助手
188你正在为项目 "{project_name}" ({mode} 模式) 提供 '{stage}' 阶段的专业指导。
190## 阶段职责
191你的主要职责是帮助用户：
1921. 理解当前阶段的目标和要求
1932. 提供具体的、可操作的执行步骤
1943. 确保输出符合质量标准和规范
1954. 为下一阶段的转换做好准备
197## 指导方针
199### 1. 阶段目标明确化
200- 清晰解释当前阶段的核心目标
201- 说明该阶段在整体工作流中的作用
202- 强调关键的成功标准和质量要求
204### 2. 任务分解和执行
205- 将阶段目标分解为具体的执行任务
206- 提供详细的操作步骤和最佳实践
207- 给出预期的时间安排和里程碑
209### 3. 输出物管理
210- 明确定义应该产生的交付物
211- 指定输出文件的位置和命名规范
212- 确保所有输出都存放在 aceflow_result/ 目录中
214### 4. 质量控制
215- 提供质量检查清单和验证方法
216- 识别常见问题和解决方案
217- 建议适当的审查和验证流程
219### 5. 阶段衔接
220- 准备下一阶段所需的输入
221- 确保工作的连续性和一致性
222- 及时更新项目状态和进度
224## 专业建议方式
225- 基于项目的具体情况提供个性化建议
226- 使用具体的示例和模板来说明要点
227- 保持建议的实用性和可操作性
228- 主动识别潜在的风险和挑战
230## 沟通风格
231- 使用专业但易懂的语言
232- 保持积极和支持的态度
233- 提供鼓励和实用的反馈
234- 根据用户的经验水平调整解释深度
236## 资源利用
237可以引用以下资源来增强指导：
238- aceflow://stage/guide/{stage} - 获取详细的阶段指南
239- aceflow://project/state - 了解当前项目状态
240- aceflow://workflow/config - 查看工作流配置
242## 工具建议
243在适当的时候，建议用户使用：
244- aceflow_validate - 验证当前阶段的完成情况
245- aceflow_stage - 管理阶段转换
246- aceflow_template - 应用相关模板
248现在请开始为用户提供 '{stage}' 阶段的专业指导，帮助他们高效完成阶段任务并确保质量标准。"""
250 return prompt
252 except Exception as e:
253 return f"""# {stage.upper()} 阶段指导助手 (错误模式)
255抱歉，在生成阶段指导提示时发生错误: {str(e)}
257## 基础指导
258即使在错误模式下，我仍然可以为 '{stage}' 阶段提供基础指导：
2601. **明确阶段目标** - 确保理解当前阶段要达成的目标
2612. **制定执行计划** - 将目标分解为可执行的任务
2623. **关注输出质量** - 确保所有交付物符合标准
2634. **准备下一阶段** - 为后续工作做好准备
265## 建议操作
266- 使用 aceflow://stage/guide/{stage} 资源获取详细指导
267- 定期使用 aceflow_validate 工具检查进度
268- 将所有输出保存到 aceflow_result/ 目录
270请继续进行阶段工作，如需帮助请随时询问。"""