@lobehub/lobehub 2.0.0-next.187 → 2.0.0-next.189

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;
-    }
-  },
-});

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

@@ -1,241 +0,0 @@
-import {
-  type EditLocalFileParams,
-  type GetCommandOutputParams,
-  type GlobFilesParams,
-  type GrepContentParams,
-  type KillCommandParams,
-  type ListLocalFileParams,
-  type LocalReadFileParams,
-  type LocalReadFilesParams,
-  type LocalSearchFilesParams,
-  type MoveLocalFilesParams,
-  type RenameLocalFileParams,
-  type RunCommandParams,
-  type WriteLocalFileParams,
-} from '@lobechat/electron-client-ipc';
-import debug from 'debug';
-import { type StateCreator } from 'zustand/vanilla';
-
-import { type ChatStore } from '@/store/chat/store';
-import { LocalSystemExecutionRuntime } from '@/tools/local-system/ExecutionRuntime';
-
-const log = debug('lobe-store:builtin-tool');
-
-/* eslint-disable typescript-sort-keys/interface */
-export interface LocalFileAction {
-  internal_triggerLocalFileToolCalling: (
-    id: string,
-    callingService: () => Promise<{ content: string; error?: any; state?: any; success: boolean }>,
-  ) => Promise<boolean>;
-
-  // File Operations
-  listLocalFiles: (id: string, params: ListLocalFileParams) => Promise<boolean>;
-  moveLocalFiles: (id: string, params: MoveLocalFilesParams) => Promise<boolean>;
-  readLocalFile: (id: string, params: LocalReadFileParams) => Promise<boolean>;
-  readLocalFiles: (id: string, params: LocalReadFilesParams) => Promise<boolean>;
-  renameLocalFile: (id: string, params: RenameLocalFileParams) => Promise<boolean>;
-  reSearchLocalFiles: (id: string, params: LocalSearchFilesParams) => Promise<boolean>;
-  searchLocalFiles: (id: string, params: LocalSearchFilesParams) => Promise<boolean>;
-  writeLocalFile: (id: string, params: WriteLocalFileParams) => Promise<boolean>;
-
-  // Shell Commands
-  editLocalFile: (id: string, params: EditLocalFileParams) => Promise<boolean>;
-  getCommandOutput: (id: string, params: GetCommandOutputParams) => Promise<boolean>;
-  killCommand: (id: string, params: KillCommandParams) => Promise<boolean>;
-  runCommand: (id: string, params: RunCommandParams) => Promise<boolean>;
-
-  // Search & Find
-  globLocalFiles: (id: string, params: GlobFilesParams) => Promise<boolean>;
-  grepContent: (id: string, params: GrepContentParams) => Promise<boolean>;
-}
-/* eslint-enable typescript-sort-keys/interface */
-
-const runtime = new LocalSystemExecutionRuntime();
-
-/* eslint-disable sort-keys-fix/sort-keys-fix */
-export const localSystemSlice: StateCreator<
-  ChatStore,
-  [['zustand/devtools', never]],
-  [],
-  LocalFileAction
-> = (set, get) => ({
-  // ==================== File Editing ====================
-  editLocalFile: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.editLocalFile(params);
-    });
-  },
-
-  writeLocalFile: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.writeLocalFile(params);
-    });
-  },
-  moveLocalFiles: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.moveLocalFiles(params);
-    });
-  },
-  renameLocalFile: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.renameLocalFile(params);
-    });
-  },
-
-  // ==================== Search & Find ====================
-  grepContent: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.grepContent(params);
-    });
-  },
-
-  globLocalFiles: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.globLocalFiles(params);
-    });
-  },
-
-  searchLocalFiles: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.searchLocalFiles(params);
-    });
-  },
-
-  listLocalFiles: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.listLocalFiles(params);
-    });
-  },
-
-  reSearchLocalFiles: async (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);
-  },
-
-  readLocalFile: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.readLocalFile(params);
-    });
-  },
-
-  readLocalFiles: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.readLocalFiles(params);
-    });
-  },
-
-  // ==================== Shell Commands ====================
-  runCommand: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.runCommand(params);
-    });
-  },
-  killCommand: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.killCommand(params);
-    });
-  },
-  getCommandOutput: async (id, params) => {
-    return get().internal_triggerLocalFileToolCalling(id, async () => {
-      return await runtime.getCommandOutput(params);
-    });
-  },
-
-  // ==================== utils ====================
-
-  internal_triggerLocalFileToolCalling: async (id, callingService) => {
-    // Get parent operationId from messageOperationMap (should be executeToolCall)
-    const parentOperationId = get().messageOperationMap[id];
-
-    // Create child operation for local system execution
-    // Auto-associates message with this operation via messageId in context
-    const { operationId: localSystemOpId, abortController } = get().startOperation({
-      type: 'builtinToolLocalSystem',
-      context: {
-        messageId: id,
-      },
-      parentOperationId,
-      metadata: {
-        startTime: Date.now(),
-      },
-    });
-
-    log(
-      '[localSystem] messageId=%s, parentOpId=%s, localSystemOpId=%s, aborted=%s',
-      id,
-      parentOperationId,
-      localSystemOpId,
-      abortController.signal.aborted,
-    );
-
-    const context = { operationId: localSystemOpId };
-
-    try {
-      const { state, content, success, error } = await callingService();
-
-      // Complete local system operation
-      get().completeOperation(localSystemOpId);
-
-      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('[localSystem] 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('[localSystem] Request aborted: messageId=%s', id);
-        // Fail local system operation for abort
-        get().failOperation(localSystemOpId, {
-          message: 'User cancelled the request',
-          type: 'UserAborted',
-        });
-        // Don't update error message for user aborts
-        return false;
-      }
-
-      // Fail local system operation for other errors
-      get().failOperation(localSystemOpId, {
-        message: err.message,
-        type: 'PluginServerError',
-      });
-
-      // For other errors, update message
-      await get().optimisticUpdateMessagePluginError(
-        id,
-        {
-          body: error,
-          message: err.message,
-          type: 'PluginServerError',
-        },
-        context,
-      );
-
-      return false;
-    }
-  },
-});

package/src/tools/knowledge-base/ExecutionRuntime/index.ts

@@ -1,25 +0,0 @@
-import { KnowledgeBaseExecutionRuntime as BaseKnowledgeBaseExecutionRuntime } from '@lobechat/builtin-tool-knowledge-base/executionRuntime';
-import type { ReadKnowledgeArgs, SearchKnowledgeBaseArgs } from '@lobechat/builtin-tool-knowledge-base';
-
-import { ragService } from '@/services/rag';
-
-// Create a factory that returns an instance with ragService injected
-export class KnowledgeBaseExecutionRuntime extends BaseKnowledgeBaseExecutionRuntime {
-  constructor() {
-    super(ragService);
-  }
-
-  searchKnowledgeBase(
-    args: SearchKnowledgeBaseArgs,
-    options?: Parameters<BaseKnowledgeBaseExecutionRuntime['searchKnowledgeBase']>[1],
-  ) {
-    return super.searchKnowledgeBase(args, options);
-  }
-
-  readKnowledge(args: ReadKnowledgeArgs, options?: Parameters<BaseKnowledgeBaseExecutionRuntime['readKnowledge']>[1]) {
-    return super.readKnowledge(args, options);
-  }
-}
-
-// Re-export types for convenience
-export type { ReadKnowledgeArgs, SearchKnowledgeBaseArgs } from '@lobechat/builtin-tool-knowledge-base';

package/src/tools/knowledge-base/Render/ReadKnowledge/index.tsx

@@ -1,29 +0,0 @@
-'use client';
-
-import { ReadKnowledge as BaseReadKnowledge } from '@lobechat/builtin-tool-knowledge-base/client';
-import { type BuiltinRenderProps } from '@lobechat/types';
-import { memo } from 'react';
-import { useTranslation } from 'react-i18next';
-
-import FileIcon from '@/components/FileIcon';
-
-import type { ReadKnowledgeArgs, ReadKnowledgeState } from '../../index';
-
-const ReadKnowledge = memo<BuiltinRenderProps<ReadKnowledgeArgs, ReadKnowledgeState>>(
-  ({ pluginState }) => {
-    const { t } = useTranslation('tool');
-
-    return (
-      <BaseReadKnowledge
-        FileIcon={FileIcon}
-        labels={{
-          chars: t('lobe-knowledge-base.readKnowledge.meta.chars'),
-          lines: t('lobe-knowledge-base.readKnowledge.meta.lines'),
-        }}
-        pluginState={pluginState}
-      />
-    );
-  },
-);
-
-export default ReadKnowledge;

package/src/tools/knowledge-base/Render/SearchKnowledgeBase/index.tsx

@@ -1,29 +0,0 @@
-'use client';
-
-import { SearchKnowledgeBase as BaseSearchKnowledgeBase } from '@lobechat/builtin-tool-knowledge-base/client';
-import { type BuiltinRenderProps } from '@lobechat/types';
-import { memo } from 'react';
-
-import FileIcon from '@/components/FileIcon';
-import { useIsMobile } from '@/hooks/useIsMobile';
-import { useChatStore } from '@/store/chat';
-
-import type { SearchKnowledgeBaseArgs, SearchKnowledgeBaseState } from '../../index';
-
-const SearchKnowledgeBase = memo<
-  BuiltinRenderProps<SearchKnowledgeBaseArgs, SearchKnowledgeBaseState>
->(({ pluginState }) => {
-  const openFilePreview = useChatStore((s) => s.openFilePreview);
-  const isMobile = useIsMobile();
-
-  return (
-    <BaseSearchKnowledgeBase
-      FileIcon={FileIcon}
-      isMobile={isMobile}
-      onFileClick={openFilePreview}
-      pluginState={pluginState}
-    />
-  );
-});
-
-export default SearchKnowledgeBase;

package/src/tools/knowledge-base/Render/index.ts

@@ -1,7 +0,0 @@
-import ReadKnowledge from './ReadKnowledge';
-import SearchKnowledgeBase from './SearchKnowledgeBase';
-
-export const KnowledgeBaseRenders = {
-  readKnowledge: ReadKnowledge,
-  searchKnowledgeBase: SearchKnowledgeBase,
-};

package/src/tools/knowledge-base/index.ts

@@ -1,12 +0,0 @@
-// Re-export from the package for backward compatibility
-export {
-  type FileContentDetail,
-  KnowledgeBaseApiName,
-  KnowledgeBaseIdentifier,
-  KnowledgeBaseManifest,
-  type ReadKnowledgeArgs,
-  type ReadKnowledgeState,
-  type SearchKnowledgeBaseArgs,
-  type SearchKnowledgeBaseState,
-  systemPrompt,
-} from '@lobechat/builtin-tool-knowledge-base';

package/src/tools/local-system/ExecutionRuntime/index.ts

@@ -1,9 +0,0 @@
-import { LocalSystemExecutionRuntime as BaseLocalSystemExecutionRuntime } from '@lobechat/builtin-tool-local-system/executionRuntime';
-
-import { localFileService } from '@/services/electron/localFileService';
-
-export class LocalSystemExecutionRuntime extends BaseLocalSystemExecutionRuntime {
-  constructor() {
-    super(localFileService);
-  }
-}

package/src/tools/local-system/systemRole.ts

@@ -1 +0,0 @@
-export { systemPrompt } from '@lobechat/builtin-tool-local-system';