mirror of
https://gitea.toothfairyai.com/ToothFairyAI/tf_code.git
synced 2026-03-31 14:22:27 +00:00
Co-authored-by: Github Action <action@github.com> Co-authored-by: Adam <2363879+adamdotdevin@users.noreply.github.com>
54 lines
1.3 KiB
TypeScript
54 lines
1.3 KiB
TypeScript
import z from "zod"
|
|
import { Tool } from "./tool"
|
|
import DESCRIPTION_WRITE from "./todowrite.txt"
|
|
import { Todo } from "../session/todo"
|
|
|
|
export const TodoWriteTool = Tool.define("todowrite", {
|
|
description: DESCRIPTION_WRITE,
|
|
parameters: z.object({
|
|
todos: z.array(z.object(Todo.Info.shape)).describe("The updated todo list"),
|
|
}),
|
|
async execute(params, ctx) {
|
|
await ctx.ask({
|
|
permission: "todowrite",
|
|
patterns: ["*"],
|
|
always: ["*"],
|
|
metadata: {},
|
|
})
|
|
|
|
await Todo.update({
|
|
sessionID: ctx.sessionID,
|
|
todos: params.todos,
|
|
})
|
|
return {
|
|
title: `${params.todos.filter((x) => x.status !== "completed").length} todos`,
|
|
output: JSON.stringify(params.todos, null, 2),
|
|
metadata: {
|
|
todos: params.todos,
|
|
},
|
|
}
|
|
},
|
|
})
|
|
|
|
export const TodoReadTool = Tool.define("todoread", {
|
|
description: "Use this tool to read your todo list",
|
|
parameters: z.object({}),
|
|
async execute(_params, ctx) {
|
|
await ctx.ask({
|
|
permission: "todoread",
|
|
patterns: ["*"],
|
|
always: ["*"],
|
|
metadata: {},
|
|
})
|
|
|
|
const todos = await Todo.get(ctx.sessionID)
|
|
return {
|
|
title: `${todos.filter((x) => x.status !== "completed").length} todos`,
|
|
metadata: {
|
|
todos,
|
|
},
|
|
output: JSON.stringify(todos, null, 2),
|
|
}
|
|
},
|
|
})
|