💄 style: Add VertexAI to ProviderIcon

lobehub · Jan 21, 2025 · ca8859c · ca8859c
1 parent ff7fd33
commit ca8859c
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/src/features/ProviderCombine/index.tsx b/src/features/ProviderCombine/index.tsx
@@ -21,7 +21,7 @@ const ProviderCombine = forwardRef<HTMLDivElement, ProviderCombineProps>(
       const provider = originProvider.toLowerCase();
 
       for (const item of providerMappings) {
-        if (item.keywords.some((keyword) => new RegExp(keyword, 'i').test(provider))) {
+        if (item.keywords.some((keyword) => keyword.toLowerCase() === provider)) {
           return item;
         }
       }

diff --git a/src/features/ProviderIcon/index.tsx b/src/features/ProviderIcon/index.tsx
@@ -24,7 +24,7 @@ const ProviderIcon = forwardRef<any, ProviderIconProps>(
       const provider = originProvider.toLowerCase();
 
       for (const item of providerMappings) {
-        if (item.keywords.some((keyword) => new RegExp(keyword, 'i').test(provider))) {
+        if (item.keywords.some((keyword) => keyword.toLowerCase() === provider)) {
           return item;
         }
       }

diff --git a/src/features/providerConfig.tsx b/src/features/providerConfig.tsx
@@ -42,6 +42,7 @@ import Spark from '@/Spark';
 import Stepfun from '@/Stepfun';
 import Together from '@/Together';
 import Upstage from '@/Upstage';
+import VertexAI from '@/VertexAI';
 import Wenxin from '@/Wenxin';
 import WorkersAI from '@/WorkersAI';
 import XAI from '@/XAI';
@@ -189,4 +190,5 @@ export const providerMappings: ProviderMapping[] = [
   { Icon: Higress, keywords: [ModelProvider.Higress] },
   { 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
@@ -36,10 +36,11 @@ export enum ModelProvider {
   Taichu = 'taichu',
   TogetherAI = 'togetherai',
   Upstage = 'upstage',
+  VertexAI = 'vertexai',
   Wenxin = 'wenxin',
   XAI = 'xai',
   ZeroOne = 'zeroone',
-  ZhiPu = 'zhipu',
+  ZhiPu = 'zhipu'
 }
 
 export type ModelProviderKey = Lowercase<keyof typeof ModelProvider>;