atridad/himbot
1
0
Fork 0
Code Projects Releases Packages 1 Activity Actions

this is literally the most useless command why did I make it

Browse source
This commit is contained in:
Atridad Lahiji 2023-06-29 21:38:15 -06:00
parent 1dba6498f6
commit 28b77131cf
No known key found for this signature in database
1 changed files with 0 additions and 65 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

65
src/commands/prompthelper.ts
View file

@ -1,65 +0,0 @@
import { ApplyOptions } from '@sapphire/decorators';
import { Args, Command } from '@sapphire/framework';
import { Message, blockQuote, codeBlock } from 'discord.js';
import { Configuration, OpenAIApi } from 'openai';
const configuration = new Configuration({
apiKey: process.env.OPENAI_API_KEY
});
const openai = new OpenAIApi(configuration);
@ApplyOptions<Command.Options>({
description: 'AI will help you with AI!',
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('AI will help you with AI!').setRequired(true))
);
}
// Message command
public async messageRun(message: Message, args: Args) {
return this.promptHelper(message, args.getOption('prompt') || message.content.split('!wryna ')[1]);
}
// Chat Input (slash) command
public async chatInputRun(interaction: Command.ChatInputCommandInteraction) {
return this.promptHelper(interaction, interaction.options.getString('prompt') || 'NOTHING');
}
private async promptHelper(
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.createChatCompletion({
model: 'gpt-3.5-turbo',
messages: [
{
role: 'user',
content: `Can you optimize the following prompt to be used for an image generation model?: ${prompt}`
}
]
});
const content = blockQuote(`> ${prompt}\n${codeBlock(`${chatCompletion.data.choices[0].message?.content}`)}`);
if (interactionOrMessage instanceof Message) {
return askMessage.edit({ content: content.length <= 2000 ? content : 'Sorry... AI no work good...' });
}
return interactionOrMessage.editReply({
content: content.length <= 2000 ? content : 'Sorry... AI no work good...'
});
}
}