@@ -5,18 +5,10 @@ import {
55 sendMessageFeedback ,
66} from '../services/apis/chatgpt-web'
77import { generateAnswersWithBingWebApi } from '../services/apis/bing-web.mjs'
8- import {
9- generateAnswersWithOpenAiApi ,
10- generateAnswersWithGptCompletionApi ,
11- } from '../services/apis/openai-api'
12- import { generateAnswersWithCustomApi } from '../services/apis/custom-api.mjs'
13- import { generateAnswersWithOllamaApi } from '../services/apis/ollama-api.mjs'
8+ import { generateAnswersWithOpenAICompatibleApi } from '../services/apis/openai-api'
149import { generateAnswersWithAzureOpenaiApi } from '../services/apis/azure-openai-api.mjs'
1510import { generateAnswersWithClaudeApi } from '../services/apis/claude-api.mjs'
16- import { generateAnswersWithChatGLMApi } from '../services/apis/chatglm-api.mjs'
1711import { generateAnswersWithWaylaidwandererApi } from '../services/apis/waylaidwanderer-api.mjs'
18- import { generateAnswersWithOpenRouterApi } from '../services/apis/openrouter-api.mjs'
19- import { generateAnswersWithAimlApi } from '../services/apis/aiml-api.mjs'
2012import {
2113 defaultConfig ,
2214 getUserConfig ,
@@ -52,10 +44,8 @@ import { refreshMenu } from './menus.mjs'
5244import { registerCommands } from './commands.mjs'
5345import { generateAnswersWithBardWebApi } from '../services/apis/bard-web.mjs'
5446import { generateAnswersWithClaudeWebApi } from '../services/apis/claude-web.mjs'
55- import { generateAnswersWithMoonshotCompletionApi } from '../services/apis/moonshot-api.mjs'
5647import { generateAnswersWithMoonshotWebApi } from '../services/apis/moonshot-web.mjs'
5748import { isUsingModelName } from '../utils/model-name-convert.mjs'
58- import { generateAnswersWithDeepSeekApi } from '../services/apis/deepseek-api.mjs'
5949import { redactSensitiveFields } from './redact.mjs'
6050
6151const RECONNECT_CONFIG = {
@@ -345,6 +335,20 @@ function setPortProxy(port, proxyTabId) {
345335 }
346336}
347337
338+ function isUsingOpenAICompatibleApiSession ( session ) {
339+ return (
340+ isUsingCustomModel ( session ) ||
341+ isUsingChatgptApiModel ( session ) ||
342+ isUsingMoonshotApiModel ( session ) ||
343+ isUsingChatGLMApiModel ( session ) ||
344+ isUsingDeepSeekApiModel ( session ) ||
345+ isUsingOllamaApiModel ( session ) ||
346+ isUsingOpenRouterApiModel ( session ) ||
347+ isUsingAimlApiModel ( session ) ||
348+ isUsingGptCompletionApiModel ( session )
349+ )
350+ }
351+
348352async function executeApi ( session , port , config ) {
349353 console . log (
350354 `[background] executeApi called for model: ${ session . modelName } , apiMode: ${ session . apiMode } ` ,
@@ -360,29 +364,7 @@ async function executeApi(session, port, config) {
360364 )
361365 }
362366 try {
363- if ( isUsingCustomModel ( session ) ) {
364- console . debug ( '[background] Using Custom Model API' )
365- if ( ! session . apiMode )
366- await generateAnswersWithCustomApi (
367- port ,
368- session . question ,
369- session ,
370- config . customModelApiUrl . trim ( ) || 'http://localhost:8000/v1/chat/completions' ,
371- config . customApiKey ,
372- config . customModelName ,
373- )
374- else
375- await generateAnswersWithCustomApi (
376- port ,
377- session . question ,
378- session ,
379- session . apiMode . customUrl ?. trim ( ) ||
380- config . customModelApiUrl . trim ( ) ||
381- 'http://localhost:8000/v1/chat/completions' ,
382- session . apiMode . apiKey ?. trim ( ) || config . customApiKey ,
383- session . apiMode . customName ,
384- )
385- } else if ( isUsingChatgptWebModel ( session ) ) {
367+ if ( isUsingChatgptWebModel ( session ) ) {
386368 console . debug ( '[background] Using ChatGPT Web Model' )
387369 let tabId
388370 if (
@@ -507,46 +489,15 @@ async function executeApi(session, port, config) {
507489 console . debug ( '[background] Using Gemini Web Model' )
508490 const cookies = await getBardCookies ( )
509491 await generateAnswersWithBardWebApi ( port , session . question , session , cookies )
510- } else if ( isUsingChatgptApiModel ( session ) ) {
511- console . debug ( '[background] Using OpenAI API Model ' )
512- await generateAnswersWithOpenAiApi ( port , session . question , session , config . apiKey )
492+ } else if ( isUsingOpenAICompatibleApiSession ( session ) ) {
493+ console . debug ( '[background] Using OpenAI-compatible API provider ' )
494+ await generateAnswersWithOpenAICompatibleApi ( port , session . question , session , config )
513495 } else if ( isUsingClaudeApiModel ( session ) ) {
514496 console . debug ( '[background] Using Anthropic API Model' )
515497 await generateAnswersWithClaudeApi ( port , session . question , session )
516- } else if ( isUsingMoonshotApiModel ( session ) ) {
517- console . debug ( '[background] Using Moonshot API Model' )
518- await generateAnswersWithMoonshotCompletionApi (
519- port ,
520- session . question ,
521- session ,
522- config . moonshotApiKey ,
523- )
524- } else if ( isUsingChatGLMApiModel ( session ) ) {
525- console . debug ( '[background] Using ChatGLM API Model' )
526- await generateAnswersWithChatGLMApi ( port , session . question , session )
527- } else if ( isUsingDeepSeekApiModel ( session ) ) {
528- console . debug ( '[background] Using DeepSeek API Model' )
529- await generateAnswersWithDeepSeekApi ( port , session . question , session , config . deepSeekApiKey )
530- } else if ( isUsingOllamaApiModel ( session ) ) {
531- console . debug ( '[background] Using Ollama API Model' )
532- await generateAnswersWithOllamaApi ( port , session . question , session )
533- } else if ( isUsingOpenRouterApiModel ( session ) ) {
534- console . debug ( '[background] Using OpenRouter API Model' )
535- await generateAnswersWithOpenRouterApi (
536- port ,
537- session . question ,
538- session ,
539- config . openRouterApiKey ,
540- )
541- } else if ( isUsingAimlApiModel ( session ) ) {
542- console . debug ( '[background] Using AIML API Model' )
543- await generateAnswersWithAimlApi ( port , session . question , session , config . aimlApiKey )
544498 } else if ( isUsingAzureOpenAiApiModel ( session ) ) {
545499 console . debug ( '[background] Using Azure OpenAI API Model' )
546500 await generateAnswersWithAzureOpenaiApi ( port , session . question , session )
547- } else if ( isUsingGptCompletionApiModel ( session ) ) {
548- console . debug ( '[background] Using GPT Completion API Model' )
549- await generateAnswersWithGptCompletionApi ( port , session . question , session , config . apiKey )
550501 } else if ( isUsingGithubThirdPartyApiModel ( session ) ) {
551502 console . debug ( '[background] Using Github Third Party API Model' )
552503 await generateAnswersWithWaylaidwandererApi ( port , session . question , session )
0 commit comments