AI Chat: Add mixtral model

brave · Jan 9, 2024 · f217fb1 · f217fb1
1 parent c74b8b7
commit f217fb1
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 7 deletions.
diff --git a/components/ai_chat/core/browser/models.cc b/components/ai_chat/core/browser/models.cc
@@ -37,20 +37,25 @@ const base::flat_map<std::string_view, mojom::Model> kAllModels = {
      {"chat-basic", "llama-2-13b-chat", "llama2 13b", "Meta",
       mojom::ModelEngineType::LLAMA_REMOTE, mojom::ModelCategory::CHAT,
       mojom::ModelAccess::BASIC, 9000, 9700}},
-    {"chat-leo-expanded",
-     {"chat-leo-expanded", "llama-2-70b-chat", "llama2 70b", "Meta",
-      mojom::ModelEngineType::LLAMA_REMOTE, mojom::ModelCategory::CHAT,
-      mojom::ModelAccess::PREMIUM, 9000, 9700}},
+    // {"chat-leo-expanded",
+    //  {"chat-leo-expanded", "llama-2-70b-chat", "llama2 70b", "Meta",
+    //   mojom::ModelEngineType::LLAMA_REMOTE, mojom::ModelCategory::CHAT,
+    //   mojom::ModelAccess::PREMIUM, 9000, 9700}},
     {"chat-claude-instant",
      {"chat-claude-instant", "claude-instant-v1", "Claude Instant", "Anthropic",
       mojom::ModelEngineType::CLAUDE_REMOTE, mojom::ModelCategory::CHAT,
       mojom::ModelAccess::BASIC_AND_PREMIUM, 200000, 320000}},
+    {"chat-leo-expanded",
+     {"chat-leo-expanded", "mixtral-8x7b-instruct", "Mixtral", "Mistral AI",
+      mojom::ModelEngineType::LLAMA_REMOTE, mojom::ModelCategory::CHAT,
+      mojom::ModelAccess::BASIC_AND_PREMIUM, 9000, 9700}},
 };
 
 const std::vector<std::string_view> kAllModelKeysDisplayOrder = {
+    // "chat-mixtral-basic"
+    "chat-leo-expanded",
     "chat-claude-instant",
     "chat-basic",
-    "chat-leo-expanded",
 };
 
 }  // namespace ai_chat
diff --git a/components/ai_chat/core/common/features.cc b/components/ai_chat/core/common/features.cc
@@ -22,7 +22,7 @@ BASE_FEATURE(kAIChat,
 #endif
 );
 const base::FeatureParam<std::string> kAIModelsDefaultKey{&kAIChat, "default_model",
-                                                   "chat-claude-instant"};
+                                                   "chat-leo-expanded"};
 const base::FeatureParam<std::string> kAIModelsPremiumDefaultKey{&kAIChat, "default_premium_model",
                                                    "chat-claude-instant"};
 const base::FeatureParam<bool> kAIChatSSE{&kAIChat, "ai_chat_sse", true};

diff --git a/components/resources/ai_chat_ui_strings.grdp b/components/resources/ai_chat_ui_strings.grdp
@@ -55,7 +55,7 @@
     Hi, I'm Leo. I'm a fully hosted AI assistant by Brave. I'm powered by Llama 13B, a model created by Meta to be performant and applicable to many use cases.
   </message>
   <message name="IDS_CHAT_UI_INTRO_MESSAGE_CHAT_LEO_EXPANDED" desc="AI Chat intro message for the expanded model">
-    Hi, I'm Leo. I'm a fully hosted AI assistant by Brave. I'm powered by Llama 70B, a model created by Meta to handle more advanced tasks than 13B.
+    Hi, I'm Leo. I'm a fully hosted AI assistant by Brave. I'm powered by Mixtral 7B, a model created by Mistral AI to handle advanced tasks.
   </message>
   <message name="IDS_CHAT_UI_INTRO_MESSAGE_CHAT_LEO_CLAUDE_INSTANT" desc="AI Chat intro message for the Claude Instant model">
     Hi, I'm Leo. I'm proxied by Brave and powered by Claude Instant, a model created by Anthropic to power conversational and text processing tasks.