cloudaxe-opencode/packages/opencode/src/tool/todo.ts

54 lines
1.3 KiB
TypeScript
Raw Permalink Normal View History

import z from "zod"
2025-06-04 19:09:40 +00:00
import { Tool } from "./tool"
import DESCRIPTION_WRITE from "./todowrite.txt"
import { Todo } from "../session/todo"
2025-06-04 19:09:40 +00:00
export const TodoWriteTool = Tool.define("todowrite", {
2025-06-04 19:09:40 +00:00
description: DESCRIPTION_WRITE,
parameters: z.object({
2025-11-16 01:19:36 +00:00
todos: z.array(z.object(Todo.Info.shape)).describe("The updated todo list"),
2025-06-04 19:09:40 +00:00
}),
async execute(params, ctx) {
await ctx.ask({
permission: "todowrite",
patterns: ["*"],
always: ["*"],
metadata: {},
})
await Todo.update({
sessionID: ctx.sessionID,
todos: params.todos,
})
2025-06-04 19:09:40 +00:00
return {
title: `${params.todos.filter((x) => x.status !== "completed").length} todos`,
2025-06-04 19:09:40 +00:00
output: JSON.stringify(params.todos, null, 2),
metadata: {
todos: params.todos,
},
}
},
})
export const TodoReadTool = Tool.define("todoread", {
2025-06-04 19:09:40 +00:00
description: "Use this tool to read your todo list",
2025-06-04 22:03:04 +00:00
parameters: z.object({}),
async execute(_params, ctx) {
await ctx.ask({
permission: "todoread",
patterns: ["*"],
always: ["*"],
metadata: {},
})
const todos = await Todo.get(ctx.sessionID)
2025-06-04 19:09:40 +00:00
return {
title: `${todos.filter((x) => x.status !== "completed").length} todos`,
2025-06-04 19:09:40 +00:00
metadata: {
todos,
},
output: JSON.stringify(todos, null, 2),
}
},
})