const { OpenAI } = require('openai');

const api = new OpenAI({
  baseURL: 'https://api.aimlapi.com/v1,
  apiKey: '<YOUR_API_KEY>',
});

const main = async () => {
  const result = await api.chat.completions.create({
    model: 'Salesforce/codegen2-7B',
    messages: [
      {
        role: 'system',
        content: 'You are SQL code assistant.',
      },
      {
        role: 'user',
        content: 'Could you please provide me with an example of a database structure that I could use for a project in MySQL?'
      }
    ],
  });

  const message = result.choices[0].message.content;
  console.log(\`Assistant: \${message}\`);
};

main();