feat: add vicuna 7b & 13b from LMSYS

18533930525 · Sep 18, 2023 · 70ea7a7 · 70ea7a7
1 parent 3aa0ec8
commit 70ea7a7
Show file tree

Hide file tree

Showing 9 changed files with 35 additions and 1 deletion.
diff --git a/public/bots/vicuna-13b-logo.png b/public/bots/vicuna-13b-logo.png
diff --git a/public/bots/vicuna-33b-logo.png b/public/bots/vicuna-33b-logo.png
diff --git a/public/bots/vicuna-7b-logo.png b/public/bots/vicuna-7b-logo.png
diff --git a/src/bots/index.js b/src/bots/index.js
@@ -44,6 +44,8 @@ import Llama270bBot from "./lmsys/Llama270bBot";
 import Falcon180bBot from "./huggingface/Falcon180bBot";
 import ChatGLM6bBot from "./lmsys/ChatGLM6bBot";
 import CodeLlamaBot from "./lmsys/CodeLlamaBot";
+import Vicuna7bBot from "./lmsys/Vicuna7bBot";
+import Vicuna13bBot from "./lmsys/Vicuna13bBot";
 
 const all = [
   Qihoo360AIBrainBot.getInstance(),
@@ -88,6 +90,8 @@ const all = [
   SageBot.getInstance(),
   SkyWorkBot.getInstance(),
   SparkBot.getInstance(),
+  Vicuna7bBot.getInstance(),
+  Vicuna13bBot.getInstance(),
   VicunaBot.getInstance(),
   YouChatBot.getInstance(),
 ];
@@ -138,6 +142,8 @@ export const botTags = {
     bots.getBotByClassName("GradioAppBot"),
     bots.getBotByClassName("AlpacaBot"),
     bots.getBotByClassName("VicunaBot"),
+    bots.getBotByClassName("Vicuna7bBot"),
+    bots.getBotByClassName("Vicuna13bBot"),
     bots.getBotByClassName("CharacterAIBot"),
     bots.getBotByClassName("ClaudeAIBot"),
     bots.getBotByClassName("PiBot"),
@@ -163,6 +169,8 @@ export const botTags = {
     bots.getBotByClassName("MOSSBot"),
     bots.getBotByClassName("OpenAssistantBot"),
     bots.getBotByClassName("VicunaBot"),
+    bots.getBotByClassName("Vicuna7bBot"),
+    bots.getBotByClassName("Vicuna13bBot"),
     bots.getBotByClassName("Falcon180bBot"),
     bots.getBotByClassName("ChatGLM6bBot"),
     bots.getBotByClassName("CodeLlamaBot"),

diff --git a/src/bots/lmsys/Vicuna13bBot.js b/src/bots/lmsys/Vicuna13bBot.js
@@ -0,0 +1,12 @@
+import LMSYSBot from "./LMSYSBot";
+
+export default class Vicuna13bBot extends LMSYSBot {
+  static _brandId = "lmsys"; // Brand id of the bot, should be unique. Used in i18n.
+  static _className = "Vicuna13bBot"; // Class name of the bot
+  static _logoFilename = "vicuna-13b-logo.png"; // Place it in public/bots/
+  static _model = "vicuna-13b";
+
+  constructor() {
+    super();
+  }
+}
diff --git a/src/bots/lmsys/Vicuna7bBot.js b/src/bots/lmsys/Vicuna7bBot.js
@@ -0,0 +1,12 @@
+import LMSYSBot from "./LMSYSBot";
+
+export default class Vicuna7bBot extends LMSYSBot {
+  static _brandId = "lmsys"; // Brand id of the bot, should be unique. Used in i18n.
+  static _className = "Vicuna7bBot"; // Class name of the bot
+  static _logoFilename = "vicuna-7b-logo.png"; // Place it in public/bots/
+  static _model = "vicuna-7b";
+
+  constructor() {
+    super();
+  }
+}
diff --git a/src/bots/lmsys/VicunaBot.js b/src/bots/lmsys/VicunaBot.js
@@ -3,7 +3,7 @@ import LMSYSBot from "./LMSYSBot";
 export default class VicunaBot extends LMSYSBot {
   static _brandId = "lmsys"; // Brand id of the bot, should be unique. Used in i18n.
   static _className = "VicunaBot"; // Class name of the bot
-  static _logoFilename = "vicuna-logo.jpeg"; // Place it in public/bots/
+  static _logoFilename = "vicuna-33b-logo.png"; // Place it in public/bots/
   static _model = "vicuna-33b";
 
   constructor() {

diff --git a/src/i18n/locales/en.json b/src/i18n/locales/en.json
@@ -144,6 +144,7 @@
   },
   "lmsys": {
     "name": "LMSYS",
+    "vicuna-7b": "vicuna-7b",
     "vicuna-13b": "vicuna-13b",
     "vicuna-33b": "vicuna-33b",
     "chatglm-6b": "chatglm-6b",

diff --git a/src/i18n/locales/zh.json b/src/i18n/locales/zh.json
@@ -144,6 +144,7 @@
   },
   "lmsys": {
     "name": "LMSYS",
+    "vicuna-7b": "vicuna-7b",
     "vicuna-13b": "vicuna-13b",
     "vicuna-33b": "vicuna-33b",
     "chatglm-6b": "chatglm-6b",