From aa56baad377adc41e32f10b8dc7547fbed1363d9 Mon Sep 17 00:00:00 2001
From: Roman Rizzi <roman@discourse.org>
Date: Thu, 4 Jan 2024 12:22:43 -0300
Subject: [PATCH] FEATURE: Add Mixtral support for AI Bot (#396)

---
 assets/javascripts/initializers/ai-bot-replies.js | 2 +-
 config/locales/client.en.yml                      | 2 ++
 config/settings.yml                               | 1 +
 lib/ai_bot/bot.rb                                 | 2 ++
 lib/ai_bot/entry_point.rb                         | 4 ++++
 5 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/assets/javascripts/initializers/ai-bot-replies.js b/assets/javascripts/initializers/ai-bot-replies.js
index 050e1edc..5ea5e359 100644
--- a/assets/javascripts/initializers/ai-bot-replies.js
+++ b/assets/javascripts/initializers/ai-bot-replies.js
@@ -12,7 +12,7 @@ import copyConversation from "../discourse/lib/copy-conversation";
 const AUTO_COPY_THRESHOLD = 4;
 
 function isGPTBot(user) {
-  return user && [-110, -111, -112, -113].includes(user.id);
+  return user && [-110, -111, -112, -113, -114].includes(user.id);
 }
 
 function attachHeaderIcon(api) {
diff --git a/config/locales/client.en.yml b/config/locales/client.en.yml
index 97ecf500..f46df038 100644
--- a/config/locales/client.en.yml
+++ b/config/locales/client.en.yml
@@ -197,6 +197,8 @@ en:
           gpt-3:
             5-turbo: "GPT-3.5"
           claude-2: "Claude 2"
+          mixtral-8x7B-Instruct-V0:
+            "1": "Mixtral-8x7B V0.1"
       sentiments:
         dashboard:
           title: "Sentiment"
diff --git a/config/settings.yml b/config/settings.yml
index 2cd9846b..2c7a2367 100644
--- a/config/settings.yml
+++ b/config/settings.yml
@@ -274,6 +274,7 @@ discourse_ai:
       - gpt-4
       - gpt-4-turbo
       - claude-2
+      - mixtral-8x7B-Instruct-V0.1
   ai_bot_add_to_header:
     default: true
     client: true
diff --git a/lib/ai_bot/bot.rb b/lib/ai_bot/bot.rb
index f0d4178b..ecdede94 100644
--- a/lib/ai_bot/bot.rb
+++ b/lib/ai_bot/bot.rb
@@ -123,6 +123,8 @@ module DiscourseAi
             "gpt-4-turbo"
           when DiscourseAi::AiBot::EntryPoint::GPT3_5_TURBO_ID
             "gpt-3.5-turbo-16k"
+          when DiscourseAi::AiBot::EntryPoint::MIXTRAL_ID
+            "mistralai/Mixtral-8x7B-Instruct-v0.1"
           else
             nil
           end
diff --git a/lib/ai_bot/entry_point.rb b/lib/ai_bot/entry_point.rb
index 0567490e..eaa750b3 100644
--- a/lib/ai_bot/entry_point.rb
+++ b/lib/ai_bot/entry_point.rb
@@ -9,11 +9,13 @@ module DiscourseAi
       GPT3_5_TURBO_ID = -111
       CLAUDE_V2_ID = -112
       GPT4_TURBO_ID = -113
+      MIXTRAL_ID = -114
       BOTS = [
         [GPT4_ID, "gpt4_bot", "gpt-4"],
         [GPT3_5_TURBO_ID, "gpt3.5_bot", "gpt-3.5-turbo"],
         [CLAUDE_V2_ID, "claude_bot", "claude-2"],
         [GPT4_TURBO_ID, "gpt4t_bot", "gpt-4-turbo"],
+        [MIXTRAL_ID, "mixtral_bot", "mixtral-8x7B-Instruct-V0.1"],
       ]
 
       def self.map_bot_model_to_user_id(model_name)
@@ -26,6 +28,8 @@ module DiscourseAi
           GPT4_ID
         in "claude-2"
           CLAUDE_V2_ID
+        in "mixtral-8x7B-Instruct-V0.1"
+          MIXTRAL_ID
         else
           nil
         end