✨ feat: Add LM Studio to ProviderIcon

lobehub · Oct 29, 2024 · e310622 · e310622
1 parent 9723053
commit e310622
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 3 deletions.
diff --git a/src/features/ProviderIcon/DefaultIcon.tsx b/src/features/ProviderIcon/DefaultIcon.tsx
@@ -1,6 +1,6 @@
 import { Icon } from '@lobehub/ui';
 import { useTheme } from 'antd-style';
-import { BrainCircuitIcon } from 'lucide-react';
+import { Radar } from 'lucide-react';
 import { CSSProperties, forwardRef } from 'react';
 
 interface DefaultIconProps {
@@ -15,7 +15,7 @@ const DefaultIcon = forwardRef<any, DefaultIconProps>(({ color, size = 12, ...re
   return (
     <Icon
       color={color || theme.colorTextDescription}
-      icon={BrainCircuitIcon}
+      icon={Radar}
       ref={ref}
       size={{ fontSize: size }}
       {...rest}

diff --git a/src/features/providerConfig.tsx b/src/features/providerConfig.tsx
@@ -20,6 +20,7 @@ import Google from '@/Google';
 import Groq from '@/Groq';
 import HuggingFace from '@/HuggingFace';
 import Hunyuan from '@/Hunyuan';
+import LmStudio from '@/LmStudio';
 import LobeHub from '@/LobeHub';
 import Minimax from '@/Minimax';
 import Mistral from '@/Mistral';
@@ -162,4 +163,5 @@ export const providerMappings: ProviderMapping[] = [
   },
   { Icon: SenseNova, combineMultiple: 0.95, keywords: [ModelProvider.SenseNova] },
   { Icon: HuggingFace, combineMultiple: 1.16, keywords: [ModelProvider.HuggingFace] },
+  { Icon: LmStudio, keywords: [ModelProvider.LmStudio] },
 ];
diff --git a/src/features/providerEnum.ts b/src/features/providerEnum.ts
@@ -13,6 +13,7 @@ export enum ModelProvider {
   Groq = 'groq',
   HuggingFace = 'huggingface',
   Hunyuan = 'hunyuan',
+  LmStudio = 'lmstudio',
   LobeHub = 'lobehub',
   Minimax = 'minimax',
   Mistral = 'mistral',
@@ -32,7 +33,7 @@ export enum ModelProvider {
   Upstage = 'upstage',
   Wenxin = 'wenxin',
   ZeroOne = 'zeroone',
-  ZhiPu = 'zhipu'
+  ZhiPu = 'zhipu',
 }
 
 export type ModelProviderKey = Lowercase<keyof typeof ModelProvider>;