🐛 fix: fix model list

lobehub · Jul 12, 2024 · 4112e0b · 4112e0b
1 parent b4c7257
commit 4112e0b
Show file tree

Hide file tree

Showing 4 changed files with 9 additions and 0 deletions.
diff --git a/src/config/llm.ts b/src/config/llm.ts
@@ -73,6 +73,7 @@ export const getLLMConfig = () => {
       AWS_ACCESS_KEY_ID: z.string().optional(),
       AWS_SECRET_ACCESS_KEY: z.string().optional(),
 
+      ENABLED_WENXIN: z.boolean(),
       WENXIN_ACCESS_KEY: z.string().optional(),
       WENXIN_SECRET_KEY: z.string().optional(),
 
@@ -158,6 +159,7 @@ export const getLLMConfig = () => {
       AWS_ACCESS_KEY_ID: process.env.AWS_ACCESS_KEY_ID,
       AWS_SECRET_ACCESS_KEY: process.env.AWS_SECRET_ACCESS_KEY,
 
+      ENABLED_WENXIN: !!process.env.WENXIN_ACCESS_KEY && !!process.env.WENXIN_SECRET_KEY,
       WENXIN_ACCESS_KEY: process.env.WENXIN_ACCESS_KEY,
       WENXIN_SECRET_KEY: process.env.WENXIN_SECRET_KEY,
 

diff --git a/src/config/modelProviders/index.ts b/src/config/modelProviders/index.ts
@@ -19,6 +19,7 @@ import QwenProvider from './qwen';
 import StepfunProvider from './stepfun';
 import TaichuProvider from './taichu';
 import TogetherAIProvider from './togetherai';
+import WenxinProvider from './wenxin';
 import ZeroOneProvider from './zeroone';
 import ZhiPuProvider from './zhipu';
 
@@ -60,6 +61,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   MinimaxProvider,
   MistralProvider,
   GroqProvider,
+  WenxinProvider,
   MoonshotProvider,
   ZeroOneProvider,
   ZhiPuProvider,

diff --git a/src/config/modelProviders/wenxin.ts b/src/config/modelProviders/wenxin.ts
@@ -4,6 +4,7 @@ import { ModelProviderCard } from '@/types/llm';
 const BaiduWenxin: ModelProviderCard = {
   chatModels: [
     {
+      displayName: 'ERNIE-4.0-8K',
       enabled: true,
       id: 'ERNIE-4.0-8K',
       tokens: 8192,
@@ -72,6 +73,7 @@ const BaiduWenxin: ModelProviderCard = {
     },
   ],
   checkModel: 'ERNIE-4.0-8K',
+  disableBrowserRequest: true,
   id: 'wenxin',
   name: 'Wenxin',
 };

diff --git a/src/server/globalConfig/index.ts b/src/server/globalConfig/index.ts
@@ -51,6 +51,8 @@ export const getServerGlobalConfig = () => {
     ENABLED_ZEROONE,
     ENABLED_TOGETHERAI,
     TOGETHERAI_MODEL_LIST,
+
+    ENABLED_WENXIN,
   } = getLLMConfig();
 
   const config: GlobalServerConfig = {
@@ -121,6 +123,7 @@ export const getServerGlobalConfig = () => {
           modelString: TOGETHERAI_MODEL_LIST,
         }),
       },
+      wenxin: { enabled: ENABLED_WENXIN },
       zeroone: { enabled: ENABLED_ZEROONE },
       zhipu: { enabled: ENABLED_ZHIPU },
     },