feat: add ChatGLM3-6b (#637)

* 支持ChatCLM3-6b * add chatglm3-6b to en * update chatglm logo * update chatglm logo js --------- Co-authored-by: yaoyilin <[email protected]>
ai-shifu · Nov 27, 2023 · b16f8e2 · b16f8e2 · vercel · Nov 27, 2023
1 parent 72aad41
commit b16f8e2
Show file tree

Hide file tree

Showing 8 changed files with 20 additions and 1 deletion.
diff --git a/public/bots/chatglm-logo.png b/public/bots/chatglm-logo.png
diff --git a/public/bots/chatglm2-logo.png b/public/bots/chatglm2-logo.png
diff --git a/public/bots/chatglm3-logo.png b/public/bots/chatglm3-logo.png
diff --git a/src/bots/index.js b/src/bots/index.js
@@ -44,6 +44,7 @@ import Llama27bBot from "./lmsys/Llama27bBot";
 import Llama270bBot from "./lmsys/Llama270bBot";
 import Falcon180bBot from "./huggingface/Falcon180bBot";
 import ChatGLM6bBot from "./lmsys/ChatGLM6bBot";
+import ChatGLM36bBot from "./lmsys/ChatGLM36bBot";
 import CodeLlamaBot from "./lmsys/CodeLlamaBot";
 import Vicuna7bBot from "./lmsys/Vicuna7bBot";
 import Vicuna13bBot from "./lmsys/Vicuna13bBot";
@@ -65,6 +66,7 @@ const all = [
   CharacterAIBot.getInstance(),
   ChatGLMBot.getInstance(),
   ChatGLM6bBot.getInstance(),
+  ChatGLM36bBot.getInstance(),
   ClaudeInstantPoeBot.getInstance(),
   ClaudeInstant100kPoeBot.getInstance(),
   ClaudeBot.getInstance(),
@@ -136,6 +138,7 @@ export const botTags = {
     bots.getBotByClassName("BingChatPreciseBot"),
     bots.getBotByClassName("ChatGLMBot"),
     bots.getBotByClassName("ChatGLM6bBot"),
+    bots.getBotByClassName("ChatGLM36bBot"),
     bots.getBotByClassName("ChatGPT35Bot"),
     bots.getBotByClassName("ChatGPT35PoeBot"),
     bots.getBotByClassName("ClaudeBot"),
@@ -193,6 +196,7 @@ export const botTags = {
     bots.getBotByClassName("Vicuna13bBot"),
     bots.getBotByClassName("Falcon180bBot"),
     bots.getBotByClassName("ChatGLM6bBot"),
+    bots.getBotByClassName("ChatGLM36bBot"),
     bots.getBotByClassName("CodeLlamaBot"),
     bots.getBotByClassName("CodeLlamaHCBot"),
     bots.getBotByClassName("Wizardlm13bBot"),
@@ -220,6 +224,7 @@ export const botTags = {
     bots.getBotByClassName("MOSSBot"),
     bots.getBotByClassName("ChatGLMBot"),
     bots.getBotByClassName("ChatGLM6bBot"),
+    bots.getBotByClassName("ChatGLM36bBot"),
   ],
 };
 export default bots;
diff --git a/src/bots/lmsys/ChatGLM36bBot.js b/src/bots/lmsys/ChatGLM36bBot.js
@@ -0,0 +1,12 @@
+import LMSYSBot from "./LMSYSBot";
+
+export default class ChatGLM36bBot extends LMSYSBot {
+  static _brandId = "lmsys"; // Brand id of the bot, should be unique. Used in i18n.
+  static _className = "ChatGLM36bBot"; // Class name of the bot
+  static _logoFilename = "chatglm3-logo.png"; // Place it in public/bots/
+  static _model = "chatglm3-6b";
+
+  constructor() {
+    super();
+  }
+}
diff --git a/src/bots/lmsys/ChatGLM6bBot.js b/src/bots/lmsys/ChatGLM6bBot.js
@@ -3,7 +3,7 @@ import LMSYSBot from "./LMSYSBot";
 export default class ChatGLM6bBot extends LMSYSBot {
   static _brandId = "lmsys"; // Brand id of the bot, should be unique. Used in i18n.
   static _className = "ChatGLM6bBot"; // Class name of the bot
-  static _logoFilename = "chatglm-logo.svg"; // Place it in public/bots/
+  static _logoFilename = "chatglm2-logo.png"; // Place it in public/bots/
   static _model = "chatglm2-6b";
 
   constructor() {

diff --git a/src/i18n/locales/en.json b/src/i18n/locales/en.json
@@ -164,6 +164,7 @@
     "vicuna-33b": "vicuna-33b",
     "chatglm-6b": "chatglm-6b",
     "chatglm2-6b": "chatglm2-6b",
+    "chatglm3-6b": "chatglm3-6b",
     "alpaca-13b": "alpaca-13b",
     "claude-1": "claude-1",
     "wizardlm-13b": "WizardLM-13B",

diff --git a/src/i18n/locales/zh.json b/src/i18n/locales/zh.json
@@ -164,6 +164,7 @@
     "vicuna-33b": "vicuna-33b",
     "chatglm-6b": "chatglm-6b",
     "chatglm2-6b": "chatglm2-6b",
+    "chatglm3-6b": "chatglm3-6b",
     "alpaca-13b": "alpaca-13b",
     "claude-1": "claude-1",
     "wizardlm-13b": "WizardLM-13B",