From 8d53ab85a5b6cf5aa9c5b2729c58db95ba47171f Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Mon, 4 Dec 2023 11:58:21 +0800
Subject: [PATCH] #573's patch

---
 src/background/index.mjs         |  8 +++++++-
 src/services/apis/claude-web.mjs | 12 +++++++++++-
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/background/index.mjs b/src/background/index.mjs
index 5d27aa0..f65c13d 100644
--- 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
index 7a8a8d5..d1525ff 100644
--- 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,
     })
 }