feat: independent Claude.ai model support (#457, #476, #477, #503)

Author: stacksculptor

package-lock.json

@@ -22,6 +22,7 @@
     "@nem035/gpt-3-encoder": "^1.1.7",
     "@picocss/pico": "^1.5.9",
     "@primer/octicons-react": "^18.3.0",
+    "claude-ai": "^1.2.2",
     "countries-list": "^2.6.1",
     "diff": "^5.1.0",
     "eventsource-parser": "^1.0.0",

package.json

@@ -19,6 +19,7 @@ import {
   bingWebModelKeys,
   chatgptApiModelKeys,
   chatgptWebModelKeys,
+  claudeWebModelKeys,
   customApiModelKeys,
   defaultConfig,
   getUserConfig,
@@ -34,11 +35,13 @@ import {
   getBardCookies,
   getBingAccessToken,
   getChatGptAccessToken,
+  getClaudeSessionKey,
   registerPortListener,
 } from '../services/wrappers.mjs'
 import { refreshMenu } from './menus.mjs'
 import { registerCommands } from './commands.mjs'
 import { generateAnswersWithBardWebApi } from '../services/apis/bard-web.mjs'
+import { generateAnswersWithClaudeWebApi } from '../services/apis/claude-web.mjs'
 
 function setPortProxy(port, proxyTabId) {
   port.proxy = Browser.tabs.connect(proxyTabId)
@@ -123,6 +126,9 @@ async function executeApi(session, port, config) {
   } else if (bardWebModelKeys.includes(session.modelName)) {
     const cookies = await getBardCookies()
     await generateAnswersWithBardWebApi(port, session.question, session, cookies)
+  } else if (claudeWebModelKeys.includes(session.modelName)) {
+    const sessionKey = await getClaudeSessionKey()
+    await generateAnswersWithClaudeWebApi(port, session.question, session, sessionKey)
   }
 }
 

src/background/index.mjs

@@ -30,6 +30,7 @@ export const chatgptWebModelKeys = [
 ]
 export const bingWebModelKeys = ['bingFree4', 'bingFreeSydney']
 export const bardWebModelKeys = ['bardWebFree']
+export const claudeWebModelKeys = ['claude2WebFree']
 export const gptApiModelKeys = ['gptApiDavinci']
 export const chatgptApiModelKeys = [
   'chatgptApi35',
@@ -81,6 +82,7 @@ export const Models = {
     value: 'gpt-3.5-turbo-16k-0613',
     desc: 'ChatGPT (GPT-3.5-turbo-16k 0613)',
   },
+  claude2WebFree: { value: 'claude-2', desc: 'Claude.ai (Web, Claude 2)' },
   bingFree4: { value: '', desc: 'Bing (Web, GPT-4)' },
   bingFreeSydney: { value: '', desc: 'Bing (Web, GPT-4, Sydney)' },
   bardWebFree: { value: '', desc: 'Bard (Web)' },
@@ -175,21 +177,22 @@ export const defaultConfig = {
   activeApiModes: [
     'chatgptFree35',
     'chatgptFree35Mobile',
-    'chatgptPlus4',
-    'chatgptPlus4Mobile',
+    // 'chatgptPlus4',
+    // 'chatgptPlus4Mobile',
     'chatgptApi35',
     'chatgptApi35_16k',
+    'claude2WebFree',
     'bingFree4',
     'bingFreeSydney',
-    'poeAiWebSage', //poe.com/Assistant
-    'poeAiWebGPT4',
-    'poeAiWebGPT4_32k',
-    'poeAiWebClaudePlus',
-    'poeAiWebClaude100k',
+    // 'poeAiWebSage', //poe.com/Assistant
+    // 'poeAiWebGPT4',
+    // 'poeAiWebGPT4_32k',
+    // 'poeAiWebClaudePlus',
+    // 'poeAiWebClaude100k',
     'chatgptApi4_8k',
     'customModel',
     'azureOpenAi',
-    'poeAiWebCustom',
+    // 'poeAiWebCustom',
   ],
   activeSelectionTools: ['translate', 'summary', 'polish', 'code', 'ask'],
   activeSiteAdapters: [

src/config/index.mjs

@@ -0,0 +1,42 @@
+import { pushRecord } from './shared.mjs'
+import Claude from 'claude-ai'
+
+/**
+ * @param {Runtime.Port} port
+ * @param {string} question
+ * @param {Session} session
+ * @param {string} sessionKey
+ */
+export async function generateAnswersWithClaudeWebApi(port, question, session, sessionKey) {
+  const bot = new Claude({ sessionKey })
+  await bot.init()
+
+  let answer = ''
+  const progressFunc = ({ completion }) => {
+    answer = completion
+    port.postMessage({ answer: answer, done: false, session: null })
+  }
+
+  const doneFunc = () => {
+    pushRecord(session, question, answer)
+    console.debug('conversation history', { content: session.conversationRecords })
+    port.postMessage({ answer: answer, done: true, session: session })
+  }
+
+  if (!session.claude_conversation)
+    await bot
+      .startConversation(question, {
+        progress: progressFunc,
+        done: doneFunc,
+      })
+      .then((conversation) => {
+        session.claude_conversation = conversation
+        port.postMessage({ answer: answer, done: true, session: session })
+      })
+  else
+    await bot.sendMessage(question, {
+      conversation: session.claude_conversation,
+      progress: progressFunc,
+      done: doneFunc,
+    })
+}

src/services/apis/claude-web.mjs

@@ -25,6 +25,7 @@ import { v4 as uuidv4 } from 'uuid'
  * @property {Object|null} bingWeb_jailbreakConversationCache
  * @property {number|null} poe_chatId
  * @property {object|null} bard_conversationObj
+ * @property {object|null} claude_conversation
  */
 /**
  * @param {string|null} question
@@ -78,5 +79,8 @@ export function initSession({
 
     // bard
     bard_conversationObj: null,
+
+    // claude.ai
+    claude_conversation: null,
   }
 }

src/services/init-session.mjs

@@ -40,6 +40,10 @@ export async function getBardCookies() {
   return '__Secure-1PSID=' + token
 }
 
+export async function getClaudeSessionKey() {
+  return (await Browser.cookies.get({ url: 'https://claude.ai/', name: 'sessionKey' }))?.value
+}
+
 export function registerPortListener(executor) {
   Browser.runtime.onConnect.addListener((port) => {
     console.debug('connected')