mirror of
https://github.com/kolbytn/mindcraft.git
synced 2025-04-21 21:52:07 +02:00
36 lines
879 B
JavaScript
36 lines
879 B
JavaScript
import OpenAIApi from 'openai';
|
|
|
|
|
|
let openAiConfig = null;
|
|
if (process.env.OPENAI_ORG_ID) {
|
|
openAiConfig = {
|
|
organization: process.env.OPENAI_ORG_ID,
|
|
apiKey: process.env.OPENAI_API_KEY,
|
|
};
|
|
} else {
|
|
openAiConfig = {
|
|
apiKey: process.env.OPENAI_API_KEY,
|
|
};
|
|
}
|
|
const openai = new OpenAIApi(openAiConfig);
|
|
|
|
|
|
export async function sendRequest(turns, systemMessage, stop_seq='***') {
|
|
|
|
let messages = [{'role': 'system', 'content': systemMessage}].concat(turns);
|
|
|
|
let res = null;
|
|
try {
|
|
let completion = await openai.chat.completions.create({
|
|
model: 'gpt-3.5-turbo',
|
|
messages: messages,
|
|
stop: stop_seq,
|
|
});
|
|
res = completion.choices[0].message.content;
|
|
}
|
|
catch (err) {
|
|
console.log(err);
|
|
res = 'My brain disconnected, try again.';
|
|
}
|
|
return res;
|
|
}
|