This commit is contained in:
josc146
2023-12-04 11:58:21 +08:00
parent d002aa8f2e
commit 8d53ab85a5
2 changed files with 18 additions and 2 deletions
+7 -1
View File
@@ -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,
)
}
}
+11 -1
View File
@@ -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,
})
}