rashi-discord-bot-lib/dist/ai/ImageGenerator.js

🚀 Powerful Discord bot framework with built-in database, event handling, and utilities

"use strict";
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.ImageGenerator = void 0;
const openai_1 = __importDefault(require("openai"));
class ImageGenerator {
    openai;
    constructor(apiKey) {
        this.openai = new openai_1.default({
            apiKey: apiKey || process.env.OPENAI_API_KEY,
        });
    }
    async generateImage(options) {
        const response = await this.openai.images.generate({
            prompt: options.prompt,
            n: options.n || 1,
            size: options.size || '512x512',
            response_format: options.response_format || 'url',
            user: options.user,
        });
        if (!response.data || !Array.isArray(response.data)) {
            throw new Error('OpenAI did not return image data.');
        }
        return response.data.map((item) => ({
            url: item.url ?? undefined,
            base64: item.b64_json ?? undefined,
        }));
    }
}
exports.ImageGenerator = ImageGenerator;
//# sourceMappingURL=ImageGenerator.js.map