diff --git a/src/features/providerConfig.tsx b/src/features/providerConfig.tsx index 282db93..a9a100b 100644 --- a/src/features/providerConfig.tsx +++ b/src/features/providerConfig.tsx @@ -44,6 +44,7 @@ import TencentCloud from '@/TencentCloud'; import Together from '@/Together'; import Upstage from '@/Upstage'; import VertexAI from '@/VertexAI'; +import Vllm from '@/Vllm'; import Wenxin from '@/Wenxin'; import WorkersAI from '@/WorkersAI'; import XAI from '@/XAI'; @@ -192,6 +193,7 @@ export const providerMappings: ProviderMapping[] = [ }, { Icon: InternLM, combineMultiple: 0.95, keywords: [ModelProvider.InternLM] }, { Icon: Higress, keywords: [ModelProvider.Higress] }, + { Icon: Vllm, keywords: [ModelProvider.VLLM] }, { Icon: GiteeAI, combineMultiple: 0.95, keywords: [ModelProvider.GiteeAI] }, { Icon: ModelScope, combineMultiple: 0.95, keywords: [ModelProvider.ModelScope] }, { Icon: VertexAI, keywords: [ModelProvider.VertexAI] }, diff --git a/src/features/providerEnum.ts b/src/features/providerEnum.ts index 35e70f7..e13b0f5 100644 --- a/src/features/providerEnum.ts +++ b/src/features/providerEnum.ts @@ -38,6 +38,7 @@ export enum ModelProvider { TencentCloud = 'tencentcloud', TogetherAI = 'togetherai', Upstage = 'upstage', + VLLM = 'vllm', VertexAI = 'vertexai', Wenxin = 'wenxin', XAI = 'xai',