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

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')

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

@@ -17,15 +17,12 @@ import { StateCreator } from 'zustand/vanilla';
 
 import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
-import { useAgentStore } from '@/store/agent';
 import { agentChatConfigSelectors, agentSelectors } from '@/store/agent/selectors';
 import { getAgentStoreState } from '@/store/agent/store';
 import { aiModelSelectors, aiProviderSelectors } from '@/store/aiInfra';
 import { getAiInfraStoreState } from '@/store/aiInfra/store';
 import { ChatStore } from '@/store/chat/store';
-import { messageMapKey } from '@/store/chat/utils/messageMapKey';
 import { getFileStoreState } from '@/store/file/store';
-import { useSessionStore } from '@/store/session';
 import { WebBrowsingManifest } from '@/tools/web-browsing';
 import { Action, setNamespace } from '@/utils/storeDebug';
 
@@ -158,13 +155,7 @@ export const generateAIChat: StateCreator<
   },
 
   sendMessage: async ({ message, files, onlyAddUserMessage, isWelcomeQuestion }) => {
-    const {
-      internal_coreProcessMessage,
-      activeTopicId,
-      activeId,
-      activeThreadId,
-      sendMessageInServer,
-    } = get();
+    const { activeId, sendMessageInServer } = get();
     if (!activeId) return;
 
     const fileIdList = files?.map((f) => f.id);
@@ -177,134 +168,6 @@ export const generateAIChat: StateCreator<
     // router to server mode send message
     if (isServerMode)
       return sendMessageInServer({ message, files, onlyAddUserMessage, isWelcomeQuestion });
-
-    set({ isCreatingMessage: true }, false, n('creatingMessage/start'));
-
-    const newMessage: CreateMessageParams = {
-      content: message,
-      // if message has attached with files, then add files to message and the agent
-      files: fileIdList,
-      role: 'user',
-      sessionId: activeId,
-      // if there is activeTopicId，then add topicId to message
-      topicId: activeTopicId,
-      threadId: activeThreadId,
-    };
-
-    const agentConfig = agentChatConfigSelectors.currentChatConfig(getAgentStoreState());
-
-    let tempMessageId: string | undefined = undefined;
-    let newTopicId: string | undefined = undefined;
-
-    // it should be the default topic, then
-    // if autoCreateTopic is enabled, check to whether we need to create a topic
-    if (!onlyAddUserMessage && !activeTopicId && agentConfig.enableAutoCreateTopic) {
-      // check activeTopic and then auto create topic
-      const chats = chatSelectors.activeBaseChats(get());
-
-      // we will add two messages (user and assistant), so the finial length should +2
-      const featureLength = chats.length + 2;
-
-      // if there is no activeTopicId and the feature length is greater than the threshold
-      // then create a new topic and active it
-      if (!activeTopicId && featureLength >= agentConfig.autoCreateTopicThreshold) {
-        // we need to create a temp message for optimistic update
-        tempMessageId = get().internal_createTmpMessage(newMessage);
-        get().internal_toggleMessageLoading(true, tempMessageId);
-
-        const topicId = await get().createTopic();
-
-        if (topicId) {
-          newTopicId = topicId;
-          newMessage.topicId = topicId;
-
-          // we need to copy the messages to the new topic or the message will disappear
-          const mapKey = chatSelectors.currentChatKey(get());
-          const newMaps = {
-            ...get().messagesMap,
-            [messageMapKey(activeId, topicId)]: get().messagesMap[mapKey],
-          };
-          set({ messagesMap: newMaps }, false, n('moveMessagesToNewTopic'));
-
-          // make the topic loading
-          get().internal_updateTopicLoading(topicId, true);
-        }
-      }
-    }
-    //  update assistant update to make it rerank
-    useSessionStore.getState().triggerSessionUpdate(get().activeId);
-
-    const id = await get().internal_createMessage(newMessage, {
-      tempMessageId,
-      skipRefresh: !onlyAddUserMessage && newMessage.fileList?.length === 0,
-    });
-
-    if (!id) {
-      set({ isCreatingMessage: false }, false, n('creatingMessage/start'));
-      if (!!newTopicId) get().internal_updateTopicLoading(newTopicId, false);
-      return;
-    }
-
-    if (tempMessageId) get().internal_toggleMessageLoading(false, tempMessageId);
-
-    // switch to the new topic if create the new topic
-    if (!!newTopicId) {
-      await get().switchTopic(newTopicId, true);
-      await get().internal_fetchMessages();
-
-      // delete previous messages
-      // remove the temp message map
-      const newMaps = { ...get().messagesMap, [messageMapKey(activeId, null)]: [] };
-      set({ messagesMap: newMaps }, false, 'internal_copyMessages');
-    }
-
-    // if only add user message, then stop
-    if (onlyAddUserMessage) {
-      set({ isCreatingMessage: false }, false, 'creatingMessage/start');
-      return;
-    }
-
-    // Get the current messages to generate AI response
-    const messages = chatSelectors.activeBaseChats(get());
-    const userFiles = chatSelectors.currentUserFiles(get()).map((f) => f.id);
-
-    await internal_coreProcessMessage(messages, id, {
-      isWelcomeQuestion,
-      ragQuery: get().internal_shouldUseRAG() ? message : undefined,
-      threadId: activeThreadId,
-    });
-
-    set({ isCreatingMessage: false }, false, n('creatingMessage/stop'));
-
-    const summaryTitle = async () => {
-      // if autoCreateTopic is false, then stop
-      if (!agentConfig.enableAutoCreateTopic) return;
-
-      // check activeTopic and then auto update topic title
-      if (newTopicId) {
-        const chats = chatSelectors.getBaseChatsByKey(messageMapKey(activeId, newTopicId))(get());
-        await get().summaryTopicTitle(newTopicId, chats);
-        return;
-      }
-
-      if (!activeTopicId) return;
-      const topic = topicSelectors.getTopicById(activeTopicId)(get());
-
-      if (topic && !topic.title) {
-        const chats = chatSelectors.getBaseChatsByKey(messageMapKey(activeId, topic.id))(get());
-        await get().summaryTopicTitle(topic.id, chats);
-      }
-    };
-
-    // if there is relative files, then add files to agent
-    // only available in server mode
-    const addFilesToAgent = async () => {
-      if (userFiles.length === 0 || !isServerMode) return;
-
-      await useAgentStore.getState().addFilesToAgent(userFiles, false);
-    };
-
-    await Promise.all([summaryTitle(), addFilesToAgent()]);
   },
   stopGenerateMessage: () => {
     const { chatLoadingIdsAbortController, internal_toggleChatLoading } = get();

package/src/store/user/slices/common/action.test.ts

@@ -4,7 +4,6 @@ import { withSWR } from '~test-utils';
 
 import { DEFAULT_PREFERENCE } from '@/const/user';
 import { userService } from '@/services/user';
-import { ClientService } from '@/services/user/_deprecated';
 import { useUserStore } from '@/store/user';
 import { preferenceSelectors } from '@/store/user/selectors';
 import { GlobalServerConfig } from '@/types/serverConfig';
@@ -31,9 +30,7 @@ describe('createCommonSlice', () => {
       const avatar = 'data:image/png;base64,';
 
       const spyOn = vi.spyOn(result.current, 'refreshUserState');
-      const updateAvatarSpy = vi
-        .spyOn(ClientService.prototype, 'updateAvatar')
-        .mockResolvedValue(undefined);
+      const updateAvatarSpy = vi.spyOn(userService, 'updateAvatar').mockResolvedValue(undefined);
 
       await act(async () => {
         await result.current.updateAvatar(avatar);

package/src/app/(backend)/trpc/edge/[trpc]/route.ts

@@ -1,26 +0,0 @@
-import { fetchRequestHandler } from '@trpc/server/adapters/fetch';
-import type { NextRequest } from 'next/server';
-
-import { pino } from '@/libs/logger';
-import { createEdgeContext } from '@/libs/trpc/edge/context';
-import { edgeRouter } from '@/server/routers/edge';
-
-const handler = (req: NextRequest) =>
-  fetchRequestHandler({
-    /**
-     * @link https://trpc.io/docs/v11/context
-     */
-    createContext: () => createEdgeContext(req),
-
-    endpoint: '/trpc/edge',
-
-    onError: ({ error, path }) => {
-      pino.info(`Error in tRPC handler (edge) on path: ${path}`);
-      console.error(error);
-    },
-
-    req,
-    router: edgeRouter,
-  });
-
-export { handler as GET, handler as POST };

package/src/app/[variants]/(main)/(mobile)/me/data/features/Category.tsx

@@ -1,48 +0,0 @@
-'use client';
-
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import Cell, { CellProps } from '@/components/Cell';
-import DataImporter from '@/features/DataImporter';
-import { configService } from '@/services/export/_deprecated';
-
-const Category = memo(() => {
-  const { t } = useTranslation('common');
-  const items: CellProps[] = [
-    {
-      key: 'allAgent',
-      label: t('exportType.allAgent'),
-      onClick: configService.exportAgents,
-    },
-    {
-      key: 'allAgentWithMessage',
-      label: t('exportType.allAgentWithMessage'),
-      onClick: configService.exportSessions,
-    },
-    {
-      key: 'globalSetting',
-      label: t('exportType.globalSetting'),
-      onClick: configService.exportSettings,
-    },
-    {
-      type: 'divider',
-    },
-    {
-      key: 'all',
-      label: t('exportType.all'),
-      onClick: configService.exportAll,
-    },
-    {
-      type: 'divider',
-    },
-    {
-      key: 'import',
-      label: <DataImporter>{t('importData')}</DataImporter>,
-    },
-  ];
-
-  return items?.map(({ key, ...item }, index) => <Cell key={key || index} {...item} />);
-});
-
-export default Category;

package/src/app/[variants]/(main)/(mobile)/me/data/features/Header.tsx

@@ -1,33 +0,0 @@
-'use client';
-
-import { ChatHeader } from '@lobehub/ui/mobile';
-import { useRouter } from 'next/navigation';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import { mobileHeaderSticky } from '@/styles/mobileHeader';
-
-const Header = memo(() => {
-  const { t } = useTranslation('common');
-
-  const router = useRouter();
-  return (
-    <ChatHeader
-      center={
-        <ChatHeader.Title
-          title={
-            <Flexbox align={'center'} gap={4} horizontal>
-              {t('userPanel.data')}
-            </Flexbox>
-          }
-        />
-      }
-      onBackClick={() => router.push('/me')}
-      showBackButton
-      style={mobileHeaderSticky}
-    />
-  );
-});
-
-export default Header;

package/src/app/[variants]/(main)/(mobile)/me/data/layout.tsx

@@ -1,13 +0,0 @@
-import { PropsWithChildren } from 'react';
-
-import MobileContentLayout from '@/components/server/MobileNavLayout';
-
-import Header from './features/Header';
-
-const Layout = ({ children }: PropsWithChildren) => {
-  return <MobileContentLayout header={<Header />}>{children}</MobileContentLayout>;
-};
-
-Layout.displayName = 'MeDataLayout';
-
-export default Layout;

package/src/app/[variants]/(main)/(mobile)/me/data/loading.tsx

@@ -1,5 +0,0 @@
-import SkeletonLoading from '@/components/Loading/SkeletonLoading';
-
-export default () => {
-  return <SkeletonLoading paragraph={{ rows: 8 }} />;
-};

package/src/app/[variants]/(main)/(mobile)/me/data/page.tsx

@@ -1,29 +0,0 @@
-import { redirect } from 'next/navigation';
-
-import { metadataModule } from '@/server/metadata';
-import { translation } from '@/server/translation';
-import { DynamicLayoutProps } from '@/types/next';
-import { RouteVariants } from '@/utils/server/routeVariants';
-
-import Category from './features/Category';
-
-export const generateMetadata = async (props: DynamicLayoutProps) => {
-  const locale = await RouteVariants.getLocale(props);
-  const { t } = await translation('common', locale);
-  return metadataModule.generate({
-    title: t('userPanel.data'),
-    url: '/me/data',
-  });
-};
-
-const Page = async (props: DynamicLayoutProps) => {
-  const mobile = await RouteVariants.getIsMobile(props);
-
-  if (!mobile) return redirect('/chat');
-
-  return <Category />;
-};
-
-Page.displayName = 'MeData';
-
-export default Page;