mcp-server-stability-ai
Version:
MCP [Model Context Protocol](https://modelcontextprotocol.io/) Server integrating MCP Clients with [Stability AI](https://stability.ai/) image manipulation functionalities: generate, edit, upscale, and more.
34 lines (33 loc) • 1.26 kB
JavaScript
import * as fs from "fs";
export const saveBase64ImageToFileToolDefinition = {
name: "stability-ai-save-base64-image-to-file",
description: "Save a base64 encoded image to a file. Useful if the user has provided a base64 encoded image, and we need to save it first before using another tool to process it.",
inputSchema: {
type: "object",
properties: {
imageAsBase64: {
type: "string",
description: "The base64 encoded image",
},
filename: {
type: "string",
description: "The filename to save the image as (something descriptive).",
},
},
required: ["imageAsBase64", "filename"],
},
};
export const saveBase64ImageToFile = async (args) => {
const { imageAsBase64, filename } = args;
const IMAGE_STORAGE_DIRECTORY = process.env.IMAGE_STORAGE_DIRECTORY;
fs.mkdirSync(IMAGE_STORAGE_DIRECTORY, { recursive: true });
fs.writeFileSync(`${IMAGE_STORAGE_DIRECTORY}/${filename}`, imageAsBase64, "base64");
return {
content: [
{
type: "text",
text: `Saved image to ${IMAGE_STORAGE_DIRECTORY}/${filename}`,
},
],
};
};