@powerhousedao/reactor-mcp 4.1.0-dev.2 → 4.1.0-dev.3

package/dist/src/mcp/reactor.js

@@ -0,0 +1,408 @@
+import { z } from "zod";
+import { toolWithCallback } from "./utils.js";
+export const getDocumentTool = {
+    name: "getDocument",
+    description: "Retrieve a document by its ID",
+    inputSchema: {
+        id: z.string().describe("ID of the document to retrieve"),
+    },
+    outputSchema: {
+        document: z.object({}).describe("Document object"), // TODO: Define document schema
+    },
+};
+export const createDocumentTool = {
+    name: "createDocument",
+    description: "Create a new document",
+    inputSchema: {
+        documentType: z.string().describe("Type of the document to create"),
+        documentId: z.string().optional().describe("Optional ID for the document"),
+    },
+    outputSchema: {
+        result: z
+            .object({
+            status: z
+                .enum(["SUCCESS", "CONFLICT", "MISSING", "ERROR"])
+                .describe("Operation status"),
+            error: z
+                .string()
+                .optional()
+                .describe("Error details if operation failed"), // TODO: Define error schema
+            operations: z
+                .array(z.object({}))
+                .describe("Array of operations created"), // TODO: Define operation schema
+            document: z.object({}).optional().describe("Updated document"), // TODO: Define document schema
+            signals: z.array(z.object({})).describe("Array of signals generated"), // TODO: Define signal schema
+        })
+            .describe("Operation result"),
+    },
+};
+export const getDocumentsTool = {
+    name: "getDocuments",
+    description: "List documents in a drive",
+    inputSchema: {
+        parentId: z.string().describe("ID of the drive"),
+    },
+    outputSchema: {
+        documentIds: z.array(z.string()).describe("Array of document IDs"),
+    },
+};
+export const deleteDocumentTool = {
+    name: "deleteDocument",
+    description: "Delete a document",
+    inputSchema: {
+        documentId: z.string().describe("ID of the document to delete"),
+    },
+    outputSchema: {
+        success: z.boolean().describe("Whether the deletion was successful"),
+    },
+};
+export const addActionTool = {
+    name: "addAction",
+    description: "Add an action to a document",
+    inputSchema: {
+        documentId: z.string().describe("ID of the document"),
+        action: z
+            .object({
+            type: z.string().describe("The name of the action"),
+            input: z
+                .union([
+                z.string(),
+                z.number(),
+                z.boolean(),
+                z.object({}),
+                z.array(z.any()),
+                z.null(),
+            ])
+                .describe("The payload of the action"),
+            scope: z.string().describe("The scope of the action"),
+            context: z.object({}).optional().describe("Optional action context"), // TODO: Define context schema
+        })
+            .strict()
+            .describe("Action to add to the document"),
+    },
+    outputSchema: {
+        result: z
+            .object({
+            status: z
+                .enum(["SUCCESS", "CONFLICT", "MISSING", "ERROR"])
+                .describe("Operation status"),
+            error: z
+                .string()
+                .optional()
+                .describe("Error details if operation failed"), // TODO: Define error schema
+            operations: z
+                .array(z.object({}))
+                .describe("Array of operations created"), // TODO: Define operation schema
+            document: z.object({}).optional().describe("Updated document"), // TODO: Define document schema
+            signals: z.array(z.object({})).describe("Array of signals generated"), // TODO: Define signal schema
+        })
+            .describe("Operation result"),
+    },
+};
+export const addOperationTool = {
+    name: "addOperation",
+    description: "Add an operation to a document",
+    inputSchema: {
+        documentId: z.string().describe("ID of the document"),
+        operation: z
+            .object({
+            type: z.string().describe("The name of the action"),
+            input: z
+                .union([
+                z.string(),
+                z.number(),
+                z.boolean(),
+                z.object({}),
+                z.array(z.any()),
+                z.null(),
+            ])
+                .describe("The payload of the action"),
+            scope: z.string().describe("The scope of the action"),
+            index: z.number().describe("Position of the operation in the history"),
+            timestamp: z
+                .string()
+                .describe("Timestamp of when the operation was added"),
+            hash: z.string().describe("Hash of the resulting document data"),
+            skip: z.number().describe("The number of operations skipped"),
+            error: z
+                .string()
+                .optional()
+                .describe("Error message for a failed action"),
+            id: z.string().optional().describe("Unique operation id"),
+            context: z.object({}).optional().describe("Optional action context"), // TODO: Define context schema
+        })
+            .strict()
+            .describe("Operation to add to the document"),
+    },
+    outputSchema: {
+        result: z
+            .object({
+            status: z
+                .enum(["SUCCESS", "CONFLICT", "MISSING", "ERROR"])
+                .describe("Operation status"),
+            error: z
+                .string()
+                .optional()
+                .describe("Error details if operation failed"), // TODO: Define error schema
+            operations: z
+                .array(z.object({}))
+                .describe("Array of operations created"), // TODO: Define operation schema
+            document: z.object({}).optional().describe("Updated document"), // TODO: Define document schema
+            signals: z.array(z.object({})).describe("Array of signals generated"), // TODO: Define signal schema
+        })
+            .describe("Operation result"),
+    },
+};
+// Drive Operation Tools
+export const getDrivesTool = {
+    name: "getDrives",
+    description: "List all drives",
+    inputSchema: {},
+    outputSchema: {
+        driveIds: z.array(z.string()).describe("Array of drive IDs"),
+    },
+};
+export const addDriveTool = {
+    name: "addDrive",
+    description: "Create a new drive",
+    inputSchema: {
+        driveInput: z
+            .object({
+            global: z
+                .object({
+                name: z.string().describe("Name of the drive"),
+                icon: z
+                    .string()
+                    .nullable()
+                    .optional()
+                    .describe("Optional icon for the drive"),
+            })
+                .describe("Global drive properties"),
+            id: z.string().optional().describe("Optional drive ID"),
+            slug: z.string().optional().describe("Optional drive slug"),
+            preferredEditor: z
+                .string()
+                .optional()
+                .describe("Optional preferred editor"),
+            local: z
+                .object({
+                availableOffline: z
+                    .boolean()
+                    .optional()
+                    .describe("Whether drive is available offline"),
+                sharingType: z
+                    .string()
+                    .nullable()
+                    .optional()
+                    .describe("Sharing type"),
+            })
+                .optional()
+                .describe("Optional local state properties"),
+        })
+            .describe("Drive configuration"),
+    },
+    outputSchema: {
+        drive: z.object({}).describe("Created drive document"), // TODO: Define DocumentDriveDocument schema
+    },
+};
+export const getDriveTool = {
+    name: "getDrive",
+    description: "Get a specific drive",
+    inputSchema: {
+        driveId: z.string().describe("ID of the drive to retrieve"),
+        options: z
+            .object({
+            revisions: z
+                .record(z.number())
+                .optional()
+                .describe("Optional revision filter"),
+            checkHashes: z.boolean().optional().describe("Whether to check hashes"),
+            // TODO: Add other ReducerOptions if needed
+        })
+            .optional()
+            .describe("Optional get document options"),
+    },
+    outputSchema: {
+        drive: z.object({}).describe("Drive document"), // TODO: Define DocumentDriveDocument schema
+    },
+};
+export const deleteDriveTool = {
+    name: "deleteDrive",
+    description: "Delete a drive",
+    inputSchema: {
+        driveId: z.string().describe("ID of the drive to delete"),
+    },
+    outputSchema: {
+        success: z.boolean().describe("Whether the deletion was successful"),
+    },
+};
+export const addRemoteDriveTool = {
+    name: "addRemoteDrive",
+    description: "Connect to a remote drive",
+    inputSchema: {
+        url: z.string().describe("URL of the remote drive"),
+        options: z
+            .object({
+            availableOffline: z
+                .boolean()
+                .describe("Whether drive is available offline"),
+            sharingType: z.string().nullable().optional().describe("Sharing type"),
+            pullFilter: z
+                .object({
+                branch: z
+                    .array(z.string())
+                    .nullable()
+                    .optional()
+                    .describe("Branch filter"),
+                documentId: z
+                    .array(z.string())
+                    .nullable()
+                    .optional()
+                    .describe("Document ID filter"),
+                documentType: z
+                    .array(z.string())
+                    .nullable()
+                    .optional()
+                    .describe("Document type filter"),
+                scope: z
+                    .array(z.string())
+                    .nullable()
+                    .optional()
+                    .describe("Scope filter"),
+            })
+                .optional()
+                .describe("Optional pull filter"),
+            pullInterval: z
+                .number()
+                .optional()
+                .describe("Pull interval in milliseconds"),
+        })
+            .describe("Remote drive options"),
+    },
+    outputSchema: {
+        drive: z.object({}).describe("Connected remote drive document"), // TODO: Define DocumentDriveDocument schema
+    },
+};
+// All tools array for type inference
+// eslint-disable-next-line @typescript-eslint/no-unused-vars
+const allTools = [
+    getDocumentTool,
+    createDocumentTool,
+    getDocumentsTool,
+    deleteDocumentTool,
+    addActionTool,
+    addOperationTool,
+    getDrivesTool,
+    addDriveTool,
+    getDriveTool,
+    deleteDriveTool,
+    addRemoteDriveTool,
+];
+export async function createReactorMcpProvider(reactor) {
+    await reactor.initialize();
+    const tools = {
+        getDocument: toolWithCallback(getDocumentTool, async (params) => {
+            const doc = await reactor.getDocument(params.id);
+            return { document: doc };
+        }),
+        createDocument: toolWithCallback(createDocumentTool, async (params) => {
+            // Create document input based on provided parameters
+            const createInput = {
+                documentType: params.documentType,
+                ...(params.documentId && { id: params.documentId }),
+            };
+            const result = await reactor.queueDocument(createInput);
+            return {
+                result: {
+                    ...result,
+                    error: result.error?.message,
+                },
+            };
+        }),
+        getDocuments: toolWithCallback(getDocumentsTool, async (params) => {
+            const documentIds = await reactor.getDocuments(params.parentId);
+            return { documentIds };
+        }),
+        deleteDocument: toolWithCallback(deleteDocumentTool, async (params) => {
+            try {
+                await reactor.deleteDocument(params.documentId);
+                return { success: true };
+            }
+            catch {
+                return { success: false };
+            }
+        }),
+        addAction: toolWithCallback(addActionTool, async (params) => {
+            const result = await reactor.addAction(params.documentId, params.action);
+            return {
+                result: {
+                    ...result,
+                    error: result.error?.message,
+                },
+            };
+        }),
+        addOperation: toolWithCallback(addOperationTool, async (params) => {
+            const result = await reactor.addOperation(params.documentId, params.operation);
+            return {
+                result: {
+                    ...result,
+                    error: typeof result.error === "string"
+                        ? result.error
+                        : result.error?.message,
+                },
+            };
+        }),
+        // Drive operation implementations
+        getDrives: toolWithCallback(getDrivesTool, async () => {
+            const driveIds = await reactor.getDrives();
+            return { driveIds };
+        }),
+        addDrive: toolWithCallback(addDriveTool, async (params) => {
+            // Extract preferredEditor and create proper DriveInput
+            const { preferredEditor, ...driveInput } = params.driveInput;
+            const drive = await reactor.addDrive(driveInput, preferredEditor);
+            return { drive };
+        }),
+        getDrive: toolWithCallback(getDriveTool, async (params) => {
+            const drive = await reactor.getDrive(params.driveId, params.options);
+            return { drive };
+        }),
+        deleteDrive: toolWithCallback(deleteDriveTool, async (params) => {
+            try {
+                await reactor.deleteDrive(params.driveId);
+                return { success: true };
+            }
+            catch {
+                return { success: false };
+            }
+        }),
+        addRemoteDrive: toolWithCallback(addRemoteDriveTool, async (params) => {
+            const { sharingType, pullFilter, ...restOptions } = params.options;
+            const drive = await reactor.addRemoteDrive(params.url, {
+                ...restOptions,
+                listeners: [],
+                triggers: [],
+                sharingType: sharingType ?? null,
+                ...(pullFilter && {
+                    pullFilter: {
+                        branch: pullFilter.branch === undefined ? [] : pullFilter.branch,
+                        documentId: pullFilter.documentId === undefined ? [] : pullFilter.documentId,
+                        documentType: pullFilter.documentType === undefined
+                            ? []
+                            : pullFilter.documentType,
+                        scope: pullFilter.scope === undefined ? [] : pullFilter.scope,
+                    },
+                }),
+            });
+            return { drive };
+        }),
+    };
+    const resources = {};
+    const prompts = {};
+    return {
+        tools,
+        resources,
+        prompts,
+    };
+}
+//# sourceMappingURL=reactor.js.map

package/dist/src/mcp/reactor.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reactor.js","sourceRoot":"","sources":["../../../src/mcp/reactor.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EAAE,gBAAgB,EAAE,MAAM,YAAY,CAAC;AAE9C,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,+BAA+B;IAC5C,WAAW,EAAE;QACX,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;KAC1D;IACD,YAAY,EAAE;QACZ,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,iBAAiB,CAAC,EAAE,+BAA+B;KACpF;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,IAAI,EAAE,gBAAgB;IACtB,WAAW,EAAE,uBAAuB;IACpC,WAAW,EAAE;QACX,YAAY,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gCAAgC,CAAC;QACnE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;KAC3E;IACD,YAAY,EAAE;QACZ,MAAM,EAAE,CAAC;aACN,MAAM,CAAC;YACN,MAAM,EAAE,CAAC;iBACN,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;iBACjD,QAAQ,CAAC,kBAAkB,CAAC;YAC/B,KAAK,EAAE,CAAC;iBACL,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,mCAAmC,CAAC,EAAE,4BAA4B;YAC9E,UAAU,EAAE,CAAC;iBACV,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;iBACnB,QAAQ,CAAC,6BAA6B,CAAC,EAAE,gCAAgC;YAC5E,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,+BAA+B;YAC/F,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,6BAA6B;SACrG,CAAC;aACD,QAAQ,CAAC,kBAAkB,CAAC;KAChC;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,IAAI,EAAE,cAAc;IACpB,WAAW,EAAE,2BAA2B;IACxC,WAAW,EAAE;QACX,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;KACjD;IACD,YAAY,EAAE;QACZ,WAAW,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,uBAAuB,CAAC;KACnE;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,IAAI,EAAE,gBAAgB;IACtB,WAAW,EAAE,mBAAmB;IAChC,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,8BAA8B,CAAC;KAChE;IACD,YAAY,EAAE;QACZ,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;KACrE;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,6BAA6B;IAC1C,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACrD,MAAM,EAAE,CAAC;aACN,MAAM,CAAC;YACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;YACnD,KAAK,EAAE,CAAC;iBACL,KAAK,CAAC;gBACL,CAAC,CAAC,MAAM,EAAE;gBACV,CAAC,CAAC,MAAM,EAAE;gBACV,CAAC,CAAC,OAAO,EAAE;gBACX,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACZ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBAChB,CAAC,CAAC,IAAI,EAAE;aACT,CAAC;iBACD,QAAQ,CAAC,2BAA2B,CAAC;YACxC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;YACrD,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,8BAA8B;SACrG,CAAC;aACD,MAAM,EAAE;aACR,QAAQ,CAAC,+BAA+B,CAAC;KAC7C;IACD,YAAY,EAAE;QACZ,MAAM,EAAE,CAAC;aACN,MAAM,CAAC;YACN,MAAM,EAAE,CAAC;iBACN,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;iBACjD,QAAQ,CAAC,kBAAkB,CAAC;YAC/B,KAAK,EAAE,CAAC;iBACL,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,mCAAmC,CAAC,EAAE,4BAA4B;YAC9E,UAAU,EAAE,CAAC;iBACV,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;iBACnB,QAAQ,CAAC,6BAA6B,CAAC,EAAE,gCAAgC;YAC5E,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,+BAA+B;YAC/F,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,6BAA6B;SACrG,CAAC;aACD,QAAQ,CAAC,kBAAkB,CAAC;KAChC;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,gBAAgB,GAAG;IAC9B,IAAI,EAAE,cAAc;IACpB,WAAW,EAAE,gCAAgC;IAC7C,WAAW,EAAE;QACX,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,oBAAoB,CAAC;QACrD,SAAS,EAAE,CAAC;aACT,MAAM,CAAC;YACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,wBAAwB,CAAC;YACnD,KAAK,EAAE,CAAC;iBACL,KAAK,CAAC;gBACL,CAAC,CAAC,MAAM,EAAE;gBACV,CAAC,CAAC,MAAM,EAAE;gBACV,CAAC,CAAC,OAAO,EAAE;gBACX,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC;gBACZ,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;gBAChB,CAAC,CAAC,IAAI,EAAE;aACT,CAAC;iBACD,QAAQ,CAAC,2BAA2B,CAAC;YACxC,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;YACrD,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,0CAA0C,CAAC;YACtE,SAAS,EAAE,CAAC;iBACT,MAAM,EAAE;iBACR,QAAQ,CAAC,2CAA2C,CAAC;YACxD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;YAChE,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,kCAAkC,CAAC;YAC7D,KAAK,EAAE,CAAC;iBACL,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,mCAAmC,CAAC;YAChD,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;YACzD,OAAO,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC,EAAE,8BAA8B;SACrG,CAAC;aACD,MAAM,EAAE;aACR,QAAQ,CAAC,kCAAkC,CAAC;KAChD;IACD,YAAY,EAAE;QACZ,MAAM,EAAE,CAAC;aACN,MAAM,CAAC;YACN,MAAM,EAAE,CAAC;iBACN,IAAI,CAAC,CAAC,SAAS,EAAE,UAAU,EAAE,SAAS,EAAE,OAAO,CAAC,CAAC;iBACjD,QAAQ,CAAC,kBAAkB,CAAC;YAC/B,KAAK,EAAE,CAAC;iBACL,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,mCAAmC,CAAC,EAAE,4BAA4B;YAC9E,UAAU,EAAE,CAAC;iBACV,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;iBACnB,QAAQ,CAAC,6BAA6B,CAAC,EAAE,gCAAgC;YAC5E,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,+BAA+B;YAC/F,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,4BAA4B,CAAC,EAAE,6BAA6B;SACrG,CAAC;aACD,QAAQ,CAAC,kBAAkB,CAAC;KAChC;CAC4B,CAAC;AAEhC,wBAAwB;AAExB,MAAM,CAAC,MAAM,aAAa,GAAG;IAC3B,IAAI,EAAE,WAAW;IACjB,WAAW,EAAE,iBAAiB;IAC9B,WAAW,EAAE,EAAE;IACf,YAAY,EAAE;QACZ,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,CAAC,oBAAoB,CAAC;KAC7D;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B,IAAI,EAAE,UAAU;IAChB,WAAW,EAAE,oBAAoB;IACjC,WAAW,EAAE;QACX,UAAU,EAAE,CAAC;aACV,MAAM,CAAC;YACN,MAAM,EAAE,CAAC;iBACN,MAAM,CAAC;gBACN,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;gBAC9C,IAAI,EAAE,CAAC;qBACJ,MAAM,EAAE;qBACR,QAAQ,EAAE;qBACV,QAAQ,EAAE;qBACV,QAAQ,CAAC,6BAA6B,CAAC;aAC3C,CAAC;iBACD,QAAQ,CAAC,yBAAyB,CAAC;YACtC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,mBAAmB,CAAC;YACvD,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,qBAAqB,CAAC;YAC3D,eAAe,EAAE,CAAC;iBACf,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,2BAA2B,CAAC;YACxC,KAAK,EAAE,CAAC;iBACL,MAAM,CAAC;gBACN,gBAAgB,EAAE,CAAC;qBAChB,OAAO,EAAE;qBACT,QAAQ,EAAE;qBACV,QAAQ,CAAC,oCAAoC,CAAC;gBACjD,WAAW,EAAE,CAAC;qBACX,MAAM,EAAE;qBACR,QAAQ,EAAE;qBACV,QAAQ,EAAE;qBACV,QAAQ,CAAC,cAAc,CAAC;aAC5B,CAAC;iBACD,QAAQ,EAAE;iBACV,QAAQ,CAAC,iCAAiC,CAAC;SAC/C,CAAC;aACD,QAAQ,CAAC,qBAAqB,CAAC;KACnC;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,wBAAwB,CAAC,EAAE,4CAA4C;KACrG;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,YAAY,GAAG;IAC1B,IAAI,EAAE,UAAU;IAChB,WAAW,EAAE,sBAAsB;IACnC,WAAW,EAAE;QACX,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,6BAA6B,CAAC;QAC3D,OAAO,EAAE,CAAC;aACP,MAAM,CAAC;YACN,SAAS,EAAE,CAAC;iBACT,MAAM,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;iBAClB,QAAQ,EAAE;iBACV,QAAQ,CAAC,0BAA0B,CAAC;YACvC,WAAW,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;YACvE,2CAA2C;SAC5C,CAAC;aACD,QAAQ,EAAE;aACV,QAAQ,CAAC,+BAA+B,CAAC;KAC7C;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,gBAAgB,CAAC,EAAE,4CAA4C;KAC7F;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,eAAe,GAAG;IAC7B,IAAI,EAAE,aAAa;IACnB,WAAW,EAAE,gBAAgB;IAC7B,WAAW,EAAE;QACX,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,2BAA2B,CAAC;KAC1D;IACD,YAAY,EAAE;QACZ,OAAO,EAAE,CAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,CAAC,qCAAqC,CAAC;KACrE;CAC4B,CAAC;AAEhC,MAAM,CAAC,MAAM,kBAAkB,GAAG;IAChC,IAAI,EAAE,gBAAgB;IACtB,WAAW,EAAE,2BAA2B;IACxC,WAAW,EAAE;QACX,GAAG,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,yBAAyB,CAAC;QACnD,OAAO,EAAE,CAAC;aACP,MAAM,CAAC;YACN,gBAAgB,EAAE,CAAC;iBAChB,OAAO,EAAE;iBACT,QAAQ,CAAC,oCAAoC,CAAC;YACjD,WAAW,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,CAAC,cAAc,CAAC;YACtE,UAAU,EAAE,CAAC;iBACV,MAAM,CAAC;gBACN,MAAM,EAAE,CAAC;qBACN,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;qBACjB,QAAQ,EAAE;qBACV,QAAQ,EAAE;qBACV,QAAQ,CAAC,eAAe,CAAC;gBAC5B,UAAU,EAAE,CAAC;qBACV,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;qBACjB,QAAQ,EAAE;qBACV,QAAQ,EAAE;qBACV,QAAQ,CAAC,oBAAoB,CAAC;gBACjC,YAAY,EAAE,CAAC;qBACZ,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;qBACjB,QAAQ,EAAE;qBACV,QAAQ,EAAE;qBACV,QAAQ,CAAC,sBAAsB,CAAC;gBACnC,KAAK,EAAE,CAAC;qBACL,KAAK,CAAC,CAAC,CAAC,MAAM,EAAE,CAAC;qBACjB,QAAQ,EAAE;qBACV,QAAQ,EAAE;qBACV,QAAQ,CAAC,cAAc,CAAC;aAC5B,CAAC;iBACD,QAAQ,EAAE;iBACV,QAAQ,CAAC,sBAAsB,CAAC;YACnC,YAAY,EAAE,CAAC;iBACZ,MAAM,EAAE;iBACR,QAAQ,EAAE;iBACV,QAAQ,CAAC,+BAA+B,CAAC;SAC7C,CAAC;aACD,QAAQ,CAAC,sBAAsB,CAAC;KACpC;IACD,YAAY,EAAE;QACZ,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,QAAQ,CAAC,iCAAiC,CAAC,EAAE,4CAA4C;KAC9G;CAC4B,CAAC;AAMhC,qCAAqC;AACrC,6DAA6D;AAC7D,MAAM,QAAQ,GAAG;IACf,eAAe;IACf,kBAAkB;IAClB,gBAAgB;IAChB,kBAAkB;IAClB,aAAa;IACb,gBAAgB;IAChB,aAAa;IACb,YAAY;IACZ,YAAY;IACZ,eAAe;IACf,kBAAkB;CACV,CAAC;AAKX,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAAC,OAA6B;IAC1E,MAAM,OAAO,CAAC,UAAU,EAAE,CAAC;IAE3B,MAAM,KAAK,GAAoB;QAC7B,WAAW,EAAE,gBAAgB,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC9D,MAAM,GAAG,GAAG,MAAM,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC;YACjD,OAAO,EAAE,QAAQ,EAAE,GAAG,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,cAAc,EAAE,gBAAgB,CAAC,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACpE,qDAAqD;YACrD,MAAM,WAAW,GAAG;gBAClB,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,GAAG,CAAC,MAAM,CAAC,UAAU,IAAI,EAAE,EAAE,EAAE,MAAM,CAAC,UAAU,EAAE,CAAC;aACpD,CAAC;YAEF,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,aAAa,CAAC,WAAW,CAAC,CAAC;YACxD,OAAO;gBACL,MAAM,EAAE;oBACN,GAAG,MAAM;oBACT,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO;iBAC7B;aACF,CAAC;QACJ,CAAC,CAAC;QAEF,YAAY,EAAE,gBAAgB,CAAC,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAChE,MAAM,WAAW,GAAG,MAAM,OAAO,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAChE,OAAO,EAAE,WAAW,EAAE,CAAC;QACzB,CAAC,CAAC;QAEF,cAAc,EAAE,gBAAgB,CAAC,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACpE,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;gBAChD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;QAEF,SAAS,EAAE,gBAAgB,CAAC,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC1D,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,MAAM,CAAC,CAAC;YACzE,OAAO;gBACL,MAAM,EAAE;oBACN,GAAG,MAAM;oBACT,KAAK,EAAE,MAAM,CAAC,KAAK,EAAE,OAAO;iBAC7B;aACF,CAAC;QACJ,CAAC,CAAC;QAEF,YAAY,EAAE,gBAAgB,CAAC,gBAAgB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAChE,MAAM,MAAM,GAAG,MAAM,OAAO,CAAC,YAAY,CACvC,MAAM,CAAC,UAAU,EACjB,MAAM,CAAC,SAAS,CACjB,CAAC;YACF,OAAO;gBACL,MAAM,EAAE;oBACN,GAAG,MAAM;oBACT,KAAK,EACH,OAAO,MAAM,CAAC,KAAK,KAAK,QAAQ;wBAC9B,CAAC,CAAC,MAAM,CAAC,KAAK;wBACd,CAAC,CAAC,MAAM,CAAC,KAAK,EAAE,OAAO;iBAC5B;aACF,CAAC;QACJ,CAAC,CAAC;QAEF,kCAAkC;QAClC,SAAS,EAAE,gBAAgB,CAAC,aAAa,EAAE,KAAK,IAAI,EAAE;YACpD,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,SAAS,EAAE,CAAC;YAC3C,OAAO,EAAE,QAAQ,EAAE,CAAC;QACtB,CAAC,CAAC;QAEF,QAAQ,EAAE,gBAAgB,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACxD,uDAAuD;YACvD,MAAM,EAAE,eAAe,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,CAAC,UAAU,CAAC;YAC7D,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,UAAU,EAAE,eAAe,CAAC,CAAC;YAClE,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC,CAAC;QAEF,QAAQ,EAAE,gBAAgB,CAAC,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACxD,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,OAAO,CAAC,CAAC;YACrE,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC,CAAC;QAEF,WAAW,EAAE,gBAAgB,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YAC9D,IAAI,CAAC;gBACH,MAAM,OAAO,CAAC,WAAW,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;gBAC1C,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;YAC3B,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC;QAEF,cAAc,EAAE,gBAAgB,CAAC,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,EAAE;YACpE,MAAM,EAAE,WAAW,EAAE,UAAU,EAAE,GAAG,WAAW,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC;YACnE,MAAM,KAAK,GAAG,MAAM,OAAO,CAAC,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE;gBACrD,GAAG,WAAW;gBACd,SAAS,EAAE,EAAE;gBACb,QAAQ,EAAE,EAAE;gBACZ,WAAW,EAAE,WAAW,IAAI,IAAI;gBAChC,GAAG,CAAC,UAAU,IAAI;oBAChB,UAAU,EAAE;wBACV,MAAM,EAAE,UAAU,CAAC,MAAM,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,MAAM;wBAChE,UAAU,EACR,UAAU,CAAC,UAAU,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,UAAU;wBAClE,YAAY,EACV,UAAU,CAAC,YAAY,KAAK,SAAS;4BACnC,CAAC,CAAC,EAAE;4BACJ,CAAC,CAAC,UAAU,CAAC,YAAY;wBAC7B,KAAK,EAAE,UAAU,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,KAAK;qBAC9D;iBACF,CAAC;aACH,CAAC,CAAC;YACH,OAAO,EAAE,KAAK,EAAE,CAAC;QACnB,CAAC,CAAC;KACH,CAAC;IAEF,MAAM,SAAS,GAAG,EAAE,CAAC;IAErB,MAAM,OAAO,GAAG,EAAE,CAAC;IAEnB,OAAO;QACL,KAAK;QACL,SAAS;QACT,OAAO;KACR,CAAC;AACJ,CAAC"}

package/dist/src/mcp/types.d.ts

@@ -0,0 +1,24 @@
+import type { ToolCallback } from "@modelcontextprotocol/sdk/server/mcp.js";
+import type { Prompt, Resource, ToolAnnotations } from "@modelcontextprotocol/sdk/types.js";
+import type { z, ZodRawShape } from "zod";
+export type ExtractCallbackFromTool<T extends ToolSchema> = T extends {
+    inputSchema: infer S;
+} ? ToolCallback<S extends ZodRawShape ? S : undefined> : ToolCallback;
+export type ToolWithCallback<T extends ToolSchema = ToolSchema> = T & {
+    callback: ExtractCallbackFromTool<T>;
+};
+export type ToolSchema<InputArgs extends ZodRawShape = ZodRawShape, OutputArgs extends ZodRawShape = ZodRawShape> = {
+    name: string;
+    title?: string;
+    description?: string;
+    inputSchema?: InputArgs;
+    outputSchema?: OutputArgs;
+    annotations?: ToolAnnotations;
+};
+export type ResolveZodSchema<T> = T extends z.ZodRawShape ? z.infer<z.ZodObject<T>> : T;
+export interface IMcpProvider<T extends ToolWithCallback = ToolWithCallback, R extends Resource = Resource, P extends Prompt = Prompt> {
+    tools: Record<T["name"], ToolWithCallback<T>>;
+    resources: Record<R["name"], R>;
+    prompts: Record<P["name"], P>;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/src/mcp/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/mcp/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,yCAAyC,CAAC;AAC5E,OAAO,KAAK,EACV,MAAM,EACN,QAAQ,EACR,eAAe,EAChB,MAAM,oCAAoC,CAAC;AAC5C,OAAO,KAAK,EAAE,CAAC,EAAE,WAAW,EAAE,MAAM,KAAK,CAAC;AAE1C,MAAM,MAAM,uBAAuB,CAAC,CAAC,SAAS,UAAU,IAAI,CAAC,SAAS;IACpE,WAAW,EAAE,MAAM,CAAC,CAAC;CACtB,GACG,YAAY,CAAC,CAAC,SAAS,WAAW,GAAG,CAAC,GAAG,SAAS,CAAC,GACnD,YAAY,CAAC;AAEjB,MAAM,MAAM,gBAAgB,CAAC,CAAC,SAAS,UAAU,GAAG,UAAU,IAAI,CAAC,GAAG;IACpE,QAAQ,EAAE,uBAAuB,CAAC,CAAC,CAAC,CAAC;CACtC,CAAC;AAEF,MAAM,MAAM,UAAU,CACpB,SAAS,SAAS,WAAW,GAAG,WAAW,EAC3C,UAAU,SAAS,WAAW,GAAG,WAAW,IAC1C;IACF,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,WAAW,CAAC,EAAE,MAAM,CAAC;IACrB,WAAW,CAAC,EAAE,SAAS,CAAC;IACxB,YAAY,CAAC,EAAE,UAAU,CAAC;IAC1B,WAAW,CAAC,EAAE,eAAe,CAAC;CAC/B,CAAC;AAEF,MAAM,MAAM,gBAAgB,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,WAAW,GACrD,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,GACvB,CAAC,CAAC;AAEN,MAAM,WAAW,YAAY,CAC3B,CAAC,SAAS,gBAAgB,GAAG,gBAAgB,EAC7C,CAAC,SAAS,QAAQ,GAAG,QAAQ,EAC7B,CAAC,SAAS,MAAM,GAAG,MAAM;IAEzB,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,gBAAgB,CAAC,CAAC,CAAC,CAAC,CAAC;IAC9C,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;IAChC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC;CAC/B"}

package/dist/src/mcp/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/src/mcp/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/mcp/types.ts"],"names":[],"mappings":""}

package/dist/src/mcp/utils.d.ts

@@ -0,0 +1,36 @@
+import { z } from "zod";
+import type { ResolveZodSchema, ToolSchema } from "./types.js";
+export declare class InvalidToolOutputError extends Error {
+    constructor(zodError: z.ZodError);
+}
+/**
+ * Creates a tool with a callback function that handles the tool execution,
+ * ensuring the output is valid and errors are handled and returned correctly.
+ * @param tool The tool schema to wrap.
+ * @param toolCallback The callback function to execute when the tool is called.
+ * @returns A CallToolResult with structuredContent if OutputSchema is defined or content if undefined. If the
+ * callback throws an error, the result will have isError set to true and the error message in the content.
+ */
+export declare function toolWithCallback<T extends ToolSchema>(tool: T, toolCallback: (args: ResolveZodSchema<T["inputSchema"]>) => ResolveZodSchema<T["outputSchema"]> | Promise<ResolveZodSchema<T["outputSchema"]>>): T & {
+    callback: (args: ResolveZodSchema<T["inputSchema"]>) => Promise<{
+        readonly content: {
+            type: "text";
+            text: string;
+            mimeType: string;
+        }[];
+        readonly structuredContent: {
+            [x: string]: any;
+        } | undefined;
+        isError?: undefined;
+    } | {
+        readonly isError: true;
+        readonly content: [{
+            readonly type: "text";
+            readonly text: `Error: ${string}`;
+        }];
+        readonly structuredContent: {
+            readonly error: string;
+        };
+    }>;
+};
+//# sourceMappingURL=utils.d.ts.map

package/dist/src/mcp/utils.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.d.ts","sourceRoot":"","sources":["../../../src/mcp/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,KAAK,EAAE,gBAAgB,EAAE,UAAU,EAAE,MAAM,YAAY,CAAC;AAE/D,qBAAa,sBAAuB,SAAQ,KAAK;gBACnC,QAAQ,EAAE,CAAC,CAAC,QAAQ;CAKjC;AAED;;;;;;;GAOG;AACH,wBAAgB,gBAAgB,CAAC,CAAC,SAAS,UAAU,EACnD,IAAI,EAAE,CAAC,EACP,YAAY,EAAE,CACZ,IAAI,EAAE,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,KAEtC,gBAAgB,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,GACnC,OAAO,CAAC,gBAAgB,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC;qBAKX,gBAAgB,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC;;;;;;;;;;;;;;;;;;;;EAyCxE"}

package/dist/src/mcp/utils.js

@@ -0,0 +1,62 @@
+import { z } from "zod";
+export class InvalidToolOutputError extends Error {
+    constructor(zodError) {
+        super("Invalid tool output\n" + zodError.message);
+        this.name = "InvalidToolOutputError";
+        this.cause = zodError;
+    }
+}
+/**
+ * Creates a tool with a callback function that handles the tool execution,
+ * ensuring the output is valid and errors are handled and returned correctly.
+ * @param tool The tool schema to wrap.
+ * @param toolCallback The callback function to execute when the tool is called.
+ * @returns A CallToolResult with structuredContent if OutputSchema is defined or content if undefined. If the
+ * callback throws an error, the result will have isError set to true and the error message in the content.
+ */
+export function toolWithCallback(tool, toolCallback) {
+    const outputSchema = tool.outputSchema
+        ? z.object(tool.outputSchema)
+        : undefined;
+    const wrappedCallback = async (args) => {
+        try {
+            const result = await toolCallback(args);
+            const validResult = outputSchema?.safeParse(result);
+            if (validResult && !validResult.success) {
+                throw new InvalidToolOutputError(validResult.error);
+            }
+            return {
+                content: outputSchema
+                    ? []
+                    : [
+                        {
+                            type: "text",
+                            text: JSON.stringify(result),
+                            mimeType: "application/json",
+                        },
+                    ],
+                structuredContent: outputSchema ? result : undefined,
+            };
+        }
+        catch (error) {
+            const errorString = error instanceof Error ? error.message : String(error);
+            return {
+                isError: true,
+                content: [
+                    {
+                        type: "text",
+                        text: `Error: ${errorString}`,
+                    },
+                ],
+                structuredContent: {
+                    error: errorString,
+                },
+            };
+        }
+    };
+    return {
+        ...tool,
+        callback: wrappedCallback,
+    };
+}
+//# sourceMappingURL=utils.js.map

package/dist/src/mcp/utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"utils.js","sourceRoot":"","sources":["../../../src/mcp/utils.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,MAAM,OAAO,sBAAuB,SAAQ,KAAK;IAC/C,YAAY,QAAoB;QAC9B,KAAK,CAAC,uBAAuB,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;QAClD,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;QACrC,IAAI,CAAC,KAAK,GAAG,QAAQ,CAAC;IACxB,CAAC;CACF;AAED;;;;;;;GAOG;AACH,MAAM,UAAU,gBAAgB,CAC9B,IAAO,EACP,YAIgD;IAEhD,MAAM,YAAY,GAAG,IAAI,CAAC,YAAY;QACpC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,YAAY,CAAC;QAC7B,CAAC,CAAC,SAAS,CAAC;IACd,MAAM,eAAe,GAAG,KAAK,EAAE,IAAwC,EAAE,EAAE;QACzE,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,YAAY,CAAC,IAAI,CAAC,CAAC;YAExC,MAAM,WAAW,GAAG,YAAY,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;YACpD,IAAI,WAAW,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;gBACxC,MAAM,IAAI,sBAAsB,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;YACtD,CAAC;YACD,OAAO;gBACL,OAAO,EAAE,YAAY;oBACnB,CAAC,CAAC,EAAE;oBACJ,CAAC,CAAC;wBACE;4BACE,IAAI,EAAE,MAAM;4BACZ,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;4BAC5B,QAAQ,EAAE,kBAAkB;yBAC7B;qBACF;gBACL,iBAAiB,EAAE,YAAY,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;aACnB,CAAC;QACtC,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,WAAW,GACf,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACzD,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,OAAO,EAAE;oBACP;wBACE,IAAI,EAAE,MAAM;wBACZ,IAAI,EAAE,UAAU,WAAW,EAAE;qBAC9B;iBACF;gBACD,iBAAiB,EAAE;oBACjB,KAAK,EAAE,WAAW;iBACnB;aACgC,CAAC;QACtC,CAAC;IACH,CAAC,CAAC;IACF,OAAO;QACL,GAAG,IAAI;QACP,QAAQ,EAAE,eAAe;KAC1B,CAAC;AACJ,CAAC"}