openai-node

1
#!/usr/bin/env -S npm run tsn -T
2

3
import OpenAI from 'openai';
4

5
// gets API Key from environment variable OPENAI_API_KEY
6
const openai = new OpenAI();
7

8
async function main() {
9
  // Non-streaming:
10
  const completion = await openai.chat.completions.create({
11
    model: 'gpt-4',
12
    messages: [{ role: 'user', content: 'Say this is a test' }],
13
  });
14
  console.log(completion.choices[0]?.message?.content);
15

16
  // Streaming:
17
  const stream = await openai.chat.completions.create({
18
    model: 'gpt-4',
19
    messages: [{ role: 'user', content: 'Say this is a test' }],
20
    stream: true,
21
  });
22
  for await (const part of stream) {
23
    process.stdout.write(part.choices[0]?.delta?.content || '');
24
  }
25
  process.stdout.write('\n');
26
}
27

28
main();
29