From c7a0becf29c79efa7e460defd0cc7178c7ecedc9 Mon Sep 17 00:00:00 2001
From: Sunner Sun <sunner@gmail.com>
Date: Mon, 1 May 2023 22:03:26 +0800
Subject: [PATCH] Fix broken availability dialog in release-packages

---
 src/App.vue                     | 10 +++++-----
 src/bots/BardBot.js             |  1 +
 src/bots/BingChatBalancedBot.js |  1 +
 src/bots/BingChatBot.js         |  1 +
 src/bots/BingChatCreativeBot.js |  1 +
 src/bots/BingChatPreciseBot.js  |  1 +
 src/bots/Bot.js                 |  5 +++--
 src/bots/ChatGPT35Bot.js        |  1 +
 src/bots/ChatGPT4Bot.js         |  1 +
 src/bots/ChatGPTBot.js          |  1 +
 src/bots/ERNIEBot.js            |  1 +
 src/bots/OpenAIAPI35Bot.js      |  1 +
 src/bots/OpenAIAPI4Bot.js       |  1 +
 src/bots/OpenAIAPIBot.js        |  1 +
 src/bots/SparkBot.js            |  1 +
 15 files changed, 21 insertions(+), 7 deletions(-)

diff --git a/src/App.vue b/src/App.vue
index 942d78a615..332eb36a1d 100644
--- a/src/App.vue
+++ b/src/App.vue
@@ -59,7 +59,7 @@
             <div class="bot-logos margin-bottom">
                 <img
                     v-for="(bot, index) in bots"
-                    :class="{ 'selected': activeBots[bot.constructor.name] }"
+                    :class="{ 'selected': activeBots[bot.constructor._className] }"
                     :key="index"
                     :src="bot.getLogo()"
                     :alt="bot.getFullname()"
@@ -138,7 +138,7 @@ export default {
 
             // Send the prompt to all the bots and update the message with the response
             for (const bot of this.bots) {
-                if (!this.activeBots[bot.constructor.name])
+                if (!this.activeBots[bot.constructor._className])
                     continue;
 
                 var message = {
@@ -161,7 +161,7 @@ export default {
         ...mapMutations(["changeColumns"]),
         ...mapMutations(["SET_BOT_SELECTED"]),
         toggleSelected(bot) {
-            const botId = bot.constructor.name;
+            const botId = bot.constructor._className;
             var selected = false;
             if (!bot.isAvailable()) {
                 this.clickedBot = bot;
@@ -169,14 +169,14 @@ export default {
                 this.$refs.makeAvailableModal.open();
                 selected = true;
             } else {
-                selected = !this.selectedBots[bot.constructor.name];
+                selected = !this.selectedBots[bot.constructor._className];
             }
             this.SET_BOT_SELECTED({ botId, selected});
             this.updateActiveBots();
         },
         updateActiveBots() {
             for (const bot of this.bots) {
-                this.activeBots[bot.constructor.name] = bot.isAvailable() && this.selectedBots[bot.constructor.name];
+                this.activeBots[bot.constructor._className] = bot.isAvailable() && this.selectedBots[bot.constructor._className];
             }
         },
         async checkAllBotsAvailability(specifiedBot) {
diff --git a/src/bots/BardBot.js b/src/bots/BardBot.js
index 87a5750e5e..a897020350 100644
--- a/src/bots/BardBot.js
+++ b/src/bots/BardBot.js
@@ -29,6 +29,7 @@ function parseBartResponse(resp) {
 
 export default class BardBot extends Bot {
   static _brandId = "bard";
+  static _className = "BardBot"; // Class name of the bot
   static _logoFilename = "bard-logo.svg"; // Place it in assets/bots/
   static _loginUrl = "https://bard.google.com/";
 
diff --git a/src/bots/BingChatBalancedBot.js b/src/bots/BingChatBalancedBot.js
index caca95608e..d1e97c79a5 100644
--- a/src/bots/BingChatBalancedBot.js
+++ b/src/bots/BingChatBalancedBot.js
@@ -3,6 +3,7 @@ import AsyncLock from "async-lock";
 import BingChatBot from "./BingChatBot";
 
 export default class BingChatBalancedBot extends BingChatBot {
+  static _className = "BingChatBalancedBot"; // Class name of the bot
   static _logoFilename = "bing-balanced-logo.png"; // Place it in assets/bots/
   static _model = "harmonyv3"; // Bing styles: h3imaginative, harmonyv3, h3precise
   static _lock = new AsyncLock(); // Must process requests in queue
diff --git a/src/bots/BingChatBot.js b/src/bots/BingChatBot.js
index e866d0edd2..065df7c79f 100644
--- a/src/bots/BingChatBot.js
+++ b/src/bots/BingChatBot.js
@@ -15,6 +15,7 @@ function randomIP() {
 }
 export default class BingChatBot extends Bot {
   static _brandId = "bingChat";
+  static _className = "BingChatBot"; // Class name of the bot
   static _model = "h3precise"; // Bing styles: h3imaginative, harmonyv3, h3precise
   static _logoFilename = "bing-logo.svg"; // Place it in assets/bots/
   static _loginUrl = "https://www.bing.com/new";
diff --git a/src/bots/BingChatCreativeBot.js b/src/bots/BingChatCreativeBot.js
index 0038c40e80..510f485c95 100644
--- a/src/bots/BingChatCreativeBot.js
+++ b/src/bots/BingChatCreativeBot.js
@@ -3,6 +3,7 @@ import AsyncLock from "async-lock";
 import BingChatBot from "./BingChatBot";
 
 export default class BingChatCreativeBot extends BingChatBot {
+  static _className = "BingChatCreativeBot"; // Class name of the bot
   static _logoFilename = "bing-creative-logo.png"; // Place it in assets/bots/
   static _model = "h3imaginative"; // Bing styles: h3imaginative, harmonyv3, h3precise
   static _lock = new AsyncLock(); // Must process requests in queue
diff --git a/src/bots/BingChatPreciseBot.js b/src/bots/BingChatPreciseBot.js
index 5592f19507..f35dd46383 100644
--- a/src/bots/BingChatPreciseBot.js
+++ b/src/bots/BingChatPreciseBot.js
@@ -3,6 +3,7 @@ import AsyncLock from "async-lock";
 import BingChatBot from "./BingChatBot";
 
 export default class BingChatPreciseBot extends BingChatBot {
+  static _className = "BingChatPreciseBot"; // Class name of the bot
   static _logoFilename = "bing-precise-logo.png"; // Place it in assets/bots/
   static _model = "h3precise"; // Bing styles: h3imaginative, harmonyv3, h3precise
   static _lock = new AsyncLock(); // Must process requests in queue
diff --git a/src/bots/Bot.js b/src/bots/Bot.js
index e073ec72ab..334643f9de 100644
--- a/src/bots/Bot.js
+++ b/src/bots/Bot.js
@@ -15,6 +15,7 @@ export default class Bot {
   static _isAvailable = false;
 
   static _brandId = "bot"; // Brand id of the bot, should be unique. Used in i18n.
+  static _className = "Bot"; // Class name of the bot
   static _model = ""; // Model of the bot (eg. "text-davinci-002-render-sha")
   static _logoFilename = "default-logo.svg"; // Place it in assets/bots/
   static _loginUrl = "undefined";
@@ -82,11 +83,11 @@ export default class Bot {
     } else {
       let currentClass = this.constructor;
       let parentClass = Object.getPrototypeOf(currentClass);
-      while (parentClass && parentClass.name !== "Bot") {
+      while (parentClass && parentClass._className !== "Bot") {
         currentClass = parentClass;
         parentClass = Object.getPrototypeOf(currentClass);
       }
-      const componentName = currentClass.name + "Settings";
+      const componentName = currentClass._className + "Settings";
       component = await import(`@/components/BotSettings/${componentName}.vue`);
     }
 
diff --git a/src/bots/ChatGPT35Bot.js b/src/bots/ChatGPT35Bot.js
index 5633383b87..0c0d538568 100644
--- a/src/bots/ChatGPT35Bot.js
+++ b/src/bots/ChatGPT35Bot.js
@@ -1,6 +1,7 @@
 import ChatGPTBot from "./ChatGPTBot";
 
 export default class ChatGPT35Bot extends ChatGPTBot {
+  static _className = "ChatGPT35Bot"; // Class name of the bot
   static _logoFilename = "chatgpt-35-logo.png"; // Place it in assets/bots/
   static _model = "text-davinci-002-render-sha";
 
diff --git a/src/bots/ChatGPT4Bot.js b/src/bots/ChatGPT4Bot.js
index 2e81506f8c..c703f7052b 100644
--- a/src/bots/ChatGPT4Bot.js
+++ b/src/bots/ChatGPT4Bot.js
@@ -1,6 +1,7 @@
 import ChatGPTBot from "./ChatGPTBot";
 
 export default class ChatGPT4Bot extends ChatGPTBot {
+  static _className = "ChatGPT4Bot"; // Class name of the bot
   static _logoFilename = "chatgpt-4-logo.png"; // Place it in assets/bots/
   static _model = "gpt-4";
 
diff --git a/src/bots/ChatGPTBot.js b/src/bots/ChatGPTBot.js
index de1e440b17..51bf790519 100644
--- a/src/bots/ChatGPTBot.js
+++ b/src/bots/ChatGPTBot.js
@@ -12,6 +12,7 @@ const REFRESH_SESSION_URL =
 
 export default class ChatGPTBot extends Bot {
   static _brandId = "chatGpt";
+  static _className = "ChatGPTBot"; // Class name of the bot
   static _logoFilename = "chatgpt-logo.svg"; // Place it in assets/bots/
   static _loginUrl = "https://chat.openai.com/";
   static _model = "";
diff --git a/src/bots/ERNIEBot.js b/src/bots/ERNIEBot.js
index 4071d56d5d..5dd3ed0379 100644
--- a/src/bots/ERNIEBot.js
+++ b/src/bots/ERNIEBot.js
@@ -3,6 +3,7 @@ import axios from "axios";
 
 export default class ERNIEBot extends Bot {
   static _brandId = "ernie"; // ID of the bot, should be unique
+  static _className = "ERNIEBot"; // Class name of the bot
   static _logoFilename = "ernie-logo.png"; // Place it in assets/bots/
   static _loginUrl = "https://yiyan.baidu.com/";
 
diff --git a/src/bots/OpenAIAPI35Bot.js b/src/bots/OpenAIAPI35Bot.js
index 6bfe0c0f05..5831f0f489 100644
--- a/src/bots/OpenAIAPI35Bot.js
+++ b/src/bots/OpenAIAPI35Bot.js
@@ -1,6 +1,7 @@
 import OpenAIAPIBot from "./OpenAIAPIBot";
 
 export default class OpenAIAPI35Bot extends OpenAIAPIBot {
+  static _className = "OpenAIAPI35Bot"; // Class name of the bot
   static _logoFilename = "openai-35-logo.png"; // Place it in assets/bots/
   static _model = "gpt-3.5-turbo";
 
diff --git a/src/bots/OpenAIAPI4Bot.js b/src/bots/OpenAIAPI4Bot.js
index 3fe165f761..f667825e51 100644
--- a/src/bots/OpenAIAPI4Bot.js
+++ b/src/bots/OpenAIAPI4Bot.js
@@ -1,6 +1,7 @@
 import OpenAIAPIBot from "./OpenAIAPIBot";
 
 export default class OpenAIAPI4Bot extends OpenAIAPIBot {
+  static _className = "OpenAIAPI4Bot"; // Class name of the bot
   static _logoFilename = "openai-4-logo.png"; // Place it in assets/bots/
   static _model = "gpt-4";
 
diff --git a/src/bots/OpenAIAPIBot.js b/src/bots/OpenAIAPIBot.js
index d1fbb7bda5..6c40f67326 100644
--- a/src/bots/OpenAIAPIBot.js
+++ b/src/bots/OpenAIAPIBot.js
@@ -4,6 +4,7 @@ import store from "@/store";
 
 export default class OpenAIAPIBot extends Bot {
   static _brandId = "openaiApi";
+  static _className = "OpenAIAPIBot";
   static _logoFilename = "";
   static _loginUrl = ""; // URL for the login button on the bots page
   static _apiUrl = "https://api.openai.com/v1/chat/completions";
diff --git a/src/bots/SparkBot.js b/src/bots/SparkBot.js
index cbbbb5ceb1..628e5d36fd 100644
--- a/src/bots/SparkBot.js
+++ b/src/bots/SparkBot.js
@@ -7,6 +7,7 @@ import Bot from "./Bot";
 
 export default class SparkBot extends Bot {
   static _brandId = "spark";
+  static _className = "SparkBot"; // Class name of the bot
   static _logoFilename = "spark-logo.png"; // Place it in assets/bots/
   static _loginUrl = "https://xinghuo.xfyun.cn/";
   static _lock = new AsyncLock(); // All Spark bots share the same lock