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

package/packages/builtin-tool-knowledge-base/src/client/Inspector/SearchKnowledgeBase/index.tsx

@@ -0,0 +1,75 @@
+'use client';
+
+import { type BuiltinInspectorProps } from '@lobechat/types';
+import { Text } from '@lobehub/ui';
+import { createStaticStyles, cssVar, cx } from 'antd-style';
+import { memo } from 'react';
+import { useTranslation } from 'react-i18next';
+
+import { highlightTextStyles, shinyTextStyles } from '@/styles';
+
+import { type SearchKnowledgeBaseArgs, type SearchKnowledgeBaseState } from '../../..';
+
+const styles = createStaticStyles(({ css, cssVar }) => ({
+  root: css`
+    overflow: hidden;
+    display: -webkit-box;
+    -webkit-box-orient: vertical;
+    -webkit-line-clamp: 1;
+
+    color: ${cssVar.colorTextSecondary};
+  `,
+}));
+
+export const SearchKnowledgeBaseInspector = memo<
+  BuiltinInspectorProps<SearchKnowledgeBaseArgs, SearchKnowledgeBaseState>
+>(({ args, partialArgs, isArgumentsStreaming, isLoading, pluginState }) => {
+  const { t } = useTranslation('plugin');
+
+  const query = args?.query || partialArgs?.query || '';
+  // Use fileResults length for display (aggregated by file)
+  const resultCount = pluginState?.fileResults?.length ?? 0;
+  const hasResults = resultCount > 0;
+
+  // During argument streaming
+  if (isArgumentsStreaming) {
+    if (!query)
+      return (
+        <div className={cx(styles.root, shinyTextStyles.shinyText)}>
+          <span>{t('builtins.lobe-knowledge-base.apiName.searchKnowledgeBase')}</span>
+        </div>
+      );
+
+    return (
+      <div className={cx(styles.root, shinyTextStyles.shinyText)}>
+        <span>{t('builtins.lobe-knowledge-base.apiName.searchKnowledgeBase')}: </span>
+        <span className={highlightTextStyles.gold}>{query}</span>
+      </div>
+    );
+  }
+
+  return (
+    <div className={cx(styles.root, isLoading && shinyTextStyles.shinyText)}>
+      <span style={{ marginInlineStart: 2 }}>
+        <span>{t('builtins.lobe-knowledge-base.apiName.searchKnowledgeBase')}: </span>
+        {query && <span className={highlightTextStyles.gold}>{query}</span>}
+        {!isLoading &&
+          pluginState?.fileResults &&
+          (hasResults ? (
+            <span style={{ marginInlineStart: 4 }}>({resultCount})</span>
+          ) : (
+            <Text
+              as={'span'}
+              color={cssVar.colorTextDescription}
+              fontSize={12}
+              style={{ marginInlineStart: 4 }}
+            >
+              ({t('builtins.lobe-knowledge-base.inspector.noResults')})
+            </Text>
+          ))}
+      </span>
+    </div>
+  );
+});
+
+SearchKnowledgeBaseInspector.displayName = 'SearchKnowledgeBaseInspector';

package/packages/builtin-tool-knowledge-base/src/client/Inspector/index.ts

@@ -0,0 +1,11 @@
+import { KnowledgeBaseApiName } from '../../types';
+import { ReadKnowledgeInspector } from './ReadKnowledge';
+import { SearchKnowledgeBaseInspector } from './SearchKnowledgeBase';
+
+/**
+ * Knowledge Base Inspector Components Registry
+ */
+export const KnowledgeBaseInspectors = {
+  [KnowledgeBaseApiName.readKnowledge]: ReadKnowledgeInspector,
+  [KnowledgeBaseApiName.searchKnowledgeBase]: SearchKnowledgeBaseInspector,
+};

package/packages/builtin-tool-knowledge-base/src/client/Render/ReadKnowledge/FileCard.tsx

@@ -1,9 +1,9 @@
 'use client';
 
-import { Alert, Flexbox, Text } from '@lobehub/ui';
+import { Alert, Flexbox, MaterialFileTypeIcon, Text } from '@lobehub/ui';
 import { Descriptions } from 'antd';
 import { createStaticStyles } from 'antd-style';
-import { ComponentType, memo } from 'react';
+import { memo } from 'react';
 
 import { FileContentDetail } from '../../../types';
 
@@ -63,21 +63,21 @@ const styles = createStaticStyles(({ css, cssVar }) => ({
 }));
 
 interface FileCardProps {
-  FileIcon: ComponentType<{ fileName: string; size: number }>;
   file: FileContentDetail;
-  labels: {
-    chars: string;
-    lines: string;
-  };
 }
 
-const FileCard = memo<FileCardProps>(({ file, FileIcon, labels }) => {
+const FileCard = memo<FileCardProps>(({ file }) => {
   if (file.error) {
     return (
       <Flexbox className={styles.container} gap={8}>
         <Flexbox className={styles.cardBody} gap={8}>
           <Flexbox align={'center'} className={styles.titleRow} gap={8} horizontal>
-            <FileIcon fileName={file.filename} size={16} />
+            <MaterialFileTypeIcon
+              filename={file.filename}
+              size={16}
+              type={'file'}
+              variant={'raw'}
+            />
             <div className={styles.title}>{file.filename}</div>
           </Flexbox>
         </Flexbox>
@@ -92,7 +92,7 @@ const FileCard = memo<FileCardProps>(({ file, FileIcon, labels }) => {
     <Flexbox className={styles.container} justify={'space-between'}>
       <Flexbox className={styles.cardBody} gap={8}>
         <Flexbox align={'center'} className={styles.titleRow} gap={8} horizontal>
-          <FileIcon fileName={file.filename} size={16} />
+          <MaterialFileTypeIcon filename={file.filename} size={16} type={'file'} variant={'raw'} />
           <div className={styles.title}>{file.filename}</div>
         </Flexbox>
         {file.preview && (
@@ -118,11 +118,11 @@ const FileCard = memo<FileCardProps>(({ file, FileIcon, labels }) => {
           items={[
             {
               children: file.totalCharCount?.toLocaleString(),
-              label: labels.chars,
+              label: 'Chars',
             },
             {
               children: file.totalLineCount?.toLocaleString(),
-              label: labels.lines,
+              label: 'Lines',
             },
           ]}
           size="small"

package/packages/builtin-tool-knowledge-base/src/client/Render/ReadKnowledge/index.tsx

@@ -2,36 +2,27 @@
 
 import { BuiltinRenderProps } from '@lobechat/types';
 import { Flexbox } from '@lobehub/ui';
-import { ComponentType, memo } from 'react';
+import { memo } from 'react';
 
 import { ReadKnowledgeArgs, ReadKnowledgeState } from '../../../types';
 import FileCard from './FileCard';
 
-export interface ReadKnowledgeRenderProps extends Pick<
-  BuiltinRenderProps<ReadKnowledgeArgs, ReadKnowledgeState>,
-  'pluginState'
-> {
-  FileIcon: ComponentType<{ fileName: string; size: number }>;
-  labels: {
-    chars: string;
-    lines: string;
-  };
-}
+const ReadKnowledge = memo<BuiltinRenderProps<ReadKnowledgeArgs, ReadKnowledgeState>>(
+  ({ pluginState }) => {
+    const { files } = pluginState || {};
 
-const ReadKnowledge = memo<ReadKnowledgeRenderProps>(({ pluginState, FileIcon, labels }) => {
-  const { files } = pluginState || {};
+    if (!files || files.length === 0) {
+      return null;
+    }
 
-  if (!files || files.length === 0) {
-    return null;
-  }
-
-  return (
-    <Flexbox gap={12} horizontal style={{ flexWrap: 'wrap' }}>
-      {files.map((file) => (
-        <FileCard FileIcon={FileIcon} file={file} key={file.fileId} labels={labels} />
-      ))}
-    </Flexbox>
-  );
-});
+    return (
+      <Flexbox gap={12} horizontal style={{ flexWrap: 'wrap' }}>
+        {files.map((file) => (
+          <FileCard file={file} key={file.fileId} />
+        ))}
+      </Flexbox>
+    );
+  },
+);
 
 export default ReadKnowledge;

package/packages/builtin-tool-knowledge-base/src/client/Render/SearchKnowledgeBase/Item/index.tsx

@@ -1,62 +1,36 @@
 'use client';
 
 import { FileSearchResult } from '@lobechat/types';
-import { Center, Flexbox, Text, Tooltip } from '@lobehub/ui';
+import { Center, Flexbox, MaterialFileTypeIcon, Text, Tooltip } from '@lobehub/ui';
 import { cx, useThemeMode } from 'antd-style';
-import { ComponentType, memo } from 'react';
+import { memo } from 'react';
 
 import { styles } from './style';
 
 export interface FileItemProps extends FileSearchResult {
-  FileIcon: ComponentType<{
-    fileName: string;
-    size: number;
-    variant?: 'raw' | 'file' | 'folder';
-  }>;
   index: number;
-  isMobile?: boolean;
-  onFileClick?: (params: { chunkId: string; chunkText: string; fileId: string }) => void;
 }
 
-const FileItem = memo<FileItemProps>(
-  ({ fileId, fileName, relevanceScore, topChunks, FileIcon, isMobile, onFileClick }) => {
-    const { isDarkMode } = useThemeMode();
+const FileItem = memo<FileItemProps>(({ fileId, fileName, relevanceScore }) => {
+  const { isDarkMode } = useThemeMode();
 
-    // Use the first chunk for preview
-    const firstChunk = topChunks[0];
-
-    return (
-      <Flexbox
-        align={'center'}
-        className={cx(
-          styles.container,
-          isDarkMode ? styles.containerDark : styles.containerLight,
-          isMobile && styles.mobile,
-        )}
-        gap={4}
-        horizontal
-        key={fileId}
-        onClick={(e) => {
-          e.stopPropagation();
-          if (firstChunk && onFileClick) {
-            onFileClick({
-              chunkId: firstChunk.id,
-              chunkText: firstChunk.text,
-              fileId,
-            });
-          }
-        }}
-      >
-        <FileIcon fileName={fileName} size={20} variant={'raw'} />
-        <Flexbox gap={12} horizontal justify={'space-between'} style={{ maxWidth: 200 }}>
-          <Text ellipsis>{fileName}</Text>
-          <Tooltip title={`Relevance: ${(relevanceScore * 100).toFixed(1)}%`}>
-            <Center className={styles.badge}>{relevanceScore.toFixed(2)}</Center>
-          </Tooltip>
-        </Flexbox>
+  return (
+    <Flexbox
+      align={'center'}
+      className={cx(styles.container, isDarkMode ? styles.containerDark : styles.containerLight)}
+      gap={4}
+      horizontal
+      key={fileId}
+    >
+      <MaterialFileTypeIcon filename={fileName} size={20} type={'file'} variant={'raw'} />
+      <Flexbox gap={12} horizontal justify={'space-between'} style={{ maxWidth: 200 }}>
+        <Text ellipsis>{fileName}</Text>
+        <Tooltip title={`Relevance: ${(relevanceScore * 100).toFixed(1)}%`}>
+          <Center className={styles.badge}>{relevanceScore.toFixed(2)}</Center>
+        </Tooltip>
       </Flexbox>
-    );
-  },
-);
+    </Flexbox>
+  );
+});
 
 export default FileItem;

package/packages/builtin-tool-knowledge-base/src/client/Render/SearchKnowledgeBase/index.tsx

@@ -1,42 +1,30 @@
 'use client';
 
 import { BuiltinRenderProps } from '@lobechat/types';
-import { Flexbox } from '@lobehub/ui';
-import { ComponentType, memo } from 'react';
+import { Empty, Flexbox } from '@lobehub/ui';
+import { memo } from 'react';
+import { useTranslation } from 'react-i18next';
 
 import { SearchKnowledgeBaseArgs, SearchKnowledgeBaseState } from '../../../types';
 import FileItem from './Item';
 
-export interface SearchKnowledgeBaseRenderProps extends Pick<
-  BuiltinRenderProps<SearchKnowledgeBaseArgs, SearchKnowledgeBaseState>,
-  'pluginState'
-> {
-  FileIcon: ComponentType<{ fileName: string; size: number; variant?: 'raw' | 'file' | 'folder' }>;
-  isMobile?: boolean;
-  onFileClick?: (params: { chunkId: string; chunkText: string; fileId: string }) => void;
-}
+const SearchKnowledgeBase = memo<
+  BuiltinRenderProps<SearchKnowledgeBaseArgs, SearchKnowledgeBaseState>
+>(({ pluginState }) => {
+  const { t } = useTranslation('plugin');
+  const { fileResults } = pluginState || {};
 
-const SearchKnowledgeBase = memo<SearchKnowledgeBaseRenderProps>(
-  ({ pluginState, FileIcon, isMobile, onFileClick }) => {
-    const { fileResults } = pluginState || {};
+  if (!fileResults || fileResults.length === 0) {
+    return <Empty description={t('builtins.lobe-knowledge-base.inspector.noResults')} />;
+  }
 
-    return (
-      <Flexbox gap={8} horizontal wrap={'wrap'}>
-        {fileResults?.map((file, index) => {
-          return (
-            <FileItem
-              FileIcon={FileIcon}
-              index={index}
-              isMobile={isMobile}
-              key={file.fileId}
-              onFileClick={onFileClick}
-              {...file}
-            />
-          );
-        })}
-      </Flexbox>
-    );
-  },
-);
+  return (
+    <Flexbox gap={8} horizontal wrap={'wrap'}>
+      {fileResults.map((file, index) => {
+        return <FileItem index={index} key={file.fileId} {...file} />;
+      })}
+    </Flexbox>
+  );
+});
 
 export default SearchKnowledgeBase;

package/packages/builtin-tool-knowledge-base/src/client/Render/index.ts

@@ -5,8 +5,3 @@ export const KnowledgeBaseRenders = {
   readKnowledge: ReadKnowledge,
   searchKnowledgeBase: SearchKnowledgeBase,
 };
-
-
-
-export {default as ReadKnowledge} from './ReadKnowledge';
-export {default as SearchKnowledgeBase} from './SearchKnowledgeBase';

package/packages/builtin-tool-knowledge-base/src/client/index.ts

@@ -1,4 +1,8 @@
-export { KnowledgeBaseRenders, ReadKnowledge, SearchKnowledgeBase } from './Render';
+// Inspector components (customized tool call headers)
+export { KnowledgeBaseInspectors } from './Inspector';
+
+// Render components (read-only snapshots)
+export { KnowledgeBaseRenders } from './Render';
 
 // Re-export types and manifest for convenience
 export { KnowledgeBaseManifest } from '../manifest';

package/packages/builtin-tool-knowledge-base/src/executor/index.ts

@@ -0,0 +1,119 @@
+import { formatSearchResults, promptFileContents, promptNoSearchResults } from '@lobechat/prompts';
+import { BaseExecutor, type BuiltinToolContext, type BuiltinToolResult } from '@lobechat/types';
+
+import { ragService } from '@/services/rag';
+import { agentSelectors } from '@/store/agent/selectors';
+import { getAgentStoreState } from '@/store/agent/store';
+
+import {
+  type FileContentDetail,
+  KnowledgeBaseIdentifier,
+  type ReadKnowledgeArgs,
+  type ReadKnowledgeState,
+  type SearchKnowledgeBaseArgs,
+  type SearchKnowledgeBaseState,
+} from '../types';
+
+/**
+ * Knowledge Base Tool Executor
+ *
+ * Handles knowledge base search and retrieval operations.
+ */
+class KnowledgeBaseExecutor extends BaseExecutor<{
+  readKnowledge: 'readKnowledge';
+  searchKnowledgeBase: 'searchKnowledgeBase';
+}> {
+  readonly identifier = KnowledgeBaseIdentifier;
+  protected readonly apiEnum = {
+    readKnowledge: 'readKnowledge' as const,
+    searchKnowledgeBase: 'searchKnowledgeBase' as const,
+  };
+
+  /**
+   * Search knowledge base and return file summaries with relevant chunks
+   */
+  searchKnowledgeBase = async (
+    params: SearchKnowledgeBaseArgs,
+    ctx: BuiltinToolContext,
+  ): Promise<BuiltinToolResult> => {
+    try {
+      const { query, topK = 20 } = 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;
+
+      const { chunks, fileResults } = await ragService.semanticSearchForChat(
+        { knowledgeIds: knowledgeBaseIds, query, topK },
+        ctx.signal,
+      );
+
+      if (chunks.length === 0) {
+        const state: SearchKnowledgeBaseState = { chunks: [], fileResults: [], totalResults: 0 };
+
+        return { content: promptNoSearchResults(query), state, success: true };
+      }
+
+      // Format search results for AI
+      const formattedContent = formatSearchResults(fileResults, query);
+
+      const state: SearchKnowledgeBaseState = { chunks, fileResults, totalResults: chunks.length };
+
+      return { content: formattedContent, state, success: true };
+    } catch (e) {
+      return {
+        content: `Error searching knowledge base: ${(e as Error).message}`,
+        error: { body: e, message: (e as Error).message, type: 'PluginServerError' },
+        success: false,
+      };
+    }
+  };
+
+  /**
+   * Read full content of specific files from knowledge base
+   */
+  readKnowledge = async (params: ReadKnowledgeArgs): Promise<BuiltinToolResult> => {
+    try {
+      const { fileIds } = params;
+
+      if (!fileIds || fileIds.length === 0) {
+        return {
+          content: 'Error: No file IDs provided',
+          success: false,
+        };
+      }
+
+      const fileContents = await ragService.getFileContents(fileIds);
+
+      const formattedContent = promptFileContents(fileContents);
+
+      const state: ReadKnowledgeState = {
+        files: fileContents.map(
+          (file): FileContentDetail => ({
+            error: file.error,
+            fileId: file.fileId,
+            filename: file.filename,
+            preview: file.preview,
+            totalCharCount: file.totalCharCount,
+            totalLineCount: file.totalLineCount,
+          }),
+        ),
+      };
+
+      return { content: formattedContent, state, success: true };
+    } catch (e) {
+      return {
+        content: `Error reading knowledge: ${(e as Error).message}`,
+        error: { body: e, message: (e as Error).message, type: 'PluginServerError' },
+        success: false,
+      };
+    }
+  };
+}
+
+// Export the executor instance for registration
+export const knowledgeBaseExecutor = new KnowledgeBaseExecutor();

package/packages/builtin-tool-local-system/package.json

@@ -5,6 +5,7 @@
   "exports": {
     ".": "./src/index.ts",
     "./client": "./src/client/index.ts",
+    "./executor": "./src/executor/index.ts",
     "./executionRuntime": "./src/ExecutionRuntime/index.ts"
   },
   "main": "./src/index.ts",

package/packages/builtin-tool-local-system/src/client/Inspector/EditLocalFile/index.tsx

@@ -2,15 +2,16 @@
 
 import { type EditLocalFileParams } from '@lobechat/electron-client-ipc';
 import { type BuiltinInspectorProps } from '@lobechat/types';
+import { Icon, Text } from '@lobehub/ui';
 import { createStaticStyles, cssVar, cx } from 'antd-style';
-import { Check, X } from 'lucide-react';
-import path from 'path-browserify-esm';
-import { memo } from 'react';
+import { Minus, Plus } from 'lucide-react';
+import { type ReactNode, memo } from 'react';
 import { useTranslation } from 'react-i18next';
 
-import { highlightTextStyles, shinyTextStyles } from '@/styles';
+import { shinyTextStyles } from '@/styles';
 
 import { type EditLocalFileState } from '../../../types';
+import { FilePathDisplay } from '../../components/FilePathDisplay';
 
 const styles = createStaticStyles(({ css, cssVar }) => ({
   root: css`
@@ -21,9 +22,9 @@ const styles = createStaticStyles(({ css, cssVar }) => ({
 
     color: ${cssVar.colorTextSecondary};
   `,
-  statusIcon: css`
-    margin-block-end: -2px;
-    margin-inline-start: 4px;
+  separator: css`
+    margin-inline: 2px;
+    color: ${cssVar.colorTextQuaternary};
   `,
 }));
 
@@ -32,18 +33,11 @@ export const EditLocalFileInspector = memo<
 >(({ args, partialArgs, isArgumentsStreaming, pluginState, isLoading }) => {
   const { t } = useTranslation('plugin');
 
-  // Show filename with parent directory for context
   const filePath = args?.file_path || partialArgs?.file_path || '';
-  let displayPath = '';
-  if (filePath) {
-    const { base, dir } = path.parse(filePath);
-    const parentDir = path.basename(dir);
-    displayPath = parentDir ? `${parentDir}/${base}` : base;
-  }
 
   // During argument streaming
   if (isArgumentsStreaming) {
-    if (!displayPath)
+    if (!filePath)
       return (
         <div className={cx(styles.root, shinyTextStyles.shinyText)}>
           <span>{t('builtins.lobe-local-system.apiName.editLocalFile')}</span>
@@ -53,27 +47,48 @@ export const EditLocalFileInspector = memo<
     return (
       <div className={cx(styles.root, shinyTextStyles.shinyText)}>
         <span>{t('builtins.lobe-local-system.apiName.editLocalFile')}: </span>
-        <span className={highlightTextStyles.primary}>{displayPath}</span>
+        <FilePathDisplay filePath={filePath} />
       </div>
     );
   }
 
-  // Check if edit was successful (has replacements count)
-  const isSuccess = pluginState?.replacements !== undefined && pluginState.replacements >= 0;
+  // Build stats parts with colors and icons
+  const linesAdded = pluginState?.linesAdded ?? 0;
+  const linesDeleted = pluginState?.linesDeleted ?? 0;
+
+  const statsParts: ReactNode[] = [];
+  if (linesAdded > 0) {
+    statsParts.push(
+      <Text as={'span'} code color={cssVar.colorSuccess} fontSize={12} key="added">
+        <Icon icon={Plus} size={12} />
+        {linesAdded}
+      </Text>,
+    );
+  }
+  if (linesDeleted > 0) {
+    statsParts.push(
+      <Text as={'span'} code color={cssVar.colorError} fontSize={12} key="deleted">
+        <Icon icon={Minus} size={12} />
+        {linesDeleted}
+      </Text>,
+    );
+  }
 
   return (
     <div className={cx(styles.root, isLoading && shinyTextStyles.shinyText)}>
-      <span style={{ marginInlineStart: 2 }}>
-        <span>{t('builtins.lobe-local-system.apiName.editLocalFile')}: </span>
-        {displayPath && <span className={highlightTextStyles.primary}>{displayPath}</span>}
-        {isLoading ? null : pluginState ? (
-          isSuccess ? (
-            <Check className={styles.statusIcon} color={cssVar.colorSuccess} size={14} />
-          ) : (
-            <X className={styles.statusIcon} color={cssVar.colorError} size={14} />
-          )
-        ) : null}
-      </span>
+      <span>{t('builtins.lobe-local-system.apiName.editLocalFile')}: </span>
+      <FilePathDisplay filePath={filePath} />
+      {!isLoading && statsParts.length > 0 && (
+        <>
+          {' '}
+          {statsParts.map((part, index) => (
+            <span key={index}>
+              {index > 0 && <span className={styles.separator}> / </span>}
+              {part}
+            </span>
+          ))}
+        </>
+      )}
     </div>
   );
 });