apexify.js 4.0.7 → 4.0.8

package/dist/ai/modals-chat/Gemini-pro.js

@@ -9,11 +9,27 @@ const path_1 = __importDefault(require("path"));
 const fs_1 = __importDefault(require("fs"));
 const config_1 = __importDefault(require("./config"));
 const general_functions_1 = require("../../canvas/utils/general functions");
+const verse_db_1 = require("verse.db");
+const axios_1 = __importDefault(require("axios"));
 let currentApiKeyIndex = 0;
 async function geminiPro(message, AI) {
-    var _a;
+    var _a, _b, _c, _d, _e;
     try {
-        const genAI = new generative_ai_1.GoogleGenerativeAI(AI.API_KEY || config_1.default.apiKeys[currentApiKeyIndex]);
+        let apiKeyIndex = currentApiKeyIndex;
+        let genAI;
+        while (apiKeyIndex < config_1.default.apiKeys.length) {
+            const validateKey = await axios_1.default.get(`https://generativelanguage.googleapis.com/v1beta/models?key=${config_1.default.apiKeys[apiKeyIndex]}`);
+            if (validateKey.status === 200) {
+                genAI = new generative_ai_1.GoogleGenerativeAI(config_1.default.apiKeys[apiKeyIndex]);
+                break;
+            }
+            else {
+                apiKeyIndex++;
+            }
+        }
+        if (apiKeyIndex === config_1.default.apiKeys.length) {
+            return 'All provided API keys are invalid.';
+        }
         let personalityString = '';
         if (AI.AiPersonality) {
             const personalityFilePath = path_1.default.join(process.cwd(), AI.AiPersonality);
@@ -23,47 +39,77 @@ async function geminiPro(message, AI) {
         const generationConfig = {
             maxOutputTokens: 750,
         };
-        const systemInstruction = `${personalityString}\n And working on discord in serverName: ${message.guild.name} at channelName: ${message.channel.name} and responding to the userName: ${message.author.username}`;
-        const model = genAI.getGenerativeModel({ model: "gemini-1.5-pro-latest", systemInstruction, generationConfig });
+        const safetySettings = [
+            {
+                category: generative_ai_1.HarmCategory.HARM_CATEGORY_HARASSMENT,
+                threshold: generative_ai_1.HarmBlockThreshold.BLOCK_ONLY_HIGH,
+            },
+            {
+                category: generative_ai_1.HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+                threshold: generative_ai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+            },
+            {
+                category: generative_ai_1.HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+                threshold: generative_ai_1.HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+            },
+            {
+                category: generative_ai_1.HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+                threshold: generative_ai_1.HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+            },
+        ];
+        const systemInstruction = `${personalityString}\n And working on discord in serverName: ${message.serverName} at channelName: ${message.channelName} and responding to the userName:<@ ${message.userId}>`;
+        const model = genAI.getGenerativeModel({ model: "gemini-1.5-pro-latest", systemInstruction, generationConfig, safetySettings });
+        let historyData = undefined;
+        const defaultHistory = [
+            {
+                role: "user",
+                parts: [{ text: `${personalityString}\n And working on discord in serverName: ${message.serverName} at channelName: ${message.channelName} and responding to the userName: <@${message.userId}>` }],
+            },
+            {
+                role: "model",
+                parts: [{ text: `I will greet the user with their name: <@${message.userId}>. Feel free to ask me anything! 😊` }],
+            },
+            {
+                role: "user",
+                parts: [{ text: "Any question regarding JEDI STUDIO, verse.db, apexify.js, or djs-builder?" }],
+            },
+            {
+                role: "model",
+                parts: [{ text: `You should reply with visiting documentation online at jedi-studio.com. The development team consists of 8 people, led by Marco. The CEO/founder is Kimo.` }],
+            },
+            {
+                role: "user",
+                parts: [{ text: "How was I developed or programmed?" }],
+            },
+            {
+                role: "model",
+                parts: [{ text: `You were made by JEDI STUDIO under the development of jedi.tsx, one of the CEOs for JEDI STUDIO.` }],
+            },
+            {
+                role: "user",
+                parts: [{ text: "Tell me about verse.db." }],
+            },
+            {
+                role: "model",
+                parts: [{ text: `verse.db is a database that enables users to manage data for SQL, JSON, YAML, and SQOL. It's a new brand by JEDI STUDIO, and SOOL will support more database management with comprehensive features and security.` }],
+            },
+        ];
+        let db;
+        if (message.db) {
+            db = new verse_db_1.connect({ adapter: 'json', dataPath: `${message.serverId}_ChatHistory` });
+            const data = await db.find(`${message.userId}_chatHistory`, { userId: message.userId });
+            if (!((_a = data.results) === null || _a === void 0 ? void 0 : _a.history) || ((_c = (_b = data.results) === null || _b === void 0 ? void 0 : _b.history) === null || _c === void 0 ? void 0 : _c.length) === 0) {
+                await db.update(`${message.userId}_chatHistory`, { userId: message.userId }, { $set: { history: [] } }, true);
+            }
+            else {
+                historyData = ((_d = data.results) === null || _d === void 0 ? void 0 : _d.history) || undefined;
+            }
+        }
         const chat = model.startChat({
-            history: [
-                {
-                    role: "user",
-                    parts: [{ text: `${personalityString}\n And working on discord in serverName: ${message.guild.name} at channelName: ${message.channel.name} and responding to the userName: ${message.author.username}` }],
-                },
-                {
-                    role: "model",
-                    parts: [{ text: `I will greet the user with their name: <@${message.author.id}>. Feel free to ask me anything! 😊` }],
-                },
-                {
-                    role: "user",
-                    parts: [{ text: "Any question regarding JEDI STUDIO, verse.db, apexify.js, or djs-builder?" }],
-                },
-                {
-                    role: "model",
-                    parts: [{ text: `You should reply with visiting documentation online at jedi-studio.com. The development team consists of 8 people, led by Marco. The CEO/founder is Kimo.` }],
-                },
-                {
-                    role: "user",
-                    parts: [{ text: "How was I developed or programmed?" }],
-                },
-                {
-                    role: "model",
-                    parts: [{ text: `You were made by JEDI STUDIO under the development of jedi.tsx, one of the CEOs for JEDI STUDIO.` }],
-                },
-                {
-                    role: "user",
-                    parts: [{ text: "Tell me about verse.db." }],
-                },
-                {
-                    role: "model",
-                    parts: [{ text: `verse.db is a database that enables users to manage data for SQL, JSON, YAML, and SQOL. It's a new brand by JEDI STUDIO, and SOOL will support more database management with comprehensive features and security.` }],
-                },
-            ],
+            history: historyData || defaultHistory,
             generationConfig
         });
-        const attachment = (_a = message.attachments) === null || _a === void 0 ? void 0 : _a.first();
-        const imgURL = (attachment === null || attachment === void 0 ? void 0 : attachment.url) || null;
+        const imgURL = ((_e = message.attachment) === null || _e === void 0 ? void 0 : _e.url) || null;
         let result;
         if (imgURL) {
             const imageData = await urlToBase64(imgURL);
@@ -78,22 +124,40 @@ async function geminiPro(message, AI) {
         else {
             result = await chat.sendMessage(AI.userMsg);
         }
-        const response = result.response;
-        return response.text();
+        const response = await result.response.text();
+        if (message.db) {
+            const updateQuery_1 = {
+                $push: {
+                    "history": {
+                        role: "user",
+                        parts: [{ text: `${AI.userMsg}` }]
+                    }
+                },
+            };
+            historyData = await db.update(`${message.userId}_chatHistory`, { userId: message.userId }, updateQuery_1, true);
+            const updateQuery_2 = {
+                $push: {
+                    "history": {
+                        role: "model",
+                        parts: [{ text: `${response}` }]
+                    }
+                },
+            };
+            historyData = await db.update(`${message.userId}_chatHistory`, { userId: message.userId }, updateQuery_2, true);
+        }
+        return response;
     }
     catch (e) {
         if (e.message) {
             if (e.message === '[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent: [400 Bad Request] User location is not supported for the API use.') {
-                return `<${message.author.id}> your location isn't supported by gemini AI.`;
+                return `The hoster/bot owner/the used host isn't supported by gemini.`;
+            }
+            else if (e.response && (e.response.status === 429 || e.response.status === 403)) {
+                return 'Ai is on a cooldown for the rest of the day. Either provide your own API key or wait for tomorrow. Check ai.google.dev for free apikeys';
             }
-            else if (e.response && (e.response.status === 429 || e.response.status === 400)) {
-                currentApiKeyIndex++;
-                if (currentApiKeyIndex < config_1.default.apiKeys.length) {
-                    return geminiPro(message, { ...AI, API_KEY: config_1.default.apiKeys[currentApiKeyIndex] });
-                }
-                else {
-                    return 'Ai is on a cooldown for the rest of the day. Either provide your API key or wait for tomorrow. Check ai.google.dev for apikeys';
-                }
+            else if (e.message === '[GoogleGenerativeAI Error]: Candidate was blocked due to SAFETY') {
+                console.error(e);
+                return `Due to safety enabled by gemini you have been blocked.`;
             }
             else {
                 console.error(e);

package/dist/ai/modals-chat/Gemini-pro.js.map

@@ -1 +1 @@
-{"version":3,"file":"Gemini-pro.js","sourceRoot":"","sources":["../../../lib/ai/modals-chat/Gemini-pro.ts"],"names":[],"mappings":";;;;;;AAAA,yDAA2D;AAC3D,gDAAwB;AACxB,4CAAoB;AACpB,sDAA8B;AAC9B,4EAAiE;AAEjE,IAAI,kBAAkB,GAAG,CAAC,CAAC;AAEpB,KAAK,UAAU,SAAS,CAAC,OAAY,EAAE,EAA6E;;IAEzH,IAAI,CAAC;QACH,MAAM,KAAK,GAAG,IAAI,kCAAkB,CAAC,EAAE,CAAC,OAAO,IAAI,gBAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,CAAC;QAEvF,IAAI,iBAAiB,GAAW,EAAE,CAAC;QACnC,IAAI,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,mBAAmB,GAAG,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC;YACvE,MAAM,kBAAkB,GAAG,YAAE,CAAC,YAAY,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;YACzE,iBAAiB,GAAG,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjE,CAAC;QAGD,MAAM,gBAAgB,GAAG;YACrB,eAAe,EAAE,GAAG;SACvB,CAAC;QAEF,MAAM,iBAAiB,GAAG,GAAG,iBAAiB,4CAA4C,OAAO,CAAC,KAAK,CAAC,IAAI,oBAAoB,OAAO,CAAC,OAAO,CAAC,IAAI,oCAAoC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,CAAC;QAClN,MAAM,KAAK,GAAG,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,gBAAgB,EAAC,CAAC,CAAC;QAE/G,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAC;YAC3B,OAAO,EAAE;gBACL;oBACI,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,iBAAiB,4CAA4C,OAAO,CAAC,KAAK,CAAC,IAAI,oBAAoB,OAAO,CAAC,OAAO,CAAC,IAAI,oCAAoC,OAAO,CAAC,MAAM,CAAC,QAAQ,EAAE,EAAE,CAAC;iBAC7M;gBACD;oBACI,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,4CAA4C,OAAO,CAAC,MAAM,CAAC,EAAE,qCAAqC,EAAE,CAAC;iBACxH;gBACD;oBACI,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,2EAA2E,EAAE,CAAC;iBACjG;gBACD;oBACI,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,2JAA2J,EAAE,CAAC;iBACjL;gBACD;oBACI,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,oCAAoC,EAAE,CAAC;iBAC1D;gBACD;oBACI,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kGAAkG,EAAE,CAAC;iBACxH;gBACD;oBACI,IAAI,EAAE,MAAM;oBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;iBAC/C;gBACD;oBACI,IAAI,EAAE,OAAO;oBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,mNAAmN,EAAE,CAAC;iBACzO;aACJ;YACD,gBAAgB;SACnB,CAAC,CAAC;QAED,MAAM,UAAU,GAAG,MAAA,OAAO,CAAC,WAAW,0CAAE,KAAK,EAAE,CAAC;QAChD,MAAM,MAAM,GAAG,CAAA,UAAU,aAAV,UAAU,uBAAV,UAAU,CAAE,GAAG,KAAI,IAAI,CAAC;QACvC,IAAI,MAAW,CAAC;QAEhB,IAAI,MAAM,EAAE,CAAC;YACT,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAA;YAC3C,MAAM,KAAK,GAAG;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,WAAW;iBACtB;aACF,CAAC;YACJ,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACJ,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC;QACD,MAAM,QAAQ,GAAI,MAAM,CAAC,QAAQ,CAAC;QAElC,OAAO,QAAQ,CAAC,IAAI,EAAE,CAAA;IACxB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QAChB,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,CAAC,OAAO,KAAK,mMAAmM,EAAE,CAAC;gBACtN,OAAO,IAAI,OAAO,CAAC,MAAM,CAAC,EAAE,+CAA+C,CAAC;YAC9E,CAAC;iBAAM,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC,EAAE,CAAC;gBAClF,kBAAkB,EAAE,CAAC;gBACrB,IAAI,kBAAkB,GAAG,gBAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;oBAC7C,OAAO,SAAS,CAAC,OAAO,EAAE,EAAE,GAAG,EAAE,EAAE,OAAO,EAAE,gBAAM,CAAC,OAAO,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;gBACtF,CAAC;qBAAM,CAAC;oBACJ,OAAO,gIAAgI,CAAC;gBAC5I,CAAC;YACH,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACjB,OAAO,iLAAiL,CAAC;YAC7L,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACjB,OAAO,gCAAgC,CAAC;QAC5C,CAAC;IACH,CAAC;AACH,CAAC;AAjGD,8BAiGC;AAED,KAAK,UAAU,WAAW,CAAC,QAAgB;IACzC,IAAI,CAAC;QACD,MAAM,eAAe,GAAG,MAAM,IAAA,6BAAS,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAEzD,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACxD,OAAO,YAAY,CAAC;IACxB,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,0CAA0C,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAC/E,CAAC;AACH,CAAC"}
+{"version":3,"file":"Gemini-pro.js","sourceRoot":"","sources":["../../../lib/ai/modals-chat/Gemini-pro.ts"],"names":[],"mappings":";;;;;;AAAA,yDAA6F;AAC7F,gDAAwB;AACxB,4CAAoB;AACpB,sDAA8B;AAC9B,4EAAiE;AACjE,uCAAoC;AACpC,kDAA0B;AAE1B,IAAI,kBAAkB,GAAG,CAAC,CAAC;AAEpB,KAAK,UAAU,SAAS,CAAC,OAAqH,EAAE,EAA6E;;IAGlO,IAAI,CAAC;QAEH,IAAI,WAAW,GAAG,kBAAkB,CAAC;QACrC,IAAI,KAAU,CAAC;QACX,OAAO,WAAW,GAAG,gBAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACzC,MAAM,WAAW,GAAG,MAAM,eAAK,CAAC,GAAG,CAAC,+DAA+D,gBAAM,CAAC,OAAO,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;YAClI,IAAI,WAAW,CAAC,MAAM,KAAK,GAAG,EAAE,CAAC;gBAC7B,KAAK,GAAG,IAAI,kCAAkB,CAAC,gBAAM,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;gBAE5D,MAAM;YACV,CAAC;iBAAM,CAAC;gBACJ,WAAW,EAAE,CAAC;YAClB,CAAC;QACL,CAAC;QAED,IAAI,WAAW,KAAK,gBAAM,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YACxC,OAAO,oCAAoC,CAAC;QAChD,CAAC;QAEL,IAAI,iBAAiB,GAAW,EAAE,CAAC;QACnC,IAAI,EAAE,CAAC,aAAa,EAAE,CAAC;YACnB,MAAM,mBAAmB,GAAG,cAAI,CAAC,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC;YACvE,MAAM,kBAAkB,GAAG,YAAE,CAAC,YAAY,CAAC,mBAAmB,EAAE,OAAO,CAAC,CAAC;YACzE,iBAAiB,GAAG,kBAAkB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QACjE,CAAC;QAGD,MAAM,gBAAgB,GAAG;YACrB,eAAe,EAAE,GAAG;SACvB,CAAC;QAEF,MAAM,cAAc,GAAG;YACnB;gBACI,QAAQ,EAAE,4BAAY,CAAC,wBAAwB;gBAC/C,SAAS,EAAE,kCAAkB,CAAC,eAAe;aAChD;YACD;gBACI,QAAQ,EAAE,4BAAY,CAAC,yBAAyB;gBAChD,SAAS,EAAE,kCAAkB,CAAC,sBAAsB;aACvD;YACD;gBACI,QAAQ,EAAE,4BAAY,CAAC,+BAA+B;gBACtD,SAAS,EAAE,kCAAkB,CAAC,mBAAmB;aACpD;YACD;gBACI,QAAQ,EAAE,4BAAY,CAAC,+BAA+B;gBACtD,SAAS,EAAE,kCAAkB,CAAC,sBAAsB;aACvD;SACJ,CAAC;QAEF,MAAM,iBAAiB,GAAG,GAAG,iBAAiB,4CAA4C,OAAO,CAAC,UAAU,oBAAoB,OAAO,CAAC,WAAW,sCAAsC,OAAO,CAAC,MAAM,GAAG,CAAC;QAC3M,MAAM,KAAK,GAAG,KAAK,CAAC,kBAAkB,CAAC,EAAE,KAAK,EAAE,uBAAuB,EAAE,iBAAiB,EAAE,gBAAgB,EAAE,cAAc,EAAE,CAAC,CAAC;QAEhI,IAAI,WAAW,GAAsB,SAAS,CAAC;QAE/C,MAAM,cAAc,GAAG;YACnB;gBACI,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,iBAAiB,4CAA4C,OAAO,CAAC,UAAU,oBAAoB,OAAO,CAAC,WAAW,sCAAsC,OAAO,CAAC,MAAM,GAAG,EAAE,CAAC;aACtM;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,4CAA4C,OAAO,CAAC,MAAM,qCAAqC,EAAE,CAAC;aACrH;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,2EAA2E,EAAE,CAAC;aACjG;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,2JAA2J,EAAE,CAAC;aACjL;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,oCAAoC,EAAE,CAAC;aAC1D;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,kGAAkG,EAAE,CAAC;aACxH;YACD;gBACI,IAAI,EAAE,MAAM;gBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,CAAC;aAC/C;YACD;gBACI,IAAI,EAAE,OAAO;gBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,mNAAmN,EAAE,CAAC;aACzO;SACJ,CAAC;QAEF,IAAI,EAAO,CAAC;QAEZ,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;YACb,EAAE,GAAG,IAAI,kBAAO,CAAC,EAAE,OAAO,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,OAAO,CAAC,QAAQ,cAAc,EAAE,CAAC,CAAC;YAEpF,MAAM,IAAI,GAAG,MAAM,EAAE,CAAC,IAAI,CAAC,GAAG,OAAO,CAAC,MAAM,cAAc,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAExF,IAAI,CAAC,CAAA,MAAA,IAAI,CAAC,OAAO,0CAAE,OAAO,CAAA,IAAI,CAAA,MAAA,MAAA,IAAI,CAAC,OAAO,0CAAE,OAAO,0CAAE,MAAM,MAAK,CAAC,EAAE,CAAC;gBACjE,MAAM,EAAE,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,cAAc,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAAE,EAAE,IAAI,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;YACjH,CAAC;iBAAM,CAAC;gBACL,WAAW,GAAG,CAAA,MAAA,IAAI,CAAC,OAAO,0CAAE,OAAO,KAAI,SAAS,CAAC;YACpD,CAAC;QACJ,CAAC;QAED,MAAM,IAAI,GAAG,KAAK,CAAC,SAAS,CAAC;YAC3B,OAAO,EAAE,WAAW,IAAI,cAAc;YACtC,gBAAgB;SACjB,CAAC,CAAC;QAEH,MAAM,MAAM,GAAG,CAAA,MAAA,OAAO,CAAC,UAAU,0CAAE,GAAG,KAAI,IAAI,CAAC;QAC/C,IAAI,MAAW,CAAC;QAEhB,IAAI,MAAM,EAAE,CAAC;YACT,MAAM,SAAS,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC,CAAA;YAC3C,MAAM,KAAK,GAAG;gBACV,UAAU,EAAE;oBACV,IAAI,EAAE,SAAS;oBACf,QAAQ,EAAE,WAAW;iBACtB;aACF,CAAC;YACJ,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC;QACzD,CAAC;aAAM,CAAC;YACJ,MAAM,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,OAAO,CAAC,CAAC;QAChD,CAAC;QAED,MAAM,QAAQ,GAAI,MAAM,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QAE/C,IAAI,OAAO,CAAC,EAAE,EAAE,CAAC;YAEb,MAAM,aAAa,GAAG;gBAClB,KAAK,EAAE;oBACL,SAAS,EAAE;wBACT,IAAI,EAAE,MAAM;wBACZ,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,OAAO,EAAE,EAAE,CAAC;qBACnC;iBACF;aACJ,CAAC;YAEF,WAAW,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,cAAc,EAC5D,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAC1B,aAAa,EAAE,IAAI,CACnB,CAAC;YAEF,MAAM,aAAa,GAAG;gBAClB,KAAK,EAAE;oBACL,SAAS,EAAE;wBACT,IAAI,EAAE,OAAO;wBACb,KAAK,EAAE,CAAC,EAAE,IAAI,EAAE,GAAG,QAAQ,EAAE,EAAE,CAAC;qBACjC;iBACF;aACJ,CAAC;YAEF,WAAW,GAAG,MAAM,EAAE,CAAC,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,cAAc,EAC7D,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,EAC1B,aAAa,EACb,IAAI,CACH,CAAC;QACN,CAAC;QAED,OAAO,QAAQ,CAAC;IACpB,CAAC;IAAC,OAAO,CAAM,EAAE,CAAC;QACd,IAAI,CAAC,CAAC,OAAO,EAAE,CAAC;YACZ,IAAI,CAAC,CAAC,OAAO,KAAK,mMAAmM,EAAE,CAAC;gBACtN,OAAO,+DAA+D,CAAC;YACzE,CAAC;iBAAM,IAAI,CAAC,CAAC,QAAQ,IAAI,CAAC,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,KAAK,GAAG,CAAC,EAAE,CAAC;gBAC9E,OAAO,yIAAyI,CAAC;YACvJ,CAAC;iBAAM,IAAI,CAAC,CAAC,OAAO,KAAK,iEAAiE,EAAE,CAAC;gBACzF,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACjB,OAAO,wDAAwD,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACJ,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;gBACjB,OAAO,iLAAiL,CAAC;YAC7L,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;YACjB,OAAO,gCAAgC,CAAC;QAC5C,CAAC;IACH,CAAC;AACH,CAAC;AArLD,8BAqLC;AAED,KAAK,UAAU,WAAW,CAAC,QAAgB;IACzC,IAAI,CAAC;QACD,MAAM,eAAe,GAAG,MAAM,IAAA,6BAAS,EAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;QAEzD,MAAM,YAAY,GAAG,eAAe,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACxD,OAAO,YAAY,CAAC;IACxB,CAAC;IAAC,OAAO,KAAU,EAAE,CAAC;QAClB,MAAM,IAAI,KAAK,CAAC,0CAA0C,KAAK,CAAC,OAAO,EAAE,CAAC,CAAC;IAC/E,CAAC;AACH,CAAC"}

package/lib/ai/ApexAI.ts

@@ -22,9 +22,6 @@ import { imageTools } from "./buttons/drawMenu";
 import { ImageModals } from './functions/validOptions';
 import { apexai, geminiFlash, geminiPro, facebook_ai, yi_34b, starChat } from "./modals-chat/modals";
 
-/**
- * Configuration options for the Gemini Flash functionality.
- */
 export interface Options {
   /**
    * Configuration options related to voice functionality.
@@ -106,8 +103,8 @@ export interface Options {
       personality?: string | any;
       API_KEY?: string;
       memory?: {
-          memoryOn: boolean;
-          id: string;
+          memoryOn?: boolean;
+          id?: string;
       };
       typeWriting?: {
           enable?: boolean;
@@ -138,8 +135,9 @@ export interface Options {
    */
   others?: {
       messageType?: {
-          type: string;
-          intialContent: string;
+          type?: string;
+          intialContent?: string;
+          sendAs?: string | "embed" | "content";
       };
       buttons?: any[];
       keywords?: string[];
@@ -275,7 +273,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
 
      await message.channel?.sendTyping();
 
-        usermsg = message.content;  
+      usermsg = message.content;  
 
      if (
         message.attachments.some((attachment: any) =>
@@ -319,10 +317,10 @@ export async function ApexAI (message: any, aiOptions: Options) {
          if (message.attachments.size > 0) {
             if (attachment.name.endsWith('.pdf')) {
                 const pdfContent = await readPdf(attachment.url);
-                usermsg += pdfContent;
+                usermsg += `\n\n This is the pdf file content:\n\n ${pdfContent}`;
             } else {
                 const txtContent = await readTextFile(attachment.url);
-                usermsg += txtContent;
+                usermsg += `\n\n This is the .txt file content:\n\n ${txtContent}`;
             }
          }
       }
@@ -390,9 +388,10 @@ export async function ApexAI (message: any, aiOptions: Options) {
 
       if (drawValid) {
 
-        usermsg = `Rewrite this text below in more descriptive way make it clear to be visualized correctly and enhance it and use stronger words please\n\n\n ${usermsg}`
-        if (enhancerOn) {
-            usermsg += await geminiFlash(message, { userMsg: usermsg, API_KEY: API_KEY, AiPersonality: personality, })
+        usermsg = `Rewrite this text below in more descriptive way make it clear to be visualized correctly and enhance it and use stronger words please and please return the response with nothing else in just the enhanced prompt nothing more nothing less\n\n\n ${usermsg}`
+       
+        if (enhancerOn) { 
+          usermsg += await hercai.question({ model: "v3", content: usermsg});
         }
 
         return await aiImagine(
@@ -456,9 +455,37 @@ export async function ApexAI (message: any, aiOptions: Options) {
         } else if (chatModal === 'starChat') {
             response = await starChat(usermsg);
         } else if (chatModal === 'gemini-flash') {
-            response = await geminiFlash(message, { userMsg: usermsg, API_KEY: API_KEY, AiPersonality: personality, });
+            response = await geminiFlash(
+              {
+                  userId: message.author.id,
+                  serverName: message.guild.name,
+                  serverId: message.guild.id,
+                  channelName: message.channel.name,
+                  attachment: attachment,
+                  db: memoryOn
+              },
+              {
+                  userMsg: usermsg,
+                  API_KEY: API_KEY,
+                  AiPersonality: personality
+              }
+            );
         } else if (chatModal === 'gemini-pro') {
-          response = await geminiPro(message, { userMsg: usermsg, API_KEY: API_KEY, AiPersonality: personality, })
+          response = await geminiPro(
+            {
+                userId: message.author.id,
+                serverName: message.guild.name,
+                serverId: message.guild.id,
+                channelName: message.channel.name,
+                attachment: attachment,
+                db: memoryOn
+            },
+            {
+                userMsg: usermsg,
+                API_KEY: API_KEY,
+                AiPersonality: personality
+            }
+          );
         } else if (chatModal === 'v3' || chatModal === 'v3-32k' || chatModal === 'turbo' || chatModal === 'turbo-16k' || chatModal === 'gemini') {
           if (!memoryOn) {
 

package/lib/ai/functions/draw.ts

@@ -740,7 +740,7 @@ async function aiImagine(
         }
       }
     }
-    const allRows = [...buttons,  row1, row2]
+    const allRows = [...buttons, row1, row2]
     await message.reply({
       files: attachData,
       components: allRows,

package/lib/ai/modals-chat/Gemini-flash.ts

@@ -1,17 +1,40 @@
-import { GoogleGenerativeAI } from "@google/generative-ai";
+import { GoogleGenerativeAI, HarmCategory, HarmBlockThreshold } from "@google/generative-ai";
 import path from 'path';
 import fs from 'fs';
 import config from './config'; 
 import { converter } from "../../canvas/utils/general functions";
+import  { connect } from "verse.db";
+import axios from "axios";
 
 let currentApiKeyIndex = 0;
 
-export async function geminiFlash(message: any, AI: { AiPersonality: string | null, userMsg: string, API_KEY: string | null }): Promise<any> {
+export async function geminiFlash(message: { userId: string, serverName: string, serverId: string, channelName: string, attachment: any, db: boolean  }, AI: { AiPersonality: string | null, userMsg: string, API_KEY: string | null }): Promise<any> {
+
+    let db: any;
+
+    if (message.db) {
+       db = new connect({ adapter: 'json', dataPath: `${message.serverId}_ChatHistory` });
+    }
 
   try { 
-    const genAI = new GoogleGenerativeAI(AI.API_KEY || config.apiKeys[currentApiKeyIndex]);
+    let apiKeyIndex = currentApiKeyIndex;
+    let genAI: any;
+        while (apiKeyIndex < config.apiKeys.length) {
+            const validateKey = await axios.get(`https://generativelanguage.googleapis.com/v1beta/models?key=${config.apiKeys[apiKeyIndex]}`);
+            if (validateKey.status === 200) {
+                genAI = new GoogleGenerativeAI(config.apiKeys[apiKeyIndex]);
+                
+                break;
+            } else {
+                apiKeyIndex++;
+            }
+        }
 
+        if (apiKeyIndex === config.apiKeys.length) {
+            return 'All provided API keys are invalid.';
+        }
     let personalityString: string = '';
+    
     if (AI.AiPersonality) {
         const personalityFilePath = path.join(process.cwd(), AI.AiPersonality);
         const personalityContent = fs.readFileSync(personalityFilePath, 'utf-8');
@@ -22,49 +45,83 @@ export async function geminiFlash(message: any, AI: { AiPersonality: string | nu
         maxOutputTokens: 750,
     };
 
-    const systemInstruction = `${personalityString}\n And working on discord in serverName: ${message.guild.name} at channelName: ${message.channel.name} and responding to the userName: ${message.author.username}`;
-    const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash-latest", systemInstruction, generationConfig});
+    const safetySettings = [
+        {
+            category: HarmCategory.HARM_CATEGORY_HARASSMENT,
+            threshold: HarmBlockThreshold.BLOCK_ONLY_HIGH,
+        },
+        {
+            category: HarmCategory.HARM_CATEGORY_HATE_SPEECH,
+            threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+        },
+        {
+            category: HarmCategory.HARM_CATEGORY_SEXUALLY_EXPLICIT,
+            threshold: HarmBlockThreshold.BLOCK_LOW_AND_ABOVE,
+        },
+        {
+            category: HarmCategory.HARM_CATEGORY_DANGEROUS_CONTENT,
+            threshold: HarmBlockThreshold.BLOCK_MEDIUM_AND_ABOVE,
+        },
+    ];
+
+    const systemInstruction = `${personalityString}\n And working on discord in serverName: ${message.serverName} at channelName: ${message.channelName} and responding to the userName: <@${message.userId}>`;
+    const model = genAI.getGenerativeModel({ model: "gemini-1.5-flash-latest", systemInstruction, generationConfig, safetySettings });
+
+    const defaultHistory = [
+        {
+            role: "user",
+            parts: [{ text: `${personalityString}\n And working on discord in serverName: ${message.serverName} at channelName: ${message.channelName} and responding to the userName: <@${message.userId}>` }],
+        },
+        {
+            role: "model",
+            parts: [{ text: `I will greet the user with their name: <@${message.userId}>. Feel free to ask me anything! 😊` }],
+        },
+        {
+            role: "user",
+            parts: [{ text: "Any question regarding JEDI STUDIO, verse.db, apexify.js, or djs-builder?" }],
+        },
+        {
+            role: "model",
+            parts: [{ text: `You should reply with visiting documentation online at jedi-studio.com. The development team consists of 8 people, led by Marco. The CEO/founder is Kimo.` }],
+        },
+        {
+            role: "user",
+            parts: [{ text: "How was I developed or programmed?" }],
+        },
+        {
+            role: "model",
+            parts: [{ text: `You were made by JEDI STUDIO under the development of jedi.tsx, one of the CEOs for JEDI STUDIO.` }],
+        },
+        {
+            role: "user",
+            parts: [{ text: "Tell me about verse.db." }],
+        },
+        {
+            role: "model",
+            parts: [{ text: `verse.db is a database that enables users to manage data for SQL, JSON, YAML, and SQOL. It's a new brand by JEDI STUDIO, and SOOL will support more database management with comprehensive features and security.` }],
+        },
+    ];
+
+    let historyData: any[] | undefined = undefined;
+
+    if (message.db) {
+        db = new connect({ adapter: 'json', dataPath: `${message.serverId}_ChatHistory` });
+
+       const data = await db.find(`${message.userId}_chatHistory`, { userId: message.userId });
+       
+       if (!data.results?.history || data.results?.history?.length === 0) {
+          await db.update(`${message.userId}_chatHistory`, { userId: message.userId }, { $set: { history: [] } }, true);
+       } else {
+          historyData = data.results?.history || undefined;
+       }
+    }
 
     const chat = model.startChat({
-      history: [
-          {
-              role: "user",
-              parts: [{ text: `${personalityString}\n And working on discord in serverName: ${message.guild.name} at channelName: ${message.channel.name} and responding to the userName: ${message.author.username}` }],
-          },
-          {
-              role: "model",
-              parts: [{ text: `I will greet the user with their name: <@${message.author.id}>. Feel free to ask me anything! 😊` }],
-          },
-          {
-              role: "user",
-              parts: [{ text: "Any question regarding JEDI STUDIO, verse.db, apexify.js, or djs-builder?" }],
-          },
-          {
-              role: "model",
-              parts: [{ text: `You should reply with visiting documentation online at jedi-studio.com. The development team consists of 8 people, led by Marco. The CEO/founder is Kimo.` }],
-          },
-          {
-              role: "user",
-              parts: [{ text: "How was I developed or programmed?" }],
-          },
-          {
-              role: "model",
-              parts: [{ text: `You were made by JEDI STUDIO under the development of jedi.tsx, one of the CEOs for JEDI STUDIO.` }],
-          },
-          {
-              role: "user",
-              parts: [{ text: "Tell me about verse.db." }],
-          },
-          {
-              role: "model",
-              parts: [{ text: `verse.db is a database that enables users to manage data for SQL, JSON, YAML, and SQOL. It's a new brand by JEDI STUDIO, and SOOL will support more database management with comprehensive features and security.` }],
-          },
-      ],
+      history: historyData || defaultHistory,
       generationConfig
-  });
+    });
   
-    const attachment = message.attachments?.first();
-    const imgURL = attachment?.url || null;
+    const imgURL = message.attachment?.url || null;
     let result: any;
 
     if (imgURL) {
@@ -79,20 +136,51 @@ export async function geminiFlash(message: any, AI: { AiPersonality: string | nu
     } else {
         result = await chat.sendMessage(AI.userMsg);
     }
-    const response =  result.response;
 
-    return response.text()
+    const response =  await result.response.text();
+
+    if (message.db) {
+        
+        const updateQuery_1 = {
+            $push: {
+              "history": {
+                role: "user",
+                parts: [{ text: `${AI.userMsg}` }]
+              }
+            },
+        };
+
+        historyData = await db.update(`${message.userId}_chatHistory`,
+         { userId: message.userId },
+         updateQuery_1, true
+        );
+
+        const updateQuery_2 = {
+            $push: {
+              "history": {
+                role: "model",
+                parts: [{ text: `${response}` }]
+              }
+            },
+        };
+        
+        historyData = await db.update(`${message.userId}_chatHistory`,
+        { userId: message.userId },
+        updateQuery_2,
+        true
+        );
+    }
+
+    return response;
 } catch (e: any) {
     if (e.message) {
         if (e.message === '[GoogleGenerativeAI Error]: Error fetching from https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent: [400 Bad Request] User location is not supported for the API use.') {
-          return `<${message.author.id}> your location isn't supported by gemini AI.`;
-        } else if (e.response && (e.response.status === 429 || e.response.status === 400)) {
-          currentApiKeyIndex++;
-          if (currentApiKeyIndex < config.apiKeys.length) {
-              return geminiFlash(message, { ...AI, API_KEY: config.apiKeys[currentApiKeyIndex] });
-          } else {
-              return 'Ai is on a cooldown for the rest of the day. Either provide your API key or wait for tomorrow. Check ai.google.dev for apikeys';
-          }
+          return `The hoster/bot owner/the used host isn't supported by gemini.`;
+        } else if (e.response && (e.response.status === 429 || e.response.status === 403)) {
+              return 'Ai is on a cooldown for the rest of the day. Either provide your own API key or wait for tomorrow. Check ai.google.dev for free apikeys';
+        } else if (e.message === '[GoogleGenerativeAI Error]: Candidate was blocked due to SAFETY') {
+            console.error(e);
+            return `Due to safety enabled by gemini you have been blocked.`;
         } else {
             console.error(e);
             return `Try again later please... Either API is on a cooldown or an internal server error has occurred. If issue persists please contact the bot developer or owner of the npm package.`;