apexify.js 3.3.11 → 4.0.0

package/lib/ai/ApexAI.ts

@@ -1,5 +1,6 @@
 import { Hercai } from "hercai";
-const hercai = new Hercai()
+import fs from "fs";
+import path from "path";
 import {
   imageReader,
   toDraw,
@@ -10,78 +11,163 @@ import {
   typeWriter,
   readImage,
 } from "./utils";
-import axios from "axios";
 import { 
   ModalBuilder,
   TextInputBuilder,
   TextInputStyle,
-  ActionRowBuilder 
+  ActionRowBuilder
 } from "discord.js";
 import { filters } from "./buttons/tools";
 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 {
-    voice?: {
-       textVoice?:{
-        enable?: boolean;
-        voiceModal?: string;
-        voice_code?: string;
-        apiKey?: string;
-        type?: string;
-       };
-    };
-    imagine?: {
-        enable?: boolean;
-        drawTrigger?: string[];
-        imageModal?: string;
-        numOfImages?: number;
-        nsfw?: {
+  /**
+   * Configuration options related to voice functionality.
+   * @param voice.textVoice Configuration options for text-to-voice functionality.
+   * @param voice.textVoice.enable Whether text-to-voice functionality is enabled.
+   * @param voice.textVoice.voiceModal The voice modal to be used.
+   * @param voice.textVoice.voice_code The voice code only for (apexAI and zenithAI modal).
+   * @param voice.textVoice.apiKey The API key for accessing the voice service only for (apexAI and zenithAI modal).
+   * @param voice.textVoice.type The type of voice only for (apexAI and zenithAI modal).
+   */
+  voice?: {
+      textVoice?: {
+          enable?: boolean;
+          voiceModal?: string;
+          voice_code?: string;
+          apiKey?: string;
+          type?: string;
+      };
+  };
+  /**
+   * Configuration options related to image generation.
+   * @param imagine.enable Whether image generation is enabled.
+   * @param imagine.drawTrigger The trigger phrases for initiating image generation.
+   * @param imagine.imageModal The modal for the image generation.
+   * @param imagine.numOfImages The number of images to generate.
+   * @param imagine.nsfw Configuration options for NSFW filtering.
+   * @param imagine.nsfw.enable Whether NSFW filtering is enabled.
+   * @param imagine.nsfw.keywords Keywords for NSFW filtering.
+   * @param imagine.enhancer Configuration options for image enhancement.
+   * @param imagine.enhancer.enable Whether image enhancement is enabled (rewrites your prompt in more descriptive way).
+   * @param imagine.enhancer.enhancerModal The modal for image enhancement only for (Prodia modals).
+   * @param imagine.enhancer.cfg_scale The scale for image enhancement only for (Prodia modals).
+   * @param imagine.enhancer.steps The number of enhancement steps only for (Prodia modals).
+   * @param imagine.enhancer.seed The seed for image enhancement only for (Prodia modals).
+   * @param imagine.enhancer.imgStyle The style of the image only for (Prodia modals).
+   * @param imagine.enhancer.negative_prompt The negative prompt for image enhancement only for (Prodia modals).
+   * @param imagine.enhancer.sampler The sampler for image enhancement only for (Prodia modals).
+   */
+  imagine?: {
+      enable?: boolean;
+      drawTrigger?: string[];
+      imageModal?: ImageModals;
+      numOfImages?: number;
+      nsfw?: {
           enable?: boolean;
           keywords?: string[];
-        };
-        enhancer?: boolean;
-    };
-    chat?: {
-        chatModal?: string;
-        readFiles?: boolean;
-        readImages?: boolean;
-        typeWriting?:{
+          deepCheck?: boolean;
+      };
+      enhancer?: {
+          enable: boolean;
+          enhancerModal?: string | "ESRGAN_4x" | "Lanczos" | "Nearest" | "LDSR" | "R-ESRGAN 4x+" | "R-ESRGAN 4x+ Anime6B" | "ScuNET GAN" | "ScuNET PSNR" | "SwinIR 4x" | undefined;
+          cfg_scale?: number;
+          steps?: number;
+          seed?: number;
+          imgStyle?: string | "3d-model" | "analog-film" | "anime" | "cinematic" | "comic-book" | "digital-art" | "enhance" | "isometric" | "fantasy-art" | "isometric" | "line-art" | "low-poly" | "neon-punk" | "origami" | "photographic" | "pixel-art" | "texture" | "craft-clay";
+          negative_prompt?: string;
+          sampler?: string | "DPM++ 2M Karras" | "DPM++ SDE Karras" | "DPM++ 2M SDE Exponential" | "DPM++ 2M SDE Karras" | "Euler a" | "Euler" | "LMS" | "Heun" | "DPM2" | "DPM2 a" | "DPM++ 2S a" | "DPM++ 2M" | "DPM++ SDE" | "DPM++ 2M SDE" | "DPM++ 2M SDE Heun" | "DPM++ 2M SDE Heun Karras" | "DPM++ 2M SDE Heun Exponential" | "DPM++ 3M SDE" | "DPM++ 3M SDE Karras" | "DPM++ 3M SDE Exponential" | "DPM fast" | "DPM adaptive" | "LMS Karras" | "DPM2 Karras" | "DPM2 a Karras" | "DPM++ 2S a Karras" | "Restart" | "DDIM" | "PLMS" | "UniPC";
+      };
+  };
+  /**
+   * Configuration options related to chat functionality.
+   * @param chat.chatModal The chat modal to be used.
+   * @param chat.readFiles Whether to read files.
+   * @param chat.readImages Whether to read images.
+   * @param chat.personality The personality for the chat.
+   * @param chat.API_KEY The API key for accessing the chat service.
+   * @param chat.memory Configuration options for memory.
+   * @param chat.memory.memoryOn Whether memory is enabled.
+   * @param chat.memory.id The ID for memory.
+   * @param chat.typeWriting Configuration options for typing effect.
+   * @param chat.typeWriting.enable Whether the typing effect is enabled.
+   * @param chat.typeWriting.speed The speed of typing.
+   * @param chat.typeWriting.delay The delay for typing.
+   */
+  chat?: {
+      chatModal?: string;
+      readFiles?: boolean;
+      readImages?: boolean;
+      personality?: string | any;
+      API_KEY?: string;
+      memory?: {
+          memoryOn: boolean;
+          id: string;
+      };
+      typeWriting?: {
           enable?: boolean;
           speed?: number;
           delay?: number;
-        };
-    };
-    others?: {
-        messageType?: {
-            type: string;
-            intialContent: string;
-        };
-        keywords?: string[];
-        keywordResponses?: Record<string, string>;
-        loader?: {
-            enable?: boolean;
-            loadingMessage?: string;
-            loadingTimer?: number;
-        };
-        channel?: {
+      };
+  };
+  /**
+   * Additional configuration options.
+   * @param others.messageType Configuration options for message types.
+   * @param others.messageType.type The type of message.
+   * @param others.messageType.intialContent The initial content of the message.
+   * @param others.buttons Buttons configuration.
+   * @param others.keywords Keywords for response.
+   * @param others.keywordResponses Responses for keywords.
+   * @param others.loader Configuration options for loader.
+   * @param others.loader.enable Whether the loader is enabled.
+   * @param others.loader.loadingMessage The loading message.
+   * @param others.loader.loadingTimer The loading timer.
+   * @param others.channel Configuration options for channels.
+   * @param others.channel.enable Whether channels are enabled.
+   * @param others.channel.id The ID of the channels.
+   * @param others.permissions Configuration options for permissions.
+   * @param others.permissions.enable Whether permissions are enabled.
+   * @param others.permissions.role The role for permissions.
+   * @param others.permissions.permission The permission.
+   * @param others.permissions.blockedUsers Blocked users.
+   */
+  others?: {
+      messageType?: {
+          type: string;
+          intialContent: string;
+      };
+      buttons?: any[];
+      keywords?: string[];
+      keywordResponses?: Record<string, string>;
+      loader?: {
+          enable?: boolean;
+          loadingMessage?: string;
+          loadingTimer?: number;
+      };
+      channel?: {
           enable?: boolean;
           id?: string[];
-        };
-        permissions?: {
+      };
+      permissions?: {
           enable?: boolean;
           role?: string[];
           permission?: string[];
           blockedUsers?: string[];
-        };
-    };
+      };
+  };
 }
 
-  type Response = string | { content?: string; reply?: string } | any;
+type Response = string | { content?: string; reply?: string } | any;
+const hercai = new Hercai('6eZZOdDwm6Epdzn8mnhcX9SBDkxvoNYcNj9ILS0P44=');
 
 
 export async function ApexAI (message: any, aiOptions: Options) {
 
-  
   await imageTools(
     message.client,
     ModalBuilder,
@@ -89,11 +175,11 @@ export async function ApexAI (message: any, aiOptions: Options) {
     TextInputStyle,
     ActionRowBuilder
   );
+
   await filters(message.client);
   
     let usermsg: string = ''; 
 
-
     const {
       voice: {
           textVoice: {
@@ -111,14 +197,30 @@ export async function ApexAI (message: any, aiOptions: Options) {
         numOfImages = 2,
         nsfw: {
           enable: nsfwEnabled = false,
-          keywords: nsfwKeyWords = []
+          keywords: nsfwKeyWords = [],
+          deepCheck: deepCheck = false
       } = {}, 
-        enhancer = false
+        enhancer: {
+          enable: enhancerOn = false,
+          enhancerModal: enhanceModal = 'ESRGAN_4x',
+          negative_prompt:  negative_prompt = '',
+          cfg_scale: cfgScale = 7,
+          sampler: sampler = 'DDIM',
+          steps: steps = 20,
+          seed: seed = -1,
+          imgStyle: imgStyle = 'enhance'
+        } = {},
     } = {},
       chat: {
           chatModal = "v3",
           readFiles = false,
           readImages = false,
+          API_KEY = null,
+          personality = null,
+          memory: {
+            memoryOn: memoryOn = false,
+            id: memoryId = ''
+            } = {},
           typeWriting: {
               enable: typeWritingEnable = false,
               speed = 70,
@@ -130,6 +232,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
             type: msgType = 'reply',
             intialContent: msgContent = ''
           } = {},
+          buttons: buttons = [],
           channel: {
               enable: channelEnable = false,
               id: channelIds = []
@@ -151,7 +254,6 @@ export async function ApexAI (message: any, aiOptions: Options) {
   } = aiOptions;
 
     if (permissionEnable) {
-
       if (role.length > 0) { 
           const userRoles = message.member?.roles.cache.map((role: any) => role.id);
           const hasPermission = userRoles.some((roleId: any) => role.includes(roleId));
@@ -213,8 +315,8 @@ export async function ApexAI (message: any, aiOptions: Options) {
     }
   }
 
-     if (aiOptions.chat && readFiles) {
-        if (message.attachments.size > 0) {
+      if (aiOptions.chat && readFiles) {
+         if (message.attachments.size > 0) {
             if (attachment.name.endsWith('.pdf')) {
                 const pdfContent = await readPdf(attachment.url);
                 usermsg += pdfContent;
@@ -222,8 +324,8 @@ export async function ApexAI (message: any, aiOptions: Options) {
                 const txtContent = await readTextFile(attachment.url);
                 usermsg += txtContent;
             }
-        }
-    }
+         }
+      }
 
      if (aiOptions.others?.loader !== null && loaderEnable === true) {
       if (msgType === 'reply') {
@@ -270,10 +372,8 @@ export async function ApexAI (message: any, aiOptions: Options) {
         usermsg = `${usermsg}\n\n Read previous message: ${replied}`;
       }
 
-
       let response: Response = '';
 
-  
       for (const keyword of keywords) {
         if (usermsg.toLowerCase().includes(keyword.toLowerCase())) {
           response = keywordResponses[keyword] || "";
@@ -290,9 +390,10 @@ export async function ApexAI (message: any, aiOptions: Options) {
 
       if (drawValid) {
 
-        if (enhancer) {
-            usermsg += await gemmaAi_4(usermsg)
-          }
+        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, })
+        }
 
         return await aiImagine(
           message,
@@ -301,7 +402,10 @@ export async function ApexAI (message: any, aiOptions: Options) {
           hercai,
           modal,
           nsfwEnabled,
-          nsfwKeyWords
+          nsfwKeyWords,
+          deepCheck,
+          aiOptions.imagine?.enhancer,
+          buttons
         );
 
     } else if (aiOptions.voice) {
@@ -320,9 +424,11 @@ export async function ApexAI (message: any, aiOptions: Options) {
           textVoiceApiKey, 
           textVoiceType,
           nsfwEnabled,
-          nsfwKeyWords
+          nsfwKeyWords,
+          deepCheck,
+          aiOptions.imagine?.enhancer,
+          buttons
         );
-
       }
  
     if (msgType === 'reply') {
@@ -343,20 +449,37 @@ export async function ApexAI (message: any, aiOptions: Options) {
         try { 
         if (chatModal === 'apexChat') {
             response = await apexai(usermsg);
-        } else if (chatModal === 'gemma-v3') { 
-            response = await gemmaAi_3(usermsg);
-        } else if (chatModal === 'gemma-v4') { 
-            response = await gemmaAi_4(usermsg);
+        } else if (chatModal === 'yi-ai') { 
+            response = await yi_34b(usermsg);
+        } else if (chatModal === 'facebook-ai') { 
+            response = await facebook_ai(usermsg);
         } else if (chatModal === 'starChat') {
             response = await starChat(usermsg);
-        } else if (chatModal === 'zephyr-beta') {
-            response = await zephyr_beta(usermsg);
+        } else if (chatModal === 'gemini-flash') {
+            response = await geminiFlash(message, { 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, })
         } else if (chatModal === 'v3' || chatModal === 'v3-32k' || chatModal === 'turbo' || chatModal === 'turbo-16k' || chatModal === 'gemini') {
-                response = await hercai.question({
-                    model: chatModal,
-                    content: usermsg,
-                });
-                response = response.reply
+          if (!memoryOn) {
+
+            const personalityFilePath = path.join(process.cwd(), personality);
+            const personalityContent = fs.readFileSync(personalityFilePath, 'utf-8');
+            const personalityString = personalityContent.split('\n').join(' ');
+            
+            response = await hercai.question({
+                model: chatModal,
+                content: usermsg,
+                personality: personalityString
+            });
+
+            response = response.reply;
+              } else {
+                response = await hercai.betaQuestion({
+                  content: usermsg,
+                  user: memoryId
+              });
+              response = response.reply
+              }
             } else {
                 throw new Error('Invalid chat modal. Check documentation for valid chat modals.')
             }
@@ -364,23 +487,23 @@ export async function ApexAI (message: any, aiOptions: Options) {
           if (msgType === 'reply') {
             if (error.response && error.response.status === 429) {
                 console.error("Too many requests. Please try again later.");
-                return message.reply(`Please wait i am in a cool down for a minute`);
+                return await message.reply({ content: `Please wait i am in a cool down for a minute`, components: buttons });
             } else if (error.response && error.response.status === 500) {
                 console.error("Internal server error. Please try again later.");
-                return message.reply(`Please wait i am in a cool down for a minute`);
+                return await message.reply({ content: `Please wait i am in a cool down for a minute`, components: buttons });
             } else {
-                await message.reply(`Please wait i am in a cool down for a minute`);
                 console.error("The Api is on a cool down for 10 seconds", error.message);
+                return await message.reply({ content: `Please wait i am in a cool down for a minute`, components: buttons });
             }
           } else if (msgType === 'send') {
             if (error.response && error.response.status === 429) {
               console.error("Too many requests. Please try again later.");
-              return message.channel.send(`Please wait i am in a cool down for a minute`);
+              return message.channel.send({ content: `Please wait i am in a cool down for a minute`, components: buttons });
           } else if (error.response && error.response.status === 500) {
               console.error("Internal server error. Please try again later.");
-              return message.channel.send(`Please wait i am in a cool down for a minute`);
+              return message.channel.send({ content: `Please wait i am in a cool down for a minute`, components: buttons });
           } else {
-              await message.channel.send(`Please wait i am in a cool down for a minute`);
+              await message.channel.send({ content: `Please wait i am in a cool down for a minute`, components: buttons });
               console.error("The Api is on a cool down for 10 seconds", error.message);
           }
           }
@@ -389,7 +512,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
       if (msgType === 'reply') {
         if (typeWritingEnable) {
           if (response.length <= 2000) {
-              await typeWriter(message.channel, response, speed, delay);
+              await typeWriter(message.channel, response, speed, delay, buttons);
           } else {
               let parts: string[] = [];
               while (typeof response === 'string' && response.length > 0) {
@@ -402,7 +525,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
                   }
               }
               for (const part of parts) {
-                await typeWriter(message.channel, part, speed, delay);
+                await typeWriter(message.channel, part, speed, delay, buttons);
               }
           }
 
@@ -410,6 +533,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
           if (response.length <= 2000) {
               await message.reply({
                   content: response,
+                  components: buttons,
                   allowedMentions: { repliedUser: false },
               });
           } else {
@@ -426,6 +550,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
               for (const part of parts) {
                   await message.reply({
                       content: part,
+                      components: buttons,
                       allowedMentions: { repliedUser: false },
                   });
               }
@@ -434,7 +559,7 @@ export async function ApexAI (message: any, aiOptions: Options) {
       } else if (msgType === 'send') {
         if (typeWritingEnable) {
             if (response.length <= 2000) {
-                await typeWriter(message.channel, response, speed, delay);
+                await typeWriter(message.channel, response, speed, delay, buttons);
             } else {
                 let parts: string[] = [];
                 while (typeof response === 'string' && response.length > 0) {
@@ -447,13 +572,14 @@ export async function ApexAI (message: any, aiOptions: Options) {
                     }
                 }
                 for (const part of parts) {
-                    await typeWriter(message.channel, part, speed, delay);
+                    await typeWriter(message.channel, part, speed, delay, buttons);
                 }
             }
         } else {
             if (response.length <= 2000) {
                 await message.channel.send({
-                    content: response
+                    content: response,
+                    components: buttons
                 });
             } else {
                 let parts: string[] = [];
@@ -468,116 +594,12 @@ export async function ApexAI (message: any, aiOptions: Options) {
                 }
                 for (const part of parts) {
                     await message.channel.send({
-                        content: part
+                        content: part,
+                        components: buttons
                     });
                 }
             }
         }
-    }    
-  }
-}
-
-export async function gemmaAi_4(prompt: string) {
-    try {
-        const response = await axios.post('https://api-inference.huggingface.co/models/google/gemma-7b-it', { inputs: prompt }, {
-            headers: { 'Authorization': `Bearer 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` }
-        });
-        return response.data[0].generated_text;
-    } catch (error: any) {
-        console.error('Error fetching response:', error.message);
-        return null;
-    }
-}
-
-export async function gemmaAi_3(prompt: string) {
-    try {
-        const response = await axios.post('https://api-inference.huggingface.co/models/google/gemma-2b-it', { inputs: prompt }, {
-            headers: { 'Authorization': `Bearer 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` }
-        });
-        return response.data[0].generated_text;
-    } catch (error: any) {
-        console.error('Error fetching response:', error.message);
-        return null;
-    }
-}
-
-export async function apexai(prompt: string) {
-    try {
-      const messages = [
-        {"role": "user", "content": `${prompt}`}
-      ];
-        const formattedMessages = messages.map(message => `[${message.role}] ${message.content}`).join('\n');
-
-        const response = await axios.post(`https://api-inference.huggingface.co/models/mistralai/Mixtral-8x7B-Instruct-v0.1`, {
-            inputs: formattedMessages
-        }, {
-            headers: {
-                'Authorization': `Bearer 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`,
-                'Content-Type': 'application/json'
-            }
-        });
-
-        const generatedText = response.data[0].generated_text;
-
-        const lines = generatedText.split('\n').slice(1);
-
-        const output = lines.join('\n');
-
-     return output
-    } catch (error: any) {
-      console.error('Error:', error.response.data);
-      return 'Please wait i am on cooldown.'
     }
-}
-
-export async function starChat(prompt: string) {
-  const messages = [{"role":"user","content": `${prompt}`}]
-  
-    try {
-        const response = await axios.post('https://api-inference.huggingface.co/models/HuggingFaceH4/starchat2-15b-v0.1', {
-            inputs: JSON.stringify(messages), 
-        }, {
-            headers: {
-                'Authorization': `Bearer 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`,
-            },
-        });
-
-        const chatbotReply = response.data[0];
-        const chatbotResponseText = chatbotReply.generated_text.replace(/^.*?\n.*?\n/, '');
-        const chatbotResponseArray = JSON.parse(chatbotResponseText);
-        const chatbotResponseString = chatbotResponseArray.join(' ');
-
-        return chatbotResponseString;
-    } catch (error: any) {
-        console.error('Error fetching response:', error.message);
-        return null;
-    }
-}
-
-export async function zephyr_beta(prompt: string) {
-
-  const messages = [{"role":"user","content": `${prompt}` }]
-    try {
-        const response = await axios.post('https://api-inference.huggingface.co/models/HuggingFaceH4/zephyr-7b-beta', {
-            inputs: JSON.stringify(messages), 
-        }, {
-            headers: {
-                'Authorization': `Bearer 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`,
-            },
-        });
-
-        const chatbotReply = response.data[0];
-
-        const textParts = chatbotReply.generated_text.split('\n');
-
-        const secondArrayString = textParts[2];
-        const chatbotResponseArray = JSON.parse(secondArrayString);
-
-        const chatbotResponseString = chatbotResponseArray.map((obj: any) => obj.content).join(' ');
-
-        return chatbotResponseString;
-    } catch (error: any) {
-        console.error('Error fetching response:', error.message);
-        return null;
-    }
-}
+  }
+};

package/lib/ai/{models.ts → direct-use.ts}

@@ -3,8 +3,8 @@ import axios from 'axios';
 import api  from "api";
  const sdk = api("@prodia/v1.3.0#be019b2kls0gqss3");
 sdk.auth("43435e1c-cab1-493f-a224-f51e4b97ce8d");
-const herc = new Hercai();
-import { apexai, starChat, gemmaAi_3, gemmaAi_4, zephyr_beta } from './ApexAI';
+const hercai = new Hercai('6eZZOdDwm6Epdzn8mnhcX9SBDkxvoNYcNj9ILS0P44=');
+import { apexai, starChat, facebook_ai, yi_34b } from './modals-chat/modals';
 interface ApexImagineOptions {
 negative?: string;
 number?: number;
@@ -100,7 +100,7 @@ const validProdiaModalsP = [
 if (allowedModelsH.includes(model)) {
     for (let i = 0; i < count; i++) {
         try {
-            const result = await herc.drawImage({ model: model as ImagineModelOption, prompt: prompt, negative_prompt: neg });
+            const result = await hercai.drawImage({ model: model as ImagineModelOption, prompt: prompt, negative_prompt: neg });
             resultUrls.push(result.url);
         } catch (error) {
             console.error("Failed to draw image with Hercai. Retrying...", error);
@@ -513,22 +513,19 @@ async function processChunk(model: string, prompt: string): Promise<string> {
     case 'turbo':
     case 'turbo-16k':
     case 'gemini':
-      response = (await herc.question({ model: model as ChatModelOption, content: prompt })).reply;
+      response = (await hercai.question({ model: model as ChatModelOption, content: prompt })).reply;
       break;
     case 'apexChat':
       response = await apexai(prompt);
       break;
-    case 'gemma-v3':
-      response = await gemmaAi_3(prompt);
-      break;
-    case 'gemma-v4':
-      response = await gemmaAi_4(prompt);
-      break;
     case 'starChat':
       response = await starChat(prompt);
       break;
-    case 'zephyr-beta':
-      response = await zephyr_beta(prompt);
+      case 'facebook-ai':
+        response = await facebook_ai(prompt);
+      break;
+      case 'yi-ai':
+        response = await yi_34b(prompt);
       break;
     default:
       throw new Error('Invalid model.');