notion-mcp-server 1.0.1 → 2.4.2

package/build/tools/generateImageVariants.js

@@ -1,105 +0,0 @@
-import { replicate } from "../services/replicate.js";
-import { handleError } from "../utils/error.js";
-import { outputToBase64 } from "../utils/image.js";
-import { CONFIG } from "../config/index.js";
-export const registerGenerateImageVariantsTool = async (input) => {
-    const { prompt, num_variants, seed, support_image_mcp_response_type, prompt_variations, variation_mode, ...commonParams } = input;
-    try {
-        let effectiveVariants = num_variants;
-        let usingPromptVariations = false;
-        // Decide if we're using prompt variations
-        if (prompt_variations && prompt_variations.length > 0) {
-            usingPromptVariations = true;
-            // If using prompt variations, number of variants is limited by available variations
-            effectiveVariants = Math.min(num_variants, prompt_variations.length);
-        }
-        // Process all variants in parallel
-        const generationPromises = Array.from({ length: effectiveVariants }, (_, index) => {
-            // If seed is provided, create deterministic variants by adding the index
-            const variantSeed = seed !== undefined ? seed + index : undefined;
-            // Determine which prompt to use for this variant
-            let variantPrompt = prompt;
-            if (usingPromptVariations) {
-                const variation = prompt_variations[index];
-                if (variation_mode === "append") {
-                    variantPrompt = `${prompt} ${variation}`;
-                }
-                else {
-                    // 'replace' mode
-                    variantPrompt = variation;
-                }
-            }
-            return replicate
-                .run(CONFIG.imageModelId, {
-                input: {
-                    prompt: variantPrompt,
-                    seed: variantSeed,
-                    ...commonParams,
-                },
-            })
-                .then((outputs) => {
-                const [output] = outputs;
-                const imageUrl = output.url();
-                if (support_image_mcp_response_type) {
-                    return outputToBase64(output).then((imageBase64) => ({
-                        variantIndex: index + 1,
-                        imageUrl,
-                        imageBase64,
-                        usedPrompt: variantPrompt,
-                    }));
-                }
-                return {
-                    variantIndex: index + 1,
-                    imageUrl,
-                    usedPrompt: variantPrompt,
-                };
-            });
-        });
-        // Wait for all variant generation to complete
-        const results = (await Promise.all(generationPromises));
-        // Build response content
-        const responseContent = [];
-        // Add intro text - different based on whether we're using prompt variations
-        if (usingPromptVariations) {
-            responseContent.push({
-                type: "text",
-                text: `Generated ${results.length} variants of "${prompt}" using custom prompt variations (${variation_mode} mode)`,
-            });
-        }
-        else {
-            responseContent.push({
-                type: "text",
-                text: `Generated ${results.length} variants of: "${prompt}" using seed variations`,
-            });
-        }
-        // Add each variant with its index and prompt info
-        for (const result of results) {
-            // Build an appropriate description based on variant type
-            let variantDescription = `Variant #${result.variantIndex}`;
-            if (usingPromptVariations) {
-                variantDescription += `\nPrompt: "${result.usedPrompt}"`;
-            }
-            else if (seed !== undefined) {
-                variantDescription += ` (seed: ${seed + (result.variantIndex - 1)})`;
-            }
-            variantDescription += `\nImage URL: ${result.imageUrl}`;
-            responseContent.push({
-                type: "text",
-                text: `\n\n${variantDescription}`,
-            });
-            if (support_image_mcp_response_type && result.imageBase64) {
-                responseContent.push({
-                    type: "image",
-                    data: result.imageBase64,
-                    mimeType: `image/${input.output_format === "jpg" ? "jpeg" : input.output_format}`,
-                });
-            }
-        }
-        return {
-            content: responseContent,
-        };
-    }
-    catch (error) {
-        handleError(error);
-    }
-};

package/build/tools/generateMultipleImages.js

@@ -1,60 +0,0 @@
-import { replicate } from "../services/replicate.js";
-import { handleError } from "../utils/error.js";
-import { outputToBase64 } from "../utils/image.js";
-import { CONFIG } from "../config/index.js";
-export const registerGenerateMultipleImagesTool = async (input) => {
-    const { prompts, support_image_mcp_response_type, ...commonParams } = input;
-    try {
-        // Process all prompts in parallel
-        const generationPromises = prompts.map(async (prompt) => {
-            const [output] = (await replicate.run(CONFIG.imageModelId, {
-                input: {
-                    prompt,
-                    ...commonParams,
-                },
-            }));
-            const imageUrl = output.url();
-            if (support_image_mcp_response_type) {
-                const imageBase64 = await outputToBase64(output);
-                return {
-                    prompt,
-                    imageUrl,
-                    imageBase64,
-                };
-            }
-            return {
-                prompt,
-                imageUrl,
-            };
-        });
-        // Wait for all image generation to complete
-        const results = await Promise.all(generationPromises);
-        // Build response content
-        const responseContent = [];
-        // Add intro text
-        responseContent.push({
-            type: "text",
-            text: `Generated ${results.length} images based on your prompts:`,
-        });
-        // Add each image with its prompt
-        for (const result of results) {
-            responseContent.push({
-                type: "text",
-                text: `\n\nPrompt: "${result.prompt}"\nImage URL: ${result.imageUrl}`,
-            });
-            if (support_image_mcp_response_type && result.imageBase64) {
-                responseContent.push({
-                    type: "image",
-                    data: result.imageBase64,
-                    mimeType: `image/${input.output_format === "jpg" ? "jpeg" : input.output_format}`,
-                });
-            }
-        }
-        return {
-            content: responseContent,
-        };
-    }
-    catch (error) {
-        handleError(error);
-    }
-};

package/build/tools/generateSVG.js

@@ -1,43 +0,0 @@
-import { replicate } from "../services/replicate.js";
-import { handleError } from "../utils/error.js";
-import { urlToSvg } from "../utils/image.js";
-import { CONFIG } from "../config/index.js";
-export const registerGenerateSvgTool = async (input) => {
-    try {
-        const output = (await replicate.run(CONFIG.svgModelId, {
-            input,
-        }));
-        const svgUrl = output.url();
-        if (!svgUrl) {
-            throw new Error("Failed to generate SVG URL");
-        }
-        try {
-            const svg = await urlToSvg(svgUrl);
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: `This is a generated SVG url: ${svgUrl}`,
-                    },
-                    {
-                        type: "text",
-                        text: svg,
-                    },
-                ],
-            };
-        }
-        catch (error) {
-            return {
-                content: [
-                    {
-                        type: "text",
-                        text: `This is a generated SVG url: ${svgUrl}`,
-                    },
-                ],
-            };
-        }
-    }
-    catch (error) {
-        return handleError(error);
-    }
-};

package/build/tools/getPrediction.js

@@ -1,22 +0,0 @@
-// import { replicate } from "../services/replicate.js";
-// import { handleError } from "../utils/error.js";
-// import { GetPredictionParams } from "../types/index.js";
-// import { CallToolResult } from "@modelcontextprotocol/sdk/types.js";
-export {};
-// export const registerGetPredictionTool = async ({
-//   predictionId,
-// }: GetPredictionParams): Promise<CallToolResult> => {
-//   try {
-//     const prediction = await replicate.predictions.get(predictionId);
-//     return {
-//       content: [
-//         {
-//           type: "text",
-//           text: JSON.stringify(prediction, null, 2),
-//         },
-//       ],
-//     };
-//   } catch (error) {
-//     handleError(error);
-//   }
-// };

package/build/tools/pages.js

@@ -1,22 +0,0 @@
-import { handleNotionError } from "../utils/error.js";
-import { registerCreatePageTool } from "./createPage.js";
-import { archivePage } from "./updatePage.js";
-import { restorePage } from "./updatePage.js";
-import { searchPages } from "./searchPage.js";
-import { updatePageProperties } from "./updatePageProperties.js";
-export const registerPagesOperationTool = async (params) => {
-    switch (params.payload.action) {
-        case "create_page":
-            return registerCreatePageTool(params.payload.params);
-        case "archive_page":
-            return archivePage(params.payload.params);
-        case "restore_page":
-            return restorePage(params.payload.params);
-        case "search_pages":
-            return searchPages(params.payload.params);
-        case "update_page_properties":
-            return updatePageProperties(params.payload.params);
-        default:
-            return handleNotionError(new Error(`Unsupported action, use one of the following: "create_page", "archive_page", "restore_page", "search_pages", "update_page_properties"`));
-    }
-};

package/build/tools/predictionList.js

@@ -1,30 +0,0 @@
-import { replicate } from "../services/replicate.js";
-import { handleError } from "../utils/error.js";
-export const registerPredictionListTool = async ({ limit, }) => {
-    try {
-        const predictions = [];
-        for await (const page of replicate.paginate(replicate.predictions.list)) {
-            predictions.push(...page);
-            if (predictions.length >= limit) {
-                break;
-            }
-        }
-        const limitedPredictions = predictions.slice(0, limit);
-        const totalPages = Math.ceil(predictions.length / limit);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Found ${limitedPredictions.length} predictions (showing ${limitedPredictions.length} of ${predictions.length} total, page 1 of ${totalPages})`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(limitedPredictions, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        handleError(error);
-    }
-};

package/build/tools/queryDatabase.js

@@ -1,22 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const queryDatabase = async (params) => {
-    try {
-        const response = await notion.databases.query(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Database queried successfully. Found ${response.results.length} results.`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/retrieveBlock.js

@@ -1,24 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const retrieveBlock = async (params) => {
-    try {
-        const response = await notion.blocks.retrieve({
-            block_id: params.blockId,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: "Block retrieved successfully! Note: If this block has children, use the retrieve_block_children endpoint to get the list of child blocks.",
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/retrieveBlockChildren.js

@@ -1,32 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const retrieveBlockChildren = async (params) => {
-    try {
-        const response = await notion.blocks.children.list({
-            block_id: params.blockId,
-            start_cursor: params.start_cursor,
-            page_size: params.page_size,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Successfully retrieved ${response.results.length} children of block ${params.blockId}`,
-                },
-                {
-                    type: "text",
-                    text: `Has more: ${response.has_more ? "Yes" : "No"}${response.has_more && response.next_cursor
-                        ? `, Next cursor: ${response.next_cursor}`
-                        : ""}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response.results, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/searchPage.js

@@ -1,24 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export async function searchPages(params) {
-    try {
-        const response = await notion.search({
-            query: params.query || "",
-            sort: params.sort,
-            start_cursor: params.start_cursor,
-            page_size: params.page_size || 10,
-        });
-        const resultsText = JSON.stringify(response, null, 2);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Found ${response.results.length} results. ${response.has_more ? "More results available." : ""}\n\n${resultsText}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-}

package/build/tools/updateBlock.js

@@ -1,25 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const updateBlock = async (params) => {
-    try {
-        const response = await notion.blocks.update({
-            block_id: params.blockId,
-            ...params.data,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Block ${response.id} updated successfully`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/updateDatabase.js

@@ -1,18 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const updateDatabase = async (params) => {
-    try {
-        const response = await notion.databases.update(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Database updated successfully: ${response.id}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};

package/build/tools/updatePage.js

@@ -1,40 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export async function archivePage(params) {
-    try {
-        const response = await notion.pages.update({
-            page_id: params.pageId,
-            archived: true,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Page archived successfully: ${response.id}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-}
-export async function restorePage(params) {
-    try {
-        const response = await notion.pages.update({
-            page_id: params.pageId,
-            archived: false,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Page restored successfully: ${response.id}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-}

package/build/tools/updatePageProperties.js

@@ -1,21 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export async function updatePageProperties(params) {
-    try {
-        const response = await notion.pages.update({
-            page_id: params.pageId,
-            properties: params.properties,
-        });
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Page properties updated successfully: ${response.id}`,
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-}

package/build/tools/users.js

@@ -1,75 +0,0 @@
-import { notion } from "../services/notion.js";
-import { handleNotionError } from "../utils/error.js";
-export const registerGetListUsersTool = async (params) => {
-    try {
-        const response = await notion.users.list(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Users retrieved successfully: ${response.results.length}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};
-export const registerGetUserTool = async (params) => {
-    try {
-        const response = await notion.users.retrieve(params);
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `User retrieved successfully: ${response.id}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};
-export const registerGetBotUserTool = async () => {
-    try {
-        const response = await notion.users.me({});
-        return {
-            content: [
-                {
-                    type: "text",
-                    text: `Bot user retrieved successfully: ${response.id}`,
-                },
-                {
-                    type: "text",
-                    text: JSON.stringify(response, null, 2),
-                },
-            ],
-        };
-    }
-    catch (error) {
-        return handleNotionError(error);
-    }
-};
-// Combined tool function that handles all user operations
-export const registerUsersOperationTool = async (params) => {
-    switch (params.payload.action) {
-        case "list_users":
-            return registerGetListUsersTool(params.payload.params);
-        case "get_user":
-            return registerGetUserTool(params.payload.params);
-        case "get_bot_user":
-            return registerGetBotUserTool();
-        default:
-            return handleNotionError(new Error(`Unsupported action, use one of the following: "list_users", "get_user", "get_bot_user"`));
-    }
-};

package/build/types/blocks.js

@@ -1,12 +0,0 @@
-import { z } from "zod";
-import { APPEND_BLOCK_CHILDREN_SCHEMA, RETRIEVE_BLOCK_SCHEMA, RETRIEVE_BLOCK_CHILDREN_SCHEMA, UPDATE_BLOCK_SCHEMA, DELETE_BLOCK_SCHEMA, BATCH_APPEND_BLOCK_CHILDREN_SCHEMA, BATCH_UPDATE_BLOCKS_SCHEMA, BATCH_DELETE_BLOCKS_SCHEMA, BATCH_MIXED_OPERATIONS_SCHEMA, BLOCKS_OPERATION_SCHEMA, } from "../schema/blocks.js";
-export const appendBlockChildrenSchema = z.object(APPEND_BLOCK_CHILDREN_SCHEMA);
-export const retrieveBlockSchema = z.object(RETRIEVE_BLOCK_SCHEMA);
-export const retrieveBlockChildrenSchema = z.object(RETRIEVE_BLOCK_CHILDREN_SCHEMA);
-export const updateBlockSchema = z.object(UPDATE_BLOCK_SCHEMA);
-export const deleteBlockSchema = z.object(DELETE_BLOCK_SCHEMA);
-export const batchAppendBlockChildrenSchema = z.object(BATCH_APPEND_BLOCK_CHILDREN_SCHEMA);
-export const batchUpdateBlocksSchema = z.object(BATCH_UPDATE_BLOCKS_SCHEMA);
-export const batchDeleteBlocksSchema = z.object(BATCH_DELETE_BLOCKS_SCHEMA);
-export const batchMixedOperationsSchema = z.object(BATCH_MIXED_OPERATIONS_SCHEMA);
-export const blocksOperationSchema = z.object(BLOCKS_OPERATION_SCHEMA);

package/build/types/comments.js

@@ -1,7 +0,0 @@
-import { z } from "zod";
-import { GET_COMMENTS_SCHEMA, ADD_PAGE_COMMENT_SCHEMA, ADD_DISCUSSION_COMMENT_SCHEMA, COMMENTS_OPERATION_SCHEMA, } from "../schema/comments.js";
-// Types inferred from the schemas
-export const getCommentsSchema = z.object(GET_COMMENTS_SCHEMA);
-export const addPageCommentSchema = z.object(ADD_PAGE_COMMENT_SCHEMA);
-export const addDiscussionCommentSchema = z.object(ADD_DISCUSSION_COMMENT_SCHEMA);
-export const commentsOperationSchema = z.object(COMMENTS_OPERATION_SCHEMA);

package/build/types/database.js

@@ -1,6 +0,0 @@
-import { z } from "zod";
-import { CREATE_DATABASE_SCHEMA, QUERY_DATABASE_SCHEMA, UPDATE_DATABASE_SCHEMA, DATABASE_OPERATION_SCHEMA, } from "../schema/database.js";
-export const createDatabaseSchema = z.object(CREATE_DATABASE_SCHEMA);
-export const queryDatabaseSchema = z.object(QUERY_DATABASE_SCHEMA);
-export const updateDatabaseSchema = z.object(UPDATE_DATABASE_SCHEMA);
-export const databaseOperationSchema = z.object(DATABASE_OPERATION_SCHEMA);

package/build/types/notion.js

@@ -1 +0,0 @@
-export {};

package/build/types/page.js

@@ -1,8 +0,0 @@
-import { z } from "zod";
-import { ARCHIVE_PAGE_SCHEMA, CREATE_PAGE_SCHEMA, RESTORE_PAGE_SCHEMA, SEARCH_PAGES_SCHEMA, UPDATE_PAGE_PROPERTIES_SCHEMA, PAGES_OPERATION_SCHEMA, } from "../schema/page.js";
-export const createPageSchema = z.object(CREATE_PAGE_SCHEMA);
-export const archivePageSchema = z.object(ARCHIVE_PAGE_SCHEMA);
-export const restorePageSchema = z.object(RESTORE_PAGE_SCHEMA);
-export const searchPagesSchema = z.object(SEARCH_PAGES_SCHEMA);
-export const updatePagePropertiesSchema = z.object(UPDATE_PAGE_PROPERTIES_SCHEMA);
-export const pagesOperationSchema = z.object(PAGES_OPERATION_SCHEMA);

package/build/types/richText.js

@@ -1 +0,0 @@
-export {};

package/build/types/tools.js

@@ -1 +0,0 @@
-export {};

package/build/types/users.js

@@ -1,6 +0,0 @@
-import { z } from "zod";
-import { LIST_USERS_SCHEMA, GET_USER_SCHEMA, USERS_OPERATION_SCHEMA, } from "../schema/users.js";
-// Types inferred from schemas
-const listUsersSchema = z.object(LIST_USERS_SCHEMA);
-const getUserSchema = z.object(GET_USER_SCHEMA);
-const usersOperationSchema = z.object(USERS_OPERATION_SCHEMA);

package/build/utils/blob.js

@@ -1,5 +0,0 @@
-export async function outputToBase64(output) {
-    const blob = await output.blob();
-    const buffer = Buffer.from(await blob.arrayBuffer());
-    return buffer.toString("base64");
-}