@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
JavaScript
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