UNPKG

@favdevs/payload-ai

Version:

<p align="center"> <img alt="Payload AI Plugin" src="assets/payload-ai-intro.gif" width="100%" /> </p>

27 lines (26 loc) 907 B
import { GoogleGenAI } from '@google/genai'; export const generateImage = async (prompt, { aspectRatio = '1:1', model = 'imagen-4.0-fast-generate-001', outputMimeType = 'image/png' } = {})=>{ const ai = new GoogleGenAI({ apiKey: process.env.GOOGLE_GENERATIVE_AI_API_KEY }); const response = await ai.models.generateImages({ model, prompt, config: { numberOfImages: 1, aspectRatio, outputMimeType } }); const generatedImage = response.generatedImages?.[0]; if (!generatedImage?.image?.imageBytes) { throw new Error('No image generated'); } const base64ImageBytes = generatedImage.image.imageBytes; const buffer = Buffer.from(base64ImageBytes, 'base64'); return { alt: generatedImage.enhancedPrompt || prompt, buffer }; }; //# sourceMappingURL=generateImage.js.map