1 files changed, 46 insertions, 0 deletions
diff --git a/bot/src/events/MessageCreate.js b/bot/src/events/MessageCreate.js
new file mode 100644
index 0000000..9394a20
--- /dev/null
+++ b/bot/src/events/MessageCreate.js
@@ -0,0 +1,46 @@
+import { Events } from 'discord.js';
+import { ollama } from '../utils/ollama.js';
+import { ollamaEnabled, ollamaModel } from '../storage/consts.js';
+
+export default {
+    name: Events.MessageCreate,
+    async execute(msg) {
+        if (!msg.client.application?.owner) await msg.client.application?.fetch();
+        if (msg.author.bot) return;
+        if (!msg.guild) return;
+
+        const args = msg.content.slice(`<@${msg.client.user.id}>`.length).trim();
+
+        if (msg.mentions.has(msg.client.user)) {
+            if (ollamaEnabled) {
+                if (!args) return msg.reply('Sorry? What was that?');
+
+                try {
+                    const response = await ollama.chat({
+                        model: ollamaModel,
+                        messages: [{ role: 'user', content: args }],
+                    });
+
+                    let content = response.message.content;
+                    content = content.replace(/<think>.*?<\/think>/g, '');
+
+                    if (content.length > 2000) {
+                        const chunks = content.match(/[\s\S]{1,2000}/g) || [];
+                        for (const chunk of chunks) {
+                            await msg.reply({ content: chunk });
+                        }
+                    } else {
+                        msg.reply({ content });
+                    }
+
+                } catch (err) {
+                    console.error(err);
+                    msg.reply('Something went wrong.');
+                }
+            } else {
+                return msg.reply('Sorry, this feature has been turned off.');
+            }
+
+        }
+    }
+};