Add tongyi qianwen bot

Author: wong2

manifest.config.ts

@@ -62,6 +62,11 @@ export default defineManifest(async (env) => {
           enabled: true,
           path: 'src/rules/ddg.json',
         },
+        {
+          id: 'ruleset_qianwen',
+          enabled: true,
+          path: 'src/rules/qianwen.json',
+        },
       ],
     },
   }

src/app/bots/index.ts

@@ -4,6 +4,7 @@ import { ChatGPTBot } from './chatgpt'
 import { ClaudeBot } from './claude'
 import { LMSYSBot } from './lmsys'
 import { PiBot } from './pi'
+import { QianwenWebBot } from './tongyi'
 import { XunfeiBot } from './xunfei'
 
 export type BotId =
@@ -21,6 +22,7 @@ export type BotId =
   | 'pi'
   | 'guanaco'
   | 'wizardlm'
+  | 'qianwen'
 
 export function createBotInstance(botId: BotId) {
   switch (botId) {
@@ -52,6 +54,8 @@ export function createBotInstance(botId: BotId) {
       return new LMSYSBot('wizardlm-13b')
     case 'pi':
       return new PiBot()
+    case 'qianwen':
+      return new QianwenWebBot()
   }
 }
 

src/app/bots/tongyi/api.ts

@@ -0,0 +1,28 @@
+import { ofetch } from 'ofetch'
+import { ChatError, ErrorCode } from '~utils/errors'
+
+interface CreationResponse {
+  data: {
+    sessionId: string
+  }
+  success: boolean
+  errorMsg: string | null
+  errorCode: string | null
+}
+
+export async function createConversation(firstQuery: string) {
+  const resp = await ofetch<CreationResponse>('https://qianwen.aliyun.com/addSession', {
+    method: 'POST',
+    body: { firstQuery },
+    headers: {
+      'X-Xsrf-Token': '7a3dae93-1d29-4eb6-9940-34efad5a2b78',
+    },
+  })
+  if (!resp.success) {
+    if (resp.errorCode === '4000') {
+      throw new ChatError('请先登录通义千问账号', ErrorCode.QIANWEN_WEB_UNAUTHORIZED)
+    }
+    throw new Error(`Error: ${resp.errorCode} ${resp.errorMsg}`)
+  }
+  return resp.data.sessionId
+}

src/app/bots/tongyi/index.ts

@@ -0,0 +1,71 @@
+import { parseSSEResponse } from '~utils/sse'
+import { AbstractBot, SendMessageParams } from '../abstract-bot'
+import { requestHostPermission } from '~app/utils/permissions'
+import { ChatError, ErrorCode } from '~utils/errors'
+import { createConversation } from './api'
+import { uuid } from '~utils'
+
+function generateMessageId() {
+  return uuid().replace(/-/g, '')
+}
+
+interface ConversationContext {
+  conversationId: string
+  lastMessageId?: string
+}
+
+export class QianwenWebBot extends AbstractBot {
+  private conversationContext?: ConversationContext
+
+  async doSendMessage(params: SendMessageParams) {
+    if (!(await requestHostPermission('https://qianwen.aliyun.com/'))) {
+      throw new ChatError('Missing qianwen.aliyun.com permission', ErrorCode.MISSING_HOST_PERMISSION)
+    }
+
+    if (!this.conversationContext) {
+      const conversationId = await createConversation(params.prompt)
+      this.conversationContext = { conversationId }
+    }
+
+    const resp = await fetch('https://qianwen.aliyun.com/conversation', {
+      method: 'POST',
+      signal: params.signal,
+      headers: {
+        'Content-Type': 'application/json',
+        'X-Xsrf-Token': '7a3dae93-1d29-4eb6-9940-34efad5a2b78',
+      },
+      body: JSON.stringify({
+        action: 'next',
+        msgId: generateMessageId(),
+        parentMsgId: this.conversationContext.lastMessageId || '0',
+        contents: [{ contentType: 'text', content: params.prompt }],
+        timeout: 17,
+        sessionId: this.conversationContext.conversationId,
+        model: '',
+        userAction: 'chat',
+        openSearch: true,
+      }),
+    })
+
+    await parseSSEResponse(resp, (message) => {
+      console.debug('qianwen sse', message)
+      const data = JSON.parse(message)
+      const text = data.content[0]
+      if (text) {
+        params.onEvent({ type: 'UPDATE_ANSWER', data: { text } })
+      }
+      if (data.stopReason === 'stop') {
+        this.conversationContext!.lastMessageId = data.msgId
+        params.onEvent({ type: 'DONE' })
+      }
+    })
+  }
+
+  resetConversation() {
+    this.conversationContext = undefined
+  }
+
+  get name() {
+    return 'Tongyi Qianwen'
+  }
+}

src/app/components/Chat/ErrorAction.tsx

@@ -74,6 +74,13 @@ const ErrorAction: FC<{ error: ChatError }> = ({ error }) => {
       </a>
     )
   }
+  if (error.code === ErrorCode.QIANWEN_WEB_UNAUTHORIZED) {
+    return (
+      <a href="https://qianwen.aliyun.com" target="_blank" rel="noreferrer">
+        <ActionButton text={t('Login at qianwen.aliyun.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/consts.ts

@@ -12,6 +12,7 @@ import rwkvLogo from '~/assets/rwkv-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 { BotId } from './bots'
 
 export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
@@ -71,6 +72,10 @@ export const CHATBOTS: Record<BotId, { name: string; avatar: any }> = {
     name: 'WizardLM',
     avatar: wizardlmLogo,
   },
+  qianwen: {
+    name: 'Qianwen',
+    avatar: qianwenLogo,
+  },
 }
 
 export const CHATGPT_HOME_URL = 'https://chat.openai.com'

src/assets/qianwen-logo.png

@@ -0,0 +1,25 @@
+[
+  {
+    "id": 1,
+    "priority": 1,
+    "action": {
+      "type": "modifyHeaders",
+      "requestHeaders": [
+        {
+          "header": "origin",
+          "operation": "set",
+          "value": "https://qianwen.aliyun.com"
+        },
+        {
+          "header": "referer",
+          "operation": "set",
+          "value": "https://qianwen.aliyun.com/chat"
+        }
+      ]
+    },
+    "condition": {
+      "requestDomains": ["qianwen.aliyun.com"],
+      "resourceTypes": ["xmlhttprequest"]
+    }
+  }
+]

src/rules/qianwen.json

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