@lobehub/lobehub 2.0.0-next.13 → 2.0.0-next.15

package/src/server/routers/tools/search.ts

@@ -1,12 +1,9 @@
-import { isServerMode } from '@lobechat/const';
 import { z } from 'zod';
 
-import { passwordProcedure } from '@/libs/trpc/edge';
 import { authedProcedure, router } from '@/libs/trpc/lambda';
 import { searchService } from '@/server/services/search';
 
-// TODO: password procedure 未来的处理方式可能要思考下
-const searchProcedure = isServerMode ? authedProcedure : passwordProcedure;
+const searchProcedure = authedProcedure;
 
 export const searchRouter = router({
   crawlPages: searchProcedure

package/src/services/__tests__/tool.test.ts

@@ -1,8 +1,5 @@
 import { Mock, beforeEach, describe, expect, it, vi } from 'vitest';
 
-import { edgeClient } from '@/libs/trpc/client';
-import { globalHelpers } from '@/store/global/helpers';
-
 import { toolService } from '../tool';
 import openAPIV3 from './openai/OpenAPI_V3.json';
 import OpenAIPlugin from './openai/plugin.json';

package/src/services/aiModel/index.test.ts

@@ -1,10 +1,7 @@
 import { testService } from '~test-utils';
 
-import { ClientService } from './client';
 import { ServerService } from './server';
 
 describe('aiModelService', () => {
   testService(ServerService);
-
-  testService(ClientService);
 });

package/src/services/aiModel/index.ts

@@ -1,9 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-export const aiModelService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : new ClientService();
+export const aiModelService = new ServerService();

package/src/services/aiProvider/index.test.ts

@@ -1,10 +1,7 @@
 import { testService } from '~test-utils';
 
-import { ClientService } from './client';
 import { ServerService } from './server';
 
 describe('aiProviderService', () => {
   testService(ServerService);
-
-  testService(ClientService);
 });

package/src/services/aiProvider/index.ts

@@ -1,9 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-export const aiProviderService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : new ClientService();
+export const aiProviderService = new ServerService();

package/src/services/chatGroup/index.ts

@@ -1,12 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new ServerService();
-
-export const chatGroupService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
+export const chatGroupService = new ServerService();

package/src/services/config.ts

@@ -1,23 +1,13 @@
-import { BRANDING_NAME, isDeprecatedEdition, isServerMode } from '@lobechat/const';
+import { BRANDING_NAME, isServerMode } from '@lobechat/const';
 import { downloadFile, exportJSONFile } from '@lobechat/utils/client';
 import dayjs from 'dayjs';
 
-import { CURRENT_CONFIG_VERSION } from '@/migrations';
 import { ImportPgDataStructure } from '@/types/export';
 
 import { exportService } from './export';
-import { configService as deprecatedExportService } from './export/_deprecated';
 
 class ConfigService {
   exportAll = async () => {
-    // TODO: remove this in V2
-    if (isDeprecatedEdition) {
-      const config = await deprecatedExportService.exportAll();
-      const filename = `${BRANDING_NAME}-config-v${CURRENT_CONFIG_VERSION}.json`;
-      exportJSONFile(config, filename);
-      return;
-    }
-
     const { data, url } = await exportService.exportData();
     const filename = `${dayjs().format('YYYY-MM-DD-hh-mm')}_${BRANDING_NAME}-data.json`;
 
@@ -34,60 +24,6 @@ class ConfigService {
     exportJSONFile(result, filename);
   };
 
-  exportAgents = async () => {
-    // TODO: remove this in V2
-    if (isDeprecatedEdition) {
-      const config = await deprecatedExportService.exportAgents();
-      const filename = `${BRANDING_NAME}-agents-v${CURRENT_CONFIG_VERSION}.json`;
-      exportJSONFile(config, filename);
-      return;
-    }
-  };
-
-  exportSingleAgent = async (agentId: string) => {
-    // TODO: remove this in V2
-    if (isDeprecatedEdition) {
-      const result = await deprecatedExportService.exportSingleAgent(agentId);
-      if (!result) return;
-
-      const filename = `${BRANDING_NAME}-${result.title}-v${CURRENT_CONFIG_VERSION}.json`;
-      exportJSONFile(result.config, filename);
-      return;
-    }
-  };
-
-  exportSessions = async () => {
-    // TODO: remove this in V2
-    if (isDeprecatedEdition) {
-      const config = await deprecatedExportService.exportSessions();
-      const filename = `${BRANDING_NAME}-sessions-v${CURRENT_CONFIG_VERSION}.json`;
-      exportJSONFile(config, filename);
-      return;
-    }
-  };
-
-  exportSettings = async () => {
-    // TODO: remove this in V2
-    if (isDeprecatedEdition) {
-      const config = await deprecatedExportService.exportSessions();
-      const filename = `${BRANDING_NAME}-settings-v${CURRENT_CONFIG_VERSION}.json`;
-      exportJSONFile(config, filename);
-      return;
-    }
-  };
-
-  exportSingleSession = async (sessionId: string) => {
-    // TODO: remove this in V2
-    if (isDeprecatedEdition) {
-      const data = await deprecatedExportService.exportSingleSession(sessionId);
-      if (!data) return;
-
-      const filename = `${BRANDING_NAME}-${data.title}-v${CURRENT_CONFIG_VERSION}.json`;
-      exportJSONFile(data.config, filename);
-      return;
-    }
-  };
-
   private createDataStructure = async (
     data: any,
     mode: 'pglite' | 'postgres',

package/src/services/export/index.ts

@@ -1,6 +1,3 @@
-import { isServerMode } from '@/const/version';
-
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-export const exportService = isServerMode ? new ServerService() : new ClientService();
+export const exportService = new ServerService();

package/src/services/file/index.ts

@@ -1,13 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const fileService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
+export const fileService = new ServerService();

package/src/services/import/index.ts

@@ -1,9 +1,3 @@
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const importService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' ? new ServerService() : clientService;
+export const importService = new ServerService();

package/src/services/message/index.ts

@@ -1,13 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const messageService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
+export const messageService = new ServerService();

package/src/services/plugin/index.ts

@@ -1,13 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const pluginService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
+export const pluginService = new ServerService();

package/src/services/session/index.ts

@@ -1,13 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const sessionService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
+export const sessionService = new ServerService();

package/src/services/tableViewer/client.ts

@@ -1,16 +1,13 @@
-import { clientDB } from '@/database/client/db';
-import { TableViewerRepo } from '@/database/repositories/tableViewer';
-import { BaseClientService } from '@/services/baseClientService';
-
-export class ClientService extends BaseClientService {
-  private get tableViewerRepo(): TableViewerRepo {
-    return new TableViewerRepo(clientDB as any, this.userId);
-  }
-
-  getAllTables = async () => this.tableViewerRepo.getAllTables();
-
-  getTableDetails = async (tableName: string) => this.tableViewerRepo.getTableDetails(tableName);
-
-  getTableData = async (tableName: string) =>
-    this.tableViewerRepo.getTableData(tableName, { page: 1, pageSize: 300 });
+export class ClientService {
+  getAllTables = async () => [];
+
+  getTableDetails = async (tableName: string) => {
+    console.log('getTableDetails:', tableName);
+    return [];
+  };
+
+  getTableData = async (tableName: string) => {
+    console.log('getTableData:', tableName);
+    return [];
+  };
 }

package/src/services/thread/index.ts

@@ -1,9 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-export const threadService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : new ClientService();
+export const threadService = new ServerService();

package/src/services/topic/index.ts

@@ -1,13 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const topicService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
+export const topicService = new ServerService();

package/src/services/user/index.ts

@@ -1,15 +1,3 @@
-import { isDesktop } from '@/const/version';
-
-import { ClientService as DeprecatedService } from './_deprecated';
-import { ClientService } from './client';
 import { ServerService } from './server';
 
-const clientService =
-  process.env.NEXT_PUBLIC_CLIENT_DB === 'pglite' ? new ClientService() : new DeprecatedService();
-
-export const userService =
-  process.env.NEXT_PUBLIC_SERVICE_MODE === 'server' || isDesktop
-    ? new ServerService()
-    : clientService;
-
-export const userClientService = clientService;
+export const userService = new ServerService();

package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChat.test.ts

@@ -87,57 +87,6 @@ describe('chatMessage actions', () => {
     });
 
     describe('message creation', () => {
-      it('should create user message and trigger AI processing', async () => {
-        const { result } = renderHook(() => useChatStore());
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE });
-        });
-
-        expect(messageService.createMessage).toHaveBeenCalledWith({
-          content: TEST_CONTENT.USER_MESSAGE,
-          files: undefined,
-          role: 'user',
-          sessionId: TEST_IDS.SESSION_ID,
-          topicId: TEST_IDS.TOPIC_ID,
-        });
-        expect(result.current.internal_coreProcessMessage).toHaveBeenCalled();
-      });
-
-      it('should send message with files attached', async () => {
-        const { result } = renderHook(() => useChatStore());
-        const files = [{ id: TEST_IDS.FILE_ID } as UploadFileItem];
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE, files });
-        });
-
-        expect(messageService.createMessage).toHaveBeenCalledWith({
-          content: TEST_CONTENT.USER_MESSAGE,
-          files: [TEST_IDS.FILE_ID],
-          role: 'user',
-          sessionId: TEST_IDS.SESSION_ID,
-          topicId: TEST_IDS.TOPIC_ID,
-        });
-      });
-
-      it('should send files without message content', async () => {
-        const { result } = renderHook(() => useChatStore());
-        const files = [{ id: TEST_IDS.FILE_ID } as UploadFileItem];
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.EMPTY, files });
-        });
-
-        expect(messageService.createMessage).toHaveBeenCalledWith({
-          content: TEST_CONTENT.EMPTY,
-          files: [TEST_IDS.FILE_ID],
-          role: 'user',
-          sessionId: TEST_IDS.SESSION_ID,
-          topicId: TEST_IDS.TOPIC_ID,
-        });
-      });
-
       it('should not process AI when onlyAddUserMessage is true', async () => {
         const { result } = renderHook(() => useChatStore());
 
@@ -169,196 +118,6 @@ describe('chatMessage actions', () => {
         expect(result.current.internal_coreProcessMessage).not.toHaveBeenCalled();
       });
     });
-
-    describe('auto-create topic', () => {
-      const TOPIC_THRESHOLD = 5;
-
-      it('should create topic when threshold is reached and feature is enabled', async () => {
-        const { result } = renderHook(() => useChatStore());
-
-        const createTopicMock = vi.fn(() => Promise.resolve(TEST_IDS.NEW_TOPIC_ID));
-        const switchTopicMock = vi.fn();
-
-        act(() => {
-          setupMockSelectors({
-            agentConfig: {
-              enableAutoCreateTopic: true,
-              autoCreateTopicThreshold: TOPIC_THRESHOLD,
-            },
-          });
-
-          useChatStore.setState({
-            activeTopicId: undefined,
-            messagesMap: {
-              [messageMapKey(TEST_IDS.SESSION_ID)]: createMockMessages(TOPIC_THRESHOLD),
-            },
-            createTopic: createTopicMock,
-            switchTopic: switchTopicMock,
-          });
-        });
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE });
-        });
-
-        expect(createTopicMock).toHaveBeenCalled();
-        expect(switchTopicMock).toHaveBeenCalledWith(TEST_IDS.NEW_TOPIC_ID, true);
-      });
-    });
-
-    describe('RAG integration', () => {
-      it('should include RAG query when RAG is enabled', async () => {
-        const { result } = renderHook(() => useChatStore());
-        vi.spyOn(result.current, 'internal_shouldUseRAG').mockReturnValue(true);
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.RAG_QUERY });
-        });
-
-        expect(result.current.internal_coreProcessMessage).toHaveBeenCalledWith(
-          expect.any(Array),
-          expect.any(String),
-          expect.objectContaining({
-            ragQuery: TEST_CONTENT.RAG_QUERY,
-          }),
-        );
-      });
-
-      it('should not use RAG when feature is disabled', async () => {
-        const { result } = renderHook(() => useChatStore());
-        vi.spyOn(result.current, 'internal_shouldUseRAG').mockReturnValue(false);
-        const retrieveChunksSpy = vi.spyOn(result.current, 'internal_retrieveChunks');
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE });
-        });
-
-        expect(retrieveChunksSpy).not.toHaveBeenCalled();
-        expect(result.current.internal_coreProcessMessage).toHaveBeenCalledWith(
-          expect.any(Array),
-          expect.any(String),
-          expect.not.objectContaining({
-            ragQuery: expect.anything(),
-          }),
-        );
-      });
-    });
-
-    describe('special flags', () => {
-      it('should pass isWelcomeQuestion flag to processing', async () => {
-        const { result } = renderHook(() => useChatStore());
-
-        await act(async () => {
-          await result.current.sendMessage({
-            message: TEST_CONTENT.USER_MESSAGE,
-            isWelcomeQuestion: true,
-          });
-        });
-
-        expect(result.current.internal_coreProcessMessage).toHaveBeenCalledWith(
-          expect.anything(),
-          expect.anything(),
-          { isWelcomeQuestion: true },
-        );
-      });
-
-      it('should return early when onlyAddUserMessage is true', async () => {
-        const { result } = renderHook(() => useChatStore());
-
-        await act(async () => {
-          await result.current.sendMessage({
-            message: TEST_CONTENT.USER_MESSAGE,
-            onlyAddUserMessage: true,
-          });
-        });
-
-        expect(result.current.internal_coreProcessMessage).not.toHaveBeenCalled();
-      });
-    });
-
-    describe('topic creation flow', () => {
-      it('should handle tempMessage during topic creation', async () => {
-        setupMockSelectors({
-          chatConfig: { enableAutoCreateTopic: true, autoCreateTopicThreshold: 2 },
-        });
-
-        act(() => {
-          useChatStore.setState({ activeTopicId: undefined });
-          setupStoreWithMessages(createMockMessages(5));
-        });
-
-        const { result } = renderHook(() => useChatStore());
-        const createTopicMock = vi
-          .spyOn(result.current, 'createTopic')
-          .mockResolvedValue(TEST_IDS.NEW_TOPIC_ID);
-        const toggleMessageLoadingSpy = vi.spyOn(result.current, 'internal_toggleMessageLoading');
-        const createTmpMessageSpy = vi
-          .spyOn(result.current, 'internal_createTmpMessage')
-          .mockReturnValue('temp-id');
-        vi.spyOn(result.current, 'internal_fetchMessages').mockResolvedValue();
-        vi.spyOn(result.current, 'switchTopic').mockResolvedValue();
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE });
-        });
-
-        expect(createTmpMessageSpy).toHaveBeenCalled();
-        expect(toggleMessageLoadingSpy).toHaveBeenCalledWith(true, 'temp-id');
-        expect(createTopicMock).toHaveBeenCalled();
-      });
-
-      it('should call summaryTopicTitle after processing when new topic created', async () => {
-        setupMockSelectors({
-          chatConfig: { enableAutoCreateTopic: true, autoCreateTopicThreshold: 2 },
-        });
-
-        act(() => {
-          useChatStore.setState({ activeTopicId: undefined });
-          setupStoreWithMessages(createMockMessages(5));
-        });
-
-        const { result } = renderHook(() => useChatStore());
-        vi.spyOn(result.current, 'createTopic').mockResolvedValue(TEST_IDS.NEW_TOPIC_ID);
-        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-id');
-        vi.spyOn(result.current, 'internal_fetchMessages').mockResolvedValue();
-        vi.spyOn(result.current, 'switchTopic').mockResolvedValue();
-
-        const summaryTopicTitleSpy = vi
-          .spyOn(result.current, 'summaryTopicTitle')
-          .mockResolvedValue();
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE });
-        });
-
-        expect(summaryTopicTitleSpy).toHaveBeenCalledWith(TEST_IDS.NEW_TOPIC_ID, expect.any(Array));
-      });
-
-      it('should handle topic creation failure gracefully', async () => {
-        setupMockSelectors({
-          chatConfig: { enableAutoCreateTopic: true, autoCreateTopicThreshold: 2 },
-        });
-
-        act(() => {
-          useChatStore.setState({ activeTopicId: undefined });
-          setupStoreWithMessages(createMockMessages(5));
-        });
-
-        const { result } = renderHook(() => useChatStore());
-        vi.spyOn(result.current, 'createTopic').mockResolvedValue(undefined);
-        vi.spyOn(result.current, 'internal_createTmpMessage').mockReturnValue('temp-id');
-        const toggleLoadingSpy = vi.spyOn(result.current, 'internal_toggleMessageLoading');
-        const updateTopicLoadingSpy = vi.spyOn(result.current, 'internal_updateTopicLoading');
-
-        await act(async () => {
-          await result.current.sendMessage({ message: TEST_CONTENT.USER_MESSAGE });
-        });
-
-        // Should still call the AI processing even if topic creation fails
-        expect(result.current.internal_coreProcessMessage).toHaveBeenCalled();
-        expect(updateTopicLoadingSpy).not.toHaveBeenCalled();
-      });
-    });
   });
 
   describe('regenerateMessage', () => {

package/src/store/chat/slices/aiChat/actions/__tests__/generateAIChatV2.test.ts

@@ -19,8 +19,33 @@ import { resetTestEnvironment, setupMockSelectors, spyOnMessageService } from '.
 // Keep zustand mock as it's needed globally
 vi.mock('zustand/traditional');
 
+// Mock AntdStaticMethods
+vi.mock('@/components/AntdStaticMethods', () => ({
+  notification: {
+    error: vi.fn(),
+    success: vi.fn(),
+    info: vi.fn(),
+    warning: vi.fn(),
+  },
+  message: {
+    error: vi.fn(),
+    success: vi.fn(),
+    info: vi.fn(),
+    warning: vi.fn(),
+  },
+}));
+
+// Mock sessionService to prevent TRPC requests
+vi.mock('@/services/session', () => ({
+  sessionService: {
+    updateSession: vi.fn(),
+    updateSessionConfig: vi.fn(),
+    updateSessionChatConfig: vi.fn(),
+  },
+}));
+
 // Mock server mode for V2 tests
-vi.mock('@/const/version', async (importOriginal) => {
+vi.mock('@lobechat/const', async (importOriginal) => {
   const module = await importOriginal();
   return {
     ...(module as any),

package/src/store/chat/slices/aiChat/actions/__tests__/helpers.ts

@@ -60,7 +60,9 @@ export const spyOnMessageService = () => {
   const createMessageSpy = vi
     .spyOn(messageService, 'createMessage')
     .mockResolvedValue(TEST_IDS.NEW_MESSAGE_ID);
-  const updateMessageSpy = vi.spyOn(messageService, 'updateMessage').mockResolvedValue(undefined);
+  const updateMessageSpy = vi
+    .spyOn(messageService, 'updateMessage')
+    .mockResolvedValue({ messages: [], success: true });
   const removeMessageSpy = vi.spyOn(messageService, 'removeMessage').mockResolvedValue(undefined);
   const updateMessageErrorSpy = vi
     .spyOn(messageService, 'updateMessageError')