#573's patch

josStorer · Dec 4, 2023 · 8d53ab8 · 8d53ab8
1 parent d002aa8
commit 8d53ab8
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/src/background/index.mjs b/src/background/index.mjs
@@ -144,7 +144,13 @@ async function executeApi(session, port, config) {
     await generateAnswersWithBardWebApi(port, session.question, session, cookies)
   } else if (claudeWebModelKeys.includes(session.modelName)) {
     const sessionKey = await getClaudeSessionKey()
-    await generateAnswersWithClaudeWebApi(port, session.question, session, sessionKey)
+    await generateAnswersWithClaudeWebApi(
+      port,
+      session.question,
+      session,
+      sessionKey,
+      session.modelName,
+    )
   }
 }
 

diff --git a/src/services/apis/claude-web.mjs b/src/services/apis/claude-web.mjs
@@ -1,13 +1,21 @@
 import { pushRecord } from './shared.mjs'
 import Claude from 'claude-ai'
+import { Models } from '../../config/index.mjs'
 
 /**
  * @param {Runtime.Port} port
  * @param {string} question
  * @param {Session} session
  * @param {string} sessionKey
+ * @param {string} modelName
  */
-export async function generateAnswersWithClaudeWebApi(port, question, session, sessionKey) {
+export async function generateAnswersWithClaudeWebApi(
+  port,
+  question,
+  session,
+  sessionKey,
+  modelName,
+) {
   const bot = new Claude({ sessionKey })
   await bot.init()
 
@@ -28,6 +36,7 @@ export async function generateAnswersWithClaudeWebApi(port, question, session, s
       .startConversation(question, {
         progress: progressFunc,
         done: doneFunc,
+        model: Models[modelName].value,
       })
       .then((conversation) => {
         session.claude_conversation = conversation
@@ -38,5 +47,6 @@ export async function generateAnswersWithClaudeWebApi(port, question, session, s
       conversation: session.claude_conversation,
       progress: progressFunc,
       done: doneFunc,
+      model: Models[modelName].value,
     })
 }