Shrimp Task Manager

import { z } from "zod"; import { getAllTasks } from "../../models/taskModel.js"; import { TaskStatus } from "../../types/index.js"; import { getListTasksPrompt } from "../../prompts/index.js"; export const listTasksSchema = z.object({ status: z .enum(["all", "pending", "in_progress", "completed"]) .describe("要列出的任務狀態，可選擇 'all' 列出所有任務，或指定具體狀態"), // Task status to list, choose 'all' to list all tasks, or specify a specific status }); // 列出任務工具 // List tasks tool export async function listTasks({ status }: z.infer<typeof listTasksSchema>) { const tasks = await getAllTasks(); let filteredTasks = tasks; switch (status) { case "all": break; case "pending": filteredTasks = tasks.filter( (task) => task.status === TaskStatus.PENDING ); break; case "in_progress": filteredTasks = tasks.filter( (task) => task.status === TaskStatus.IN_PROGRESS ); break; case "completed": filteredTasks = tasks.filter( (task) => task.status === TaskStatus.COMPLETED ); break; } if (filteredTasks.length === 0) { return { content: [ { type: "text" as const, text: `## 系統通知\n\n目前系統中沒有${ // ## System Notification\n\nCurrently there are no ${ status === "all" ? "任何" : `任何 ${status} 的` // status === "all" ? "any" : `any ${status}` }任務。請查詢其他狀態任務或先使用「split_tasks」工具創建任務結構，再進行後續操作。`, // }tasks. Please query other status tasks or first use the "split_tasks" tool to create task structure, then proceed with subsequent operations. }, ], }; } const tasksByStatus = tasks.reduce((acc, task) => { if (!acc[task.status]) { acc[task.status] = []; } acc[task.status].push(task); return acc; }, {} as Record<string, typeof tasks>); // 使用prompt生成器獲取最終prompt // Use prompt generator to get the final prompt const prompt = await getListTasksPrompt({ status, tasks: tasksByStatus, allTasks: filteredTasks, }); return { content: [ { type: "text" as const, text: prompt, }, ], }; }