support moonshot

This commit is contained in:
Jialei
2024-03-03 21:45:22 +09:00
committed by josc146
parent 6b494c03b7
commit 0d7d7354d8
5 changed files with 101 additions and 14 deletions
+10
View File
@@ -30,6 +30,7 @@ import {
gptApiModelKeys,
poeWebModelKeys,
setUserConfig,
moonshotApiModelKeys,
} from '../config/index.mjs'
import '../_locales/i18n'
import { openUrl } from '../utils/open-url'
@@ -44,6 +45,7 @@ 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'
import { generateAnswersWithMoonshotCompletionApi } from '../services/apis/moonshot-api.mjs'
function setPortProxy(port, proxyTabId) {
port.proxy = Browser.tabs.connect(proxyTabId)
@@ -151,6 +153,14 @@ async function executeApi(session, port, config) {
sessionKey,
session.modelName,
)
} else if (moonshotApiModelKeys.includes(session.modelName)) {
await generateAnswersWithMoonshotCompletionApi(
port,
session.question,
session,
config.apiKey,
session.modelName,
)
}
}