tardis-bot/modules/chatgpt/index.js

23 lines
625 B
JavaScript
Raw Normal View History

let ChatGPTAPI = null;
module.exports = function (bot, api) {
import("chatgpt").then((mod) => {
ChatGPTAPI = mod.ChatGPTAPI;
});
api.registerCommand('chat', async (args, message) => {
const response = await getChatGPTResponse(message.content.slice(6).trim());
api.respond(message, response);
});
}
async function getChatGPTResponse(prompt) {
const api = new ChatGPTAPI({
2023-11-12 23:07:15 +00:00
apiKey: process.env.OPENAI_API_KEY,
completionParams: {
model: 'gpt-4'
}
})
const res = await api.sendMessage(prompt);
return res.text;
}