mirror of
https://github.com/wassname/chatGPTBox.git
synced 2026-07-04 18:47:19 +08:00
josc146
94 lines
3.3 KiB
JavaScript
94 lines
3.3 KiB
JavaScript
import BingAIClient from '../clients/bing/index.mjs'
|
|
import { getUserConfig } from '../../config/index.mjs'
|
|
import { pushRecord, setAbortController } from './shared.mjs'
|
|
|
|
/**
|
|
* @param {Runtime.Port} port
|
|
* @param {string} question
|
|
* @param {Session} session
|
|
* @param {string} accessToken
|
|
* @param {boolean} sydneyMode
|
|
*/
|
|
export async function generateAnswersWithBingWebApi(
|
|
port,
|
|
question,
|
|
session,
|
|
accessToken,
|
|
sydneyMode = false,
|
|
) {
|
|
const { controller, messageListener, disconnectListener } = setAbortController(port)
|
|
const config = await getUserConfig()
|
|
let modelMode
|
|
if (session.modelName.includes('-')) modelMode = session.modelName.split('-')[1]
|
|
else modelMode = config.modelMode
|
|
|
|
console.debug('mode', modelMode)
|
|
|
|
const bingAIClient = new BingAIClient({ userToken: accessToken, features: { genImage: false } })
|
|
if (session.bingWeb_jailbreakConversationCache)
|
|
bingAIClient.conversationsCache.set(
|
|
session.bingWeb_jailbreakConversationId,
|
|
session.bingWeb_jailbreakConversationCache,
|
|
)
|
|
|
|
let answer = ''
|
|
const response = await bingAIClient
|
|
.sendMessage(question, {
|
|
abortController: controller,
|
|
toneStyle: modelMode,
|
|
jailbreakConversationId: sydneyMode,
|
|
onProgress: (message) => {
|
|
answer = message
|
|
// reference markers [^number^]
|
|
answer = answer.replaceAll(/\[\^(\d+)\^\]/g, '<sup>$1</sup>')
|
|
port.postMessage({ answer: answer, done: false, session: null })
|
|
},
|
|
...(session.bingWeb_conversationId
|
|
? {
|
|
conversationId: session.bingWeb_conversationId,
|
|
encryptedConversationSignature: session.bingWeb_encryptedConversationSignature,
|
|
clientId: session.bingWeb_clientId,
|
|
invocationId: session.bingWeb_invocationId,
|
|
}
|
|
: session.bingWeb_jailbreakConversationId
|
|
? {
|
|
jailbreakConversationId: session.bingWeb_jailbreakConversationId,
|
|
parentMessageId: session.bingWeb_parentMessageId,
|
|
}
|
|
: {}),
|
|
})
|
|
.catch((err) => {
|
|
port.onMessage.removeListener(messageListener)
|
|
port.onDisconnect.removeListener(disconnectListener)
|
|
throw err
|
|
})
|
|
|
|
if (!sydneyMode) {
|
|
session.bingWeb_encryptedConversationSignature = response.encryptedConversationSignature
|
|
session.bingWeb_conversationId = response.conversationId
|
|
session.bingWeb_clientId = response.clientId
|
|
session.bingWeb_invocationId = response.invocationId
|
|
} else {
|
|
session.bingWeb_jailbreakConversationId = response.jailbreakConversationId
|
|
session.bingWeb_parentMessageId = response.messageId
|
|
session.bingWeb_jailbreakConversationCache = bingAIClient.conversationsCache.get(
|
|
response.jailbreakConversationId,
|
|
)
|
|
}
|
|
|
|
if (response.details.sourceAttributions && response.details.sourceAttributions.length > 0) {
|
|
const footnotes =
|
|
'\n\\-\n' +
|
|
response.details.sourceAttributions
|
|
.map((attr, index) => `\\[${index + 1}]: [${attr.providerDisplayName}](${attr.seeMoreUrl})`)
|
|
.join('\n')
|
|
answer += footnotes
|
|
}
|
|
|
|
pushRecord(session, question, answer)
|
|
console.debug('conversation history', { content: session.conversationRecords })
|
|
port.onMessage.removeListener(messageListener)
|
|
port.onDisconnect.removeListener(disconnectListener)
|
|
port.postMessage({ answer: answer, done: true, session: session })
|
|
}
|