@lobehub/lobehub 2.0.0-next.186 → 2.0.0-next.188

package/src/server/modules/Mecha/AgentToolsEngine/__tests__/index.test.ts

@@ -1,11 +1,11 @@
 // @vitest-environment node
+import { KnowledgeBaseManifest } from '@lobechat/builtin-tool-knowledge-base';
 import { LocalSystemManifest } from '@lobechat/builtin-tool-local-system';
 import { WebBrowsingManifest } from '@lobechat/builtin-tool-web-browsing';
 import { ToolsEngine } from '@lobechat/context-engine';
 import { describe, expect, it } from 'vitest';
 
 import { builtinTools } from '@/tools';
-import { KnowledgeBaseManifest } from '@/tools/knowledge-base';
 
 import { createServerAgentToolsEngine, createServerToolsEngine } from '../index';
 import type { InstalledPlugin, ServerAgentToolsContext } from '../types';

package/src/server/modules/Mecha/AgentToolsEngine/index.ts

@@ -9,6 +9,7 @@
  * - Gets model capabilities from provided function
  * - No dependency on frontend stores (useToolStore, useAgentStore, etc.)
  */
+import { KnowledgeBaseManifest } from '@lobechat/builtin-tool-knowledge-base';
 import { LocalSystemManifest } from '@lobechat/builtin-tool-local-system';
 import { WebBrowsingManifest } from '@lobechat/builtin-tool-web-browsing';
 import { ToolsEngine } from '@lobechat/context-engine';
@@ -16,7 +17,6 @@ import type { LobeChatPluginManifest } from '@lobehub/chat-plugin-sdk';
 import debug from 'debug';
 
 import { builtinTools } from '@/tools';
-import { KnowledgeBaseManifest } from '@/tools/knowledge-base';
 
 import type {
   ServerAgentToolsContext,

package/src/services/chat/mecha/modelParamsResolver.ts

@@ -95,6 +95,17 @@ export const resolveModelExtendParams = (ctx: ModelParamsContext): ModelExtendPa
     extendParams.reasoning_effort = chatConfig.gpt5_1ReasoningEffort;
   }
 
+  if (modelExtendParams.includes('gpt5_2ReasoningEffort') && chatConfig.gpt5_2ReasoningEffort) {
+    extendParams.reasoning_effort = chatConfig.gpt5_2ReasoningEffort;
+  }
+
+  if (
+    modelExtendParams.includes('gpt5_2ProReasoningEffort') &&
+    chatConfig.gpt5_2ProReasoningEffort
+  ) {
+    extendParams.reasoning_effort = chatConfig.gpt5_2ProReasoningEffort;
+  }
+
   // Text verbosity
   if (modelExtendParams.includes('textVerbosity') && chatConfig.textVerbosity) {
     extendParams.verbosity = chatConfig.textVerbosity;

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

@@ -5,18 +5,10 @@ import { type ChatStore } from '@/store/chat/store';
 import { type AgentBuilderAction, agentBuilderSlice } from './agentBuilder';
 import { type GroupAgentBuilderAction, groupAgentBuilderSlice } from './groupAgentBuilder';
 import { type ChatCodeInterpreterAction, codeInterpreterSlice } from './interpreter';
-import { type KnowledgeBaseAction, knowledgeBaseSlice } from './knowledgeBase';
-import { type LocalFileAction, localSystemSlice } from './localSystem';
 import { type SearchAction, searchSlice } from './search';
 
 export interface ChatBuiltinToolAction
-  extends
-    SearchAction,
-    LocalFileAction,
-    ChatCodeInterpreterAction,
-    KnowledgeBaseAction,
-    AgentBuilderAction,
-    GroupAgentBuilderAction {}
+  extends SearchAction, ChatCodeInterpreterAction, AgentBuilderAction, GroupAgentBuilderAction {}
 
 export const chatToolSlice: StateCreator<
   ChatStore,
@@ -25,9 +17,7 @@ export const chatToolSlice: StateCreator<
   ChatBuiltinToolAction
 > = (...params) => ({
   ...searchSlice(...params),
-  ...localSystemSlice(...params),
   ...codeInterpreterSlice(...params),
-  ...knowledgeBaseSlice(...params),
   ...agentBuilderSlice(...params),
   ...groupAgentBuilderSlice(...params),
 });

package/src/store/tool/slices/builtin/executors/index.ts

@@ -6,6 +6,8 @@
  */
 import { groupManagementExecutor } from '@lobechat/builtin-tool-group-management/executor';
 import { gtdExecutor } from '@lobechat/builtin-tool-gtd/executor';
+import { knowledgeBaseExecutor } from '@lobechat/builtin-tool-knowledge-base/executor';
+import { localSystemExecutor } from '@lobechat/builtin-tool-local-system/executor';
 import { memoryExecutor } from '@lobechat/builtin-tool-memory/executor';
 import { notebookExecutor } from '@lobechat/builtin-tool-notebook/executor';
 
@@ -117,6 +119,8 @@ const registerExecutors = (executors: IBuiltinToolExecutor[]): void => {
 registerExecutors([
   groupManagementExecutor,
   gtdExecutor,
+  knowledgeBaseExecutor,
+  localSystemExecutor,
   memoryExecutor,
   notebookExecutor,
   pageAgentExecutor,

package/src/styles/text.ts

@@ -33,7 +33,7 @@ export const highlightTextStyles = createStaticStyles(({ css, cssVar }) => ({
   primary: css`
     padding-block-end: 1px;
     color: ${cssVar.colorText};
-    background: linear-gradient(to top, ${cssVar.colorPrimaryBg} 40%, transparent 40%);
+    background: linear-gradient(to top, ${cssVar.colorPrimaryBgHover} 40%, transparent 40%);
   `,
   warning: css`
     padding-block-end: 1px;

package/src/tools/executionRuntimes.ts

@@ -1,19 +1,14 @@
 import { AgentBuilderManifest } from '@lobechat/builtin-tool-agent-builder';
 import { AgentBuilderExecutionRuntime } from '@lobechat/builtin-tool-agent-builder/executionRuntime';
-import { GroupAgentBuilderIdentifier } from '@lobechat/builtin-tool-group-agent-builder';
+import { GroupAgentBuilderManifest } from '@lobechat/builtin-tool-group-agent-builder';
 import { GroupAgentBuilderExecutionRuntime } from '@lobechat/builtin-tool-group-agent-builder/executionRuntime';
-import { LocalSystemManifest } from '@lobechat/builtin-tool-local-system';
 import { WebBrowsingManifest } from '@lobechat/builtin-tool-web-browsing';
 import { WebBrowsingExecutionRuntime } from '@lobechat/builtin-tool-web-browsing/executionRuntime';
 
-import { KnowledgeBaseManifest } from './knowledge-base';
-import { KnowledgeBaseExecutionRuntime } from './knowledge-base/ExecutionRuntime';
-import { LocalSystemExecutionRuntime } from './local-system/ExecutionRuntime';
+// Note: KnowledgeBase and LocalSystem now use executor pattern via toolStore/builtin/executors
 
 export const BuiltinToolServerRuntimes: Record<string, any> = {
   [AgentBuilderManifest.identifier]: AgentBuilderExecutionRuntime,
-  [GroupAgentBuilderIdentifier]: GroupAgentBuilderExecutionRuntime,
-  [KnowledgeBaseManifest.identifier]: KnowledgeBaseExecutionRuntime,
-  [LocalSystemManifest.identifier]: LocalSystemExecutionRuntime,
+  [GroupAgentBuilderManifest.identifier]: GroupAgentBuilderExecutionRuntime,
   [WebBrowsingManifest.identifier]: WebBrowsingExecutionRuntime,
 };

package/src/tools/identifiers.ts

@@ -3,6 +3,7 @@ import { CloudSandboxManifest } from '@lobechat/builtin-tool-cloud-sandbox';
 import { GroupAgentBuilderManifest } from '@lobechat/builtin-tool-group-agent-builder';
 import { GroupManagementManifest } from '@lobechat/builtin-tool-group-management';
 import { GTDManifest } from '@lobechat/builtin-tool-gtd';
+import { KnowledgeBaseManifest } from '@lobechat/builtin-tool-knowledge-base';
 import { LocalSystemManifest } from '@lobechat/builtin-tool-local-system';
 import { MemoryManifest } from '@lobechat/builtin-tool-memory';
 import { NotebookManifest } from '@lobechat/builtin-tool-notebook';
@@ -10,7 +11,6 @@ import { PageAgentManifest } from '@lobechat/builtin-tool-page-agent';
 import { WebBrowsingManifest } from '@lobechat/builtin-tool-web-browsing';
 
 import { ArtifactsManifest } from './artifacts';
-import { KnowledgeBaseManifest } from './knowledge-base';
 
 export const builtinToolIdentifiers: string[] = [
   AgentBuilderManifest.identifier,

package/src/tools/index.ts

@@ -3,6 +3,7 @@ import { CloudSandboxManifest } from '@lobechat/builtin-tool-cloud-sandbox';
 import { GroupAgentBuilderManifest } from '@lobechat/builtin-tool-group-agent-builder';
 import { GroupManagementManifest } from '@lobechat/builtin-tool-group-management';
 import { GTDManifest } from '@lobechat/builtin-tool-gtd';
+import { KnowledgeBaseManifest } from '@lobechat/builtin-tool-knowledge-base';
 import { LocalSystemManifest } from '@lobechat/builtin-tool-local-system';
 import { MemoryManifest } from '@lobechat/builtin-tool-memory';
 import { NotebookManifest } from '@lobechat/builtin-tool-notebook';
@@ -12,7 +13,6 @@ import { isDesktop } from '@lobechat/const';
 import { type LobeBuiltinTool } from '@lobechat/types';
 
 import { ArtifactsManifest } from './artifacts';
-import { KnowledgeBaseManifest } from './knowledge-base';
 
 export const builtinTools: LobeBuiltinTool[] = [
   {

package/src/tools/inspectors.ts

@@ -7,6 +7,10 @@ import {
   GroupManagementManifest,
 } from '@lobechat/builtin-tool-group-management/client';
 import { GTDInspectors, GTDManifest } from '@lobechat/builtin-tool-gtd/client';
+import {
+  KnowledgeBaseInspectors,
+  KnowledgeBaseManifest,
+} from '@lobechat/builtin-tool-knowledge-base/client';
 import {
   LocalSystemInspectors,
   LocalSystemManifest,
@@ -32,6 +36,7 @@ const BuiltinToolInspectors: Record<string, Record<string, BuiltinInspector>> =
     BuiltinInspector
   >,
   [GTDManifest.identifier]: GTDInspectors as Record<string, BuiltinInspector>,
+  [KnowledgeBaseManifest.identifier]: KnowledgeBaseInspectors as Record<string, BuiltinInspector>,
   [LocalSystemManifest.identifier]: LocalSystemInspectors as Record<string, BuiltinInspector>,
   [PageAgentManifest.identifier]: PageAgentInspectors as Record<string, BuiltinInspector>,
   [WebBrowsingManifest.identifier]: WebBrowsingInspectors as Record<string, BuiltinInspector>,

package/src/tools/renders.ts

@@ -1,31 +1,25 @@
-// agent-builder
 import { AgentBuilderManifest } from '@lobechat/builtin-tool-agent-builder';
 import { AgentBuilderRenders } from '@lobechat/builtin-tool-agent-builder/client';
-// cloud-sandbox
 import { CloudSandboxManifest } from '@lobechat/builtin-tool-cloud-sandbox';
 import { CloudSandboxRenders } from '@lobechat/builtin-tool-cloud-sandbox/client';
-// group-management
 import { GroupManagementManifest } from '@lobechat/builtin-tool-group-management';
 import { GroupManagementRenders } from '@lobechat/builtin-tool-group-management/client';
-// gtd
 import { GTDManifest, GTDRenders } from '@lobechat/builtin-tool-gtd/client';
-// local-system
 import {
-  LocalSystemIdentifier,
+  KnowledgeBaseManifest,
+  KnowledgeBaseRenders,
+} from '@lobechat/builtin-tool-knowledge-base/client';
+import {
+  LocalSystemManifest,
   LocalSystemRenders,
 } from '@lobechat/builtin-tool-local-system/client';
 import { NotebookManifest, NotebookRenders } from '@lobechat/builtin-tool-notebook/client';
-// web-browsing
 import {
   WebBrowsingManifest,
   WebBrowsingRenders,
 } from '@lobechat/builtin-tool-web-browsing/client';
 import { type BuiltinRender } from '@lobechat/types';
 
-// knowledge-base
-import { KnowledgeBaseManifest } from './knowledge-base';
-import { KnowledgeBaseRenders } from './knowledge-base/Render';
-
 /**
  * Builtin tools renders registry
  * Organized by toolset (identifier) -> API name
@@ -37,7 +31,7 @@ const BuiltinToolsRenders: Record<string, Record<string, BuiltinRender>> = {
   [GTDManifest.identifier]: GTDRenders as Record<string, BuiltinRender>,
   [NotebookManifest.identifier]: NotebookRenders as Record<string, BuiltinRender>,
   [KnowledgeBaseManifest.identifier]: KnowledgeBaseRenders as Record<string, BuiltinRender>,
-  [LocalSystemIdentifier]: LocalSystemRenders as Record<string, BuiltinRender>,
+  [LocalSystemManifest.identifier]: LocalSystemRenders as Record<string, BuiltinRender>,
   [WebBrowsingManifest.identifier]: WebBrowsingRenders as Record<string, BuiltinRender>,
 };
 

package/packages/builtin-tool-local-system/src/client/Render/RenameLocalFile/index.tsx

@@ -1,37 +0,0 @@
-import type { LocalReadFileState } from '@lobechat/builtin-tool-local-system';
-import { type RenameLocalFileParams } from '@lobechat/electron-client-ipc';
-import { type BuiltinRenderProps } from '@lobechat/types';
-import { Flexbox, Icon } from '@lobehub/ui';
-import { createStaticStyles } from 'antd-style';
-import { ArrowRightIcon } from 'lucide-react';
-import path from 'path-browserify-esm';
-import React, { memo } from 'react';
-
-import { LocalFile } from '@/features/LocalFile';
-
-const styles = createStaticStyles(({ css, cssVar }) => ({
-  container: css`
-    color: ${cssVar.colorTextQuaternary};
-  `,
-  new: css`
-    color: ${cssVar.colorTextSecondary};
-  `,
-}));
-
-const RenameLocalFile = memo<BuiltinRenderProps<RenameLocalFileParams, LocalReadFileState>>(
-  ({ args }) => {
-    const { base: oldFileName, dir } = path.parse(args.path);
-
-    return (
-      <Flexbox align={'center'} className={styles.container} gap={8} horizontal paddingInline={12}>
-        <Flexbox>{oldFileName}</Flexbox>
-        <Flexbox>
-          <Icon icon={ArrowRightIcon} />
-        </Flexbox>
-        <LocalFile name={args.newName} path={path.join(dir, args.newName)} />
-      </Flexbox>
-    );
-  },
-);
-
-export default RenameLocalFile;

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

@@ -1,201 +0,0 @@
-import { LocalFileItem, LocalMoveFilesResultItem } from '@lobechat/electron-client-ipc';
-import { describe, expect, it, vi } from 'vitest';
-
-import { localFileService } from '@/services/electron/localFileService';
-import { ChatStore } from '@/store/chat';
-
-import { localSystemSlice } from '../localSystem';
-
-vi.mock('@/services/electron/localFileService', () => ({
-  localFileService: {
-    listLocalFiles: vi.fn(),
-    moveLocalFiles: vi.fn(),
-    readLocalFile: vi.fn(),
-    readLocalFiles: vi.fn(),
-    renameLocalFile: vi.fn(),
-    searchLocalFiles: vi.fn(),
-    writeFile: vi.fn(),
-  },
-}));
-
-const mockSet = vi.fn();
-
-const mockStore = {
-  completeOperation: vi.fn(),
-  failOperation: vi.fn(),
-  internal_triggerLocalFileToolCalling: vi.fn(),
-  messageOperationMap: {},
-  optimisticUpdateMessageContent: vi.fn(),
-  optimisticUpdateMessagePluginError: vi.fn(),
-  optimisticUpdatePluginArguments: vi.fn(),
-  optimisticUpdatePluginState: vi.fn(),
-  set: mockSet,
-  startOperation: vi.fn().mockReturnValue({
-    abortController: new AbortController(),
-    operationId: 'test-op-id',
-  }),
-} as unknown as ChatStore;
-
-const createStore = () => {
-  return localSystemSlice(
-    (set) => ({
-      ...mockStore,
-      set,
-    }),
-    () => mockStore,
-    {} as any,
-  );
-};
-
-describe('localFileSlice', () => {
-  const store = createStore();
-
-  beforeEach(() => {
-    vi.clearAllMocks();
-  });
-
-  describe('internal_triggerLocalFileToolCalling', () => {
-    it('should handle successful calling', async () => {
-      const mockContent = 'result content';
-      const mockState = { state: 'test' };
-      const mockService = vi
-        .fn()
-        .mockResolvedValue({ content: mockContent, state: mockState, success: true });
-
-      await store.internal_triggerLocalFileToolCalling('test-id', mockService);
-
-      expect(mockStore.startOperation).toBeCalled();
-      expect(mockStore.completeOperation).toBeCalled();
-      expect(mockStore.optimisticUpdatePluginState).toBeCalled();
-      expect(mockStore.optimisticUpdateMessageContent).toBeCalled();
-    });
-
-    it('should handle error', async () => {
-      const mockError = new Error('test error');
-      const mockService = vi.fn().mockRejectedValue(mockError);
-
-      await store.internal_triggerLocalFileToolCalling('test-id', mockService);
-
-      expect(mockStore.optimisticUpdateMessagePluginError).toBeCalledWith(
-        'test-id',
-        {
-          body: mockError,
-          message: 'test error',
-          type: 'PluginServerError',
-        },
-        { operationId: 'test-op-id' },
-      );
-    });
-  });
-
-  describe('listLocalFiles', () => {
-    it('should call listLocalFiles service and update state', async () => {
-      const mockResult: LocalFileItem[] = [
-        {
-          name: 'test.txt',
-          path: '/test.txt',
-          isDirectory: false,
-          createdTime: new Date(),
-          lastAccessTime: new Date(),
-          modifiedTime: new Date(),
-          size: 100,
-          type: 'file',
-        },
-      ];
-      vi.mocked(localFileService.listLocalFiles).mockResolvedValue(mockResult);
-
-      await store.listLocalFiles('test-id', { path: '/test' });
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalled();
-    });
-  });
-
-  describe('moveLocalFiles', () => {
-    it('should handle successful move', async () => {
-      const mockResults = [
-        {
-          sourcePath: '/test.txt',
-          destinationPath: '/target/test.txt',
-          success: true,
-        },
-      ] as unknown as LocalMoveFilesResultItem[];
-
-      vi.mocked(localFileService.moveLocalFiles).mockResolvedValue(mockResults);
-
-      await store.moveLocalFiles('test-id', {
-        sourcePaths: ['/test.txt'],
-        destinationDir: '/target',
-      } as any);
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalled();
-    });
-  });
-
-  describe('writeLocalFile', () => {
-    it('should handle successful write', async () => {
-      vi.mocked(localFileService.writeFile).mockResolvedValue({
-        success: true,
-        newPath: '/test.txt',
-      });
-
-      await store.writeLocalFile('test-id', { path: '/test.txt', content: 'test' });
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalled();
-    });
-
-    it('should handle write error', async () => {
-      vi.mocked(localFileService.writeFile).mockResolvedValue({
-        success: false,
-        error: 'Write failed',
-        newPath: '/test.txt',
-      });
-
-      await store.writeLocalFile('test-id', { path: '/test.txt', content: 'test' });
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalled();
-    });
-  });
-
-  describe('renameLocalFile', () => {
-    it('should handle successful rename', async () => {
-      vi.mocked(localFileService.renameLocalFile).mockResolvedValue({
-        success: true,
-        newPath: '/new.txt',
-      });
-
-      await store.renameLocalFile('test-id', { path: '/test.txt', newName: 'new.txt' });
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalled();
-    });
-
-    it('should handle rename error', async () => {
-      vi.mocked(localFileService.renameLocalFile).mockResolvedValue({
-        success: false,
-        error: 'Rename failed',
-        newPath: '/test.txt',
-      });
-
-      await store.renameLocalFile('test-id', { path: '/test.txt', newName: 'new.txt' });
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalled();
-    });
-
-    it('should validate new filename', async () => {
-      vi.mocked(localFileService.renameLocalFile).mockRejectedValue(
-        new Error('Invalid new name provided'),
-      );
-
-      await store.renameLocalFile('test-id', {
-        path: '/test.txt',
-        newName: '../invalid.txt',
-      });
-
-      expect(mockStore.internal_triggerLocalFileToolCalling).toBeCalledWith(
-        'test-id',
-        expect.any(Function),
-      );
-    });
-  });
-
-  // toggleLocalFileLoading is no longer needed as we use operation-based state management
-});

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

@@ -1,163 +0,0 @@
-import debug from 'debug';
-import { type StateCreator } from 'zustand/vanilla';
-
-import { agentSelectors } from '@/store/agent/selectors';
-import { getAgentStoreState } from '@/store/agent/store';
-import { type ChatStore } from '@/store/chat/store';
-import { KnowledgeBaseExecutionRuntime } from '@/tools/knowledge-base/ExecutionRuntime';
-
-const log = debug('lobe-store:builtin-tool:knowledge-base');
-
-export interface KnowledgeBaseAction {
-  internal_triggerKnowledgeBaseToolCalling: (
-    id: string,
-    callingService: () => Promise<{ content: string; error?: any; state?: any; success: boolean }>,
-  ) => Promise<boolean>;
-
-  /**
-   * Read full content of specific files from knowledge base
-   */
-  readKnowledge: (
-    id: string,
-    params: {
-      fileIds: string[];
-    },
-  ) => Promise<boolean>;
-
-  /**
-   * Search knowledge base for relevant files and chunks
-   */
-  searchKnowledgeBase: (
-    id: string,
-    params: {
-      query: string;
-      topK?: number;
-    },
-  ) => Promise<boolean>;
-}
-
-const runtime = new KnowledgeBaseExecutionRuntime();
-
-/* eslint-disable sort-keys-fix/sort-keys-fix */
-export const knowledgeBaseSlice: StateCreator<
-  ChatStore,
-  [['zustand/devtools', never]],
-  [],
-  KnowledgeBaseAction
-> = (set, get) => ({
-  readKnowledge: async (id, params) => {
-    return get().internal_triggerKnowledgeBaseToolCalling(id, async () => {
-      return await runtime.readKnowledge(params);
-    });
-  },
-
-  searchKnowledgeBase: async (id, params) => {
-    // Get knowledge base IDs from agent store
-    const agentState = getAgentStoreState();
-    const knowledgeIds = agentSelectors.currentKnowledgeIds(agentState);
-
-    // Only search in knowledge bases, not agent files
-    // Agent files will be injected as full content in context-engine
-    const knowledgeBaseIds = knowledgeIds.knowledgeBaseIds;
-
-    return get().internal_triggerKnowledgeBaseToolCalling(id, async () => {
-      return await runtime.searchKnowledgeBase(params, {
-        knowledgeBaseIds,
-        messageId: id,
-      });
-    });
-  },
-
-  // ==================== utils ====================
-
-  internal_triggerKnowledgeBaseToolCalling: async (id, callingService) => {
-    // Get parent operationId from messageOperationMap (should be executeToolCall)
-    const parentOperationId = get().messageOperationMap[id];
-
-    // Create child operation for knowledge base execution
-    // Auto-associates message with this operation via messageId in context
-    const { operationId: knowledgeBaseOpId, abortController } = get().startOperation({
-      context: {
-        messageId: id,
-      },
-      metadata: {
-        startTime: Date.now(),
-      },
-      parentOperationId,
-      type: 'builtinToolKnowledgeBase',
-    });
-
-    log(
-      '[knowledgeBase] messageId=%s, parentOpId=%s, knowledgeBaseOpId=%s, aborted=%s',
-      id,
-      parentOperationId,
-      knowledgeBaseOpId,
-      abortController.signal.aborted,
-    );
-
-    const context = { operationId: knowledgeBaseOpId };
-
-    try {
-      const { state, content, success, error } = await callingService();
-
-      // Complete knowledge base operation
-      get().completeOperation(knowledgeBaseOpId);
-
-      if (success) {
-        if (state) {
-          await get().optimisticUpdatePluginState(id, state, context);
-        }
-        await get().optimisticUpdateMessageContent(id, content, undefined, context);
-      } else {
-        await get().optimisticUpdateMessagePluginError(
-          id,
-          {
-            body: error,
-            message: error?.message || 'Operation failed',
-            type: 'PluginServerError',
-          },
-          context,
-        );
-        // Still update content even if failed, to show error message
-        await get().optimisticUpdateMessageContent(id, content, undefined, context);
-      }
-
-      return true;
-    } catch (error) {
-      const err = error as Error;
-
-      log('[knowledgeBase] Error: messageId=%s, error=%s', id, err.message);
-
-      // Check if it's an abort error
-      if (err.message.includes('The user aborted a request.') || err.name === 'AbortError') {
-        log('[knowledgeBase] Request aborted: messageId=%s', id);
-        // Fail knowledge base operation for abort
-        get().failOperation(knowledgeBaseOpId, {
-          message: 'User cancelled the request',
-          type: 'UserAborted',
-        });
-        // Don't update error message for user aborts
-        return false;
-      }
-
-      // Fail knowledge base operation for other errors
-      get().failOperation(knowledgeBaseOpId, {
-        message: err.message,
-        type: 'PluginServerError',
-      });
-
-      // For other errors, update message
-      await get().optimisticUpdateMessagePluginError(
-        id,
-        {
-          body: error,
-          message: err.message,
-          type: 'PluginServerError',
-        },
-        context,
-      );
-
-      return false;
-    }
-  },
-});