Connect Google BigQuery and OpenAI prompt in our serverless environment

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

    async fetch(publish, done) {
        // TODO: Customize your query
        const query = `SELECT *
      FROM \`your_db.your_schema.your_table_name\`
      LIMIT 10`;

        // for all options, see https://cloud.google.com/bigquery/docs/reference/rest/v2/jobs/query
        const options = {
            query: query,
            // Location must match that of the dataset(s) referenced in the query.
            location: "US",
        };

        // run the query as a job
        const [job] = await this.googleBigQuery.createQueryJob(options);
        console.log(`Job ${job.id} started.`);

        // wait for the query to finish
        const [rows] = await job.getQueryResults();
        for (const row of rows) {
            await publish(row);
        }
        done();
    }

    async close() {}
}

class OpenaiTargetSendPrompt {
    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 consume(item) {
        // TODO: Configure your prompt
        // More info at: https://platform.openai.com/docs/api-reference/completions/create
        const completion = await this.openai.createCompletion({
            model: item.model,
            prompt: item.prompt,
            max_tokens: item.maxTokens,
            temperature: item.temperature,
        });

        // Use the completion for whatever you want
        console.log(completion.data.choices[0].text);
    }

    async close() {}
}