From a711aec42b6527d53494421d081003e26e9ac601513febc7601fe4512fd91155 Mon Sep 17 00:00:00 2001
From: aslan <jan.trencansky@protonmail.com>
Date: Sat, 24 Jan 2026 05:05:05 -0500
Subject: [PATCH] Add smart models

---
 src/modules/ai/ai.ts  | 14 +++++++++++---
 src/modules/module.ts |  1 -
 2 files changed, 11 insertions(+), 4 deletions(-)

diff --git a/src/modules/ai/ai.ts b/src/modules/ai/ai.ts
index e38a13c..078a540 100644
--- a/src/modules/ai/ai.ts
+++ b/src/modules/ai/ai.ts
@@ -4,7 +4,12 @@ import { config } from "../../config.js";
 import { getTextGemini, getImageGemini } from "../../services/ai/ai.js";
 import { alts } from "./alts.js";
 import type { IAIInstructions } from "./types.js";
-import { changePersonality, getUserById, getUserName } from "../../helpers.js";
+import {
+    changePersonality,
+    checkRoles,
+    getUserById,
+    getUserName,
+} from "../../helpers.js";
 import { prices } from "./prices.js";
 import { state } from "../../store/store.js";
 
@@ -46,7 +51,8 @@ const onAI = async (
 
     let personality = config.app.ai.personalities[state.personality.index];
 
-    const useSmartModel = text.startsWith("!aipro");
+    const useSmartModel =
+        text.startsWith("!aipro") && checkRoles(["MODERATOR", "ADMIN"], sender);
 
     let textMod = text.replace("!aipro", "").replace("!ai", "").trim();
     let instructions = {
@@ -103,7 +109,9 @@ const onAI = async (
 const onImageGen = async (text: string, roomId: string, sender: string) => {
     const user = getUserById(sender);
 
-    const useSmartModel = text.startsWith("!imgpro");
+    const useSmartModel =
+        text.startsWith("!imgpro") &&
+        checkRoles(["MODERATOR", "ADMIN"], sender);
 
     let textMod = text
         .replace("!imgpro", "")
diff --git a/src/modules/module.ts b/src/modules/module.ts
index 96582ab..bf7029f 100644
--- a/src/modules/module.ts
+++ b/src/modules/module.ts
@@ -13,7 +13,6 @@ import { registerModuleAI } from "./ai/ai.js";
 import { registerModuleGame } from "./game/game.js";
 import { checkRoles, getUserById, log } from "../helpers.js";
 import { onAnyMessage, onMissingRole } from "./global.js";
-import { config } from "../config.js";
 
 const callbacks: ICallbackStore = {
     messageCallbacks: [],