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);
|
|
|
|
|
|
|
|
|
2023-11-07 09:44:56 -06:00
|
|
|
export async function sendRequest(turns, systemMessage, stop_seq='***') {
|
2023-08-15 23:39:02 -07:00
|
|
|
|
2023-11-07 09:44:56 -06: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({
|
2023-11-08 19:24:24 -06:00
|
|
|
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);
|
2023-11-07 09:44:56 -06:00
|
|
|
res = 'My brain disconnected, try again.';
|
2023-08-15 23:39:02 -07:00
|
|
|
}
|
|
|
|
return res;
|
|
|
|
}
|