@lobehub/lobehub 2.0.0-next.135 → 2.0.0-next.136

package/src/store/aiInfra/slices/aiProvider/__tests__/selectors.test.ts

@@ -5,9 +5,10 @@ import { aiProviderSelectors } from '../selectors';
 describe('aiProviderSelectors', () => {
   const mockState: any = {
     aiProviderList: [
-      { id: 'provider1', enabled: true, sort: 1 },
-      { id: 'provider2', enabled: false, sort: 2 },
-      { id: 'provider3', enabled: true, sort: 0 },
+      { id: 'provider1', enabled: true, sort: 1, source: 'builtin' },
+      { id: 'provider2', enabled: false, sort: 2, source: 'builtin' },
+      { id: 'provider3', enabled: true, sort: 0, source: 'builtin' },
+      { id: 'custom1', enabled: false, sort: 3, source: 'custom' },
     ],
     aiProviderDetail: {
       id: 'provider1',
@@ -56,16 +57,23 @@ describe('aiProviderSelectors', () => {
     it('should return enabled providers sorted by sort', () => {
       const result = aiProviderSelectors.enabledAiProviderList(mockState);
       expect(result).toEqual([
-        { id: 'provider3', enabled: true, sort: 0 },
-        { id: 'provider1', enabled: true, sort: 1 },
+        { id: 'provider3', enabled: true, sort: 0, source: 'builtin' },
+        { id: 'provider1', enabled: true, sort: 1, source: 'builtin' },
       ]);
     });
   });
 
   describe('disabledAiProviderList', () => {
-    it('should return disabled providers', () => {
+    it('should return disabled builtin providers', () => {
       const result = aiProviderSelectors.disabledAiProviderList(mockState);
-      expect(result).toEqual([{ id: 'provider2', enabled: false, sort: 2 }]);
+      expect(result).toEqual([{ id: 'provider2', enabled: false, sort: 2, source: 'builtin' }]);
+    });
+  });
+
+  describe('disabledCustomAiProviderList', () => {
+    it('should return disabled custom providers', () => {
+      const result = aiProviderSelectors.disabledCustomAiProviderList(mockState);
+      expect(result).toEqual([{ id: 'custom1', enabled: false, sort: 3, source: 'custom' }]);
     });
   });
 

package/src/store/aiInfra/slices/aiProvider/selectors.ts

@@ -1,6 +1,6 @@
 import { isProviderDisableBrowserRequest } from '@/config/modelProviders';
 import { AIProviderStoreState } from '@/store/aiInfra/initialState';
-import { AiProviderRuntimeConfig } from '@/types/aiProvider';
+import { AiProviderRuntimeConfig, AiProviderSourceEnum } from '@/types/aiProvider';
 import { GlobalLLMProviderKey } from '@/types/user/settings';
 
 // List
@@ -8,7 +8,10 @@ const enabledAiProviderList = (s: AIProviderStoreState) =>
   s.aiProviderList.filter((item) => item.enabled).sort((a, b) => a.sort! - b.sort!);
 
 const disabledAiProviderList = (s: AIProviderStoreState) =>
-  s.aiProviderList.filter((item) => !item.enabled);
+  s.aiProviderList.filter((item) => !item.enabled && item.source !== AiProviderSourceEnum.Custom);
+
+const disabledCustomAiProviderList = (s: AIProviderStoreState) =>
+  s.aiProviderList.filter((item) => !item.enabled && item.source === AiProviderSourceEnum.Custom);
 
 const enabledImageModelList = (s: AIProviderStoreState) => s.enabledImageModelList || [];
 
@@ -116,6 +119,7 @@ const isInitAiProviderRuntimeState = (s: AIProviderStoreState) => !!s.isInitAiPr
 export const aiProviderSelectors = {
   activeProviderConfig,
   disabledAiProviderList,
+  disabledCustomAiProviderList,
   enabledAiProviderList,
   enabledImageModelList,
   isActiveProviderApiKeyNotEmpty,