@lobehub/lobehub 2.0.0-next.104 → 2.0.0-next.106

package/src/store/chat/slices/builtinTool/actions/localSystem.ts

@@ -108,7 +108,11 @@ export const localSystemSlice: StateCreator<
   },
 
   reSearchLocalFiles: async (id, params) => {
-    await get().optimisticUpdatePluginArguments(id, params);
+    // Get operationId from messageOperationMap to ensure proper context isolation
+    const operationId = get().messageOperationMap[id];
+    const context = operationId ? { operationId } : undefined;
+
+    await get().optimisticUpdatePluginArguments(id, params, false, context);
 
     return get().searchLocalFiles(id, params);
   },

package/src/store/chat/slices/builtinTool/actions/search.ts

@@ -276,7 +276,11 @@ export const searchSlice: StateCreator<
   },
 
   triggerSearchAgain: async (id, data, options) => {
-    await get().optimisticUpdatePluginArguments(id, data);
+    // Get operationId from messageOperationMap to ensure proper context isolation
+    const operationId = get().messageOperationMap[id];
+    const context = operationId ? { operationId } : undefined;
+
+    await get().optimisticUpdatePluginArguments(id, data, false, context);
 
     await get().search(id, data, options?.aiSummary);
   },

package/src/store/chat/slices/message/actions/publicApi.ts

@@ -167,14 +167,22 @@ export const messagePublicApi: StateCreator<
     const message = dbMessageSelectors.getDbMessageById(id)(get());
     if (!message || message.role !== 'tool') return;
 
+    // Get operationId from messageOperationMap to ensure proper context isolation
+    const operationId = get().messageOperationMap[id];
+    const context = operationId ? { operationId } : undefined;
+
     const removeToolInAssistantMessage = async () => {
       if (!message.parentId) return;
-      await get().optimisticRemoveToolFromAssistantMessage(message.parentId, message.tool_call_id);
+      await get().optimisticRemoveToolFromAssistantMessage(
+        message.parentId,
+        message.tool_call_id,
+        context,
+      );
     };
 
     await Promise.all([
       // 1. remove tool message
-      get().optimisticDeleteMessage(id),
+      get().optimisticDeleteMessage(id, context),
       // 2. remove the tool item in the assistant tools
       removeToolInAssistantMessage(),
     ]);

package/src/store/chat/slices/message/actions/query.ts

@@ -32,6 +32,7 @@ export interface MessageQueryAction {
     messages: UIChatMessage[],
     params?: {
       action?: any;
+      operationId?: string;
       sessionId?: string;
       topicId?: string | null;
     },
@@ -66,10 +67,22 @@ export const messageQuery: StateCreator<
   },
 
   replaceMessages: (messages, params) => {
-    const messagesKey = messageMapKey(
-      params?.sessionId ?? get().activeId,
-      params?.topicId ?? get().activeTopicId,
-    );
+    let sessionId: string;
+    let topicId: string | null | undefined;
+
+    // Priority 1: Get context from operation if operationId is provided
+    if (params?.operationId) {
+      const { sessionId: opSessionId, topicId: opTopicId } =
+        get().internal_getSessionContext(params);
+      sessionId = opSessionId;
+      topicId = opTopicId;
+    } else {
+      // Priority 2: Use explicit sessionId/topicId or fallback to global state
+      sessionId = params?.sessionId ?? get().activeId;
+      topicId = params?.topicId ?? get().activeTopicId;
+    }
+
+    const messagesKey = messageMapKey(sessionId, topicId);
 
     // Get raw messages from dbMessagesMap and apply reducer
     const nextDbMap = { ...get().dbMessagesMap, [messagesKey]: messages };

package/src/store/chat/slices/operation/__tests__/selectors.test.ts

@@ -393,6 +393,11 @@ describe('Operation Selectors', () => {
     it('isMainWindowAgentRuntimeRunning should only detect main window operations', () => {
       const { result } = renderHook(() => useChatStore());
 
+      // Set active context
+      act(() => {
+        useChatStore.setState({ activeId: 'session1', activeTopicId: undefined });
+      });
+
       expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(false);
 
       // Start a main window operation (inThread: false)
@@ -400,7 +405,7 @@ describe('Operation Selectors', () => {
       act(() => {
         mainOpId = result.current.startOperation({
           type: 'execAgentRuntime',
-          context: { sessionId: 'session1' },
+          context: { sessionId: 'session1', topicId: null },
           metadata: { inThread: false },
         }).operationId;
       });
@@ -420,12 +425,17 @@ describe('Operation Selectors', () => {
     it('isMainWindowAgentRuntimeRunning should exclude thread operations', () => {
       const { result } = renderHook(() => useChatStore());
 
+      // Set active context
+      act(() => {
+        useChatStore.setState({ activeId: 'session1', activeTopicId: undefined });
+      });
+
       // Start a thread operation (inThread: true)
       let threadOpId: string;
       act(() => {
         threadOpId = result.current.startOperation({
           type: 'execAgentRuntime',
-          context: { sessionId: 'session1', threadId: 'thread1' },
+          context: { sessionId: 'session1', topicId: null, threadId: 'thread1' },
           metadata: { inThread: true },
         }).operationId;
       });
@@ -447,6 +457,11 @@ describe('Operation Selectors', () => {
     it('isMainWindowAgentRuntimeRunning should distinguish between main and thread operations', () => {
       const { result } = renderHook(() => useChatStore());
 
+      // Set active context
+      act(() => {
+        useChatStore.setState({ activeId: 'session1', activeTopicId: undefined });
+      });
+
       let mainOpId: string;
       let threadOpId: string;
 
@@ -454,13 +469,13 @@ describe('Operation Selectors', () => {
       act(() => {
         mainOpId = result.current.startOperation({
           type: 'execAgentRuntime',
-          context: { sessionId: 'session1' },
+          context: { sessionId: 'session1', topicId: null },
           metadata: { inThread: false },
         }).operationId;
 
         threadOpId = result.current.startOperation({
           type: 'execAgentRuntime',
-          context: { sessionId: 'session1', threadId: 'thread1' },
+          context: { sessionId: 'session1', topicId: null, threadId: 'thread1' },
           metadata: { inThread: true },
         }).operationId;
       });
@@ -489,11 +504,16 @@ describe('Operation Selectors', () => {
     it('isMainWindowAgentRuntimeRunning should exclude aborting operations', () => {
       const { result } = renderHook(() => useChatStore());
 
+      // Set active context
+      act(() => {
+        useChatStore.setState({ activeId: 'session1', activeTopicId: undefined });
+      });
+
       let opId: string;
       act(() => {
         opId = result.current.startOperation({
           type: 'execAgentRuntime',
-          context: { sessionId: 'session1' },
+          context: { sessionId: 'session1', topicId: null },
           metadata: { inThread: false },
         }).operationId;
       });
@@ -509,5 +529,73 @@ describe('Operation Selectors', () => {
       expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(false);
       expect(operationSelectors.isAgentRuntimeRunning(result.current)).toBe(false);
     });
+
+    it('isMainWindowAgentRuntimeRunning should only detect operations in current active topic', () => {
+      const { result } = renderHook(() => useChatStore());
+
+      // Set active session and topic
+      act(() => {
+        useChatStore.setState({ activeId: 'session1', activeTopicId: 'topic1' });
+      });
+
+      let topic1OpId: string;
+      let topic2OpId: string;
+
+      // Start operation in topic1 (current active topic)
+      act(() => {
+        topic1OpId = result.current.startOperation({
+          type: 'execAgentRuntime',
+          context: { sessionId: 'session1', topicId: 'topic1' },
+          metadata: { inThread: false },
+        }).operationId;
+      });
+
+      // Should detect operation in current topic
+      expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(true);
+
+      // Start operation in topic2 (different topic)
+      act(() => {
+        topic2OpId = result.current.startOperation({
+          type: 'execAgentRuntime',
+          context: { sessionId: 'session1', topicId: 'topic2' },
+          metadata: { inThread: false },
+        }).operationId;
+      });
+
+      // Should still only detect topic1 operation (current active topic)
+      expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(true);
+
+      // Switch to topic2
+      act(() => {
+        useChatStore.setState({ activeTopicId: 'topic2' });
+      });
+
+      // Should now detect topic2 operation
+      expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(true);
+
+      // Complete topic2 operation
+      act(() => {
+        result.current.completeOperation(topic2OpId!);
+      });
+
+      // Should not detect any operation in topic2 now
+      expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(false);
+
+      // Switch back to topic1
+      act(() => {
+        useChatStore.setState({ activeTopicId: 'topic1' });
+      });
+
+      // Should detect topic1 operation again
+      expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(true);
+
+      // Complete topic1 operation
+      act(() => {
+        result.current.completeOperation(topic1OpId!);
+      });
+
+      // Should not detect any operation now
+      expect(operationSelectors.isMainWindowAgentRuntimeRunning(result.current)).toBe(false);
+    });
   });
 });

package/src/store/chat/slices/operation/selectors.ts

@@ -180,14 +180,27 @@ const isAgentRuntimeRunning = (s: ChatStoreState): boolean => {
 /**
  * Check if agent runtime is running in main window only
  * Used for main window UI state (e.g., send button loading)
- * Excludes thread operations to prevent cross-contamination
+ * Excludes thread operations and operations from other topics to prevent cross-contamination
  */
 const isMainWindowAgentRuntimeRunning = (s: ChatStoreState): boolean => {
   const operationIds = s.operationsByType['execAgentRuntime'] || [];
+
   return operationIds.some((id) => {
     const op = s.operations[id];
-    // Only include main window operations (not thread)
-    return op && op.status === 'running' && !op.metadata.isAborting && !op.metadata.inThread;
+    if (!op || op.status !== 'running' || op.metadata.isAborting || op.metadata.inThread) {
+      return false;
+    }
+
+    // Session must match
+    if (s.activeId !== op.context.sessionId) return false;
+
+    // Topic comparison: normalize null/undefined (both mean "default topic")
+    // activeTopicId can be null (initial state) or undefined (after topic operations)
+    // Operation context topicId can also be null or undefined
+    const activeTopicId = s.activeTopicId ?? null;
+    const opTopicId = op.context.topicId ?? null;
+
+    return activeTopicId === opTopicId;
   });
 };
 

package/src/store/chat/slices/plugin/actions/optimisticUpdate.ts

@@ -32,6 +32,7 @@ export interface PluginOptimisticUpdateAction {
     id: string,
     value: T,
     replace?: boolean,
+    context?: OptimisticUpdateContext,
   ) => Promise<void>;
 
   /**
@@ -106,7 +107,7 @@ export const pluginOptimisticUpdate: StateCreator<
     }
   },
 
-  optimisticUpdatePluginArguments: async (id, value, replace = false) => {
+  optimisticUpdatePluginArguments: async (id, value, replace = false, context) => {
     const { refreshMessages } = get();
     const toolMessage = displayMessageSelectors.getDisplayMessageById(id)(get());
     if (!toolMessage || !toolMessage?.tool_call_id) return;
@@ -121,20 +122,22 @@ export const pluginOptimisticUpdate: StateCreator<
     if (isEqual(prevJson, nextValue)) return;
 
     // optimistic update
-    get().internal_dispatchMessage({
-      id,
-      type: 'updateMessagePlugin',
-      value: { arguments: JSON.stringify(nextValue) },
-    });
+    get().internal_dispatchMessage(
+      { id, type: 'updateMessagePlugin', value: { arguments: JSON.stringify(nextValue) } },
+      context,
+    );
 
     // 同样需要更新 assistantMessage 的 pluginArguments
     if (assistantMessage) {
-      get().internal_dispatchMessage({
-        id: assistantMessage.id,
-        type: 'updateMessageTools',
-        tool_call_id: toolMessage?.tool_call_id,
-        value: { arguments: JSON.stringify(nextValue) },
-      });
+      get().internal_dispatchMessage(
+        {
+          id: assistantMessage.id,
+          type: 'updateMessageTools',
+          tool_call_id: toolMessage?.tool_call_id,
+          value: { arguments: JSON.stringify(nextValue) },
+        },
+        context,
+      );
       assistantMessage = displayMessageSelectors.getDisplayMessageById(assistantMessage?.id)(get());
     }
 
@@ -183,11 +186,14 @@ export const pluginOptimisticUpdate: StateCreator<
     if (!assistantMessage) return;
 
     const { internal_dispatchMessage, internal_refreshToUpdateMessageTools } = get();
-    internal_dispatchMessage({
-      type: 'addMessageTool',
-      value: tool,
-      id: assistantMessage.id,
-    });
+    internal_dispatchMessage(
+      {
+        type: 'addMessageTool',
+        value: tool,
+        id: assistantMessage.id,
+      },
+      context,
+    );
 
     await internal_refreshToUpdateMessageTools(id, context);
   },
@@ -199,7 +205,7 @@ export const pluginOptimisticUpdate: StateCreator<
     const { internal_dispatchMessage, internal_refreshToUpdateMessageTools } = get();
 
     // optimistic update
-    internal_dispatchMessage({ type: 'deleteMessageTool', tool_call_id, id: message.id });
+    internal_dispatchMessage({ type: 'deleteMessageTool', tool_call_id, id: message.id }, context);
 
     // update the message tools
     await internal_refreshToUpdateMessageTools(id, context);

package/src/store/user/slices/settings/selectors/keyVaults.ts

@@ -10,12 +10,7 @@ const getVaultByProvider = (provider: string) => (s: UserStore) =>
   // @ts-ignore
   (keyVaultsSettings(s)[provider] || {}) as any;
 
-const password = (s: UserStore) => keyVaultsSettings(s).password || '';
-
 export const keyVaultsConfigSelectors = {
   getVaultByProvider,
-
   keyVaultsSettings,
-
-  password,
 };

package/src/features/ChatList/Error/AccessCodeForm.tsx

@@ -1,63 +0,0 @@
-import { Button, InputPassword } from '@lobehub/ui';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-import { Flexbox } from 'react-layout-kit';
-
-import { useChatStore } from '@/store/chat';
-import { useUserStore } from '@/store/user';
-import { keyVaultsConfigSelectors } from '@/store/user/selectors';
-
-import { FormAction } from './style';
-
-interface AccessCodeFormProps {
-  id: string;
-}
-
-const AccessCodeForm = memo<AccessCodeFormProps>(({ id }) => {
-  const { t } = useTranslation('error');
-  const [password, updateKeyVaults] = useUserStore((s) => [
-    keyVaultsConfigSelectors.password(s),
-    s.updateKeyVaults,
-  ]);
-  const [resend, deleteMessage] = useChatStore((s) => [s.delAndRegenerateMessage, s.deleteMessage]);
-
-  return (
-    <>
-      <FormAction
-        avatar={'🗳'}
-        description={t('unlock.password.description')}
-        title={t('unlock.password.title')}
-      >
-        <InputPassword
-          autoComplete={'new-password'}
-          onChange={(e) => {
-            updateKeyVaults({ password: e.target.value });
-          }}
-          placeholder={t('unlock.password.placeholder')}
-          value={password}
-          variant={'filled'}
-        />
-      </FormAction>
-      <Flexbox gap={12}>
-        <Button
-          onClick={() => {
-            resend(id);
-            deleteMessage(id);
-          }}
-          type={'primary'}
-        >
-          {t('unlock.confirm')}
-        </Button>
-        <Button
-          onClick={() => {
-            deleteMessage(id);
-          }}
-        >
-          {t('unlock.closeMessage')}
-        </Button>
-      </Flexbox>
-    </>
-  );
-});
-
-export default AccessCodeForm;

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

@@ -1,61 +0,0 @@
-import type { PartialDeep } from 'type-fest';
-import { Mock, afterEach, beforeEach, describe, expect, it, vi } from 'vitest';
-
-import { LOBE_URL_IMPORT_NAME } from '@/const/url';
-import { UserSettings } from '@/types/user/settings';
-
-import { shareService } from '../share';
-
-// Mock dependencies
-vi.mock('@/utils/parseMarkdown', () => ({
-  parseMarkdown: vi.fn(),
-}));
-
-global.fetch = vi.fn();
-
-beforeEach(() => {
-  vi.clearAllMocks();
-});
-describe('ShareGPTService', () => {
-  describe('ShareViaUrl', () => {
-    describe('createShareSettingsUrl', () => {
-      it('should create a share settings URL with the provided settings', () => {
-        const settings: PartialDeep<UserSettings> = {
-          keyVaults: {
-            openai: {
-              apiKey: 'user-key',
-            },
-          },
-        };
-        const url = shareService.createShareSettingsUrl(settings);
-        expect(url).toBe(
-          `/?${LOBE_URL_IMPORT_NAME}=%7B%22keyVaults%22:%7B%22openai%22:%7B%22apiKey%22:%22user-key%22%7D%7D%7D`,
-        );
-      });
-    });
-
-    describe('decodeShareSettings', () => {
-      it('should decode share settings from search params', () => {
-        const settings = '{"languageModel":{"openai":{"apiKey":"user-key"}}}';
-        const decodedSettings = shareService.decodeShareSettings(settings);
-        expect(decodedSettings).toEqual({
-          data: {
-            languageModel: {
-              openai: {
-                apiKey: 'user-key',
-              },
-            },
-          },
-        });
-      });
-
-      it('should return an error message if decoding fails', () => {
-        const settings = '%7B%22theme%22%3A%22dark%22%2C%22fontSize%22%3A16%';
-        const decodedSettings = shareService.decodeShareSettings(settings);
-        expect(decodedSettings).toEqual({
-          message: expect.any(String),
-        });
-      });
-    });
-  });
-});