File size: 1,672 Bytes
c09f67c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 | import { getDocumentsSchema } from "@api/schemas/documents";
import { getDocumentById, getDocuments } from "@midday/db/queries";
import { z } from "zod";
import { hasScope, READ_ONLY_ANNOTATIONS, type RegisterTools } from "../types";
export const registerDocumentTools: RegisterTools = (server, ctx) => {
const { db, teamId } = ctx;
// Require documents.read scope
if (!hasScope(ctx, "documents.read")) {
return;
}
server.registerTool(
"documents_list",
{
title: "List Documents",
description:
"List documents and files stored in the vault with filtering",
inputSchema: getDocumentsSchema.shape,
annotations: READ_ONLY_ANNOTATIONS,
},
async (params) => {
const result = await getDocuments(db, {
teamId,
cursor: params.cursor ?? null,
pageSize: params.pageSize ?? 25,
q: params.q ?? null,
tags: params.tags ?? null,
});
return {
content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
};
},
);
server.registerTool(
"documents_get",
{
title: "Get Document",
description: "Get a specific document by its ID",
inputSchema: {
id: z.string().uuid().describe("Document ID"),
},
annotations: READ_ONLY_ANNOTATIONS,
},
async ({ id }) => {
const result = await getDocumentById(db, { id, teamId });
if (!result) {
return {
content: [{ type: "text", text: "Document not found" }],
isError: true,
};
}
return {
content: [{ type: "text", text: JSON.stringify(result, null, 2) }],
};
},
);
};
|