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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
@@ -5,9 +5,10 @@ import { aiProviderSelectors } from '../selectors';
5
5
  describe('aiProviderSelectors', () => {
6
6
  const mockState: any = {
7
7
  aiProviderList: [
8
- { id: 'provider1', enabled: true, sort: 1 },
9
- { id: 'provider2', enabled: false, sort: 2 },
10
- { id: 'provider3', enabled: true, sort: 0 },
8
+ { id: 'provider1', enabled: true, sort: 1, source: 'builtin' },
9
+ { id: 'provider2', enabled: false, sort: 2, source: 'builtin' },
10
+ { id: 'provider3', enabled: true, sort: 0, source: 'builtin' },
11
+ { id: 'custom1', enabled: false, sort: 3, source: 'custom' },
11
12
  ],
12
13
  aiProviderDetail: {
13
14
  id: 'provider1',
@@ -56,16 +57,23 @@ describe('aiProviderSelectors', () => {
56
57
  it('should return enabled providers sorted by sort', () => {
57
58
  const result = aiProviderSelectors.enabledAiProviderList(mockState);
58
59
  expect(result).toEqual([
59
- { id: 'provider3', enabled: true, sort: 0 },
60
- { id: 'provider1', enabled: true, sort: 1 },
60
+ { id: 'provider3', enabled: true, sort: 0, source: 'builtin' },
61
+ { id: 'provider1', enabled: true, sort: 1, source: 'builtin' },
61
62
  ]);
62
63
  });
63
64
  });
64
65
 
65
66
  describe('disabledAiProviderList', () => {
66
- it('should return disabled providers', () => {
67
+ it('should return disabled builtin providers', () => {
67
68
  const result = aiProviderSelectors.disabledAiProviderList(mockState);
68
- expect(result).toEqual([{ id: 'provider2', enabled: false, sort: 2 }]);
69
+ expect(result).toEqual([{ id: 'provider2', enabled: false, sort: 2, source: 'builtin' }]);
70
+ });
71
+ });
72
+
73
+ describe('disabledCustomAiProviderList', () => {
74
+ it('should return disabled custom providers', () => {
75
+ const result = aiProviderSelectors.disabledCustomAiProviderList(mockState);
76
+ expect(result).toEqual([{ id: 'custom1', enabled: false, sort: 3, source: 'custom' }]);
69
77
  });
70
78
  });
71
79
 
@@ -1,6 +1,6 @@
1
1
  import { isProviderDisableBrowserRequest } from '@/config/modelProviders';
2
2
  import { AIProviderStoreState } from '@/store/aiInfra/initialState';
3
- import { AiProviderRuntimeConfig } from '@/types/aiProvider';
3
+ import { AiProviderRuntimeConfig, AiProviderSourceEnum } from '@/types/aiProvider';
4
4
  import { GlobalLLMProviderKey } from '@/types/user/settings';
5
5
 
6
6
  // List
@@ -8,7 +8,10 @@ const enabledAiProviderList = (s: AIProviderStoreState) =>
8
8
  s.aiProviderList.filter((item) => item.enabled).sort((a, b) => a.sort! - b.sort!);
9
9
 
10
10
  const disabledAiProviderList = (s: AIProviderStoreState) =>
11
- s.aiProviderList.filter((item) => !item.enabled);
11
+ s.aiProviderList.filter((item) => !item.enabled && item.source !== AiProviderSourceEnum.Custom);
12
+
13
+ const disabledCustomAiProviderList = (s: AIProviderStoreState) =>
14
+ s.aiProviderList.filter((item) => !item.enabled && item.source === AiProviderSourceEnum.Custom);
12
15
 
13
16
  const enabledImageModelList = (s: AIProviderStoreState) => s.enabledImageModelList || [];
14
17
 
@@ -116,6 +119,7 @@ const isInitAiProviderRuntimeState = (s: AIProviderStoreState) => !!s.isInitAiPr
116
119
  export const aiProviderSelectors = {
117
120
  activeProviderConfig,
118
121
  disabledAiProviderList,
122
+ disabledCustomAiProviderList,
119
123
  enabledAiProviderList,
120
124
  enabledImageModelList,
121
125
  isActiveProviderApiKeyNotEmpty,