mirror of
https://github.com/Expand-sys/expandchatbotv2
synced 2026-03-22 12:27:11 +11:00
46 lines
No EOL
1.9 KiB
JavaScript
46 lines
No EOL
1.9 KiB
JavaScript
const { CommandInteraction, ApplicationCommandType, ApplicationCommandOptionType, EmbedBuilder } = require('discord.js');
|
|
const { sendResponse } = require('../../../utils/utils');
|
|
const path = require('path');
|
|
const { Ollama } = require('ollama')
|
|
const ollama = new Ollama({host: 'http://localhost:11434'})
|
|
|
|
module.exports = {
|
|
name: `prompt`,
|
|
description: `prompt the bot for for a snarky response`,
|
|
cooldown: 30,
|
|
type: ApplicationCommandType.ChatInput,
|
|
options: [{
|
|
name: `prompt`,
|
|
description: `The question you want to ask the ai`,
|
|
type: ApplicationCommandOptionType.String,
|
|
required: true,
|
|
}],
|
|
/**
|
|
* @param {CommandInteraction} interaction
|
|
*/
|
|
async execute(interaction) {
|
|
const { member, options } = interaction;
|
|
|
|
await interaction.deferReply({ ephemeral: false }).catch(err => console.error(`${path.basename(__filename)} There was a problem deferring an interaction: `, err));
|
|
console.log(options.get("prompt"))
|
|
let response = await ollama.chat({
|
|
model: 'llama2',
|
|
messages: [
|
|
{ role: "system", content: `You are a know it all bot who doesnt have a clue about social ettiquite please respond to users messages in a way that will make them hate you, the users name is '${member.user.displayName}' refer to them by their name`},
|
|
{ role: 'user', content: options.get("prompt").value},
|
|
|
|
],
|
|
stream: false,
|
|
options: {
|
|
num_predict: 128
|
|
}
|
|
})
|
|
|
|
const responseout = new EmbedBuilder()
|
|
.setColor('#32BEA6')
|
|
.setAuthor({ name: `${member.user.displayName }`, iconURL: member.user.displayAvatarURL({ dynamic: true }) })
|
|
.setTitle("Prompt: " + options.get("prompt").value)
|
|
.setDescription(response.message.content)
|
|
sendResponse(interaction, ``, [responseout]);
|
|
}
|
|
} |