-
Notifications
You must be signed in to change notification settings - Fork 0
/
openai.js
100 lines (74 loc) · 2.45 KB
/
openai.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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
import { confirm, intro, isCancel, outro, text } from '@clack/prompts';
import { Configuration, OpenAIApi } from 'openai';
import pc from 'picocolors';
import {
conf,
historyLength,
openAIMaxTokens,
openAIModel,
openAITemperature,
options,
} from './settings.js';
import { addUsage } from './usage.js';
let openai;
export let apiKey;
let conversation = [{ role: 'system', content: 'You are a helpful assistant.' }];
function initOpenAI() {
const configuration = new Configuration({ apiKey });
openai = new OpenAIApi(configuration);
}
async function inputApiKey() {
intro('No OpenAI API key found. Create one at https://platform.openai.com/account/api-keys');
const apiKey = await text({
message: 'Enter your API Key',
});
if (isCancel(apiKey) || !apiKey || apiKey.trim().length === 0)
throw new Error('No API key provided.');
const saveApiKey = await confirm({
message: 'Save API key for future use?',
});
if (saveApiKey) conf.set('apiKey', apiKey);
outro('API key saved.');
return apiKey;
}
export async function apiKeyCheck() {
apiKey =
process.env.OPENAI_API_KEY ||
options?.['api-key'] ||
conf.get('apiKey') ||
(await inputApiKey());
if (options?.['api-key']) conf.set('apiKey', apiKey);
initOpenAI(apiKey);
}
function updateConversation({ role, content }) {
conversation.push({ role, content });
if (role === 'assistant')
conversation =
historyLength <= 0
? [conversation[0]]
: [conversation[0], ...conversation.slice(1).slice(-(2 * historyLength))];
if (process.env.DEBUG) console.log({ conversation });
}
export async function askChatGPT(question) {
let response;
updateConversation({ role: 'user', content: question });
try {
response = await openai.createChatCompletion({
model: openAIModel,
messages: conversation,
temperature: openAITemperature,
max_tokens: openAIMaxTokens,
});
if (process.env.DEBUG) console.log(JSON.stringify(response.data, null, 2));
addUsage(response?.data?.usage?.total_tokens);
const answer = response?.data?.choices[0]?.message?.content;
if (answer) updateConversation({ role: 'assistant', content: answer });
return response;
} catch (error) {
if (error?.response?.status === 401) {
conf.delete('apiKey');
console.log('\nInvalid API key. Please restart and enter a new one.');
process.exit(1);
} else console.error(`${pc.red('Error')}: ${error.message}`);
}
}