Volcengine Image Generation MCP Server

import { Server } from '@modelcontextprotocol/sdk/server/index.js'; import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'; import { CallToolRequestSchema, ListToolsRequestSchema, } from '@modelcontextprotocol/sdk/types.js'; import { z } from 'zod'; import { ImageGenerationTools } from './tools'; class VolcengineMCPServer { private server: Server; private imageTools: ImageGenerationTools; constructor() { const apiKey = process.env.VOLCENGINE_API_KEY; if (!apiKey) { throw new Error('VOLCENGINE_API_KEY environment variable is required'); } this.server = new Server( { name: 'volcengine-image-generation', version: '1.0.0', }, { capabilities: { tools: {}, }, } ); this.imageTools = new ImageGenerationTools(apiKey); this.setupTools(); } private setupTools() { this.server.setRequestHandler(ListToolsRequestSchema, async () => { return { tools: [this.imageTools.getToolDefinition()], }; }); this.server.setRequestHandler(CallToolRequestSchema, async (request: any) => { const { name, arguments: args } = request.params; if (name === 'generate_image') { try { const result = await this.imageTools.generateImage(args); if (result.success) { return { content: [ { type: 'text', text: JSON.stringify(result.data, null, 2), }, ], }; } else { return { content: [ { type: 'text', text: `Error: ${result.error}`, }, ], isError: true, }; } } catch (error) { return { content: [ { type: 'text', text: `Error: ${error instanceof Error ? error.message : 'Unknown error'}`, }, ], isError: true, }; } } throw new Error(`Unknown tool: ${name}`); }); } async run() { const transport = new StdioServerTransport(); await this.server.connect(transport); console.error('Volcengine MCP server started'); } } async function main() { try { const server = new VolcengineMCPServer(); await server.run(); } catch (error) { console.error('Failed to start server:', error); process.exit(1); } } main().catch((error) => { console.error('Unhandled error:', error); process.exit(1); });