momo-ai 1.0.11 → 1.0.12

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "momo-ai",
-    "version": "1.0.11",
+    "version": "1.0.12",
     "description": "Rachel Momo ( OpenSpec )",
     "main": "src/momo.js",
     "bin": {

package/src/_agent/trae/momo-taskplan.json

@@ -0,0 +1,5 @@
+{
+    "name": "页面规划师",
+    "id": "momo-taskplan",
+    "uri": "https://s.trae.ai/a/0447bd"
+}

package/src/_template/PROMPT/trae/momo-task.ejs

@@ -1,4 +1,4 @@
-# 领域建模提示词
+# 任务规划提示词
 
 <!-- BEGIN -->
 解读 specification/changes/ 下的所有模块（必须是 MXX- 开头的目录）对应的变更内容，并生成领域模型，根据每个目录中的 tasks.md 生成对应的任务并且更新 tasks.md 文件：

package/src/_template/PROMPT/trae/momo-taskplan.ejs

@@ -0,0 +1,8 @@
+# 页面规划提示词
+
+<!-- BEGIN -->
+解读 specification/changes/ 下的所有模块子目录 tasks/ 目录下的任务信息，规划每个任务的页面开发并且更新对应的任务 md 文档。
+
+1. 每个模块的任务细节位于 specification/changes/MXX-模块/tasks/ 目录下。
+2. 参考 project.md 文件中的技术栈和风格说明。
+<!-- END -->

package/src/executor/executeAgent.js

@@ -135,25 +135,34 @@ module.exports = async (options) => {
         
         // 如果没有提供 agent 参数，显示可用的 agents 并让用户选择
         if (!agentName) {
-            Ec.waiting('可用的 Agents:');
-            agents.forEach((agent, index) => {
-                Ec.waiting(`${index + 1}. ${agent.argName.red.bold} (${agent.name}), id = ${agent.id}, uri = ${agent.uri}`);
-            });
-            
-            // 获取用户选择
-            const answer = await Ec.ask('请选择要使用的 Agent (输入编号): ');
-            const selectedIndex = parseInt(answer) - 1;
-            
-            // 验证选择
-            if (isNaN(selectedIndex) || selectedIndex < 0 || selectedIndex >= agents.length) {
-                Ec.error("❌ 无效的选择");
-                Ec.askClose();
-                process.exit(1);
+            while (true) {
+                Ec.waiting('可用的 Agents:');
+                agents.forEach((agent, index) => {
+                    Ec.waiting(`${index + 1}. ${agent.argName.red.bold} (${agent.name}), id = ${agent.id}, uri = ${agent.uri}`);
+                });
+                Ec.waiting(`${agents.length + 1}. 退出`);
+                
+                // 获取用户选择
+                const answer = await Ec.ask('请选择要使用的 Agent (输入编号): ');
+                const selectedIndex = parseInt(answer) - 1;
+                
+                // 验证选择
+                if (isNaN(selectedIndex) || selectedIndex < 0 || selectedIndex > agents.length) {
+                    Ec.error("❌ 无效的选择");
+                    continue;
+                }
+                
+                // 检查是否选择退出
+                if (selectedIndex === agents.length) {
+                    Ec.info("已退出 Agent 选择");
+                    Ec.askClose();
+                    process.exit(0);
+                }
+                
+                // 使用选择的 agent
+                const selectedAgent = agents[selectedIndex];
+                return await _processAgent(selectedAgent);
             }
-            
-            // 使用选择的 agent
-            const selectedAgent = agents[selectedIndex];
-            return await _processAgent(selectedAgent);
         }
         
         // 查找指定的 agent
@@ -164,6 +173,7 @@ module.exports = async (options) => {
             agents.forEach((a, index) => {
                 Ec.waiting(`${index + 1}. ${a.argName.red.bold} (${a.name}), id = ${a.id}, uri = ${a.uri}`);
             });
+            Ec.waiting(`${agents.length + 1}. 退出`);
             Ec.askClose();
             process.exit(1);
         }

package/src/executor/executeProject.js

@@ -14,7 +14,7 @@ const fsAsync = {
 };
 
 /**
- * 获取或创建 refer.json 配置文件路径
+ * 获取 refer.json 配置文件路径
  * @param {string} baseDir 项目根目录
  * @returns {string} refer.json 文件路径
  */
@@ -22,11 +22,27 @@ const _getReferConfigPath = (baseDir) => {
     return path.resolve(baseDir, '.momo', 'advanced', 'refer.json');
 };
 
+/**
+ * 获取模板中的 refer.json 配置文件路径
+ * @returns {string} 模板中 refer.json 文件路径
+ */
+const _getTemplateReferConfigPath = () => {
+    return path.resolve(__dirname, '../_template/LAIN/.momo/advanced/refer.json');
+};
+
 /**
  * 获取默认的 refer 配置
  * @returns {Object} 默认配置对象
  */
 const _getDefaultReferConfig = () => {
+    // 从模板文件读取默认配置
+    const templateConfigPath = _getTemplateReferConfigPath();
+    if (fs.existsSync(templateConfigPath)) {
+        const content = fs.readFileSync(templateConfigPath, 'utf8');
+        return JSON.parse(content);
+    }
+    
+    // 如果模板文件不存在，返回硬编码的默认配置
     return {
         maven: {
             file: 'pom.xml',