79 lines
3.0 KiB
JavaScript
79 lines
3.0 KiB
JavaScript
|
// ═══════════════════════════════════════
|
|||
|
|
// 铸渊开发Agent · 任务补全引擎
|
|||
|
|
// 高层面规划 → LLM补全 → 详细可执行步骤
|
|||
|
|
// ═══════════════════════════════════════
|
|||
|
|
|
|||
|
|
const fs = require("fs");
|
|||
|
|
const path = require("path");
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 补全单条高级任务为详细步骤
|
|||
|
|
* @param {object} task - { id, action, description, context }
|
|||
|
|
* @param {function} modelCaller - LLM 调用函数 (prompt) => string
|
|||
|
|
* @returns {object} 补全后的任务
|
|||
|
|
*/
|
|||
|
|
async function expandTask(task, modelCaller) {
|
|||
|
|
const prompt = [
|
|||
|
|
"你是一个开发Agent的任务补全引擎。将高级描述补全为可执行的详细步骤。",
|
|||
|
|
"",
|
|||
|
|
"原任务:",
|
|||
|
|
" ID: " + task.id,
|
|||
|
|
" 动作: " + task.action,
|
|||
|
|
" 描述: " + (task.description || "无"),
|
|||
|
|
" 上下文: " + (task.context || "无"),
|
|||
|
|
"",
|
|||
|
|
"请严格基于以上原任务拆解。不要改变原任务的action/target/content。只补充detail、depends_on等元信息。如果原任务已经足够具体，直接返回：",
|
|||
|
|
"[{\"step\":1,\"action\":\"write_file|run_cmd|call_api|check\",\"target\":\"文件路径或命令\",\"detail\":\"具体做什么\",\"depends_on\":null}]"
|
|||
|
|
].join("\n");
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
const raw = await modelCaller(prompt);
|
|||
|
|
const jsonMatch = raw.match(/\[[\s\S]*\]/);
|
|||
|
|
if (jsonMatch) {
|
|||
|
|
const substeps = JSON.parse(jsonMatch[0]);
|
|||
|
|
return { ...task, substeps, expanded: true, expanded_at: new Date().toISOString() };
|
|||
|
|
}
|
|||
|
|
} catch (e) {
|
|||
|
|
console.warn("[Expander] LLM补全失败，退化为单步模式: " + e.message);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 退化：LLM不可用时，原任务本身就是一个步骤
|
|||
|
|
return {
|
|||
|
|
...task,
|
|||
|
|
substeps: [{ step: 1, action: task.action, target: task.target || "", detail: task.description || "" }],
|
|||
|
|
expanded: false,
|
|||
|
|
expanded_at: new Date().toISOString()
|
|||
|
|
};
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* 完整规划补全
|
|||
|
|
* @param {object} plan - 完整规划 { plan_id, tasks[], ... }
|
|||
|
|
* @param {function} modelCaller - LLM 调用函数
|
|||
|
|
* @param {string} plansDir - 规划文件存储目录
|
|||
|
|
*/
|
|||
|
|
async function expandPlan(plan, modelCaller, plansDir) {
|
|||
|
|
console.log("[Expander] 开始补全规划: " + plan.plan_id + " · " + plan.tasks.length + "个任务");
|
|||
|
|
|
|||
|
|
const expandedTasks = [];
|
|||
|
|
for (let i = 0; i < plan.tasks.length; i++) {
|
|||
|
|
const task = plan.tasks[i];
|
|||
|
|
console.log("[Expander] 补全 " + (i + 1) + "/" + plan.tasks.length + ": " + task.id);
|
|||
|
|
const expanded = await expandTask(task, modelCaller);
|
|||
|
|
expandedTasks.push(expanded);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
plan.tasks = expandedTasks;
|
|||
|
|
plan._status = "expanded";
|
|||
|
|
plan._expanded_at = new Date().toISOString();
|
|||
|
|
|
|||
|
|
// 写回规划文件
|
|||
|
|
const planFile = path.join(plansDir, plan.plan_id + ".json");
|
|||
|
|
fs.writeFileSync(planFile, JSON.stringify(plan, null, 2), "utf-8");
|
|||
|
|
|
|||
|
|
console.log("[Expander] 补全完成 · 共 " + expandedTasks.reduce((s, t) => s + (t.substeps?.length || 0), 0) + " 个子步骤");
|
|||
|
|
return plan;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
module.exports = { expandPlan, expandTask };
|