replicate-flux-mcp/build/tools/generateImage.js

MCP for Replicate Flux Model

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 registerGenerateImageTool = async (input) => {
    const { support_image_mcp_response_type, ...predictionInput } = input;
    try {
        const [output] = (await replicate.run(CONFIG.imageModelId, {
            input: predictionInput,
        }));
        const imageUrl = output.url();
        if (support_image_mcp_response_type) {
            const imageBase64 = await outputToBase64(output);
            return {
                content: [
                    {
                        type: "text",
                        text: `This is a generated image link: ${imageUrl}`,
                    },
                    {
                        type: "image",
                        data: imageBase64,
                        mimeType: "image/png",
                    },
                    {
                        type: "text",
                        text: `The image above is generated by the Flux model and prompt: ${input.prompt}`,
                    },
                ],
            };
        }
        return {
            content: [
                {
                    type: "text",
                    text: `This is a generated image link: ${imageUrl}`,
                },
                {
                    type: "text",
                    text: `The image above is generated by the Flux model and prompt: ${input.prompt}`,
                },
            ],
        };
    }
    catch (error) {
        handleError(error);
    }
};