Connect DALL-E (OpenAI) images and Snowflake in our serverless environment

class OpenaiSourceGenerateImages {
    async init() {
        // TODO: Create your OpenAI credential
        // More info at https://yepcode.io/docs/integrations/openai/#credential-configuration
        this.openai = yepcode.integration.openai("your-openai-credential-name");
    }

    async fetch(publish, done) {
        // TODO: Configure your prompt
        const prompts = [
            "A cute baby sea otter",
            "A happy little cloud",
            "A grumpy potato",
            "A sleepy owl",
            "A mischievous kitten",
            "A nerdy giraffe",
            "A hipster penguin",
            "A jolly llama",
            "A clumsy elephant",
            "A sassy flamingo",
            "A quirky squirrel",
        ];

        for (const prompt of prompts) {
            // More info at: https://platform.openai.com/docs/api-reference/images
            const response = await this.openai.images.generate({
                prompt: prompt,
                n: 1,
                size: "1024x1024",
            });

            await publish(response.data);
        }
        done();
    }

    async close() {}
}

class SnowflakeTargetInsert {
    async init() {
        // TODO: Create your snowflake credential
        // More info at https://yepcode.io/docs/integrations/snowflake/#credential-configuration
        this.snowflake = yepcode.integration.snowflake(
            "your-snowflake-credential-name"
        );
        await this.snowflake.connect();
    }

    async consume(item) {
        await new Promise((resolve, reject) => {
            this.snowflake.execute({
                // TODO: Customize your SQL INSERT statement
                sqlText: "insert into YOUR_TABLE(ID, NAME) values(?, ?)",
                // TODO: Map item to any of the ? in insert sentence
                binds: [item.id, item.name],
                complete: (err, stmt, rows) => {
                    if (err) {
                        reject(err);
                        return;
                    }
                    resolve([stmt, rows]);
                },
            });
        });
    }

    async close() {
        await this.snowflake.destroy();
    }
}