Connect OpenAI prompt and Databricks in our serverless environment

class OpenaiSourceGetPromptResponse {
    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 funnyPrompts = [
            "What do you think of humans?",
            "Do you prefer tabs or spaces?",
            "What's the meaning of life, the universe, and everything?",
            "What's your favorite programming joke?",
            "What's the worst code you've ever seen?"
        ];
        // More info at: https://platform.openai.com/docs/api-reference/completions/create
        const completion = await this.openai.createCompletion({
            model: "text-davinci-003",
            prompt: funnyPrompts,
            max_tokens: 7,
            temperature: 0,
        });

        for (const choice of completion.data.choices) {
            await publish(choice);
        }
        done();
    }

    async close() {}
}

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

    async consume(item) {
        // TODO: Customize your sql query
        // More info at: https://github.com/databricks/databricks-sql-nodejs/blob/main/tests/e2e/data_types.test.js#L151
        const operation = await this.session.executeStatement(
            `INSERT INTO yourTable VALUES(${item.id}, ${item.name})`
        );
        const result = await operation.fetchAll();
        await operation.close();
        return result;
    }

    async close() {
        await this.session.close();
        await this.databricks.close();
    }
}