mcp-image-generator

by GMKR
Verified
import { Together } from "together-ai"; export const useTogether = () => { const together = new Together({ apiKey: process.env.TOGETHER_API_KEY!, }); async function generateImage(prompt: string, params?: Record<string, string | number>) { const model = process.env.MODEL_NAME || "black-forest-labs/FLUX.1-schnell-Free" const response = await together.images.create({ ...params, prompt, model, response_format: "base64", }); return response.data.map((pc) => { return { type: "image", data: pc.b64_json!, mimeType: 'image/jpeg', } }).filter((pc) => pc.data !== null && pc.data !== undefined) } return { generateImage, } }