✨ feat: Add XVERSE (元象大模型) model provider

lobehub · Jul 10, 2024 · a18f66a · a18f66a
1 parent 6e6ad3e
commit a18f66a
Show file tree

Hide file tree

Showing 16 changed files with 349 additions and 0 deletions.
diff --git a/src/app/(main)/settings/llm/ProviderList/providers.tsx b/src/app/(main)/settings/llm/ProviderList/providers.tsx
@@ -15,6 +15,7 @@ import {
   Stepfun,
   Together,
   Tongyi,
+  // Xverse, // TODO: Change to xverse
   ZeroOne,
   Zhipu,
 } from '@lobehub/icons';
@@ -39,6 +40,7 @@ import {
   StepfunProviderCard,
   TaichuProviderCard,
   TogetherAIProviderCard,
+  XverseProviderCard,
   ZeroOneProviderCard,
   ZhiPuProviderCard,
 } from '@/config/modelProviders';
@@ -184,6 +186,11 @@ export const useProviderList = (): ProviderItem[] => {
         docUrl: urlJoin(BASE_DOC_URL, 'taichu'),
         title: <AiMass.Combine size={ 28 } type={ 'color' } />,
       },
+      {
+        ...XverseProviderCard,
+        docUrl: urlJoin(BASE_DOC_URL, 'xverse'),
+        title: <Baichuan.Combine size={ 28 } type={ 'color' } />, // TODO: change icon to xverse
+      },
     ],
     [azureProvider, ollamaProvider, ollamaProvider, bedrockProvider],
   );

diff --git a/src/app/api/chat/agentRuntime.ts b/src/app/api/chat/agentRuntime.ts
@@ -184,6 +184,13 @@ const getLlmOptionsFromPayload = (provider: string, payload: JWTPayload) => {
 
       const apiKey = apiKeyManager.pick(payload?.apiKey || TAICHU_API_KEY);
 
+      return { apiKey };
+    }
+    case ModelProvider.Xverse: {
+      const { XVERSE_API_KEY } = getLLMConfig();
+
+      const apiKey = apiKeyManager.pick(payload?.apiKey || XVERSE_API_KEY);
+
       return { apiKey };
     }
   }

diff --git a/src/components/ModelIcon/index.tsx b/src/components/ModelIcon/index.tsx
@@ -30,6 +30,7 @@ import {
   Stepfun,
   Tongyi,
   Wenxin,
+  // Xverse, // TODO: change icon to xverse
   Yi,
 } from '@lobehub/icons';
 import { memo } from 'react';
@@ -68,6 +69,7 @@ const ModelIcon = memo<ModelProviderIconProps>(({ model: originModel, size = 12
   if (model.includes('dbrx')) return <Dbrx.Avatar size={size} />;
   if (model.includes('step')) return <Stepfun.Avatar size={size} />;
   if (model.includes('taichu')) return <AiMass.Avatar size={size} />;
+  if (model.includes('xverse')) return <Baichuan.Avatar size={size} />; // TODO: change icon to xverse
 
   // below: To be supported in providers, move up if supported
   if (model.includes('baichuan'))

diff --git a/src/components/ModelProviderIcon/index.tsx b/src/components/ModelProviderIcon/index.tsx
@@ -18,6 +18,7 @@ import {
   Stepfun,
   Together,
   Tongyi,
+  // Xverse, // TODO: change the icon to xverse
   ZeroOne,
   Zhipu,
 } from '@lobehub/icons';
@@ -124,6 +125,10 @@ const ModelProviderIcon = memo<ModelProviderIconProps>(({ provider }) => {
       return <AiMass size={20} />;
     }
 
+    case ModelProvider.Xverse: {
+      return <Baichuan size={20} />; // TODO: change icon to xverse
+    }
+
     default: {
       return null;
     }

diff --git a/src/components/ModelTag/ModelIcon.tsx b/src/components/ModelTag/ModelIcon.tsx
@@ -29,6 +29,7 @@ import {
   Stability,
   Tongyi,
   Wenxin,
+  // Xverse, // TODO: change icon to xverse
   Yi,
 } from '@lobehub/icons';
 import { memo } from 'react';
@@ -63,6 +64,7 @@ const ModelIcon = memo<ModelIconProps>(({ model, size = 12 }) => {
   if (model.includes('command')) return <Cohere size={size} />;
   if (model.includes('dbrx')) return <Dbrx size={size} />;
   if (model.includes('taichu')) return <AiMass size={size} />;
+  if (model.includes('xverse')) return <Baichuan size={size} />; // TODO: change icon to xverse
 
   // below: To be supported in providers, move up if supported
   if (model.includes('baichuan')) return <Baichuan size={size} />;

diff --git a/src/config/llm.ts b/src/config/llm.ts
@@ -88,6 +88,9 @@ export const getLLMConfig = () => {
 
       ENABLED_TAICHU: z.boolean(),
       TAICHU_API_KEY: z.string().optional(),
+
+      ENABLED_XVERSE: z.boolean(),
+      XVERSE_API_KEY: z.string().optional(),
     },
     runtimeEnv: {
       API_KEY_SELECT_MODE: process.env.API_KEY_SELECT_MODE,
@@ -167,6 +170,9 @@ export const getLLMConfig = () => {
 
       ENABLED_TAICHU: !!process.env.TAICHU_API_KEY,
       TAICHU_API_KEY: process.env.TAICHU_API_KEY,
+
+      ENABLED_XVERSE: !!process.env.XVERSE_API_KEY,
+      XVERSE_API_KEY: process.env.XVERSE_API_KEY,
     },
   });
 };

diff --git a/src/config/modelProviders/index.ts b/src/config/modelProviders/index.ts
@@ -18,6 +18,7 @@ import QwenProvider from './qwen';
 import StepfunProvider from './stepfun';
 import TaichuProvider from './taichu';
 import TogetherAIProvider from './togetherai';
+import XverseProvider from './xverse';
 import ZeroOneProvider from './zeroone';
 import ZhiPuProvider from './zhipu';
 
@@ -41,6 +42,7 @@ export const LOBE_DEFAULT_MODEL_LIST: ChatModelCard[] = [
   StepfunProvider.chatModels,
   BaichuanProvider.chatModels,
   TaichuProvider.chatModels,
+  XverseProvider.chatModels,
 ].flat();
 
 export const DEFAULT_MODEL_PROVIDER_LIST = [
@@ -64,6 +66,7 @@ export const DEFAULT_MODEL_PROVIDER_LIST = [
   StepfunProvider,
   BaichuanProvider,
   TaichuProvider,
+  XverseProvider,
 ];
 
 export const filterEnabledModels = (provider: ModelProviderCard) => {
@@ -93,5 +96,6 @@ export { default as QwenProviderCard } from './qwen';
 export { default as StepfunProviderCard } from './stepfun';
 export { default as TaichuProviderCard } from './taichu';
 export { default as TogetherAIProviderCard } from './togetherai';
+export { default as XverseProviderCard } from './xverse';
 export { default as ZeroOneProviderCard } from './zeroone';
 export { default as ZhiPuProviderCard } from './zhipu';
diff --git a/src/config/modelProviders/xverse.ts b/src/config/modelProviders/xverse.ts
@@ -0,0 +1,30 @@
+import { ModelProviderCard } from '@/types/llm';
+
+// ref https://chat.xverse.cn/home/index.html
+// ref https://chat.xverse.cn/docs/api-reference
+const Xverse: ModelProviderCard = {
+  chatModels: [
+    {
+      displayName: 'XVERSE-13B-2',
+      enabled: true,
+      functionCall: false,
+      id: 'XVERSE-13B-2',
+      maxOutput: 2048,
+      tokens: 16_384,
+    },
+    {
+      displayName: 'XVERSE-65B-2',
+      enabled: true,
+      functionCall: false,
+      id: 'XVERSE-65B-2',
+      maxOutput: 2048,
+      tokens: 8192,
+    },
+  ],
+  checkModel: 'XVERSE-13B-2',
+  id: 'xverse',
+  modelList: { showModelFetcher: true },
+  name: 'Xverse',
+};
+
+export default Xverse;
diff --git a/src/const/settings/llm.ts b/src/const/settings/llm.ts
@@ -16,6 +16,7 @@ import {
   StepfunProviderCard,
   TaichuProviderCard,
   TogetherAIProviderCard,
+  XverseProviderCard,
   ZeroOneProviderCard,
   ZhiPuProviderCard,
   filterEnabledModels,
@@ -96,6 +97,10 @@ export const DEFAULT_LLM_CONFIG: UserModelProviderConfig = {
     enabled: false,
     enabledModels: filterEnabledModels(TogetherAIProviderCard),
   },
+  xverse: {
+    enabled: false,
+    enabledModels: filterEnabledModels(XverseProviderCard),
+  },
   zeroone: {
     enabled: false,
     enabledModels: filterEnabledModels(ZeroOneProviderCard),

diff --git a/src/features/Conversation/Error/APIKeyForm/ProviderAvatar.tsx b/src/features/Conversation/Error/APIKeyForm/ProviderAvatar.tsx
@@ -14,6 +14,7 @@ import {
   Stepfun,
   Together,
   Tongyi,
+  // Xverse, // TODO: change icon to xverse
   ZeroOne,
   Zhipu,
 } from '@lobehub/icons';
@@ -90,6 +91,10 @@ const ProviderAvatar = memo<ProviderAvatarProps>(({ provider }) => {
       return <Together color={Together.colorPrimary} size={56} />;
     }
 
+    case ModelProvider.Xverse: {
+      return <Baichuan color={Baichuan.colorPrimary} size={56} />; // TODO: Change icon to xverse
+    }
+
     case ModelProvider.ZeroOne: {
       return <ZeroOne color={ZeroOne.colorPrimary} size={56} />;
     }

diff --git a/src/libs/agent-runtime/AgentRuntime.ts b/src/libs/agent-runtime/AgentRuntime.ts
@@ -27,6 +27,7 @@ import {
   ModelProvider,
   TextToImagePayload,
 } from './types';
+import { LobeXverseAI } from './xverse';
 import { LobeZeroOneAI } from './zeroone';
 import { LobeZhipuAI } from './zhipu';
 
@@ -121,6 +122,7 @@ class AgentRuntime {
       stepfun: Partial<ClientOptions>;
       taichu: Partial<ClientOptions>;
       togetherai: Partial<ClientOptions>;
+      xverse: Partial<ClientOptions>;
       zeroone: Partial<ClientOptions>;
       zhipu: Partial<ClientOptions>;
     }>,
@@ -233,6 +235,11 @@ class AgentRuntime {
         runtimeModel = new LobeTaichuAI(params.taichu ?? {});
         break
       }
+
+      case ModelProvider.Xverse: {
+        runtimeModel = new LobeXverseAI(params.xverse ?? {});
+        break
+      }
     }
 
     return new AgentRuntime(runtimeModel);

diff --git a/src/libs/agent-runtime/types/type.ts b/src/libs/agent-runtime/types/type.ts
@@ -40,6 +40,7 @@ export enum ModelProvider {
   Stepfun = 'stepfun',
   Taichu = 'taichu',
   TogetherAI = 'togetherai',
+  Xverse = 'xverse',
   ZeroOne = 'zeroone',
   ZhiPu = 'zhipu',
 }