@recraft-ai/mcp-recraft-server/dist/tools/GenerateImage.js

MCP Server implementation for recraft.ai API

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
    return new (P || (P = Promise))(function (resolve, reject) {
        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
        step((generator = generator.apply(thisArg, _arguments || [])).next());
    });
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.generateImageHandler = exports.generateImageTool = void 0;
const api_1 = require("../api");
const zod_1 = __importDefault(require("zod"));
const parameters_1 = require("../utils/parameters");
exports.generateImageTool = {
    name: "generate_image",
    description: "Generate an image using Recraft from a text prompt.\n" +
        "You can specify the image size, style, model, and number of images to generate.\n" +
        "You don't need to change default parameters if you don't have any specific requirements.\n" +
        "You can use styles to refine the image generation, and also to generate raster or vector images.\n" +
        "Local paths or URLs to generated images and their previews will be returned in the response.",
    inputSchema: {
        type: "object",
        properties: {
            prompt: parameters_1.PARAMETERS.promptSimple,
            size: parameters_1.PARAMETERS.imageSize,
            style: parameters_1.PARAMETERS.imageStyle,
            substyle: parameters_1.PARAMETERS.imageSubStyle,
            styleID: parameters_1.PARAMETERS.imageStyleID,
            model: parameters_1.PARAMETERS.transformModel,
            numberOfImages: parameters_1.PARAMETERS.numberOfImages,
        },
        required: ["prompt"]
    }
};
const generateImageHandler = (server, args) => __awaiter(void 0, void 0, void 0, function* () {
    try {
        const { prompt, size, style, styleID, substyle, model, numberOfImages } = zod_1.default.object({
            prompt: zod_1.default.string(),
            size: zod_1.default.nativeEnum(api_1.ImageSize).optional(),
            style: zod_1.default.nativeEnum(api_1.ImageStyle).optional(),
            substyle: zod_1.default.nativeEnum(api_1.ImageSubStyle).optional(),
            styleID: zod_1.default.string().optional(),
            model: zod_1.default.nativeEnum(api_1.TransformModel).optional(),
            numberOfImages: zod_1.default.number().min(1).max(6).optional()
        }).parse(args);
        const result = yield server.api.imageApi.generateImage({
            generateImageRequest: {
                prompt: prompt,
                size: size,
                style: styleID ? undefined : style,
                substyle: styleID ? undefined : substyle,
                styleId: styleID,
                model: model,
                responseFormat: 'url',
                n: numberOfImages,
                expire: server.isLocalResultsStorage,
            }
        });
        return yield server.transformGenerateImageResponseToCallToolResult(result);
    }
    catch (error) {
        return {
            content: [
                {
                    type: 'text',
                    text: `Error generating image: ${error}`
                }
            ],
            isError: true
        };
    }
});
exports.generateImageHandler = generateImageHandler;