@lobehub/lobehub 2.0.0-next.25 → 2.0.0-next.27

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.
Files changed (73) hide show
  1. package/CHANGELOG.md +50 -0
  2. package/changelog/v1.json +18 -0
  3. package/docs/development/database-schema.dbml +1 -0
  4. package/e2e/src/features/discover/detail-pages.feature +95 -0
  5. package/e2e/src/features/discover/interactions.feature +113 -0
  6. package/e2e/src/steps/discover/detail-pages.steps.ts +295 -0
  7. package/e2e/src/steps/discover/interactions.steps.ts +451 -0
  8. package/package.json +1 -1
  9. package/packages/database/migrations/0043_add_ai_model_settings.sql +1 -0
  10. package/packages/database/migrations/meta/0043_snapshot.json +8419 -0
  11. package/packages/database/migrations/meta/_journal.json +7 -0
  12. package/packages/database/src/core/migrations.json +10 -2
  13. package/packages/database/src/repositories/aiInfra/index.test.ts +198 -0
  14. package/packages/database/src/repositories/aiInfra/index.ts +2 -1
  15. package/packages/database/src/schemas/aiInfra.ts +2 -0
  16. package/packages/types/src/topic/topic.ts +14 -0
  17. package/src/server/routers/lambda/topic.ts +7 -1
  18. package/src/services/aiModel/index.test.ts +3 -3
  19. package/src/services/aiModel/index.ts +56 -2
  20. package/src/services/aiProvider/index.test.ts +2 -2
  21. package/src/services/aiProvider/index.ts +48 -2
  22. package/src/services/chatGroup/index.ts +66 -2
  23. package/src/services/export/index.ts +10 -2
  24. package/src/services/file/index.ts +61 -2
  25. package/src/services/import/index.ts +133 -2
  26. package/src/services/message/index.ts +176 -2
  27. package/src/services/message/{__tests__/server.test.ts → server.test.ts} +3 -3
  28. package/src/services/plugin/index.test.ts +8 -0
  29. package/src/services/plugin/index.ts +53 -2
  30. package/src/services/session/index.test.ts +8 -0
  31. package/src/services/session/index.ts +145 -2
  32. package/src/services/thread/index.test.ts +8 -0
  33. package/src/services/thread/index.ts +38 -2
  34. package/src/services/topic/index.test.ts +8 -0
  35. package/src/services/topic/index.ts +76 -2
  36. package/src/services/user/index.test.ts +8 -0
  37. package/src/services/user/index.ts +53 -2
  38. package/src/store/aiInfra/slices/aiModel/action.test.ts +17 -9
  39. package/src/store/chat/slices/aiChat/actions/__tests__/helpers.ts +4 -2
  40. package/src/store/chat/slices/topic/action.test.ts +1 -1
  41. package/src/store/chat/slices/topic/action.ts +1 -2
  42. package/src/store/chat/slices/topic/reducer.ts +1 -2
  43. package/src/store/file/slices/chat/action.ts +1 -4
  44. package/src/store/file/slices/fileManager/action.ts +2 -3
  45. package/src/store/session/slices/sessionGroup/action.test.ts +5 -5
  46. package/src/store/user/slices/common/action.test.ts +1 -1
  47. package/src/services/aiModel/server.test.ts +0 -122
  48. package/src/services/aiModel/server.ts +0 -51
  49. package/src/services/aiModel/type.ts +0 -32
  50. package/src/services/aiProvider/server.ts +0 -43
  51. package/src/services/aiProvider/type.ts +0 -27
  52. package/src/services/chatGroup/server.ts +0 -67
  53. package/src/services/chatGroup/type.ts +0 -22
  54. package/src/services/export/server.ts +0 -9
  55. package/src/services/export/type.ts +0 -5
  56. package/src/services/file/server.ts +0 -53
  57. package/src/services/file/type.ts +0 -13
  58. package/src/services/import/server.ts +0 -133
  59. package/src/services/import/type.ts +0 -17
  60. package/src/services/message/server.ts +0 -151
  61. package/src/services/message/type.ts +0 -55
  62. package/src/services/plugin/server.ts +0 -42
  63. package/src/services/plugin/type.ts +0 -23
  64. package/src/services/session/server.test.ts +0 -260
  65. package/src/services/session/server.ts +0 -125
  66. package/src/services/session/type.ts +0 -82
  67. package/src/services/thread/server.ts +0 -32
  68. package/src/services/thread/type.ts +0 -21
  69. package/src/services/topic/server.ts +0 -57
  70. package/src/services/topic/type.ts +0 -40
  71. package/src/services/user/server.test.ts +0 -149
  72. package/src/services/user/server.ts +0 -47
  73. package/src/services/user/type.ts +0 -21
@@ -0,0 +1,8 @@
1
+ import { describe } from 'vitest';
2
+ import { testService } from '~test-utils';
3
+
4
+ import { UserService } from './index';
5
+
6
+ describe('UserService', () => {
7
+ testService(UserService);
8
+ });
@@ -1,3 +1,54 @@
1
- import { ServerService } from './server';
1
+ import type { AdapterAccount } from 'next-auth/adapters';
2
+ import type { PartialDeep } from 'type-fest';
2
3
 
3
- export const userService = new ServerService();
4
+ import { lambdaClient } from '@/libs/trpc/client';
5
+ import { UserGuide, UserInitializationState, UserPreference } from '@/types/user';
6
+ import { UserSettings } from '@/types/user/settings';
7
+
8
+ export class UserService {
9
+ getUserRegistrationDuration = async (): Promise<{
10
+ createdAt: string;
11
+ duration: number;
12
+ updatedAt: string;
13
+ }> => {
14
+ return lambdaClient.user.getUserRegistrationDuration.query();
15
+ };
16
+
17
+ getUserState = async (): Promise<UserInitializationState> => {
18
+ return lambdaClient.user.getUserState.query();
19
+ };
20
+
21
+ getUserSSOProviders = async (): Promise<AdapterAccount[]> => {
22
+ return lambdaClient.user.getUserSSOProviders.query();
23
+ };
24
+
25
+ unlinkSSOProvider = async (provider: string, providerAccountId: string) => {
26
+ return lambdaClient.user.unlinkSSOProvider.mutate({ provider, providerAccountId });
27
+ };
28
+
29
+ makeUserOnboarded = async () => {
30
+ return lambdaClient.user.makeUserOnboarded.mutate();
31
+ };
32
+
33
+ updateAvatar = async (avatar: string) => {
34
+ return lambdaClient.user.updateAvatar.mutate(avatar);
35
+ };
36
+
37
+ updatePreference = async (preference: Partial<UserPreference>) => {
38
+ return lambdaClient.user.updatePreference.mutate(preference);
39
+ };
40
+
41
+ updateGuide = async (guide: Partial<UserGuide>) => {
42
+ return lambdaClient.user.updateGuide.mutate(guide);
43
+ };
44
+
45
+ updateUserSettings = async (value: PartialDeep<UserSettings>, signal?: AbortSignal) => {
46
+ return lambdaClient.user.updateSettings.mutate(value, { signal });
47
+ };
48
+
49
+ resetUserSettings = async () => {
50
+ return lambdaClient.user.resetSettings.mutate();
51
+ };
52
+ }
53
+
54
+ export const userService = new UserService();
@@ -101,7 +101,7 @@ describe('AiModelAction', () => {
101
101
  .mockResolvedValue(undefined);
102
102
  const serviceSpy = vi
103
103
  .spyOn(aiModelService, 'batchUpdateAiModels')
104
- .mockResolvedValue(undefined);
104
+ .mockResolvedValue(undefined as any);
105
105
 
106
106
  await act(async () => {
107
107
  await result.current.batchUpdateAiModels(models);
@@ -119,7 +119,7 @@ describe('AiModelAction', () => {
119
119
  const { result } = renderHook(() => useStore());
120
120
  const serviceSpy = vi
121
121
  .spyOn(aiModelService, 'batchUpdateAiModels')
122
- .mockResolvedValue(undefined);
122
+ .mockResolvedValue(undefined as any);
123
123
 
124
124
  await act(async () => {
125
125
  await result.current.batchUpdateAiModels([]);
@@ -137,7 +137,7 @@ describe('AiModelAction', () => {
137
137
  .mockResolvedValue(undefined);
138
138
  const serviceSpy = vi
139
139
  .spyOn(aiModelService, 'clearModelsByProvider')
140
- .mockResolvedValue(undefined);
140
+ .mockResolvedValue(undefined as any);
141
141
 
142
142
  await act(async () => {
143
143
  await result.current.clearModelsByProvider('test-provider');
@@ -154,7 +154,9 @@ describe('AiModelAction', () => {
154
154
  const refreshSpy = vi
155
155
  .spyOn(result.current, 'refreshAiModelList')
156
156
  .mockResolvedValue(undefined);
157
- const serviceSpy = vi.spyOn(aiModelService, 'clearRemoteModels').mockResolvedValue(undefined);
157
+ const serviceSpy = vi
158
+ .spyOn(aiModelService, 'clearRemoteModels')
159
+ .mockResolvedValue(undefined as any);
158
160
 
159
161
  await act(async () => {
160
162
  await result.current.clearRemoteModels('test-provider');
@@ -178,7 +180,9 @@ describe('AiModelAction', () => {
178
180
  const refreshSpy = vi
179
181
  .spyOn(result.current, 'refreshAiModelList')
180
182
  .mockResolvedValue(undefined);
181
- const serviceSpy = vi.spyOn(aiModelService, 'createAiModel').mockResolvedValue(undefined);
183
+ const serviceSpy = vi
184
+ .spyOn(aiModelService, 'createAiModel')
185
+ .mockResolvedValue(undefined as any);
182
186
 
183
187
  await act(async () => {
184
188
  await result.current.createNewAiModel(params);
@@ -349,7 +353,9 @@ describe('AiModelAction', () => {
349
353
  const refreshSpy = vi
350
354
  .spyOn(result.current, 'refreshAiModelList')
351
355
  .mockResolvedValue(undefined);
352
- const serviceSpy = vi.spyOn(aiModelService, 'deleteAiModel').mockResolvedValue(undefined);
356
+ const serviceSpy = vi
357
+ .spyOn(aiModelService, 'deleteAiModel')
358
+ .mockResolvedValue(undefined as any);
353
359
 
354
360
  await act(async () => {
355
361
  await result.current.removeAiModel('model-1', 'test-provider');
@@ -371,7 +377,7 @@ describe('AiModelAction', () => {
371
377
  .mockResolvedValue(undefined);
372
378
  const serviceSpy = vi
373
379
  .spyOn(aiModelService, 'toggleModelEnabled')
374
- .mockResolvedValue(undefined);
380
+ .mockResolvedValue(undefined as any);
375
381
 
376
382
  await act(async () => {
377
383
  await result.current.toggleModelEnabled({ enabled: true, id: 'model-1' });
@@ -395,7 +401,7 @@ describe('AiModelAction', () => {
395
401
  const { result } = renderHook(() => useStore());
396
402
  const serviceSpy = vi
397
403
  .spyOn(aiModelService, 'toggleModelEnabled')
398
- .mockResolvedValue(undefined);
404
+ .mockResolvedValue(undefined as any);
399
405
 
400
406
  await act(async () => {
401
407
  await result.current.toggleModelEnabled({ enabled: true, id: 'model-1' });
@@ -435,7 +441,9 @@ describe('AiModelAction', () => {
435
441
  const refreshSpy = vi
436
442
  .spyOn(result.current, 'refreshAiModelList')
437
443
  .mockResolvedValue(undefined);
438
- const serviceSpy = vi.spyOn(aiModelService, 'updateAiModel').mockResolvedValue(undefined);
444
+ const serviceSpy = vi
445
+ .spyOn(aiModelService, 'updateAiModel')
446
+ .mockResolvedValue(undefined as any);
439
447
 
440
448
  await act(async () => {
441
449
  await result.current.updateAiModelsConfig('model-1', 'test-provider', updateData);
@@ -63,10 +63,12 @@ export const spyOnMessageService = () => {
63
63
  const updateMessageSpy = vi
64
64
  .spyOn(messageService, 'updateMessage')
65
65
  .mockResolvedValue({ messages: [], success: true });
66
- const removeMessageSpy = vi.spyOn(messageService, 'removeMessage').mockResolvedValue(undefined);
66
+ const removeMessageSpy = vi
67
+ .spyOn(messageService, 'removeMessage')
68
+ .mockResolvedValue(undefined as any);
67
69
  const updateMessageErrorSpy = vi
68
70
  .spyOn(messageService, 'updateMessageError')
69
- .mockResolvedValue(undefined);
71
+ .mockResolvedValue(undefined as any);
70
72
 
71
73
  return {
72
74
  createMessageSpy,
@@ -254,7 +254,7 @@ describe('topic action', () => {
254
254
 
255
255
  const updateFavoriteSpy = vi
256
256
  .spyOn(topicService, 'updateTopic')
257
- .mockResolvedValue({ success: 1 });
257
+ .mockResolvedValue(undefined as any);
258
258
 
259
259
  const refreshTopicSpy = vi.spyOn(result.current, 'refreshTopic');
260
260
 
@@ -15,7 +15,6 @@ import { useClientDataSWR } from '@/libs/swr';
15
15
  import { chatService } from '@/services/chat';
16
16
  import { messageService } from '@/services/message';
17
17
  import { topicService } from '@/services/topic';
18
- import { CreateTopicParams } from '@/services/topic/type';
19
18
  import type { ChatStore } from '@/store/chat';
20
19
  import type { ChatStoreState } from '@/store/chat/initialState';
21
20
  import { messageMapKey } from '@/store/chat/utils/messageMapKey';
@@ -24,7 +23,7 @@ import { useSessionStore } from '@/store/session';
24
23
  import { sessionSelectors } from '@/store/session/selectors';
25
24
  import { useUserStore } from '@/store/user';
26
25
  import { systemAgentSelectors } from '@/store/user/selectors';
27
- import { ChatTopic } from '@/types/topic';
26
+ import { ChatTopic, CreateTopicParams } from '@/types/topic';
28
27
  import { merge } from '@/utils/merge';
29
28
  import { setNamespace } from '@/utils/storeDebug';
30
29
 
@@ -1,7 +1,6 @@
1
1
  import { produce } from 'immer';
2
2
 
3
- import { CreateTopicParams } from '@/services/topic/type';
4
- import { ChatTopic } from '@/types/topic';
3
+ import { ChatTopic, CreateTopicParams } from '@/types/topic';
5
4
 
6
5
  interface AddChatTopicAction {
7
6
  type: 'addTopic';
@@ -4,7 +4,6 @@ import { StateCreator } from 'zustand/vanilla';
4
4
  import { notification } from '@/components/AntdStaticMethods';
5
5
  import { FILE_UPLOAD_BLACKLIST } from '@/const/file';
6
6
  import { fileService } from '@/services/file';
7
- import { ServerService } from '@/services/file/server';
8
7
  import { ragService } from '@/services/rag';
9
8
  import { UPLOAD_NETWORK_ERROR } from '@/services/upload';
10
9
  import {
@@ -21,8 +20,6 @@ import { FileStore } from '../../store';
21
20
 
22
21
  const n = setNamespace('chat');
23
22
 
24
- const serverFileService = new ServerService();
25
-
26
23
  export interface FileAction {
27
24
  clearChatUploadFileList: () => void;
28
25
  dispatchChatUploadFileList: (payload: UploadFileListDispatch) => void;
@@ -71,7 +68,7 @@ export const createFileSlice: StateCreator<
71
68
  let fileItem: FileListItem | undefined = undefined;
72
69
 
73
70
  try {
74
- fileItem = await serverFileService.getFileItem(id);
71
+ fileItem = await fileService.getFileItem(id);
75
72
  } catch (e) {
76
73
  console.error('getFileItem Error:', e);
77
74
  continue;
@@ -4,8 +4,7 @@ import { StateCreator } from 'zustand/vanilla';
4
4
 
5
5
  import { FILE_UPLOAD_BLACKLIST, MAX_UPLOAD_FILE_COUNT } from '@/const/file';
6
6
  import { useClientDataSWR } from '@/libs/swr';
7
- import { fileService } from '@/services/file';
8
- import { ServerService } from '@/services/file/server';
7
+ import { fileService , FileService } from '@/services/file';
9
8
  import { ragService } from '@/services/rag';
10
9
  import {
11
10
  UploadFileListDispatch,
@@ -18,7 +17,7 @@ import { unzipFile } from '@/utils/unzipFile';
18
17
  import { FileStore } from '../../store';
19
18
  import { fileManagerSelectors } from './selectors';
20
19
 
21
- const serverFileService = new ServerService();
20
+ const serverFileService = new FileService();
22
21
 
23
22
  export interface FileManageAction {
24
23
  dispatchDockFileList: (payload: UploadFileListDispatch) => void;
@@ -42,7 +42,7 @@ describe('createSessionGroupSlice', () => {
42
42
  it('should clear session groups and refresh sessions', async () => {
43
43
  const spyOn = vi
44
44
  .spyOn(sessionService, 'removeSessionGroups')
45
- .mockResolvedValueOnce(undefined);
45
+ .mockResolvedValueOnce(undefined as any);
46
46
  const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
47
47
 
48
48
  const { result } = renderHook(() => useSessionStore());
@@ -59,7 +59,7 @@ describe('createSessionGroupSlice', () => {
59
59
  describe('removeSessionGroup', () => {
60
60
  it('should remove a session group and refresh sessions', async () => {
61
61
  const mockId = 'mock-id';
62
- vi.spyOn(sessionService, 'removeSessionGroup').mockResolvedValueOnce(undefined);
62
+ vi.spyOn(sessionService, 'removeSessionGroup').mockResolvedValueOnce(undefined as any);
63
63
  const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
64
64
 
65
65
  const { result } = renderHook(() => useSessionStore());
@@ -77,7 +77,7 @@ describe('createSessionGroupSlice', () => {
77
77
  it('should update a session group id and refresh sessions', async () => {
78
78
  const mockSessionId = 'session-id';
79
79
  const mockGroupId = 'group-id';
80
- vi.spyOn(sessionService, 'updateSession').mockResolvedValueOnce(undefined);
80
+ vi.spyOn(sessionService, 'updateSession').mockResolvedValueOnce(undefined as any);
81
81
  const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
82
82
 
83
83
  const { result } = renderHook(() => useSessionStore());
@@ -98,7 +98,7 @@ describe('createSessionGroupSlice', () => {
98
98
  const mockId = 'mock-id';
99
99
  const mockName = 'New Name';
100
100
  const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
101
- vi.spyOn(sessionService, 'updateSessionGroup').mockResolvedValueOnce(undefined);
101
+ vi.spyOn(sessionService, 'updateSessionGroup').mockResolvedValueOnce(undefined as any);
102
102
 
103
103
  const { result } = renderHook(() => useSessionStore());
104
104
 
@@ -117,7 +117,7 @@ describe('createSessionGroupSlice', () => {
117
117
  { id: 'id1', sort: 0 },
118
118
  { id: 'id2', sort: 1 },
119
119
  ];
120
- vi.spyOn(sessionService, 'updateSessionGroupOrder').mockResolvedValueOnce(undefined);
120
+ vi.spyOn(sessionService, 'updateSessionGroupOrder').mockResolvedValueOnce(undefined as any);
121
121
  const spyOnRefreshSessions = vi.spyOn(useSessionStore.getState(), 'refreshSessions');
122
122
 
123
123
  const { result } = renderHook(() => useSessionStore());
@@ -30,7 +30,7 @@ describe('createCommonSlice', () => {
30
30
  const avatar = 'data:image/png;base64,';
31
31
 
32
32
  const spyOn = vi.spyOn(result.current, 'refreshUserState');
33
- const updateAvatarSpy = vi.spyOn(userService, 'updateAvatar').mockResolvedValue(undefined);
33
+ const updateAvatarSpy = vi.spyOn(userService, 'updateAvatar').mockResolvedValue({} as any);
34
34
 
35
35
  await act(async () => {
36
36
  await result.current.updateAvatar(avatar);
@@ -1,122 +0,0 @@
1
- import { AiProviderModelListItem } from 'model-bank';
2
- import { describe, expect, it, vi } from 'vitest';
3
-
4
- import { lambdaClient } from '@/libs/trpc/client';
5
-
6
- import { ServerService } from './server';
7
-
8
- vi.mock('@/libs/trpc/client', () => ({
9
- lambdaClient: {
10
- aiModel: {
11
- createAiModel: { mutate: vi.fn() },
12
- getAiProviderModelList: { query: vi.fn() },
13
- getAiModelById: { query: vi.fn() },
14
- toggleModelEnabled: { mutate: vi.fn() },
15
- updateAiModel: { mutate: vi.fn() },
16
- batchUpdateAiModels: { mutate: vi.fn() },
17
- batchToggleAiModels: { mutate: vi.fn() },
18
- clearModelsByProvider: { mutate: vi.fn() },
19
- clearRemoteModels: { mutate: vi.fn() },
20
- updateAiModelOrder: { mutate: vi.fn() },
21
- removeAiModel: { mutate: vi.fn() },
22
- },
23
- },
24
- }));
25
-
26
- describe('ServerService', () => {
27
- const service = new ServerService();
28
-
29
- it('should create AI model', async () => {
30
- const params = {
31
- id: 'test-id',
32
- providerId: 'test-provider',
33
- displayName: 'Test Model',
34
- };
35
- await service.createAiModel(params);
36
- expect(vi.mocked(lambdaClient.aiModel.createAiModel.mutate)).toHaveBeenCalledWith(params);
37
- });
38
-
39
- it('should get AI provider model list', async () => {
40
- await service.getAiProviderModelList('123');
41
- expect(vi.mocked(lambdaClient.aiModel.getAiProviderModelList.query)).toHaveBeenCalledWith({
42
- id: '123',
43
- });
44
- });
45
-
46
- it('should get AI model by id', async () => {
47
- await service.getAiModelById('123');
48
- expect(vi.mocked(lambdaClient.aiModel.getAiModelById.query)).toHaveBeenCalledWith({
49
- id: '123',
50
- });
51
- });
52
-
53
- it('should toggle model enabled', async () => {
54
- const params = { id: '123', providerId: 'test', enabled: true };
55
- await service.toggleModelEnabled(params);
56
- expect(vi.mocked(lambdaClient.aiModel.toggleModelEnabled.mutate)).toHaveBeenCalledWith(params);
57
- });
58
-
59
- it('should update AI model', async () => {
60
- const value = { contextWindowTokens: 4000, displayName: 'Updated Model' };
61
- await service.updateAiModel('123', 'openai', value);
62
- expect(vi.mocked(lambdaClient.aiModel.updateAiModel.mutate)).toHaveBeenCalledWith({
63
- id: '123',
64
- providerId: 'openai',
65
- value,
66
- });
67
- });
68
-
69
- it('should batch update AI models', async () => {
70
- const models: AiProviderModelListItem[] = [
71
- {
72
- id: '123',
73
- enabled: true,
74
- type: 'chat',
75
- },
76
- ];
77
- await service.batchUpdateAiModels('provider1', models);
78
- expect(vi.mocked(lambdaClient.aiModel.batchUpdateAiModels.mutate)).toHaveBeenCalledWith({
79
- id: 'provider1',
80
- models,
81
- });
82
- });
83
-
84
- it('should batch toggle AI models', async () => {
85
- const models = ['123', '456'];
86
- await service.batchToggleAiModels('provider1', models, true);
87
- expect(vi.mocked(lambdaClient.aiModel.batchToggleAiModels.mutate)).toHaveBeenCalledWith({
88
- id: 'provider1',
89
- models,
90
- enabled: true,
91
- });
92
- });
93
-
94
- it('should clear models by provider', async () => {
95
- await service.clearModelsByProvider('provider1');
96
- expect(vi.mocked(lambdaClient.aiModel.clearModelsByProvider.mutate)).toHaveBeenCalledWith({
97
- providerId: 'provider1',
98
- });
99
- });
100
-
101
- it('should clear remote models', async () => {
102
- await service.clearRemoteModels('provider1');
103
- expect(vi.mocked(lambdaClient.aiModel.clearRemoteModels.mutate)).toHaveBeenCalledWith({
104
- providerId: 'provider1',
105
- });
106
- });
107
-
108
- it('should update AI model order', async () => {
109
- const items = [{ id: '123', sort: 1 }];
110
- await service.updateAiModelOrder('provider1', items);
111
- expect(vi.mocked(lambdaClient.aiModel.updateAiModelOrder.mutate)).toHaveBeenCalledWith({
112
- providerId: 'provider1',
113
- sortMap: items,
114
- });
115
- });
116
-
117
- it('should delete AI model', async () => {
118
- const params = { id: '123', providerId: 'openai' };
119
- await service.deleteAiModel(params);
120
- expect(vi.mocked(lambdaClient.aiModel.removeAiModel.mutate)).toHaveBeenCalledWith(params);
121
- });
122
- });
@@ -1,51 +0,0 @@
1
- import { lambdaClient } from '@/libs/trpc/client';
2
- import { IAiModelService } from '@/services/aiModel/type';
3
-
4
- export class ServerService implements IAiModelService {
5
- createAiModel: IAiModelService['createAiModel'] = async (params) => {
6
- return lambdaClient.aiModel.createAiModel.mutate(params);
7
- };
8
-
9
- getAiProviderModelList: IAiModelService['getAiProviderModelList'] = async (id) => {
10
- return lambdaClient.aiModel.getAiProviderModelList.query({ id });
11
- };
12
-
13
- getAiModelById: IAiModelService['getAiModelById'] = async (id) => {
14
- return lambdaClient.aiModel.getAiModelById.query({ id });
15
- };
16
-
17
- toggleModelEnabled: IAiModelService['toggleModelEnabled'] = async (params) => {
18
- return lambdaClient.aiModel.toggleModelEnabled.mutate(params);
19
- };
20
-
21
- updateAiModel: IAiModelService['updateAiModel'] = async (id, providerId, value) => {
22
- return lambdaClient.aiModel.updateAiModel.mutate({ id, providerId, value });
23
- };
24
-
25
- batchUpdateAiModels: IAiModelService['batchUpdateAiModels'] = async (id, models) => {
26
- return lambdaClient.aiModel.batchUpdateAiModels.mutate({ id, models });
27
- };
28
-
29
- batchToggleAiModels: IAiModelService['batchToggleAiModels'] = async (id, models, enabled) => {
30
- return lambdaClient.aiModel.batchToggleAiModels.mutate({ enabled, id, models });
31
- };
32
-
33
- clearModelsByProvider: IAiModelService['clearModelsByProvider'] = async (providerId) => {
34
- return lambdaClient.aiModel.clearModelsByProvider.mutate({ providerId });
35
- };
36
-
37
- clearRemoteModels: IAiModelService['clearRemoteModels'] = async (providerId) => {
38
- return lambdaClient.aiModel.clearRemoteModels.mutate({ providerId });
39
- };
40
-
41
- updateAiModelOrder: IAiModelService['updateAiModelOrder'] = async (providerId, items) => {
42
- return lambdaClient.aiModel.updateAiModelOrder.mutate({ providerId, sortMap: items });
43
- };
44
-
45
- deleteAiModel: IAiModelService['deleteAiModel'] = async (params: {
46
- id: string;
47
- providerId: string;
48
- }) => {
49
- return lambdaClient.aiModel.removeAiModel.mutate(params);
50
- };
51
- }
@@ -1,32 +0,0 @@
1
- /* eslint-disable typescript-sort-keys/interface */
2
- import {
3
- AiModelSortMap,
4
- AiProviderModelListItem,
5
- CreateAiModelParams,
6
- ToggleAiModelEnableParams,
7
- UpdateAiModelParams,
8
- } from 'model-bank';
9
-
10
- export interface IAiModelService {
11
- createAiModel: (params: CreateAiModelParams) => Promise<any>;
12
-
13
- getAiProviderModelList: (id: string) => Promise<AiProviderModelListItem[]>;
14
-
15
- getAiModelById: (id: string) => Promise<any>;
16
-
17
- toggleModelEnabled: (params: ToggleAiModelEnableParams) => Promise<any>;
18
-
19
- updateAiModel: (id: string, providerId: string, value: UpdateAiModelParams) => Promise<any>;
20
-
21
- batchUpdateAiModels: (id: string, models: AiProviderModelListItem[]) => Promise<any>;
22
-
23
- batchToggleAiModels: (id: string, models: string[], enabled: boolean) => Promise<any>;
24
-
25
- clearRemoteModels: (providerId: string) => Promise<any>;
26
-
27
- clearModelsByProvider: (providerId: string) => Promise<any>;
28
-
29
- updateAiModelOrder: (providerId: string, items: AiModelSortMap[]) => Promise<any>;
30
-
31
- deleteAiModel: (params: { id: string; providerId: string }) => Promise<any>;
32
- }
@@ -1,43 +0,0 @@
1
- import { lambdaClient } from '@/libs/trpc/client';
2
-
3
- import { IAiProviderService } from './type';
4
-
5
- export class ServerService implements IAiProviderService {
6
- createAiProvider: IAiProviderService['createAiProvider'] = async (params) => {
7
- return lambdaClient.aiProvider.createAiProvider.mutate(params);
8
- };
9
-
10
- getAiProviderList: IAiProviderService['getAiProviderList'] = async () => {
11
- return lambdaClient.aiProvider.getAiProviderList.query();
12
- };
13
-
14
- getAiProviderById: IAiProviderService['getAiProviderById'] = async (id) => {
15
- return lambdaClient.aiProvider.getAiProviderById.query({ id });
16
- };
17
-
18
- toggleProviderEnabled: IAiProviderService['toggleProviderEnabled'] = async (id, enabled) => {
19
- return lambdaClient.aiProvider.toggleProviderEnabled.mutate({ enabled, id });
20
- };
21
-
22
- updateAiProvider: IAiProviderService['updateAiProvider'] = async (id, value) => {
23
- return lambdaClient.aiProvider.updateAiProvider.mutate({ id, value });
24
- };
25
-
26
- updateAiProviderConfig: IAiProviderService['updateAiProviderConfig'] = async (id, value) => {
27
- return lambdaClient.aiProvider.updateAiProviderConfig.mutate({ id, value });
28
- };
29
-
30
- updateAiProviderOrder: IAiProviderService['updateAiProviderOrder'] = async (items) => {
31
- return lambdaClient.aiProvider.updateAiProviderOrder.mutate({ sortMap: items });
32
- };
33
-
34
- deleteAiProvider: IAiProviderService['deleteAiProvider'] = async (id) => {
35
- return lambdaClient.aiProvider.removeAiProvider.mutate({ id });
36
- };
37
-
38
- getAiProviderRuntimeState: IAiProviderService['getAiProviderRuntimeState'] = async (
39
- isLogin?: boolean,
40
- ) => {
41
- return lambdaClient.aiProvider.getAiProviderRuntimeState.query({ isLogin });
42
- };
43
- }
@@ -1,27 +0,0 @@
1
- import {
2
- AiProviderDetailItem,
3
- AiProviderRuntimeState,
4
- AiProviderSortMap,
5
- CreateAiProviderParams,
6
- UpdateAiProviderConfigParams,
7
- } from '@/types/aiProvider';
8
-
9
- export interface IAiProviderService {
10
- createAiProvider: (params: CreateAiProviderParams) => Promise<any>;
11
-
12
- deleteAiProvider: (id: string) => Promise<any>;
13
-
14
- getAiProviderById: (id: string) => Promise<AiProviderDetailItem | undefined>;
15
-
16
- getAiProviderList: () => Promise<any>;
17
-
18
- getAiProviderRuntimeState: (isLogin?: boolean) => Promise<AiProviderRuntimeState>;
19
-
20
- toggleProviderEnabled: (id: string, enabled: boolean) => Promise<any>;
21
-
22
- updateAiProvider: (id: string, value: any) => Promise<any>;
23
-
24
- updateAiProviderConfig: (id: string, value: UpdateAiProviderConfigParams) => Promise<any>;
25
-
26
- updateAiProviderOrder: (items: AiProviderSortMap[]) => Promise<any>;
27
- }
@@ -1,67 +0,0 @@
1
- import {
2
- ChatGroupAgentItem,
3
- ChatGroupItem,
4
- NewChatGroup,
5
- NewChatGroupAgent,
6
- } from '@/database/schemas';
7
- import { lambdaClient } from '@/libs/trpc/client';
8
-
9
- import { IChatGroupService } from './type';
10
-
11
- export class ServerService implements IChatGroupService {
12
- createGroup(params: Omit<NewChatGroup, 'userId'>): Promise<ChatGroupItem> {
13
- return lambdaClient.group.createGroup.mutate({
14
- ...params,
15
- config: params.config as any,
16
- });
17
- }
18
-
19
- updateGroup(id: string, value: Partial<ChatGroupItem>): Promise<ChatGroupItem> {
20
- return lambdaClient.group.updateGroup.mutate({
21
- id,
22
- value: {
23
- ...value,
24
- config: value.config as any,
25
- },
26
- });
27
- }
28
-
29
- deleteGroup(id: string): Promise<any> {
30
- return lambdaClient.group.deleteGroup.mutate({ id });
31
- }
32
-
33
- getGroup(id: string): Promise<ChatGroupItem | undefined> {
34
- return lambdaClient.group.getGroup.query({ id });
35
- }
36
-
37
- getGroups(): Promise<ChatGroupItem[]> {
38
- return lambdaClient.group.getGroups.query();
39
- }
40
-
41
- addAgentsToGroup(groupId: string, agentIds: string[]): Promise<ChatGroupAgentItem[]> {
42
- return lambdaClient.group.addAgentsToGroup.mutate({ agentIds, groupId });
43
- }
44
-
45
- removeAgentsFromGroup(groupId: string, agentIds: string[]): Promise<any> {
46
- return lambdaClient.group.removeAgentsFromGroup.mutate({ agentIds, groupId });
47
- }
48
-
49
- updateAgentInGroup(
50
- groupId: string,
51
- agentId: string,
52
- updates: Partial<Pick<NewChatGroupAgent, 'order' | 'role'>>,
53
- ): Promise<ChatGroupAgentItem> {
54
- return lambdaClient.group.updateAgentInGroup.mutate({
55
- agentId,
56
- groupId,
57
- updates: {
58
- order: updates.order === null ? undefined : updates.order,
59
- role: updates.role === null ? undefined : updates.role,
60
- },
61
- });
62
- }
63
-
64
- getGroupAgents(groupId: string): Promise<ChatGroupAgentItem[]> {
65
- return lambdaClient.group.getGroupAgents.query({ groupId });
66
- }
67
- }