86 lines
2.7 KiB
TypeScript
86 lines
2.7 KiB
TypeScript
import { ApplyOptions } from '@sapphire/decorators';
|
|
import { Args, Command } from '@sapphire/framework';
|
|
import { AttachmentBuilder, Message, blockQuote, codeBlock } from 'discord.js';
|
|
import OpenAI from 'openai';
|
|
|
|
const openai = new OpenAI({
|
|
apiKey: process.env.OPENAI_API_KEY
|
|
});
|
|
|
|
// @ts-ignore
|
|
// @ts-ignore
|
|
@ApplyOptions<Command.Options>({
|
|
description: 'You can ACTUALLY ask Himbot something! So cool!',
|
|
options: ['prompt']
|
|
})
|
|
export class UserCommand extends Command {
|
|
// Register Chat Input and Context Menu command
|
|
public override registerApplicationCommands(registry: Command.Registry) {
|
|
registry.registerChatInputCommand((builder) =>
|
|
builder
|
|
.setName(this.name)
|
|
.setDescription(this.description)
|
|
.addStringOption((option) =>
|
|
option.setName('prompt').setDescription('You can ACTUALLY ask Himbot something! So cool!').setRequired(true)
|
|
)
|
|
);
|
|
}
|
|
|
|
// Message command
|
|
public async messageRun(message: Message, args: Args) {
|
|
return this.ask(message, args.getOption('prompt') || message.content.split('!ask ')[1]);
|
|
}
|
|
|
|
// Chat Input (slash) command
|
|
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
|
|
return this.ask(interaction, interaction.options.getString('prompt') || 'NOTHING');
|
|
}
|
|
|
|
private async ask(interactionOrMessage: Message | Command.ChatInputCommandInteraction | Command.ContextMenuCommandInteraction, prompt: string) {
|
|
const askMessage =
|
|
interactionOrMessage instanceof Message
|
|
? await interactionOrMessage.channel.send({ content: '🤔 Thinking... 🤔' })
|
|
: await interactionOrMessage.reply({ content: '🤔 Thinking... 🤔', fetchReply: true });
|
|
|
|
const chatCompletion = await openai.chat.completions.create({
|
|
model: 'gpt-4',
|
|
messages: [
|
|
{
|
|
role: 'user',
|
|
content: prompt
|
|
}
|
|
]
|
|
});
|
|
|
|
const content = blockQuote(`> ${prompt}\n${codeBlock(`${chatCompletion.choices[0].message?.content}`)}`);
|
|
|
|
const messageAttachment: AttachmentBuilder[] = [];
|
|
|
|
if (content.length > 2000) {
|
|
messageAttachment.push(
|
|
new AttachmentBuilder(Buffer.from(`> ${prompt}\n${`${chatCompletion.choices[0].message?.content}`}`, 'utf-8'), {
|
|
name: 'response.txt',
|
|
description: "Himbot's Response"
|
|
})
|
|
);
|
|
}
|
|
|
|
if (interactionOrMessage instanceof Message) {
|
|
return askMessage.edit({
|
|
content:
|
|
content.length < 2000
|
|
? content
|
|
: `Discord only allows messages with 2000 characters or less. Please see your response in the attached file!`,
|
|
files: messageAttachment
|
|
});
|
|
}
|
|
|
|
return interactionOrMessage.editReply({
|
|
content:
|
|
content.length < 2000
|
|
? content
|
|
: `Discord only allows messages with 2000 characters or less. Please see your response in the attached file!`,
|
|
files: messageAttachment
|
|
});
|
|
}
|
|
}
|