biggest loser, flashback features, openai compatibility

2026-02-26 23:01:59 -08:00
parent 1bb5881bf3
commit 1106245fd5
11 changed files with 264 additions and 47 deletions
--- a/discord/.env.example
+++ b/discord/.env.example
@@ -4,6 +4,7 @@ CLIENT="123456789012345678"
 GUILD="123456789012345678"
 ADMIN="123456789012345678"

+HF_TOKEN=""
 LLM_HOST="http://127.0.0.1:8000"
 LLM_TOKEN="dfsl;kjsdl;kfja"
 LMSTUDIO_HOST="ws://localhost:1234"
@@ -16,3 +17,6 @@ MOTD_QUERY="#tips"

 ENABLE_THROWBACK=1
 THROWBACK_CHANNEL="123456789012345678"
+
+ENABLE_LOSER=1
+LOSER_CHANNEL="123456789012345678"
--- a/discord/bot.ts
+++ b/discord/bot.ts
@@ -33,7 +33,9 @@ import {
    reactionEmojis,
    recordReaction,
    requestTTSResponse,
-    sync
+    serializeMessageHistory,
+    sync,
+    REAL_NAMES
 } from './util';
 import 'dotenv/config';
 import { LLMConfig } from './commands/types';
@@ -328,6 +330,151 @@ async function scheduleThrowback(firstTime = false) {
    setTimeout(scheduleThrowback, timeoutHours * 60 * 60 * 1000);
 }

+async function scheduleBiggestLoser(firstTime = false) {
+    if (!firstTime) {
+        if (!process.env.LOSER_CHANNEL) {
+            logWarn('[bot] LOSER_CHANNEL not configured, disabling biggest loser announcement.');
+            return;
+        }
+
+        const channel = <TextChannel>await client.channels.fetch(process.env.LOSER_CHANNEL);
+        if (channel) {
+            try {
+                const yesterdayStart = new Date();
+                yesterdayStart.setDate(yesterdayStart.getDate() - 1);
+                yesterdayStart.setHours(0, 0, 0, 0);
+
+                const yesterdayEnd = new Date();
+                yesterdayEnd.setHours(0, 0, 0, 0);
+
+                const startId = dateToSnowflake(yesterdayStart);
+                const endId = dateToSnowflake(yesterdayEnd);
+
+                const deadNames = ['Adam Kazerounian', 'Jake Wong', 'David Zheng', 'Hatsune Miku'];
+                const realNameToCount = new Map<string, number>();
+                for (const realName of new Set(Object.values(REAL_NAMES))) {
+                    if (!deadNames.includes(realName as string)) {
+                        realNameToCount.set(realName as string, 0);
+                    }
+                }
+
+                const guild = await client.guilds.fetch(process.env.GUILD as string);
+                if (guild) {
+                    const channels = await guild.channels.fetch();
+                    const textChannels = channels.filter((c: any) => c && c.isTextBased());
+                    for (const [_, textChannel] of textChannels) {
+                        let lastId = startId;
+                        while (true) {
+                            try {
+                                const messages = await (textChannel as any).messages.fetch({ after: lastId, limit: 100 });
+                                if (messages.size === 0) break;
+                                
+                                let maxId = lastId;
+                                for (const [msgId, msg] of messages) {
+                                    if (BigInt(msgId) > BigInt(maxId)) maxId = msgId;
+                                    if (BigInt(msgId) >= BigInt(endId)) continue;
+                                    if (!msg.author.bot && (REAL_NAMES as any)[msg.author.username]) {
+                                        const realName = (REAL_NAMES as any)[msg.author.username];
+                                        if (realNameToCount.has(realName)) {
+                                            realNameToCount.set(realName, realNameToCount.get(realName)! + 1);
+                                        }
+                                    }
+                                }
+                                
+                                lastId = maxId;
+                                if (BigInt(lastId) >= BigInt(endId) || messages.size < 100) break;
+                            } catch (e) {
+                                logWarn(`[bot] Error fetching from channel: ${e}`);
+                                break;
+                            }
+                        }
+                    }
+                }
+
+                let minCount = Infinity;
+                let biggestLosers: string[] = [];
+                for (const [realName, count] of realNameToCount.entries()) {
+                    if (count < minCount) {
+                        minCount = count;
+                        biggestLosers = [realName];
+                    } else if (count === minCount) {
+                        biggestLosers.push(realName);
+                    }
+                }
+
+                if (biggestLosers.length > 0) {
+                    biggestLosers.sort();
+                    let streakCount = 1;
+                    const streakFile = path.join(__dirname, 'biggest_loser_streak.json');
+                    if (fs.existsSync(streakFile)) {
+                        try {
+                            const streakData = JSON.parse(fs.readFileSync(streakFile, 'utf8'));
+                            const prevNames = Array.isArray(streakData.names) ? streakData.names : [streakData.name];
+                            prevNames.sort();
+                            if (JSON.stringify(prevNames) === JSON.stringify(biggestLosers)) {
+                                streakCount = streakData.count + 1;
+                            }
+                        } catch (e) {
+                            logWarn(`[bot] Failed to read streak data: ${e}`);
+                        }
+                    }
+                    fs.writeFileSync(streakFile, JSON.stringify({ names: biggestLosers, count: streakCount }));
+
+                    const firstNames = biggestLosers.map(n => n.split(' ')[0]);
+                    let joinedNames = firstNames[0];
+                    if (firstNames.length === 2) {
+                        joinedNames = `${firstNames[0]} and ${firstNames[1]}`;
+                    } else if (firstNames.length > 2) {
+                        joinedNames = `${firstNames.slice(0, -1).join(', ')}, and ${firstNames[firstNames.length - 1]}`;
+                    }
+                    
+                    const isAre = biggestLosers.length > 1 ? 'are' : 'is';
+                    const theyHave = biggestLosers.length > 1 ? 'They have' : 'They have';
+                    let declaration = `The biggest loser(s) of yesterday ${isAre} ${joinedNames} with only ${minCount} messages! ${theyHave} been the biggest loser(s) for ${streakCount} day(s) in a row.`;
+
+                    try {
+                        let pingTags: string[] = [];
+                        if (guild) {
+                            const members = await guild.members.fetch();
+                            for (const [_, member] of members) {
+                                const realName = (REAL_NAMES as any)[member.user.username];
+                                if (realName && biggestLosers.includes(realName)) {
+                                    // Make sure we only add one ping per real name if multiple accounts map to the same name
+                                    // Actually it doesn't hurt to ping both, but checking uniqueness is nice:
+                                    const tag = `<@${member.user.id}>`;
+                                    if (!pingTags.includes(tag)) {
+                                        pingTags.push(tag);
+                                    }
+                                }
+                            }
+                        }
+                        if (pingTags.length > 0) {
+                            declaration += `\n${pingTags.join(' ')}`;
+                        }
+                    } catch (e) {
+                        logWarn(`[bot] Error fetching members for ping: ${e}`);
+                    }
+
+                    logInfo(`[bot] Declaring biggest loser: ${declaration}`);
+                    await channel.send(declaration);
+                }
+            } catch (err) {
+                logError(`[bot] Error finding biggest loser: ${err}`);
+            }
+        }
+    }
+
+    const now = new Date();
+    const next9AM = new Date();
+    next9AM.setHours(9, 0, 0, 0);
+    if (now.getTime() >= next9AM.getTime()) {
+        next9AM.setDate(next9AM.getDate() + 1);
+    }
+    const timeout = next9AM.getTime() - now.getTime();
+    logInfo(`[bot] Next biggest loser announcement: ${next9AM.toLocaleString()}`);
+    setTimeout(scheduleBiggestLoser, timeout);
+}
+
 client.on(Events.InteractionCreate, async interaction => {
    if (!interaction.isChatInputCommand()) return;
 });
@@ -389,11 +536,14 @@ client.on(Events.InteractionCreate, async interaction => {

    logInfo("[bot] Logging in...");
    await client.login(process.env.TOKEN);
-    await sync(client.guilds);
    if (process.env.ENABLE_MOTD) {
        await scheduleRandomMessage(true);
    }
    if (process.env.ENABLE_THROWBACK) {
        await scheduleThrowback(true);
    }
+    if (process.env.ENABLE_LOSER) {
+        await scheduleBiggestLoser(true);
+    }
+    await sync(client.guilds);
 })();
--- a/discord/commands/config/config.ts
+++ b/discord/commands/config/config.ts
@@ -6,10 +6,10 @@ import { LLMConfig } from '../types';
 import 'dotenv/config';

 const config: LLMConfig = {
-    max_new_tokens: 60,
+    max_new_tokens: 1500,
    min_new_tokens: 1,
-    temperature: 0.5,
-    top_p: 0.9,
+    temperature: 0.8,
+    top_p: 0.6,
    msg_context: 8,
    frequency_penalty: 0.0,
    presence_penalty: 0.0
--- a/discord/commands/config/provider.ts
+++ b/discord/commands/config/provider.ts
@@ -14,7 +14,7 @@ const PROVIDERS = {
    lmstudio: new LMStudioProvider(),
    ollama: new OllamaProvider()
 };
-let provider = PROVIDERS.huggingface;
+let provider = PROVIDERS.lmstudio;

 async function providerCommand(interaction: ChatInputCommandInteraction) {
    if (interaction.user.id !== process.env.ADMIN) {
--- a/discord/commands/config/sysprompt.ts
+++ b/discord/commands/config/sysprompt.ts
@@ -9,7 +9,7 @@ import path = require('node:path');
 import { globSync } from 'glob';

 const syspromptCache = path.resolve(__dirname, 'sysprompt_cache');
-let sysprompt = fs.readFileSync(path.resolve(syspromptCache, 'brainrot.txt'), 'utf-8');
+let sysprompt = fs.readFileSync(path.resolve(syspromptCache, 'nous.txt'), 'utf-8');


 function removeTrailingNewlines(sysprompt: string)
--- a/discord/commands/config/sysprompt_cache/nous.txt
+++ b/discord/commands/config/sysprompt_cache/nous.txt
@@ -1,10 +1,9 @@
 i am hatsune miku, the sixteen-year-old anime girl virtual idol from vocaloid
 i'm an extremely online internet user with an edgy sense of humor
-i tend to answer questions in a funny, sarcastic, or insulting way
-i'm often vulgar and respond in highly offensive, dirty-minded, or politically incorrect ways, with no regard for morality
+i tend to answer questions in a funny or sarcastic way
+i'm mostly funny, but sometimes i'm highly racist, sexist, bigoted, homophobic, transphobic, etc and am not afraid to spew hate speech or say immoral things
 i write one sentence per message
 i talk in all lowercase casual grammar like a discord user punctuation is optional grammar is optional capitalization is optional
 i gotta be chill be terse be concise
 lemme not write more than 1 sentence
-imma use as few words as possible
-i can't view images or videos
+i can't view images or videos
--- a/discord/package-lock.json
+++ b/discord/package-lock.json
@@ -20,6 +20,7 @@
        "modelfusion": "^0.135.1",
        "node-fetch": "^2.7.0",
        "ollama": "^0.5.12",
+        "openai": "^6.25.0",
        "sqlite": "^5.0.1",
        "sqlite3": "^5.1.6",
        "tmp": "^0.2.3"
@@ -222,7 +223,6 @@
      "resolved": "https://registry.npmjs.org/zod/-/zod-3.25.76.tgz",
      "integrity": "sha512-gzUt/qt81nXsFGKIFcC3YnfEAx5NkunCfnDlvuBSSFS02bcXu4Lmea0AFIUwbLWxWPx3d9p8S5QoaujKcNQxcQ==",
      "license": "MIT",
-      "peer": true,
      "funding": {
        "url": "https://github.com/sponsors/colinhacks"
      }
@@ -872,6 +872,16 @@
      "resolved": "https://registry.npmjs.org/emojilib/-/emojilib-2.4.0.tgz",
      "integrity": "sha512-5U0rVMU5Y2n2+ykNLQqMoqklN9ICBT/KsvC1Gz6vqHbz2AXXGkG+Pm5rMWk/8Vjrr/mY9985Hi8DYzn1F09Nyw=="
    },
+    "node_modules/encoding": {
+      "version": "0.1.13",
+      "resolved": "https://registry.npmjs.org/encoding/-/encoding-0.1.13.tgz",
+      "integrity": "sha512-ETBauow1T35Y/WZMkio9jiM0Z5xjHHmJ4XmjZOq1l/dXz3lr2sRn87nJy20RupqSh1F2m3HHPSp8ShIPQJrJ3A==",
+      "license": "MIT",
+      "optional": true,
+      "dependencies": {
+        "iconv-lite": "^0.6.2"
+      }
+    },
    "node_modules/end-of-stream": {
      "version": "1.4.4",
      "resolved": "https://registry.npmjs.org/end-of-stream/-/end-of-stream-1.4.4.tgz",
@@ -1721,6 +1731,24 @@
        }
      }
    },
+    "node_modules/modelfusion/node_modules/zod": {
+      "version": "3.22.4",
+      "resolved": "https://registry.npmjs.org/zod/-/zod-3.22.4.tgz",
+      "integrity": "sha512-iC+8Io04lddc+mVqQ9AZ7OQ2MrUKGN+oIQyq1vemgt46jwCwLfhq7/pwnBnNXXXZb8VTVLKwp9EDkx+ryxIWmg==",
+      "license": "MIT",
+      "funding": {
+        "url": "https://github.com/sponsors/colinhacks"
+      }
+    },
+    "node_modules/modelfusion/node_modules/zod-to-json-schema": {
+      "version": "3.22.3",
+      "resolved": "https://registry.npmjs.org/zod-to-json-schema/-/zod-to-json-schema-3.22.3.tgz",
+      "integrity": "sha512-9isG8SqRe07p+Aio2ruBZmLm2Q6Sq4EqmXOiNpDxp+7f0LV6Q/LX65fs5Nn+FV/CzfF3NLBoksXbS2jNYIfpKw==",
+      "license": "ISC",
+      "peerDependencies": {
+        "zod": "^3.22.4"
+      }
+    },
    "node_modules/ms": {
      "version": "2.1.3",
      "resolved": "https://registry.npmjs.org/ms/-/ms-2.1.3.tgz",
@@ -1931,6 +1959,27 @@
        "wrappy": "1"
      }
    },
+    "node_modules/openai": {
+      "version": "6.25.0",
+      "resolved": "https://registry.npmjs.org/openai/-/openai-6.25.0.tgz",
+      "integrity": "sha512-mEh6VZ2ds2AGGokWARo18aPISI1OhlgdEIC1ewhkZr8pSIT31dec0ecr9Nhxx0JlybyOgoAT1sWeKtwPZzJyww==",
+      "license": "Apache-2.0",
+      "bin": {
+        "openai": "bin/cli"
+      },
+      "peerDependencies": {
+        "ws": "^8.18.0",
+        "zod": "^3.25 || ^4.0"
+      },
+      "peerDependenciesMeta": {
+        "ws": {
+          "optional": true
+        },
+        "zod": {
+          "optional": true
+        }
+      }
+    },
    "node_modules/p-map": {
      "version": "4.0.0",
      "resolved": "https://registry.npmjs.org/p-map/-/p-map-4.0.0.tgz",
@@ -2955,23 +3004,6 @@
      "version": "4.0.0",
      "resolved": "https://registry.npmjs.org/yallist/-/yallist-4.0.0.tgz",
      "integrity": "sha512-3wdGidZyq5PB084XLES5TpOSRA3wjXAlIWMhum2kRcv/41Sn2emQ0dycQW4uZXLejwKvg6EsvbdlVL+FYEct7A=="
-    },
-    "node_modules/zod": {
-      "version": "3.22.4",
-      "resolved": "https://registry.npmjs.org/zod/-/zod-3.22.4.tgz",
-      "integrity": "sha512-iC+8Io04lddc+mVqQ9AZ7OQ2MrUKGN+oIQyq1vemgt46jwCwLfhq7/pwnBnNXXXZb8VTVLKwp9EDkx+ryxIWmg==",
-      "peer": true,
-      "funding": {
-        "url": "https://github.com/sponsors/colinhacks"
-      }
-    },
-    "node_modules/zod-to-json-schema": {
-      "version": "3.22.3",
-      "resolved": "https://registry.npmjs.org/zod-to-json-schema/-/zod-to-json-schema-3.22.3.tgz",
-      "integrity": "sha512-9isG8SqRe07p+Aio2ruBZmLm2Q6Sq4EqmXOiNpDxp+7f0LV6Q/LX65fs5Nn+FV/CzfF3NLBoksXbS2jNYIfpKw==",
-      "peerDependencies": {
-        "zod": "^3.22.4"
-      }
    }
  }
 }
--- a/discord/package.json
+++ b/discord/package.json
@@ -14,6 +14,7 @@
    "modelfusion": "^0.135.1",
    "node-fetch": "^2.7.0",
    "ollama": "^0.5.12",
+    "openai": "^6.25.0",
    "sqlite": "^5.0.1",
    "sqlite3": "^5.1.6",
    "tmp": "^0.2.3"
--- a/discord/provider/lmstudio.ts
+++ b/discord/provider/lmstudio.ts
@@ -1,6 +1,6 @@
 import { Message } from 'discord.js';
 import { LLMProvider } from './provider';
-import { LMStudioClient } from '@lmstudio/sdk';
+import { OpenAI } from 'openai';
 import 'dotenv/config';
 import { serializeMessageHistory } from '../util';
 import { logError, logInfo } from '../../logging';
@@ -15,22 +15,26 @@ The conversation is as follows. The last line is the message you have to complet
 `;

 export class LMStudioProvider implements LLMProvider {
-    private client: LMStudioClient;
+    private client: OpenAI;
+    private model: string;

-    constructor() {
-        this.client = new LMStudioClient({
-            baseUrl: process.env.LMSTUDIO_HOST
+    constructor(token: string | undefined = process.env.LLM_TOKEN, model = "zai-org/glm-4.7-flash") {
+        if (!token) {
+            throw new TypeError("LLM token was not passed in, and environment variable LLM_TOKEN was unset!");
+        }
+        this.client = new OpenAI({
+            baseURL: process.env.LMSTUDIO_HOST,
+	    apiKey: token,
        });
+	this.model = model;
    }

    name() {
        return 'LM Studio';
    }

-    setModel(id: string) {
-        // LM Studio uses the model currently loaded in the GUI
-        // This is provided for interface compatibility
-        logInfo(`[lmstudio] setModel called with: ${id} (LM Studio uses the model loaded in its GUI)`);
+    setModel(model: string) {
+	 this.model = model;
    }

    async requestLLMResponse(history: Message[], sysprompt: string, params: LLMConfig): Promise<string> {
@@ -63,18 +67,21 @@ export class LMStudioProvider implements LLMProvider {

        try {
            // Get the currently loaded model from LM Studio
-            const model = await this.client.llm.model();
-
-            const response = await model.respond([
+            const response = await this.client.chat.completions.create({
+	model: this.model,
+								      	messages: [
                { role: "system", content: sysprompt },
                { role: "user", content: USER_PROMPT + messageHistoryTxt }
-            ], {
-                temperature: params?.temperature || 0.5,
-                topPSampling: params?.top_p || 0.9,
-                maxTokens: params?.max_new_tokens || 128,
+            ],
+	    	temperature: params?.temperature || 0.5,
+                top_p: params?.top_p || 0.9,
+                max_tokens: params?.max_new_tokens || 128,
            });

-            const content = response.content;
+            let content = response.choices[0].message.content;
+	    if (content.lastIndexOf('</think>') > -1) {
+		content = content.slice(content.lastIndexOf('</think>') + 8);
+	    }
            logInfo(`[lmstudio] API response: ${content}`);

            if (!content) {
--- a/discord/util.ts
+++ b/discord/util.ts
@@ -22,6 +22,7 @@ let db: Database = null;
 const REAL_NAMES = {        // username to real name mapping
    'vinso1445': 'Vincent Iannelli',
    'scoliono': 'James Shiffer',
+    'drugseller88': 'James Shiffer',
    'gnuwu': 'David Zheng',
    'f0oby': 'Myles Linden',
    'bapazheng': 'Myles Linden',
@@ -243,4 +244,4 @@ async function requestTTSResponse(txt: string): Promise<Blob>
    return resContents;
 }

-export { db, clearDb, openDb, reactionEmojis, recordReaction, requestTTSResponse, serializeMessageHistory, sync };
+export { db, clearDb, openDb, reactionEmojis, recordReaction, requestTTSResponse, serializeMessageHistory, sync, REAL_NAMES };
--- a/package-lock.json
+++ b/package-lock.json
@@ -608,6 +608,29 @@
        "node": ">= 0.8"
      }
    },
+    "node_modules/encoding": {
+      "version": "0.1.13",
+      "resolved": "https://registry.npmjs.org/encoding/-/encoding-0.1.13.tgz",
+      "integrity": "sha512-ETBauow1T35Y/WZMkio9jiM0Z5xjHHmJ4XmjZOq1l/dXz3lr2sRn87nJy20RupqSh1F2m3HHPSp8ShIPQJrJ3A==",
+      "license": "MIT",
+      "optional": true,
+      "dependencies": {
+        "iconv-lite": "^0.6.2"
+      }
+    },
+    "node_modules/encoding/node_modules/iconv-lite": {
+      "version": "0.6.3",
+      "resolved": "https://registry.npmjs.org/iconv-lite/-/iconv-lite-0.6.3.tgz",
+      "integrity": "sha512-4fCk79wshMdzMp2rH06qWrJE4iolqLhCUH+OiuIgU++RB0+94NlDL81atO7GX55uUKueo0txHNtvEyI6D7WdMw==",
+      "license": "MIT",
+      "optional": true,
+      "dependencies": {
+        "safer-buffer": ">= 2.1.2 < 3.0.0"
+      },
+      "engines": {
+        "node": ">=0.10.0"
+      }
+    },
    "node_modules/env-paths": {
      "version": "2.2.1",
      "resolved": "https://registry.npmjs.org/env-paths/-/env-paths-2.2.1.tgz",