Remove some chatbots

Author: wong2

src/app/bots/index.ts

@@ -15,13 +15,9 @@ export type BotId =
   | 'claude'
   | 'xunfei'
   | 'vicuna'
-  | 'alpaca'
   | 'chatglm'
   | 'llama'
-  | 'oasst'
-  | 'rwkv'
   | 'pi'
-  | 'guanaco'
   | 'wizardlm'
   | 'qianwen'
   | 'baichuan'
@@ -40,18 +36,10 @@ export function createBotInstance(botId: BotId) {
       return new XunfeiBot()
     case 'vicuna':
       return new LMSYSBot('vicuna-33b')
-    case 'alpaca':
-      return new LMSYSBot('alpaca-13b')
     case 'chatglm':
       return new LMSYSBot('chatglm2-6b')
     case 'llama':
       return new LMSYSBot('llama-2-70b-chat')
-    case 'oasst':
-      return new LMSYSBot('oasst-pythia-12b')
-    case 'rwkv':
-      return new LMSYSBot('RWKV-4-Raven-14B')
-    case 'guanaco':
-      return new LMSYSBot('guanaco-33b')
     case 'wizardlm':
       return new LMSYSBot('wizardlm-13b')
     case 'pi':

src/app/consts.ts

@@ -1,19 +1,15 @@
-import alpacaLogo from '~/assets/alpaca-logo.png'
 import claudeLogo from '~/assets/anthropic-logo.png'
+import baichuanLogo from '~/assets/baichuan-logo.png'
 import bardLogo from '~/assets/bard-logo.svg'
 import bingLogo from '~/assets/bing-logo.svg'
 import chatglmLogo from '~/assets/chatglm-logo.svg'
 import chatgptLogo from '~/assets/chatgpt-logo.svg'
-import guanacoLogo from '~/assets/guanaco-logo.png'
 import llamaLogo from '~/assets/llama-logo.png'
-import oasstLogo from '~/assets/oasst-logo.svg'
 import piLogo from '~/assets/pi-logo.png'
-import rwkvLogo from '~/assets/rwkv-logo.png'
+import qianwenLogo from '~/assets/qianwen-logo.png'
 import vicunaLogo from '~/assets/vicuna-logo.jpg'
 import wizardlmLogo from '~/assets/wizardlm-logo.png'
 import xunfeiLogo from '~/assets/xunfei-logo.png'
-import qianwenLogo from '~/assets/qianwen-logo.png'
-import baichuanLogo from '~/assets/baichuan-logo.png'
 import { BotId } from './bots'
 
 export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
@@ -37,10 +33,6 @@ export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
     name: 'Llama 2',
     avatar: llamaLogo,
   },
-  alpaca: {
-    name: 'Alpaca',
-    avatar: alpacaLogo,
-  },
   vicuna: {
     name: 'Vicuna',
     avatar: vicunaLogo,
@@ -49,6 +41,10 @@ export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
     name: 'Pi',
     avatar: piLogo,
   },
+  wizardlm: {
+    name: 'WizardLM',
+    avatar: wizardlmLogo,
+  },
   chatglm: {
     name: 'ChatGLM2',
     avatar: chatglmLogo,
@@ -57,22 +53,6 @@ export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
     name: 'iFlytek Spark',
     avatar: xunfeiLogo,
   },
-  oasst: {
-    name: 'OpenAssistant',
-    avatar: oasstLogo,
-  },
-  rwkv: {
-    name: 'ChatRWKV',
-    avatar: rwkvLogo,
-  },
-  guanaco: {
-    name: 'Guanaco',
-    avatar: guanacoLogo,
-  },
-  wizardlm: {
-    name: 'WizardLM',
-    avatar: wizardlmLogo,
-  },
   qianwen: {
     name: 'Qianwen',
     avatar: qianwenLogo,

src/app/pages/MultiBotChatPanel.tsx

@@ -1,32 +1,36 @@
 import { useAtom, useAtomValue, useSetAtom } from 'jotai'
 import { atomWithStorage } from 'jotai/utils'
-import { uniqBy } from 'lodash-es'
+import { sample, uniqBy } from 'lodash-es'
 import { FC, Suspense, useCallback, useEffect, useMemo } from 'react'
 import { useTranslation } from 'react-i18next'
 import { cx } from '~/utils'
 import Button from '~app/components/Button'
 import ChatMessageInput from '~app/components/Chat/ChatMessageInput'
 import LayoutSwitch from '~app/components/Chat/LayoutSwitch'
-import { Layout } from '~app/consts'
+import { CHATBOTS, Layout } from '~app/consts'
 import { useChat } from '~app/hooks/use-chat'
 import { usePremium } from '~app/hooks/use-premium'
 import { trackEvent } from '~app/plausible'
 import { showPremiumModalAtom } from '~app/state'
 import { BotId } from '../bots'
 import ConversationPanel from '../components/Chat/ConversationPanel'
 
+const DEFAULT_BOTS: BotId[] = ['chatgpt', 'claude', 'bard', 'bing', 'llama', 'pi']
+
 const layoutAtom = atomWithStorage<Layout>('multiPanelLayout', 2, undefined, { unstable_getOnInit: true })
-const twoPanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:2', ['chatgpt', 'claude'])
-const threePanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:3', ['chatgpt', 'claude', 'bard'])
-const fourPanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:4', ['chatgpt', 'claude', 'bard', 'bing'])
-const sixPanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:6', [
-  'chatgpt',
-  'claude',
-  'bard',
-  'bing',
-  'pi',
-  'llama',
-])
+const twoPanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:2', DEFAULT_BOTS.slice(0, 2))
+const threePanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:3', DEFAULT_BOTS.slice(0, 3))
+const fourPanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:4', DEFAULT_BOTS.slice(0, 4))
+const sixPanelBotsAtom = atomWithStorage<BotId[]>('multiPanelBots:6', DEFAULT_BOTS.slice(0, 6))
+
+function replaceDeprecatedBots(bots: BotId[]): BotId[] {
+  return bots.map((bot) => {
+    if (CHATBOTS[bot]) {
+      return bot
+    }
+    return sample(DEFAULT_BOTS)!
+  })
+}
 
 const GeneralChatPanel: FC<{
   chats: ReturnType<typeof useChat>[]
@@ -131,15 +135,17 @@ const GeneralChatPanel: FC<{
 }
 
 const TwoBotChatPanel = () => {
-  const [multiPanelBotIds, setBots] = useAtom(twoPanelBotsAtom)
+  const [bots, setBots] = useAtom(twoPanelBotsAtom)
+  const multiPanelBotIds = useMemo(() => replaceDeprecatedBots(bots), [bots])
   const chat1 = useChat(multiPanelBotIds[0])
   const chat2 = useChat(multiPanelBotIds[1])
   const chats = useMemo(() => [chat1, chat2], [chat1, chat2])
   return <GeneralChatPanel chats={chats} setBots={setBots} />
 }
 
 const ThreeBotChatPanel = () => {
-  const [multiPanelBotIds, setBots] = useAtom(threePanelBotsAtom)
+  const [bots, setBots] = useAtom(threePanelBotsAtom)
+  const multiPanelBotIds = useMemo(() => replaceDeprecatedBots(bots), [bots])
   const chat1 = useChat(multiPanelBotIds[0])
   const chat2 = useChat(multiPanelBotIds[1])
   const chat3 = useChat(multiPanelBotIds[2])
@@ -148,7 +154,8 @@ const ThreeBotChatPanel = () => {
 }
 
 const FourBotChatPanel = () => {
-  const [multiPanelBotIds, setBots] = useAtom(fourPanelBotsAtom)
+  const [bots, setBots] = useAtom(fourPanelBotsAtom)
+  const multiPanelBotIds = useMemo(() => replaceDeprecatedBots(bots), [bots])
   const chat1 = useChat(multiPanelBotIds[0])
   const chat2 = useChat(multiPanelBotIds[1])
   const chat3 = useChat(multiPanelBotIds[2])
@@ -158,7 +165,8 @@ const FourBotChatPanel = () => {
 }
 
 const SixBotChatPanel = () => {
-  const [multiPanelBotIds, setBots] = useAtom(sixPanelBotsAtom)
+  const [bots, setBots] = useAtom(sixPanelBotsAtom)
+  const multiPanelBotIds = useMemo(() => replaceDeprecatedBots(bots), [bots])
   const chat1 = useChat(multiPanelBotIds[0])
   const chat2 = useChat(multiPanelBotIds[1])
   const chat3 = useChat(multiPanelBotIds[2])