Modularized LLM chatbot; Added back eval

2 files changed, 84 insertions, 2 deletions

diff --git a/bot/src/plugins/chatbot.js b/bot/src/plugins/chatbot.js
index f4095a2..3da7421 100644
--- a/bot/src/plugins/chatbot.js
+++ b/bot/src/plugins/chatbot.js
@@ -1 +1,34 @@
-// add llm feature here
+import { ollamaGlobal, ollamaModel } from '../storage/consts.js';
+import { ollama } from '../utils/ollama.js';
+import { AttachmentBuilder } from 'discord.js';
+import { guildSettings } from '../models/guild-settings.js';
+
+export async function ChatBot(msg, args) {
+    const guildSetting = await guildSettings.findOne({ where: { guildID: msg.guild.id } });
+
+    if (!guildSetting.ollamaEnabled) return;
+    if (!ollamaGlobal) return msg.reply('Sorry, the LLM chatbot feature has been turned off.');
+    if (!args) return msg.reply('Sorry? What was that?');
+
+    try {
+        const loadingMessage = await msg.reply('Thinking...');
+
+        const response = await ollama.chat({
+            model: ollamaModel,
+            messages: [{ role: 'user', content: args }],
+        });
+
+        let content = response.message.content;
+
+        if (content.length > 2000) {
+            const attachment = new AttachmentBuilder(Buffer.from(content, 'utf-8'), { name: 'output.txt' });
+            return await loadingMessage.edit({ files: [attachment] });
+        } else {
+            return await loadingMessage.edit({ content });
+        }
+
+    } catch (err) {
+        console.error(err);
+        await msg.reply(`Something went wrong. [Submit an issue at the AleeBot repository.](<https://github.com/Alee14/AleeBot/issues>)\nMessage:\n\`\`\`${err.stack}\`\`\``);
+    }
+}
diff --git a/bot/src/plugins/eval.js b/bot/src/plugins/eval.js
index 47b3988..c667dca 100644
--- a/bot/src/plugins/eval.js
+++ b/bot/src/plugins/eval.js
@@ -1,3 +1,52 @@
-export async function evaluation() {
+import { inspect } from 'util';
 
+export async function Evaluation(msg) {
+    if (!['242775871059001344'].includes(msg.author.id)) return await msg.reply('Nope! You need the person who created this bot to use this command.');
+    await msg.reply('You have entered evaluation mode. Enter the code for AleeBot to evaluate.\nType in `exit` to exit evaluation mode.');
+
+    let evaled;
+    let remove;
+
+    const filter = (i) => i.author.id === msg.author.id;
+
+    const collector = msg.channel.createMessageCollector({
+        filter,
+        time: 1000 * 600
+    });
+
+    collector.on('collect', async (msg) => {
+        if (msg.content.toLowerCase() === 'exit') {
+            return collector.stop();
+        }
+
+        try {
+            remove = (text) => {
+                if (typeof(text) === 'string') {
+                    return text.replace(/`/g, '`' + String.fromCharCode(8203)).replace(/@/g, '@' + String.fromCharCode(8203));
+                } else {
+                    return text;
+                }
+            };
+
+            evaled = eval(msg.content);
+
+            if (typeof evaled !== 'string') {
+                evaled = inspect(evaled);
+            }
+
+        } catch (err) {
+            return await msg.reply(`**Error:**\n\`\`\`\n${err.stack}\n\`\`\``);
+        }
+
+        try {
+            return await msg.reply(`**Output:**\n\`\`\`js\n${remove(evaled)}\n\`\`\``);
+        } catch (err) {
+            return await msg.reply(`**Error:**\n\`\`\`\n${err.stack}\n\`\`\``);
+        }
+
+    });
+
+    collector.on('end', async () => {
+        return await msg.reply('Exiting evaluation mode.');
+    });
 }