-
Notifications
You must be signed in to change notification settings - Fork 0
/
index.js
40 lines (32 loc) · 862 Bytes
/
index.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
import { config } from 'dotenv';
import readline from 'readline';
import { Configuration, OpenAIApi } from 'openai';
config();
const openai = new OpenAIApi(new Configuration({
apiKey: process.env.GPT_SECRET_KEY,
}));
const ui = readline.createInterface({
input: process.stdin,
output: process.stdout,
});
const messages = [];
ui.prompt('> ');
ui.on('line', async (input) => {
messages.push({
role: 'user',
content: input,
});
const result = await openai.createChatCompletion({
model: 'gpt-3.5-turbo',
messages,
temperature: 0.7,
max_tokens: 1000,
});
const answer = result.data.choices[0].message.content;
console.info(result.data.choices[0].message.content + '\r\n');
messages.push({
role: 'assistant',
content: answer,
});
ui.prompt();
});