From ad8e3115db8c409a702928a3b372f1cbc45ef497 Mon Sep 17 00:00:00 2001
From: Danil Nikolaev <lischenkodev@gmail.com>
Date: Fri, 16 Jan 2026 19:03:21 +0300
Subject: [PATCH] add gemini and mistral models to environment

---
 src/commands/gemini-chat.ts  |  2 +-
 src/commands/mistral-chat.ts |  2 +-
 src/common/environment.ts    | 28 ++++++++++++++++++----------
 3 files changed, 20 insertions(+), 12 deletions(-)

diff --git a/src/commands/gemini-chat.ts b/src/commands/gemini-chat.ts
index 3dcf736..3c836eb 100644
--- a/src/commands/gemini-chat.ts
+++ b/src/commands/gemini-chat.ts
@@ -68,7 +68,7 @@ export class GeminiChat extends ChatCommand {
             });
 
             const stream = await this.googleAi.models.generateContentStream({
-                model: "gemini-2.5-flash",
+                model: Environment.GEMINI_MODEL || "gemini-2.5-flash",
                 contents: chatContent,
             });
 
diff --git a/src/commands/mistral-chat.ts b/src/commands/mistral-chat.ts
index f36f3a6..f3ccf71 100644
--- a/src/commands/mistral-chat.ts
+++ b/src/commands/mistral-chat.ts
@@ -103,7 +103,7 @@ export class MistralChat extends ChatCommand {
             });
 
             const stream = await this.mistralAi.chat.stream({
-                model: "mistral-small-latest",
+                model: Environment.MISTRAL_MODEL || "mistral-small-latest",
                 messages: chatMessages as any
             });
 
diff --git a/src/common/environment.ts b/src/common/environment.ts
index 8791f37..cade4de 100644
--- a/src/common/environment.ts
+++ b/src/common/environment.ts
@@ -19,15 +19,19 @@ export class Environment {
 
     static USE_NAMES_IN_PROMPT: boolean;
 
-    static OLLAMA_MODEL?: string;
-    static OLLAMA_ADDRESS?: string;
-    static OLLAMA_API_KEY?: string;
-    static SYSTEM_PROMPT?: string;
-
     static MAX_PHOTO_SIZE: number;
 
+    static SYSTEM_PROMPT?: string;
+
+    static OLLAMA_ADDRESS?: string;
+    static OLLAMA_MODEL?: string;
+    static OLLAMA_API_KEY?: string;
+
     static GEMINI_API_KEY?: string;
+    static GEMINI_MODEL?: string;
+
     static MISTRAL_API_KEY?: string;
+    static MISTRAL_MODEL?: string;
 
     static waitText = "⏳ Дайте-ка подумать...";
 
@@ -47,15 +51,19 @@ export class Environment {
 
         Environment.USE_NAMES_IN_PROMPT = process.env.USE_NAMES_IN_PROMPT == "true";
 
-        Environment.OLLAMA_MODEL = process.env.OLLAMA_MODEL;
-        Environment.OLLAMA_ADDRESS = process.env.OLLAMA_ADDRESS;
-        Environment.OLLAMA_API_KEY = process.env.OLLAMA_API_KEY;
-        Environment.SYSTEM_PROMPT = process.env.SYSTEM_PROMPT?.trim();
-
         Environment.MAX_PHOTO_SIZE = Number(process.env.MAX_PHOTO_SIZE || "1280");
 
+        Environment.SYSTEM_PROMPT = process.env.SYSTEM_PROMPT?.trim();
+
+        Environment.OLLAMA_ADDRESS = process.env.OLLAMA_ADDRESS;
+        Environment.OLLAMA_MODEL = process.env.OLLAMA_MODEL;
+        Environment.OLLAMA_API_KEY = process.env.OLLAMA_API_KEY;
+
         Environment.GEMINI_API_KEY = process.env.GEMINI_API_KEY;
+        Environment.GEMINI_MODEL = process.env.GEMINI_MODEL;
+
         Environment.MISTRAL_API_KEY = process.env.MISTRAL_API_KEY;
+        Environment.MISTRAL_MODEL = process.env.MISTRAL_MODEL;
     }
 
     static setAdmins(admins: Set<number>) {