@lobehub/chat 1.36.30 → 1.36.32

package/src/database/server/models/topic.ts

@@ -29,46 +29,42 @@ export class TopicModel {
   }
   // **************** Query *************** //
 
-  async query({ current = 0, pageSize = 9999, sessionId }: QueryTopicParams = {}) {
+  query = async ({ current = 0, pageSize = 9999, sessionId }: QueryTopicParams = {}) => {
     const offset = current * pageSize;
-
-    return (
-      this.db
-        .select({
-          createdAt: topics.createdAt,
-          favorite: topics.favorite,
-          historySummary: topics.historySummary,
-          id: topics.id,
-          metadata: topics.metadata,
-          title: topics.title,
-          updatedAt: topics.updatedAt,
-        })
-        .from(topics)
-        .where(and(eq(topics.userId, this.userId), this.matchSession(sessionId)))
-        // In boolean sorting, false is considered "smaller" than true.
-        // So here we use desc to ensure that topics with favorite as true are in front.
-        .orderBy(desc(topics.favorite), desc(topics.updatedAt))
-        .limit(pageSize)
-        .offset(offset)
-    );
-  }
-
-  async findById(id: string) {
+    return this.db
+      .select({
+        createdAt: topics.createdAt,
+        favorite: topics.favorite,
+        historySummary: topics.historySummary,
+        id: topics.id,
+        metadata: topics.metadata,
+        title: topics.title,
+        updatedAt: topics.updatedAt,
+      })
+      .from(topics)
+      .where(and(eq(topics.userId, this.userId), this.matchSession(sessionId)))
+      // In boolean sorting, false is considered "smaller" than true.
+      // So here we use desc to ensure that topics with favorite as true are in front.
+      .orderBy(desc(topics.favorite), desc(topics.updatedAt))
+      .limit(pageSize)
+      .offset(offset);
+  };
+
+  findById = async (id: string) => {
     return this.db.query.topics.findFirst({
       where: and(eq(topics.id, id), eq(topics.userId, this.userId)),
     });
-  }
+  };
 
-  async queryAll(): Promise<TopicItem[]> {
+  queryAll = async (): Promise<TopicItem[]> => {
     return this.db
       .select()
       .from(topics)
       .orderBy(topics.updatedAt)
-      .where(eq(topics.userId, this.userId))
-      .execute();
-  }
+      .where(eq(topics.userId, this.userId));
+  };
 
-  async queryByKeyword(keyword: string, sessionId?: string | null): Promise<TopicItem[]> {
+  queryByKeyword = async (keyword: string, sessionId?: string | null): Promise<TopicItem[]> => {
     if (!keyword) return [];
 
     const keywordLowerCase = keyword.toLowerCase();
@@ -92,26 +88,25 @@ export class TopicModel {
         ),
       ),
     });
-  }
+  };
 
-  async count() {
+  count = async (): Promise<number> => {
     const result = await this.db
       .select({
-        count: count(),
+        count: count(topics.id),
       })
       .from(topics)
-      .where(eq(topics.userId, this.userId))
-      .execute();
+      .where(eq(topics.userId, this.userId));
 
     return result[0].count;
-  }
+  };
 
   // **************** Create *************** //
 
-  async create(
+  create = async (
     { messages: messageIds, ...params }: CreateTopicParams,
     id: string = this.genId(),
-  ): Promise<TopicItem> {
+  ): Promise<TopicItem> => {
     return this.db.transaction(async (tx) => {
       // 在 topics 表中插入新的 topic
       const [topic] = await tx
@@ -133,9 +128,9 @@ export class TopicModel {
 
       return topic;
     });
-  }
+  };
 
-  async batchCreate(topicParams: (CreateTopicParams & { id?: string })[]) {
+  batchCreate = async (topicParams: (CreateTopicParams & { id?: string })[]) => {
     // 开始一个事务
     return this.db.transaction(async (tx) => {
       // 在 topics 表中批量插入新的 topics
@@ -167,9 +162,9 @@ export class TopicModel {
 
       return createdTopics;
     });
-  }
+  };
 
-  async duplicate(topicId: string, newTitle?: string) {
+  duplicate = async (topicId: string, newTitle?: string) => {
     return this.db.transaction(async (tx) => {
       // find original topic
       const originalTopic = await tx.query.topics.findFirst({
@@ -217,48 +212,48 @@ export class TopicModel {
         topic: duplicatedTopic,
       };
     });
-  }
+  };
 
   // **************** Delete *************** //
 
   /**
    * Delete a session, also delete all messages and topics associated with it.
    */
-  async delete(id: string) {
+  delete = async (id: string) => {
     return this.db.delete(topics).where(and(eq(topics.id, id), eq(topics.userId, this.userId)));
-  }
+  };
 
   /**
    * Deletes multiple topics based on the sessionId.
    */
-  async batchDeleteBySessionId(sessionId?: string | null) {
+  batchDeleteBySessionId = async (sessionId?: string | null) => {
     return this.db
       .delete(topics)
       .where(and(this.matchSession(sessionId), eq(topics.userId, this.userId)));
-  }
+  };
 
   /**
    * Deletes multiple topics and all messages associated with them in a transaction.
    */
-  async batchDelete(ids: string[]) {
+  batchDelete = async (ids: string[]) => {
     return this.db
       .delete(topics)
       .where(and(inArray(topics.id, ids), eq(topics.userId, this.userId)));
-  }
+  };
 
-  async deleteAll() {
+  deleteAll = async () => {
     return this.db.delete(topics).where(eq(topics.userId, this.userId));
-  }
+  };
 
   // **************** Update *************** //
 
-  async update(id: string, data: Partial<TopicItem>) {
+  update = async (id: string, data: Partial<TopicItem>) => {
     return this.db
       .update(topics)
       .set({ ...data, updatedAt: new Date() })
       .where(and(eq(topics.id, id), eq(topics.userId, this.userId)))
       .returning();
-  }
+  };
 
   // **************** Helper *************** //
 

package/src/database/server/models/user.ts

@@ -26,7 +26,7 @@ export class UserModel {
     this.db = db;
   }
 
-  async getUserState() {
+  getUserState = async () => {
     const result = await this.db
       .select({
         isOnboarded: users.isOnboarded,
@@ -81,20 +81,20 @@ export class UserModel {
       settings,
       userId: this.userId,
     };
-  }
+  };
 
-  async updateUser(value: Partial<UserItem>) {
+  updateUser = async (value: Partial<UserItem>) => {
     return this.db
       .update(users)
       .set({ ...value, updatedAt: new Date() })
       .where(eq(users.id, this.userId));
-  }
+  };
 
-  async deleteSetting() {
+  deleteSetting = async () => {
     return this.db.delete(userSettings).where(eq(userSettings.id, this.userId));
-  }
+  };
 
-  async updateSetting(value: Partial<UserSettings>) {
+  updateSetting = async (value: Partial<UserSettings>) => {
     const { keyVaults, ...res } = value;
 
     // Encrypt keyVaults
@@ -120,9 +120,9 @@ export class UserModel {
     }
 
     return this.db.update(userSettings).set(newValue).where(eq(userSettings.id, this.userId));
-  }
+  };
 
-  async updatePreference(value: Partial<UserPreference>) {
+  updatePreference = async (value: Partial<UserPreference>) => {
     const user = await this.db.query.users.findFirst({ where: eq(users.id, this.userId) });
     if (!user) return;
 
@@ -130,9 +130,9 @@ export class UserModel {
       .update(users)
       .set({ preference: merge(user.preference, value) })
       .where(eq(users.id, this.userId));
-  }
+  };
 
-  async updateGuide(value: Partial<UserGuide>) {
+  updateGuide = async (value: Partial<UserGuide>) => {
     const user = await this.db.query.users.findFirst({ where: eq(users.id, this.userId) });
     if (!user) return;
 
@@ -141,7 +141,7 @@ export class UserModel {
       .update(users)
       .set({ preference: { ...prevPreference, guide: merge(prevPreference.guide || {}, value) } })
       .where(eq(users.id, this.userId));
-  }
+  };
 
   // Static method
 

package/src/features/AgentSetting/AgentPlugin/index.tsx

@@ -11,6 +11,7 @@ import { Center, Flexbox } from 'react-layout-kit';
 import { FORM_STYLE } from '@/const/layoutTokens';
 import PluginStore from '@/features/PluginStore';
 import PluginTag from '@/features/PluginStore/PluginItem/PluginTag';
+import { useFetchInstalledPlugins } from '@/hooks/useFetchInstalledPlugins';
 import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
 import { pluginHelpers, useToolStore } from '@/store/tool';
 import { toolSelectors } from '@/store/tool/selectors';
@@ -33,7 +34,6 @@ const AgentPlugin = memo(() => {
 
   const { showDalle } = useServerConfigStore(featureFlagsSelectors);
   const installedPlugins = useToolStore(toolSelectors.metaList(showDalle), isEqual);
-  const useFetchInstalledPlugins = useToolStore((s) => s.useFetchInstalledPlugins);
 
   const { isLoading } = useFetchInstalledPlugins();
 

package/src/features/ChatInput/ActionBar/Tools/Dropdown.tsx

@@ -10,6 +10,8 @@ import { Flexbox } from 'react-layout-kit';
 
 import { useWorkspaceModal } from '@/app/(main)/chat/(workspace)/features/useWorkspaceModal';
 import PluginStore from '@/features/PluginStore';
+import { useCheckPluginsIsInstalled } from '@/hooks/useCheckPluginsIsInstalled';
+import { useFetchInstalledPlugins } from '@/hooks/useFetchInstalledPlugins';
 import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/selectors';
 import { featureFlagsSelectors, useServerConfigStore } from '@/store/serverConfig';
@@ -109,13 +111,11 @@ const DropdownMenu = memo<PropsWithChildren>(({ children }) => {
 
   const plugins = useAgentStore((s) => agentSelectors.currentAgentPlugins(s));
 
-  const [useFetchPluginStore, useFetchInstalledPlugins, checkPluginsIsInstalled] = useToolStore(
-    (s) => [s.useFetchPluginStore, s.useFetchInstalledPlugins, s.useCheckPluginsIsInstalled],
-  );
+  const [useFetchPluginStore] = useToolStore((s) => [s.useFetchPluginStore]);
 
   useFetchPluginStore();
   useFetchInstalledPlugins();
-  checkPluginsIsInstalled(plugins);
+  useCheckPluginsIsInstalled(plugins);
 
   return (
     <>

package/src/features/Portal/Thread/Chat/ChatList.tsx

@@ -2,6 +2,7 @@ import React, { memo, useCallback } from 'react';
 import { Flexbox } from 'react-layout-kit';
 
 import { SkeletonList, VirtualizedList } from '@/features/Conversation';
+import { useFetchThreads } from '@/hooks/useFetchThreads';
 import { useChatStore } from '@/store/chat';
 import { threadSelectors } from '@/store/chat/selectors';
 
@@ -15,8 +16,6 @@ const ChatList = memo(({ mobile }: ChatListProps) => {
   const data = useChatStore(threadSelectors.portalDisplayChatIDs);
   const isInit = useChatStore((s) => s.threadsInit);
 
-  const useFetchThreads = useChatStore((s) => s.useFetchThreads);
-
   useFetchThreads();
 
   const itemContent = useCallback(

package/src/hooks/useCheckPluginsIsInstalled.ts

@@ -0,0 +1,10 @@
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
+import { useToolStore } from '@/store/tool';
+
+export const useCheckPluginsIsInstalled = (plugins: string[]) => {
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
+  const checkPluginsIsInstalled = useToolStore((s) => s.useCheckPluginsIsInstalled);
+
+  checkPluginsIsInstalled(isPgliteInited, plugins);
+};

package/src/hooks/useFetchInstalledPlugins.ts

@@ -0,0 +1,10 @@
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
+import { useToolStore } from '@/store/tool';
+
+export const useFetchInstalledPlugins = () => {
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
+  const [useFetchInstalledPlugins] = useToolStore((s) => [s.useFetchInstalledPlugins]);
+
+  return useFetchInstalledPlugins(isPgliteInited);
+};

package/src/hooks/useFetchMessages.ts

@@ -0,0 +1,15 @@
+import { useChatStore } from '@/store/chat';
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
+import { useSessionStore } from '@/store/session';
+
+export const useFetchMessages = () => {
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
+  const [sessionId] = useSessionStore((s) => [s.activeId]);
+  const [activeTopicId, useFetchMessages] = useChatStore((s) => [
+    s.activeTopicId,
+    s.useFetchMessages,
+  ]);
+
+  useFetchMessages(isPgliteInited, sessionId, activeTopicId);
+};

package/src/hooks/useFetchSessions.ts

@@ -0,0 +1,13 @@
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
+import { useSessionStore } from '@/store/session';
+import { useUserStore } from '@/store/user';
+import { authSelectors } from '@/store/user/slices/auth/selectors';
+
+export const useFetchSessions = () => {
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
+  const isLogin = useUserStore(authSelectors.isLogin);
+  const useFetchSessions = useSessionStore((s) => s.useFetchSessions);
+
+  useFetchSessions(isPgliteInited, isLogin);
+};

package/src/hooks/useFetchThreads.ts

@@ -0,0 +1,11 @@
+import { useChatStore } from '@/store/chat';
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
+
+export const useFetchThreads = (activeTopicId?: string) => {
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
+
+  const [useFetchThreads] = useChatStore((s) => [s.useFetchThreads]);
+
+  useFetchThreads(isPgliteInited, activeTopicId);
+};

package/src/hooks/useFetchTopics.ts

@@ -1,4 +1,7 @@
+import { useFetchThreads } from '@/hooks/useFetchThreads';
 import { useChatStore } from '@/store/chat';
+import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
 import { useSessionStore } from '@/store/session';
 
 /**
@@ -6,12 +9,9 @@ import { useSessionStore } from '@/store/session';
  */
 export const useFetchTopics = () => {
   const [sessionId] = useSessionStore((s) => [s.activeId]);
-  const [activeTopicId, useFetchTopics, useFetchThreads] = useChatStore((s) => [
-    s.activeTopicId,
-    s.useFetchTopics,
-    s.useFetchThreads,
-  ]);
-  useFetchTopics(sessionId);
+  const [activeTopicId, useFetchTopics] = useChatStore((s) => [s.activeTopicId, s.useFetchTopics]);
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
 
+  useFetchTopics(isPgliteInited, sessionId);
   useFetchThreads(activeTopicId);
 };

package/src/layout/GlobalProvider/StoreInitialization.tsx

@@ -10,6 +10,7 @@ import { useIsMobile } from '@/hooks/useIsMobile';
 import { useEnabledDataSync } from '@/hooks/useSyncData';
 import { useAgentStore } from '@/store/agent';
 import { useGlobalStore } from '@/store/global';
+import { systemStatusSelectors } from '@/store/global/selectors';
 import { useServerConfigStore } from '@/store/serverConfig';
 import { serverConfigSelectors } from '@/store/serverConfig/selectors';
 import { useUserStore } from '@/store/user';
@@ -50,7 +51,8 @@ const StoreInitialization = memo(() => {
    * But during initialization, the value of `enableAuth` might be incorrect cause of the async fetch.
    * So we need to use `isSignedIn` only to determine whether request for the default agent config and user state.
    */
-  const isLoginOnInit = enableNextAuth ? isSignedIn : isLogin;
+  const isPgliteInited = useGlobalStore(systemStatusSelectors.isPgliteInited);
+  const isLoginOnInit = isPgliteInited && (enableNextAuth ? isSignedIn : isLogin);
 
   // init inbox agent and default agent config
   useInitAgentStore(isLoginOnInit, serverConfig.defaultAgent?.config);

package/src/libs/agent-runtime/utils/streams/azureOpenai.test.ts

@@ -1,4 +1,3 @@
-import { desc } from 'drizzle-orm/expressions';
 import { describe, expect, it, vi } from 'vitest';
 
 import { AzureOpenAIStream } from './azureOpenai';

package/src/libs/next-auth/adapter/index.ts

@@ -4,7 +4,7 @@ import type {
   AdapterUser,
   VerificationToken,
 } from '@auth/core/adapters';
-import { and, eq } from 'drizzle-orm';
+import { and, eq } from 'drizzle-orm/expressions';
 import type { NeonDatabase } from 'drizzle-orm/neon-serverless';
 import { Adapter, AdapterAccount } from 'next-auth/adapters';
 

package/src/server/routers/lambda/chunk.ts

@@ -1,14 +1,14 @@
-import { inArray } from 'drizzle-orm';
+import { inArray } from 'drizzle-orm/expressions';
 import { z } from 'zod';
 
 import { DEFAULT_EMBEDDING_MODEL } from '@/const/settings';
+import { knowledgeBaseFiles } from '@/database/schemas';
 import { serverDB } from '@/database/server';
 import { AsyncTaskModel } from '@/database/server/models/asyncTask';
 import { ChunkModel } from '@/database/server/models/chunk';
 import { EmbeddingModel } from '@/database/server/models/embedding';
 import { FileModel } from '@/database/server/models/file';
 import { MessageModel } from '@/database/server/models/message';
-import { knowledgeBaseFiles } from '@/database/schemas';
 import { ModelProvider } from '@/libs/agent-runtime';
 import { authedProcedure, router } from '@/libs/trpc';
 import { keyVaults } from '@/libs/trpc/middleware/keyVaults';

package/src/services/user/client.ts

@@ -42,8 +42,8 @@ export class ClientService implements IUserService {
     return UserModel.resetSettings();
   };
 
-  updateAvatar(avatar: string) {
-    return UserModel.updateAvatar(avatar);
+  async updateAvatar(avatar: string) {
+    await UserModel.updateAvatar(avatar);
   }
 
   async updatePreference(preference: Partial<UserPreference>) {

package/src/store/agent/slices/chat/action.test.ts

@@ -3,7 +3,6 @@ import { mutate } from 'swr';
 import { describe, expect, it, vi } from 'vitest';
 
 import { INBOX_SESSION_ID } from '@/const/session';
-import { DEFAULT_AGENT_CONFIG } from '@/const/settings';
 import { globalService } from '@/services/global';
 import { sessionService } from '@/services/session';
 import { useAgentStore } from '@/store/agent';
@@ -24,7 +23,9 @@ describe('AgentSlice', () => {
     it('should call togglePlugin with the provided id and false', async () => {
       const { result } = renderHook(() => useAgentStore());
       const pluginId = 'plugin-id';
-      const togglePluginMock = vi.spyOn(result.current, 'togglePlugin');
+      const togglePluginMock = vi
+        .spyOn(result.current, 'togglePlugin')
+        .mockResolvedValue(undefined);
 
       await act(async () => {
         await result.current.removePlugin(pluginId);
@@ -39,8 +40,9 @@ describe('AgentSlice', () => {
     it('should add plugin id to plugins array if not present and open is true or undefined', async () => {
       const { result } = renderHook(() => useAgentStore());
       const pluginId = 'plugin-id';
-      const updateAgentConfigMock = vi.spyOn(result.current, 'updateAgentConfig');
-
+      const updateAgentConfigMock = vi
+        .spyOn(result.current, 'updateAgentConfig')
+        .mockResolvedValue(undefined);
       // 模拟当前配置不包含插件 ID
       vi.spyOn(agentSelectors, 'currentAgentConfig').mockReturnValue({ plugins: [] } as any);
 
@@ -57,8 +59,9 @@ describe('AgentSlice', () => {
     it('should remove plugin id from plugins array if present and open is false', async () => {
       const { result } = renderHook(() => useAgentStore());
       const pluginId = 'plugin-id';
-      const updateAgentConfigMock = vi.spyOn(result.current, 'updateAgentConfig');
-
+      const updateAgentConfigMock = vi
+        .spyOn(result.current, 'updateAgentConfig')
+        .mockResolvedValue(undefined);
       // 模拟当前配置包含插件 ID
       vi.spyOn(agentSelectors, 'currentAgentConfig').mockReturnValue({
         plugins: [pluginId],
@@ -75,7 +78,9 @@ describe('AgentSlice', () => {
     it('should not modify plugins array if plugin id is not present and open is false', async () => {
       const { result } = renderHook(() => useAgentStore());
       const pluginId = 'plugin-id';
-      const updateAgentConfigMock = vi.spyOn(result.current, 'updateAgentConfig');
+      const updateAgentConfigMock = vi
+        .spyOn(result.current, 'updateAgentConfig')
+        .mockResolvedValue(undefined);
 
       // 模拟当前配置不包含插件 ID
       vi.spyOn(agentSelectors, 'currentAgentConfig').mockReturnValue({ plugins: [] } as any);
@@ -93,7 +98,9 @@ describe('AgentSlice', () => {
     it('should update global config if current session is inbox session', async () => {
       const { result } = renderHook(() => useAgentStore());
       const config = { model: 'gpt-3.5-turbo' };
-      const updateSessionConfigMock = vi.spyOn(sessionService, 'updateSessionConfig');
+      const updateSessionConfigMock = vi
+        .spyOn(sessionService, 'updateSessionConfig')
+        .mockResolvedValue(undefined);
       const refreshMock = vi.spyOn(result.current, 'internal_refreshAgentConfig');
 
       await act(async () => {
@@ -113,7 +120,9 @@ describe('AgentSlice', () => {
     it('should update session config if current session is not inbox session', async () => {
       const { result } = renderHook(() => useAgentStore());
       const config = { model: 'gpt-3.5-turbo' };
-      const updateSessionConfigMock = vi.spyOn(sessionService, 'updateSessionConfig');
+      const updateSessionConfigMock = vi
+        .spyOn(sessionService, 'updateSessionConfig')
+        .mockResolvedValue(undefined);
       const refreshMock = vi.spyOn(result.current, 'internal_refreshAgentConfig');
 
       // 模拟当前会话不是收件箱会话
@@ -247,7 +256,9 @@ describe('AgentSlice', () => {
     it('should call sessionService.updateSessionConfig', async () => {
       const { result } = renderHook(() => useAgentStore());
 
-      const updateSessionConfigMock = vi.spyOn(sessionService, 'updateSessionConfig');
+      const updateSessionConfigMock = vi
+        .spyOn(sessionService, 'updateSessionConfig')
+        .mockResolvedValue(undefined);
 
       await act(async () => {
         await result.current.internal_updateAgentConfig('test-session-id', { foo: 'bar' } as any);

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

@@ -5,6 +5,7 @@ import { LOADING_FLAT } from '@/const/message';
 import { DEFAULT_AGENT_CHAT_CONFIG, DEFAULT_AGENT_CONFIG } from '@/const/settings';
 import { chatService } from '@/services/chat';
 import { messageService } from '@/services/message';
+import { sessionService } from '@/services/session';
 import { topicService } from '@/services/topic';
 import { useAgentStore } from '@/store/agent';
 import { agentSelectors } from '@/store/agent/selectors';
@@ -51,6 +52,15 @@ vi.mock('@/services/chat', async (importOriginal) => {
     },
   };
 });
+vi.mock('@/services/session', async (importOriginal) => {
+  const module = await importOriginal();
+
+  return {
+    sessionService: {
+      updateSession: vi.fn(),
+    },
+  };
+});
 
 const realCoreProcessMessage = useChatStore.getState().internal_coreProcessMessage;
 

package/src/store/chat/slices/builtinTool/action.ts

@@ -72,7 +72,6 @@ export const chatToolSlice: StateCreator<
 
       const data = await useFileStore.getState().uploadWithProgress({
         file: imageFile,
-        onStatusUpdate: () => {},
       });
 
       if (!data) return;

package/src/store/chat/slices/message/action.test.ts

@@ -466,7 +466,9 @@ describe('chatMessage actions', () => {
       // 设置模拟返回值
       (messageService.getMessages as Mock).mockResolvedValue(messages);
 
-      const { result } = renderHook(() => useChatStore().useFetchMessages(sessionId, topicId));
+      const { result } = renderHook(() =>
+        useChatStore().useFetchMessages(true, sessionId, topicId),
+      );
 
       // 等待异步操作完成
       await waitFor(() => {

package/src/store/chat/slices/message/action.ts

@@ -47,7 +47,11 @@ export interface ChatMessageAction {
   modifyMessageContent: (id: string, content: string) => Promise<void>;
   toggleMessageEditing: (id: string, editing: boolean) => void;
   // query
-  useFetchMessages: (sessionId: string, topicId?: string) => SWRResponse<ChatMessage[]>;
+  useFetchMessages: (
+    enable: boolean,
+    sessionId: string,
+    topicId?: string,
+  ) => SWRResponse<ChatMessage[]>;
   copyMessage: (id: string, content: string) => Promise<void>;
   refreshMessages: () => Promise<void>;
 
@@ -220,9 +224,9 @@ export const chatMessage: StateCreator<
 
     await get().internal_updateMessageContent(id, content);
   },
-  useFetchMessages: (sessionId, activeTopicId) =>
+  useFetchMessages: (enable, sessionId, activeTopicId) =>
     useClientDataSWR<ChatMessage[]>(
-      [SWR_USE_FETCH_MESSAGES, sessionId, activeTopicId],
+      enable ? [SWR_USE_FETCH_MESSAGES, sessionId, activeTopicId] : null,
       async ([, sessionId, topicId]: [string, string, string | undefined]) =>
         messageService.getMessages(sessionId, topicId),
       {

package/src/store/chat/slices/thread/action.ts

@@ -44,7 +44,7 @@ export interface ChatThreadAction {
   openThreadCreator: (messageId: string) => void;
   openThreadInPortal: (threadId: string, sourceMessageId: string) => void;
   closeThreadPortal: () => void;
-  useFetchThreads: (topicId?: string) => SWRResponse<ThreadItem[]>;
+  useFetchThreads: (enable: boolean, topicId?: string) => SWRResponse<ThreadItem[]>;
   summaryThreadTitle: (threadId: string, messages: ChatMessage[]) => Promise<void>;
   updateThreadTitle: (id: string, title: string) => Promise<void>;
   removeThread: (id: string) => Promise<void>;
@@ -209,9 +209,9 @@ export const chatThreadMessage: StateCreator<
     return data;
   },
 
-  useFetchThreads: (topicId) =>
+  useFetchThreads: (enable, topicId) =>
     useClientDataSWR<ThreadItem[]>(
-      !!topicId && isServerMode ? [SWR_USE_FETCH_THREADS, topicId] : null,
+      enable && !!topicId && isServerMode ? [SWR_USE_FETCH_THREADS, topicId] : null,
       async ([, topicId]: [string, string]) => threadService.getThreads(topicId),
       {
         suspense: true,