67 lines
2 KiB
TypeScript
67 lines
2 KiB
TypeScript
|
import DiscordApi from "discord.js";
|
||
|
|
import { Configuration as OpenAIApiConfiguration, OpenAIApi } from "openai";
|
||
|
|
|
||
|
|
import config from "./config.json";
|
||
|
|
import toOpenAIMessages from "./toOpenAIMessages";
|
||
|
|
|
||
|
|
const discord = new DiscordApi.Client({
|
||
|
|
intents: [
|
||
|
|
DiscordApi.GatewayIntentBits.Guilds,
|
||
|
|
DiscordApi.GatewayIntentBits.GuildMessages,
|
||
|
|
DiscordApi.GatewayIntentBits.MessageContent,
|
||
|
|
]
|
||
|
|
});
|
||
|
|
|
||
|
|
const openai = new OpenAIApi(new OpenAIApiConfiguration({
|
||
|
|
apiKey: config.tokens.OpenAI
|
||
|
|
}));
|
||
|
|
|
||
|
|
discord.on("ready", async event => {
|
||
|
|
console.log(`Connected to Discord as ${event.user.tag} (${event.user.id})`);
|
||
|
|
});
|
||
|
|
|
||
|
|
discord.on("messageCreate", async message => {
|
||
|
|
if (message.author.bot) return;
|
||
|
|
if (!message.mentions.has(message.client.user)) return;
|
||
|
|
|
||
|
|
try {
|
||
|
|
let messages: DiscordApi.Collection<string, DiscordApi.Message> = await message.channel.messages.fetch({ limit: 50, cache: false });
|
||
|
|
|
||
|
|
messages = messages.filter(m => message.createdTimestamp - m.createdTimestamp < 1000*60*60 );
|
||
|
|
|
||
|
|
message.channel.sendTyping();
|
||
|
|
const answer = await openai.createChatCompletion({
|
||
|
|
model: "gpt-3.5-turbo",
|
||
|
|
messages: toOpenAIMessages(messages),
|
||
|
|
max_tokens: 168
|
||
|
|
});
|
||
|
|
|
||
|
|
message.reply({
|
||
|
|
content: answer.data.choices[0].message?.content,
|
||
|
|
allowedMentions: {
|
||
|
|
repliedUser: false,
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
const usage = answer.data.usage;
|
||
|
|
if (usage != undefined) {
|
||
|
|
const channelName: string = message.inGuild() ? `${message.channel.name} (${message.guild.name})` : `@${message.author.tag}`;
|
||
|
|
console.log(`Used ${usage.total_tokens} (${usage.prompt_tokens} + ${usage.completion_tokens}) tokens for ${message.author.tag} (${message.author.id}) in #${channelName}`);
|
||
|
|
}
|
||
|
|
} catch (e) {
|
||
|
|
console.error(e);
|
||
|
|
|
||
|
|
message.reply({
|
||
|
|
embeds: [{
|
||
|
|
color: 0xff0000,
|
||
|
|
description: "Something bad happened! :frown:"
|
||
|
|
}],
|
||
|
|
allowedMentions: {
|
||
|
|
repliedUser: false,
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
discord.login(config.tokens.Discord);
|