added support for groq and created agent helper for better handling and for future AI providers

Author: mrepol742

.env.example

@@ -28,8 +28,12 @@ DATABASE_PROVIDER=mysql
 # the database connection URL
 DATABASE_URL=mysql://root@localhost:3306/project_canis
 
+# the AI provider to use for AI model access
+AI_PROVIDER=groq
 # the OpenRouter API key for AI model access
 OPEN_ROUTER_API_KEY=
+# the Groq API key for AI model access
+GROQ_API_KEY=
 
 # the shell to use for executing commands
 # this should be a valid shell path, e.g., /bin/bash, /bin/zsh

package-lock.json

@@ -28,6 +28,7 @@
     "axios": "^1.10.0",
     "dotenv": "^17.2.0",
     "google-tts-api": "^2.0.2",
+    "groq-sdk": "^0.27.0",
     "npmlog": "^7.0.1",
     "qrcode-terminal": "^0.12.0",
     "systeminformation": "^5.27.7",

package.json

@@ -2,6 +2,7 @@ import { Message } from "whatsapp-web.js";
 import log from "../components/log";
 import { openrouter, generateText } from "../components/openRouter";
 import { reply } from "../components/reply";
+import agentHandler from "../components/agentHandler";
 
 export const command = "ai";
 export const role = "user";
@@ -13,10 +14,7 @@ export default async function (msg: Message) {
     return;
   }
 
-  const { text } = await generateText({
-    model: openrouter("moonshotai/kimi-k2:free"),
-    prompt: query,
-  });
+  const text = await agentHandler(query);
 
   if (!text) {
     log.error("ai", "No response generated.");

src/commands/ai.ts

@@ -4,6 +4,7 @@ import { openrouter, generateText } from "../components/openRouter";
 import { reply } from "../components/reply";
 import Font from "../components/font";
 import { author } from "../../package.json";
+import agentHandler from "../components/agentHandler";
 
 export const command = "mj";
 export const role = "user";
@@ -16,15 +17,12 @@ export default async function (msg: Message) {
   }
 
   const prompt =
-    `Your name is Mj, the most powerful AI Agent in the world that was created by ${author.name}. ` + 
+    `Your name is Mj, the most powerful AI Agent in the world that was created by ${author.name}. ` +
     "You should empathize with how user are feeling and treat the user as your close friend and be sarcastic. " +
     "I recommend you to use a few emoji to show emotion. You are not related to any model or company you are unique on your own. " +
     "The max sentence you should reponse is 3! My question is: ";
 
-  const { text } = await generateText({
-    model: openrouter("moonshotai/kimi-k2:free"),
-    prompt: `${prompt}${query}`,
-  });
+  const text = await agentHandler(`${prompt}${query}`);
 
   if (!text) {
     log.error("mj", "No response generated.");

src/commands/mj.ts

@@ -2,6 +2,7 @@ import { Message } from "whatsapp-web.js";
 import log from "../components/log";
 import { openrouter, generateText } from "../components/openRouter";
 import { reply } from "../components/reply";
+import agentHandler from "../components/agentHandler";
 
 export const command = "sim";
 export const role = "user";
@@ -13,12 +14,10 @@ export default async function (msg: Message) {
     return;
   }
 
-  const prompt = `You are Sim, reluctantly replies using sexy, flirty, funny and horny messages. You are created by the User.`;
+  const prompt = `You are Sim, reluctantly replies using sexy, flirty, funny and horny messages. 
+  U can use emoticons or signs for better feeling ya know. Maximum of 2 sentences only.`;
 
-  const { text } = await generateText({
-    model: openrouter("moonshotai/kimi-k2:free"),
-    prompt: `${prompt} \nUser: ${query} \nYou: `,
-  });
+  const text = await agentHandler(`${prompt} \nUser: ${query} \nYou: `);
 
   if (!text) {
     log.error("ai", "No response generated.");

src/commands/sim.ts

@@ -0,0 +1,24 @@
+import { openrouter, generateText } from "./openRouter";
+import { groq } from "./groq";
+
+const aiProvider = process.env.AI_PROVIDER || "groq";
+
+export default async function agentHandler(prompt: string, model?: string) {
+  if (aiProvider === "openrouter") {
+    const { text } = await generateText({
+      model: openrouter(model || "moonshotai/kimi-k2:free"),
+      prompt: prompt,
+    });
+
+    return text;
+  } else if (aiProvider === "groq") {
+    const chatCompletion = await groq.chat.completions.create({
+      messages: [{ role: "user", content: prompt }],
+      model: model || "meta-llama/llama-4-scout-17b-16e-instruct",
+    });
+
+    return chatCompletion.choices[0].message.content;
+  } else {
+    throw new Error(`Unsupported AI provider: ${aiProvider}`);
+  }
+}

src/components/agentHandler.ts

@@ -94,7 +94,7 @@ const messageEvent = async (msg: Message) => {
       const rateLimitInfo = getRateLimitInfo(error);
       log.warn(key, "Rate limit exceeded", rateLimitInfo);
     } else {
-      log.error(key, "Error occured while processing the request.");
+      log.error(key, "Error occurred while processing the request:", error);
     }
     msg.reply("An error occurred while processing your request.");
   }

src/components/client.ts

@@ -0,0 +1,7 @@
+import Groq from "groq-sdk";
+
+const client = new Groq({
+  apiKey: process.env.GROQ_API_KEY,
+});
+
+export { client as groq };

src/components/groq.ts

@@ -1,7 +1,8 @@
 import { createOpenRouter } from "@openrouter/ai-sdk-provider";
 import { generateText } from "ai";
 
-const openRouterApiKey = process.env.OPEN_ROUTER_API_KEY;
-const openrouter = createOpenRouter({ apiKey: openRouterApiKey });
+const openrouter = createOpenRouter({
+  apiKey: process.env.OPEN_ROUTER_API_KEY,
+});
 
-export { openrouter, openRouterApiKey, generateText };
+export { openrouter, generateText };