✨ feat: Support Doubao Models

src/app/(main)/settings/llm/ProviderList/providers.tsx

@@ -6,6 +6,7 @@ import {
   AnthropicProviderCard,
   BaichuanProviderCard,
   DeepSeekProviderCard,
+  DoubaoProviderCard,
   FireworksAIProviderCard,
   GoogleProviderCard,
   GroqProviderCard,
@@ -56,6 +57,7 @@ export const useProviderList = (): ProviderItem[] => {
       BedrockProvider,
       GoogleProviderCard,
       DeepSeekProviderCard,
+      DoubaoProviderCard,
       HuggingFaceProvider,
       OpenRouterProviderCard,
       GithubProvider,
@@ -93,4 +95,4 @@ export const useProviderList = (): ProviderItem[] => {
       SenseNovaProvider,
     ],
   );
-};
+};

src/config/llm.ts

@@ -149,6 +149,10 @@ export const getLLMConfig = () => {
       SENSENOVA_ACCESS_KEY_ID: z.string().optional(),
       SENSENOVA_ACCESS_KEY_SECRET: z.string().optional(),
       SENSENOVA_MODEL_LIST: z.string().optional(),
+
+      ENABLED_DOUBAO: z.boolean(),
+      ARK_API_KEY: z.string().optional(),
+      ARK_MODEL_LIST: z.string().optional(),
     },
     runtimeEnv: {
       API_KEY_SELECT_MODE: process.env.API_KEY_SELECT_MODE,
@@ -295,6 +299,10 @@ export const getLLMConfig = () => {
       SENSENOVA_ACCESS_KEY_ID: process.env.SENSENOVA_ACCESS_KEY_ID,
       SENSENOVA_ACCESS_KEY_SECRET: process.env.SENSENOVA_ACCESS_KEY_SECRET,
       SENSENOVA_MODEL_LIST: process.env.SENSENOVA_MODEL_LIST,
+
+      ENABLED_DOUBAO: !!(process.env.ARK_API_KEY && process.env.ARK_MODEL_LIST),
+      ARK_API_KEY: process.env.ARK_API_KEY,
+      ARK_MODEL_LIST: process.env.ARK_MODEL_LIST,
     },
   });
 };

src/config/modelProviders/doubao.ts

@@ -0,0 +1,12 @@
+import { ModelProviderCard } from '@/types/llm';
+
+// ref https://www.volcengine.com/docs/82379/1330310
+const Doubao: ModelProviderCard = {
+  chatModels: [],
+  disableBrowserRequest: true, // CORS error
+  id: 'doubao',
+  name: '豆包',
+  url: 'https://www.volcengine.com/product/doubao',
+};
+
+export default Doubao;

src/config/modelProviders/index.ts

@@ -7,6 +7,7 @@ import AzureProvider from './azure';
 import BaichuanProvider from './baichuan';
 import BedrockProvider from './bedrock';
 import DeepSeekProvider from './deepseek';
+import DoubaoProvider from './doubao';
 import FireworksAIProvider from './fireworksai';
 import GithubProvider from './github';
 import GoogleProvider from './google';
@@ -65,6 +66,7 @@ export const LOBE_DEFAULT_MODEL_LIST: ChatModelCard[] = [
   HunyuanProvider.chatModels,
   WenxinProvider.chatModels,
   SenseNovaProvider.chatModels,
+  DoubaoProvider.chatModels,
 ].flat();
 
 export const DEFAULT_MODEL_PROVIDER_LIST = [
@@ -88,6 +90,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   UpstageProvider,
   QwenProvider,
   WenxinProvider,
+  DoubaoProvider,
   HunyuanProvider,
   SparkProvider,
   ZhiPuProvider,
@@ -118,6 +121,7 @@ export { default as AzureProviderCard } from './azure';
 export { default as BaichuanProviderCard } from './baichuan';
 export { default as BedrockProviderCard } from './bedrock';
 export { default as DeepSeekProviderCard } from './deepseek';
+export { default as DoubaoProviderCard } from './doubao';
 export { default as FireworksAIProviderCard } from './fireworksai';
 export { default as GithubProviderCard } from './github';
 export { default as GoogleProviderCard } from './google';

src/const/settings/llm.ts

@@ -63,6 +63,9 @@ export const DEFAULT_LLM_CONFIG: UserModelProviderConfig = {
     enabled: false,
     enabledModels: filterEnabledModels(DeepSeekProviderCard),
   },
+  doubao: {
+    enabled: false,
+  },
   fireworksai: {
     enabled: false,
     enabledModels: filterEnabledModels(FireworksAIProviderCard),

src/libs/agent-runtime/AgentRuntime.ts

@@ -10,6 +10,7 @@ import { LobeAzureOpenAI } from './azureOpenai';
 import { LobeBaichuanAI } from './baichuan';
 import { LobeBedrockAI, LobeBedrockAIParams } from './bedrock';
 import { LobeDeepSeekAI } from './deepseek';
+import { LobeDoubaoAI } from './doubao';
 import { LobeFireworksAI } from './fireworksai';
 import { LobeGithubAI } from './github';
 import { LobeGoogleAI } from './google';
@@ -132,6 +133,7 @@ class AgentRuntime {
       baichuan: Partial<ClientOptions>;
       bedrock: Partial<LobeBedrockAIParams>;
       deepseek: Partial<ClientOptions>;
+      doubao: Partial<ClientOptions>;
       fireworksai: Partial<ClientOptions>;
       github: Partial<ClientOptions>;
       google: { apiKey?: string; baseURL?: string };
@@ -321,6 +323,11 @@ class AgentRuntime {
         runtimeModel = await LobeSenseNovaAI.fromAPIKey(params.sensenova);
         break;
       }
+
+      case ModelProvider.Doubao: {
+        runtimeModel = new LobeDoubaoAI(params.doubao ?? {});
+        break;
+      }
     }
 
     return new AgentRuntime(runtimeModel);

src/libs/agent-runtime/doubao/index.ts

@@ -0,0 +1,10 @@
+import { ModelProvider } from '../types';
+import { LobeOpenAICompatibleFactory } from '../utils/openaiCompatibleFactory';
+
+export const LobeDoubaoAI = LobeOpenAICompatibleFactory({
+  baseURL: 'https://ark.cn-beijing.volces.com/api/v3',
+  debug: {
+    chatCompletion: () => process.env.DEBUG_DOUBAO_CHAT_COMPLETION === '1',
+  },
+  provider: ModelProvider.Doubao,
+});

src/libs/agent-runtime/types/type.ts

@@ -29,6 +29,7 @@ export enum ModelProvider {
   Baichuan = 'baichuan',
   Bedrock = 'bedrock',
   DeepSeek = 'deepseek',
+  Doubao = 'doubao',
   FireworksAI = 'fireworksai',
   Github = 'github',
   Google = 'google',

src/server/globalConfig/index.ts

@@ -143,6 +143,9 @@ export const getServerGlobalConfig = () => {
 
     ENABLED_HUGGINGFACE,
     HUGGINGFACE_MODEL_LIST,
+
+    ENABLED_DOUBAO,
+    ARK_MODEL_LIST,
   } = getLLMConfig();
 
   const config: GlobalServerConfig = {
@@ -210,6 +213,13 @@ export const getServerGlobalConfig = () => {
           modelString: DEEPSEEK_MODEL_LIST,
         }),
       },
+      doubao: { enabled: ENABLED_DOUBAO,
+        enabledModels: extractEnabledModels(ARK_MODEL_LIST),
+        serverModelCards: transformToChatModelCards({
+          defaultChatModels: [],
+          modelString: ARK_MODEL_LIST,
+        }),
+      },
       fireworksai: {
         enabled: ENABLED_FIREWORKSAI,
         enabledModels: extractEnabledModels(FIREWORKSAI_MODEL_LIST),

src/server/modules/AgentRuntime/index.ts

@@ -271,6 +271,13 @@ const getLlmOptionsFromPayload = (provider: string, payload: JWTPayload) => {
 
       const apiKey = sensenovaAccessKeyID + ':' + sensenovaAccessKeySecret;
 
+      return { apiKey };
+    }
+    case ModelProvider.Doubao: {
+      const { ARK_API_KEY } = getLLMConfig();
+
+      const apiKey = apiKeyManager.pick(payload?.apiKey || ARK_API_KEY);
+
       return { apiKey };
     }
   }

src/types/user/settings/keyVaults.ts

@@ -34,6 +34,7 @@ export interface UserKeyVaults {
   baichuan?: OpenAICompatibleKeyVault;
   bedrock?: AWSBedrockKeyVault;
   deepseek?: OpenAICompatibleKeyVault;
+  doubao?: OpenAICompatibleKeyVault;
   fireworksai?: OpenAICompatibleKeyVault;
   github?: OpenAICompatibleKeyVault;
   google?: OpenAICompatibleKeyVault;