@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.
- package/CHANGELOG.md +50 -0
- package/changelog/v1.json +18 -0
- package/docs/development/database-schema.dbml +1 -0
- package/e2e/src/features/discover/detail-pages.feature +95 -0
- package/e2e/src/features/discover/interactions.feature +113 -0
- package/e2e/src/steps/discover/detail-pages.steps.ts +295 -0
- package/e2e/src/steps/discover/interactions.steps.ts +451 -0
- package/package.json +1 -1
- package/packages/database/migrations/0043_add_ai_model_settings.sql +1 -0
- package/packages/database/migrations/meta/0043_snapshot.json +8419 -0
- package/packages/database/migrations/meta/_journal.json +7 -0
- package/packages/database/src/core/migrations.json +10 -2
- package/packages/database/src/repositories/aiInfra/index.test.ts +198 -0
- package/packages/database/src/repositories/aiInfra/index.ts +2 -1
- package/packages/database/src/schemas/aiInfra.ts +2 -0
- package/packages/types/src/topic/topic.ts +14 -0
- package/src/server/routers/lambda/topic.ts +7 -1
- package/src/services/aiModel/index.test.ts +3 -3
- package/src/services/aiModel/index.ts +56 -2
- package/src/services/aiProvider/index.test.ts +2 -2
- package/src/services/aiProvider/index.ts +48 -2
- package/src/services/chatGroup/index.ts +66 -2
- package/src/services/export/index.ts +10 -2
- package/src/services/file/index.ts +61 -2
- package/src/services/import/index.ts +133 -2
- package/src/services/message/index.ts +176 -2
- package/src/services/message/{__tests__/server.test.ts → server.test.ts} +3 -3
- package/src/services/plugin/index.test.ts +8 -0
- package/src/services/plugin/index.ts +53 -2
- package/src/services/session/index.test.ts +8 -0
- package/src/services/session/index.ts +145 -2
- package/src/services/thread/index.test.ts +8 -0
- package/src/services/thread/index.ts +38 -2
- package/src/services/topic/index.test.ts +8 -0
- package/src/services/topic/index.ts +76 -2
- package/src/services/user/index.test.ts +8 -0
- package/src/services/user/index.ts +53 -2
- package/src/store/aiInfra/slices/aiModel/action.test.ts +17 -9
- package/src/store/chat/slices/aiChat/actions/__tests__/helpers.ts +4 -2
- package/src/store/chat/slices/topic/action.test.ts +1 -1
- package/src/store/chat/slices/topic/action.ts +1 -2
- package/src/store/chat/slices/topic/reducer.ts +1 -2
- package/src/store/file/slices/chat/action.ts +1 -4
- package/src/store/file/slices/fileManager/action.ts +2 -3
- package/src/store/session/slices/sessionGroup/action.test.ts +5 -5
- package/src/store/user/slices/common/action.test.ts +1 -1
- package/src/services/aiModel/server.test.ts +0 -122
- package/src/services/aiModel/server.ts +0 -51
- package/src/services/aiModel/type.ts +0 -32
- package/src/services/aiProvider/server.ts +0 -43
- package/src/services/aiProvider/type.ts +0 -27
- package/src/services/chatGroup/server.ts +0 -67
- package/src/services/chatGroup/type.ts +0 -22
- package/src/services/export/server.ts +0 -9
- package/src/services/export/type.ts +0 -5
- package/src/services/file/server.ts +0 -53
- package/src/services/file/type.ts +0 -13
- package/src/services/import/server.ts +0 -133
- package/src/services/import/type.ts +0 -17
- package/src/services/message/server.ts +0 -151
- package/src/services/message/type.ts +0 -55
- package/src/services/plugin/server.ts +0 -42
- package/src/services/plugin/type.ts +0 -23
- package/src/services/session/server.test.ts +0 -260
- package/src/services/session/server.ts +0 -125
- package/src/services/session/type.ts +0 -82
- package/src/services/thread/server.ts +0 -32
- package/src/services/thread/type.ts +0 -21
- package/src/services/topic/server.ts +0 -57
- package/src/services/topic/type.ts +0 -40
- package/src/services/user/server.test.ts +0 -149
- package/src/services/user/server.ts +0 -47
- package/src/services/user/type.ts +0 -21
|
@@ -1,3 +1,54 @@
|
|
|
1
|
-
import {
|
|
1
|
+
import type { AdapterAccount } from 'next-auth/adapters';
|
|
2
|
+
import type { PartialDeep } from 'type-fest';
|
|
2
3
|
|
|
3
|
-
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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
|
|
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(
|
|
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
|
|
|
@@ -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
|
|
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
|
|
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(
|
|
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
|
-
}
|