Shrimp Task Manager

/** * planTask prompt 生成器 * planTask prompt generator * 負責將模板和參數組合成最終的 prompt * Responsible for combining templates and parameters into the final prompt */ import { loadPrompt, generatePrompt, loadPromptFromTemplate, } from "../loader.js"; import { Task, TaskDependency } from "../../types/index.js"; /** * planTask prompt 參數介面 * planTask prompt parameters interface */ export interface PlanTaskPromptParams { description: string; requirements?: string; existingTasksReference?: boolean; completedTasks?: Task[]; pendingTasks?: Task[]; memoryDir: string; } /** * 獲取 planTask 的完整 prompt * Get the complete prompt for planTask * @param params prompt 參數 * @param params prompt parameters * @returns 生成的 prompt * @returns generated prompt */ export async function getPlanTaskPrompt( params: PlanTaskPromptParams ): Promise<string> { let tasksContent = ""; if ( params.existingTasksReference && params.completedTasks && params.pendingTasks ) { const allTasks = [...params.completedTasks, ...params.pendingTasks]; // 如果存在任務，則添加相關資訊 // If tasks exist, add related information if (allTasks.length > 0) { let completeTasksContent = "no completed tasks"; // 處理已完成任務 // Process completed tasks if (params.completedTasks.length > 0) { completeTasksContent = ""; // 最多顯示10個已完成任務，避免提示詞過長 // Show at most 10 completed tasks to avoid overly long prompts const tasksToShow = params.completedTasks.length > 10 ? params.completedTasks.slice(0, 10) : params.completedTasks; tasksToShow.forEach((task, index) => { // 產生完成時間資訊 (如果有) // Generate completion time information (if available) const completedTimeText = task.completedAt ? ` - completedAt：${task.completedAt.toLocaleString()}\n` : ""; completeTasksContent += `{index}. **${task.name}** (ID: \`${ task.id }\`)\n - description：${ task.description.length > 100 ? task.description.substring(0, 100) + "..." : task.description }\n${completedTimeText}`; // 如果不是最後一個任務，添加換行 // If not the last task, add line break if (index < tasksToShow.length - 1) { completeTasksContent += "\n\n"; } }); // 如果有更多任務，顯示提示 // If there are more tasks, show hint if (params.completedTasks.length > 10) { completeTasksContent += `\n\n*（僅顯示前10個，共 ${params.completedTasks.length} 個）*\n`; // Show message indicating only first 10 tasks are displayed out of total } } let unfinishedTasksContent = "no pending tasks"; // 處理未完成任務 // Process unfinished tasks if (params.pendingTasks && params.pendingTasks.length > 0) { unfinishedTasksContent = ""; params.pendingTasks.forEach((task, index) => { const dependenciesText = task.dependencies && task.dependencies.length > 0 ? ` - dependence：${task.dependencies .map((dep: TaskDependency) => `\`${dep.taskId}\``) .join(", ")}\n` : ""; unfinishedTasksContent += `${index + 1}. **${task.name}** (ID: \`${ task.id }\`)\n - description：${ task.description.length > 150 ? task.description.substring(0, 150) + "..." : task.description }\n - status：${task.status}\n${dependenciesText}`; // 如果不是最後一個任務，添加換行 // If not the last task, add line break if (index < (params.pendingTasks?.length ?? 0) - 1) { unfinishedTasksContent += "\n\n"; } }); } const tasksTemplate = await loadPromptFromTemplate("planTask/tasks.md"); tasksContent = generatePrompt(tasksTemplate, { completedTasks: completeTasksContent, unfinishedTasks: unfinishedTasksContent, }); } } let thoughtTemplate = ""; if (process.env.ENABLE_THOUGHT_CHAIN !== "false") { thoughtTemplate = await loadPromptFromTemplate("planTask/hasThought.md"); } else { thoughtTemplate = await loadPromptFromTemplate("planTask/noThought.md"); } const indexTemplate = await loadPromptFromTemplate("planTask/index.md"); let prompt = generatePrompt(indexTemplate, { description: params.description, requirements: params.requirements || "No requirements", tasksTemplate: tasksContent, rulesPath: "shrimp-rules.md", memoryDir: params.memoryDir, thoughtTemplate: thoughtTemplate, }); // 載入可能的自定義 prompt // Load possible custom prompt return loadPrompt(prompt, "PLAN_TASK"); }