Add baichuan

Author: wong2

package.json

@@ -73,6 +73,7 @@
     "lodash-es": "^4.17.21",
     "lucide-react": "^0.264.0",
     "md5": "^2.3.0",
+    "nanoid": "^4.0.2",
     "ofetch": "^1.1.1",
     "plausible-tracker": "^0.3.8",
     "react": "^18.2.0",

src/app/bots/baichuan/api.ts

@@ -0,0 +1,34 @@
+import { ofetch } from 'ofetch'
+import { customAlphabet } from 'nanoid'
+import { ChatError, ErrorCode } from '~utils/errors'
+
+interface UserInfo {
+  id: number
+}
+
+export async function getUserInfo(): Promise<UserInfo> {
+  const resp = await ofetch<{ data?: UserInfo; code: number; msg: string }>(
+    'https://www.baichuan-ai.com/api/user/user-info',
+  )
+  if (resp.code === 401) {
+    throw new ChatError('请先登录百川账号', ErrorCode.BAICHUAN_WEB_UNAUTHORIZED)
+  }
+  if (resp.code !== 200) {
+    throw new Error(`Error: ${resp.code} ${resp.msg}`)
+  }
+  return resp.data!
+}
+
+const nanoid = customAlphabet('abcdefghijklmnopqrstuvwxyz0123456789')
+
+function randomString(length: number) {
+  return nanoid(length)
+}
+
+export function generateSessionId() {
+  return 'p' + randomString(10)
+}
+
+export function generateMessageId() {
+  return 'U' + randomString(14)
+}

src/app/bots/baichuan/index.ts

@@ -0,0 +1,122 @@
+import { AbstractBot, SendMessageParams } from '../abstract-bot'
+import { requestHostPermission } from '~app/utils/permissions'
+import { ChatError, ErrorCode } from '~utils/errors'
+import { uuid } from '~utils'
+import { generateMessageId, generateSessionId, getUserInfo } from './api'
+import { streamAsyncIterable } from '~utils/stream-async-iterable'
+
+interface Message {
+  id: string
+  createdAt: number
+  data: string
+  from: 0 | 1 // human | bot
+}
+
+interface ConversationContext {
+  conversationId: string
+  historyMessages: Message[]
+  userId: number
+  lastMessageId?: string
+}
+
+export class BaichuanWebBot extends AbstractBot {
+  private conversationContext?: ConversationContext
+
+  async doSendMessage(params: SendMessageParams) {
+    if (!(await requestHostPermission('https://*.baichuan-ai.com/'))) {
+      throw new ChatError('Missing baichuan-ai.com permission', ErrorCode.MISSING_HOST_PERMISSION)
+    }
+
+    if (!this.conversationContext) {
+      const conversationId = generateSessionId()
+      const userInfo = await getUserInfo()
+      this.conversationContext = { conversationId, historyMessages: [], userId: userInfo.id }
+    }
+
+    const { conversationId, lastMessageId, historyMessages, userId } = this.conversationContext
+
+    const message: Message = {
+      id: generateMessageId(),
+      createdAt: Date.now(),
+      data: params.prompt,
+      from: 0,
+    }
+
+    const resp = await fetch('https://www.baichuan-ai.com/api/chat/v1/chat', {
+      method: 'POST',
+      signal: params.signal,
+      headers: {
+        'Content-Type': 'application/json',
+      },
+      body: JSON.stringify({
+        stream: true,
+        request_id: uuid(),
+        app_info: { id: 10001, name: 'baichuan_web' },
+        user_info: { id: userId, status: 1 },
+        prompt: {
+          id: message.id,
+          data: message.data,
+          from: message.from,
+          parent_id: lastMessageId || 0,
+          created_at: message.createdAt,
+        },
+        session_info: { id: conversationId, name: '新的对话', created_at: Date.now() },
+        parameters: {
+          repetition_penalty: -1,
+          temperature: -1,
+          top_k: -1,
+          top_p: -1,
+          max_new_tokens: -1,
+          do_sample: -1,
+          regenerate: 0,
+        },
+        history: historyMessages,
+      }),
+    })
+
+    const decoder = new TextDecoder()
+    let result = ''
+    let answerMessageId: string | undefined
+
+    for await (const uint8Array of streamAsyncIterable(resp.body!)) {
+      const str = decoder.decode(uint8Array)
+      console.debug('baichuan', str)
+      const lines = str.split('\n')
+      for (const line of lines) {
+        if (!line) {
+          continue
+        }
+        const data = JSON.parse(line)
+        answerMessageId = data.answer.id
+        const text = data.answer.data
+        if (text) {
+          result += text
+          params.onEvent({ type: 'UPDATE_ANSWER', data: { text: result } })
+        }
+      }
+    }
+
+    this.conversationContext.historyMessages.push(message)
+    if (answerMessageId) {
+      this.conversationContext.lastMessageId = answerMessageId
+      if (result) {
+        this.conversationContext.historyMessages.push({
+          id: answerMessageId,
+          data: result,
+          createdAt: Date.now(),
+          from: 1,
+        })
+      }
+    }
+
+    params.onEvent({ type: 'DONE' })
+  }
+
+  resetConversation() {
+    this.conversationContext = undefined
+  }
+
+  get name() {
+    return '百川大模型'
+  }
+}

src/app/bots/index.ts

@@ -1,3 +1,4 @@
+import { BaichuanWebBot } from './baichuan'
 import { BardBot } from './bard'
 import { BingWebBot } from './bing'
 import { ChatGPTBot } from './chatgpt'
@@ -23,6 +24,7 @@ export type BotId =
   | 'guanaco'
   | 'wizardlm'
   | 'qianwen'
+  | 'baichuan'
 
 export function createBotInstance(botId: BotId) {
   switch (botId) {
@@ -56,6 +58,8 @@ export function createBotInstance(botId: BotId) {
       return new PiBot()
     case 'qianwen':
       return new QianwenWebBot()
+    case 'baichuan':
+      return new BaichuanWebBot()
   }
 }
 

src/app/components/Chat/ErrorAction.tsx

@@ -81,6 +81,13 @@ const ErrorAction: FC<{ error: ChatError }> = ({ error }) => {
       </a>
     )
   }
+  if (error.code === ErrorCode.BAICHUAN_WEB_UNAUTHORIZED) {
+    return (
+      <a href="https://www.baichuan-ai.com" target="_blank" rel="noreferrer">
+        <ActionButton text={t('Login at baichuan-ai.com')} />
+      </a>
+    )
+  }
   if (error.code === ErrorCode.GPT4_MODEL_WAITLIST) {
     return (
       <a href="https://openai.com/waitlist/gpt-4-api" target="_blank" rel="noreferrer">

src/app/components/Settings/EnabledBotsSettings.tsx

@@ -28,7 +28,7 @@ const EnabledBotsSettings: FC<Props> = ({ userConfig, updateConfigValue }) => {
   )
 
   return (
-    <div className="flex flex-row gap-3 flex-wrap max-w-[700px]">
+    <div className="flex flex-row gap-3 flex-wrap max-w-[720px]">
       {Object.entries(CHATBOTS).map(([botId, bot]) => {
         const enabled = userConfig.enabledBots.includes(botId as BotId)
         return (

src/app/consts.ts

@@ -13,6 +13,7 @@ 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 }> = {
@@ -76,6 +77,10 @@ export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
     name: 'Qianwen',
     avatar: qianwenLogo,
   },
+  baichuan: {
+    name: 'Baichuan',
+    avatar: baichuanLogo,
+  },
 }
 
 export const CHATGPT_HOME_URL = 'https://chat.openai.com'

src/assets/baichuan-logo.png

@@ -20,6 +20,7 @@ export enum ErrorCode {
   CLAUDE_WEB_UNAUTHORIZED = 'CLAUDE_WEB_UNAUTHORIZED',
   CLAUDE_WEB_UNAVAILABLE = 'CLAUDE_WEB_UNAVAILABLE',
   QIANWEN_WEB_UNAUTHORIZED = 'QIANWEN_WEB_UNAUTHORIZED',
+  BAICHUAN_WEB_UNAUTHORIZED = 'BAICHUAN_WEB_UNAUTHORIZED',
 }
 
 export class ChatError extends Error {

src/utils/errors.ts

@@ -3943,6 +3943,11 @@ nanoid@^3.3.6:
   resolved "https://registry.yarnpkg.com/nanoid/-/nanoid-3.3.6.tgz#443380c856d6e9f9824267d960b4236ad583ea4c"
   integrity sha512-BGcqMMJuToF7i1rt+2PWSNVnWIkGCU78jBG3RxO/bZlnZPK2Cmi2QaffxGO/2RvWi9sL+FAiRiXMgsyxQ1DIDA==
 
+nanoid@^4.0.2:
+  version "4.0.2"
+  resolved "https://registry.npmmirror.com/nanoid/-/nanoid-4.0.2.tgz#140b3c5003959adbebf521c170f282c5e7f9fb9e"
+  integrity sha512-7ZtY5KTCNheRGfEFxnedV5zFiORN1+Y1N6zvPTnHQd8ENUvfaDBeuJDZb2bN/oXwXxu3qkTXDzy57W5vAmDTBw==
+
 natural-compare-lite@^1.4.0:
   version "1.4.0"
   resolved "https://registry.yarnpkg.com/natural-compare-lite/-/natural-compare-lite-1.4.0.tgz#17b09581988979fddafe0201e931ba933c96cbb4"