mindcraft/utils/gpt.js

37 lines
879 B
JavaScript
Raw Normal View History

2023-09-04 17:01:04 -05:00
import OpenAIApi from 'openai';
2023-08-15 23:39:02 -07:00
2023-09-04 17:01:04 -05:00
let openAiConfig = null;
2023-08-15 23:39:02 -07:00
if (process.env.OPENAI_ORG_ID) {
2023-09-04 17:01:04 -05:00
openAiConfig = {
2023-08-15 23:39:02 -07:00
organization: process.env.OPENAI_ORG_ID,
apiKey: process.env.OPENAI_API_KEY,
2023-09-04 17:01:04 -05:00
};
2023-08-15 23:39:02 -07:00
} else {
2023-09-04 17:01:04 -05:00
openAiConfig = {
2023-08-15 23:39:02 -07:00
apiKey: process.env.OPENAI_API_KEY,
2023-09-04 17:01:04 -05:00
};
2023-08-15 23:39:02 -07:00
}
const openai = new OpenAIApi(openAiConfig);
export async function sendRequest(turns, systemMessage, stop_seq='***') {
2023-08-15 23:39:02 -07:00
let messages = [{'role': 'system', 'content': systemMessage}].concat(turns);
2023-08-15 23:39:02 -07:00
let res = null;
try {
2023-09-04 17:01:04 -05:00
let completion = await openai.chat.completions.create({
model: 'gpt-3.5-turbo',
2023-08-15 23:39:02 -07:00
messages: messages,
2023-08-17 00:00:57 -07:00
stop: stop_seq,
2023-08-15 23:39:02 -07:00
});
2023-09-04 17:01:04 -05:00
res = completion.choices[0].message.content;
2023-08-15 23:39:02 -07:00
}
catch (err) {
console.log(err);
res = 'My brain disconnected, try again.';
2023-08-15 23:39:02 -07:00
}
return res;
}