@lobehub/lobehub 2.0.0-next.306 → 2.0.0-next.308

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

@@ -1,22 +1,65 @@
 import { type WriteLocalFileParams } from '@lobechat/electron-client-ipc';
 import { type BuiltinRenderProps } from '@lobechat/types';
-import { Flexbox, Icon, Skeleton } from '@lobehub/ui';
+import { Flexbox, Highlighter, Icon, Markdown, Skeleton } from '@lobehub/ui';
+import { createStaticStyles } from 'antd-style';
 import { ChevronRight } from 'lucide-react';
 import path from 'path-browserify-esm';
 import { memo } from 'react';
 
 import { LocalFile, LocalFolder } from '@/features/LocalFile';
 
+const styles = createStaticStyles(({ css, cssVar }) => ({
+  container: css`
+    padding: 8px;
+    border-radius: ${cssVar.borderRadiusLG};
+    background: ${cssVar.colorFillQuaternary};
+  `,
+  previewBox: css`
+    overflow: hidden;
+    border-radius: 8px;
+    background: ${cssVar.colorBgContainer};
+  `,
+}));
+
 const WriteFile = memo<BuiltinRenderProps<WriteLocalFileParams>>(({ args }) => {
   if (!args) return <Skeleton active />;
 
   const { base, dir } = path.parse(args.path);
+  const ext = path.extname(args.path).slice(1).toLowerCase();
+
+  const renderContent = () => {
+    if (!args.content) return null;
+
+    if (ext === 'md' || ext === 'mdx') {
+      return (
+        <Markdown style={{ maxHeight: 240, overflow: 'auto', padding: '0 8px' }}>
+          {args.content}
+        </Markdown>
+      );
+    }
+
+    return (
+      <Highlighter
+        language={ext || 'text'}
+        showLanguage={false}
+        style={{ maxHeight: 240, overflow: 'auto' }}
+        variant={'borderless'}
+        wrap
+      >
+        {args.content}
+      </Highlighter>
+    );
+  };
 
   return (
-    <Flexbox horizontal>
-      <LocalFolder path={dir} />
-      <Icon icon={ChevronRight} />
-      <LocalFile name={base} path={args.path} />
+    <Flexbox className={styles.container} gap={12}>
+      <Flexbox align={'center'} horizontal>
+        <LocalFolder path={dir} />
+        <Icon icon={ChevronRight} />
+        <LocalFile name={base} path={args.path} />
+      </Flexbox>
+
+      {args.content && <Flexbox className={styles.previewBox}>{renderContent()}</Flexbox>}
     </Flexbox>
   );
 });

package/packages/builtin-tool-local-system/src/client/Streaming/WriteFile/index.tsx

@@ -0,0 +1,39 @@
+'use client';
+
+import { type WriteLocalFileParams } from '@lobechat/electron-client-ipc';
+import { type BuiltinStreamingProps } from '@lobechat/types';
+import { Highlighter, Markdown } from '@lobehub/ui';
+import path from 'path-browserify-esm';
+import { memo } from 'react';
+
+export const WriteFileStreaming = memo<BuiltinStreamingProps<WriteLocalFileParams>>(({ args }) => {
+  const { content, path: filePath } = args || {};
+
+  // Don't render if no content yet
+  if (!content) return null;
+
+  const ext = path
+    .extname(filePath || '')
+    .slice(1)
+    .toLowerCase();
+
+  // Use Markdown for .md files, Highlighter for others
+  if (ext === 'md' || ext === 'mdx') {
+    return <Markdown style={{ overflow: 'auto' }}>{content}</Markdown>;
+  }
+
+  return (
+    <Highlighter
+      animated
+      language={ext || 'text'}
+      showLanguage={false}
+      style={{ padding: '4px 8px' }}
+      variant={'outlined'}
+      wrap
+    >
+      {content}
+    </Highlighter>
+  );
+});
+
+WriteFileStreaming.displayName = 'WriteFileStreaming';

package/packages/builtin-tool-local-system/src/client/Streaming/index.ts

@@ -1,9 +1,11 @@
 import { LocalSystemApiName } from '../..';
 import { RunCommandStreaming } from './RunCommand';
+import { WriteFileStreaming } from './WriteFile';
 
 /**
  * Local System Streaming Components Registry
  */
 export const LocalSystemStreamings = {
   [LocalSystemApiName.runCommand]: RunCommandStreaming,
+  [LocalSystemApiName.writeLocalFile]: WriteFileStreaming,
 };

package/packages/builtin-tool-local-system/src/executor/index.ts

@@ -23,6 +23,21 @@ import type {
   RunCommandResult,
   WriteLocalFileParams,
 } from '@lobechat/electron-client-ipc';
+import {
+  formatCommandOutput,
+  formatCommandResult,
+  formatEditResult,
+  formatFileContent,
+  formatFileList,
+  formatFileSearchResults,
+  formatGlobResults,
+  formatGrepResults,
+  formatKillResult,
+  formatMoveResults,
+  formatMultipleFiles,
+  formatRenameResult,
+  formatWriteResult,
+} from '@lobechat/prompts';
 import { BaseExecutor, type BuiltinToolResult } from '@lobechat/types';
 
 import { localFileService } from '@/services/electron/localFileService';
@@ -76,8 +91,10 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
 
       const state: LocalFileListState = { listResults: result };
 
+      const content = formatFileList(result, params.path);
+
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: true,
       };
@@ -96,8 +113,14 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
 
       const state: LocalReadFileState = { fileContent: result };
 
+      const content = formatFileContent({
+        content: result.content,
+        lineRange: params.loc,
+        path: params.path,
+      });
+
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: true,
       };
@@ -116,8 +139,10 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
 
       const state: LocalReadFilesState = { filesContent: results };
 
+      const content = formatMultipleFiles(results);
+
       return {
-        content: JSON.stringify(results),
+        content,
         state,
         success: true,
       };
@@ -136,8 +161,10 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
 
       const state: LocalFileSearchState = { searchResults: result };
 
+      const content = formatFileSearchResults(result);
+
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: true,
       };
@@ -154,20 +181,9 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
     try {
       const results: LocalMoveFilesResultItem[] = await localFileService.moveLocalFiles(params);
 
-      const allSucceeded = results.every((r) => r.success);
-      const someFailed = results.some((r) => !r.success);
       const successCount = results.filter((r) => r.success).length;
-      const failedCount = results.length - successCount;
 
-      let message = '';
-
-      if (allSucceeded) {
-        message = `Successfully moved ${results.length} item(s).`;
-      } else if (someFailed) {
-        message = `Moved ${successCount} item(s) successfully. Failed to move ${failedCount} item(s).`;
-      } else {
-        message = `Failed to move all ${results.length} item(s).`;
-      }
+      const content = formatMoveResults(results);
 
       const state: LocalMoveFilesState = {
         results,
@@ -176,7 +192,7 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
       };
 
       return {
-        content: JSON.stringify({ message, results }),
+        content,
         state,
         success: true,
       };
@@ -202,7 +218,12 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
         };
 
         return {
-          content: JSON.stringify({ message: result.error, success: false }),
+          content: formatRenameResult({
+            error: result.error,
+            newName: params.newName,
+            oldPath: params.path,
+            success: false,
+          }),
           state,
           success: false,
         };
@@ -215,8 +236,9 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
       };
 
       return {
-        content: JSON.stringify({
-          message: `Successfully renamed file ${params.path} to ${params.newName}.`,
+        content: formatRenameResult({
+          newName: params.newName,
+          oldPath: params.path,
           success: true,
         }),
         state,
@@ -237,8 +259,9 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
 
       if (!result.success) {
         return {
-          content: JSON.stringify({
-            message: result.error || 'Failed to write file',
+          content: formatWriteResult({
+            error: result.error,
+            path: params.path,
             success: false,
           }),
           error: { message: result.error || 'Failed to write file', type: 'PluginServerError' },
@@ -247,8 +270,8 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
       }
 
       return {
-        content: JSON.stringify({
-          message: `Successfully wrote file ${params.path}`,
+        content: formatWriteResult({
+          path: params.path,
           success: true,
         }),
         success: true,
@@ -273,11 +296,12 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
         };
       }
 
-      const statsText =
-        result.linesAdded || result.linesDeleted
-          ? ` (+${result.linesAdded || 0} -${result.linesDeleted || 0})`
-          : '';
-      const message = `Successfully replaced ${result.replacements} occurrence(s) in ${params.file_path}${statsText}`;
+      const content = formatEditResult({
+        filePath: params.file_path,
+        linesAdded: result.linesAdded,
+        linesDeleted: result.linesDeleted,
+        replacements: result.replacements,
+      });
 
       const state: EditLocalFileState = {
         diffText: result.diffText,
@@ -287,7 +311,7 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
       };
 
       return {
-        content: message,
+        content,
         state,
         success: true,
       };
@@ -306,22 +330,19 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
     try {
       const result: RunCommandResult = await localFileService.runCommand(params);
 
-      let message: string;
-
-      if (result.success) {
-        if (result.shell_id) {
-          message = `Command started in background with shell_id: ${result.shell_id}`;
-        } else {
-          message = `Command completed successfully.`;
-        }
-      } else {
-        message = `Command failed: ${result.error}`;
-      }
+      const content = formatCommandResult({
+        error: result.error,
+        exitCode: result.exit_code,
+        shellId: result.shell_id,
+        stderr: result.stderr,
+        stdout: result.stdout,
+        success: result.success,
+      });
 
-      const state: RunCommandState = { message, result };
+      const state: RunCommandState = { message: content.split('\n\n')[0], result };
 
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: result.success,
       };
@@ -338,14 +359,17 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
     try {
       const result: GetCommandOutputResult = await localFileService.getCommandOutput(params);
 
-      const message = result.success
-        ? `Output retrieved. Running: ${result.running}`
-        : `Failed: ${result.error}`;
+      const content = formatCommandOutput({
+        error: result.error,
+        output: result.output,
+        running: result.running,
+        success: result.success,
+      });
 
-      const state: GetCommandOutputState = { message, result };
+      const state: GetCommandOutputState = { message: content.split('\n\n')[0], result };
 
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: result.success,
       };
@@ -362,14 +386,16 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
     try {
       const result: KillCommandResult = await localFileService.killCommand(params);
 
-      const message = result.success
-        ? `Successfully killed shell: ${params.shell_id}`
-        : `Failed to kill shell: ${result.error}`;
+      const content = formatKillResult({
+        error: result.error,
+        shellId: params.shell_id,
+        success: result.success,
+      });
 
-      const state: KillCommandState = { message, result };
+      const state: KillCommandState = { message: content, result };
 
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: result.success,
       };
@@ -388,14 +414,17 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
     try {
       const result: GrepContentResult = await localFileService.grepContent(params);
 
-      const message = result.success
-        ? `Found ${result.total_matches} matches in ${result.matches.length} locations`
+      const content = result.success
+        ? formatGrepResults({
+            matches: result.matches,
+            totalMatches: result.total_matches,
+          })
         : 'Search failed';
 
-      const state: GrepContentState = { message, result };
+      const state: GrepContentState = { message: content.split('\n')[0], result };
 
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: result.success,
       };
@@ -412,12 +441,17 @@ class LocalSystemExecutor extends BaseExecutor<typeof LocalSystemApiEnum> {
     try {
       const result: GlobFilesResult = await localFileService.globFiles(params);
 
-      const message = result.success ? `Found ${result.total_files} files` : 'Glob search failed';
+      const content = result.success
+        ? formatGlobResults({
+            files: result.files,
+            totalFiles: result.total_files,
+          })
+        : 'Glob search failed';
 
-      const state: GlobFilesState = { message, result };
+      const state: GlobFilesState = { message: content.split('\n')[0], result };
 
       return {
-        content: JSON.stringify(result),
+        content,
         state,
         success: result.success,
       };

package/packages/database/src/repositories/agentGroup/index.ts

@@ -15,8 +15,14 @@ import {
 import { LobeChatDatabase } from '../../type';
 
 export interface SupervisorAgentConfig {
+  avatar?: string;
+  backgroundColor?: string;
+  description?: string;
   model?: string;
+  params?: any;
   provider?: string;
+  systemRole?: string;
+  tags?: string[];
   title?: string;
 }
 
@@ -164,8 +170,14 @@ export class AgentGroupRepository {
     const [supervisorAgent] = await this.db
       .insert(agents)
       .values({
+        avatar: supervisorConfig?.avatar,
+        backgroundColor: supervisorConfig?.backgroundColor,
+        description: supervisorConfig?.description,
         model: supervisorConfig?.model,
+        params: supervisorConfig?.params,
         provider: supervisorConfig?.provider,
+        systemRole: supervisorConfig?.systemRole,
+        tags: supervisorConfig?.tags,
         title: supervisorConfig?.title ?? 'Supervisor',
         userId: this.userId,
         virtual: true,
@@ -356,7 +368,12 @@ export class AgentGroupRepository {
       const [newGroup] = await trx
         .insert(chatGroups)
         .values({
+          avatar: sourceGroup.avatar,
+          backgroundColor: sourceGroup.backgroundColor,
           config: sourceGroup.config,
+          content: sourceGroup.content,
+          description: sourceGroup.description,
+          editorData: sourceGroup.editorData,
           pinned: sourceGroup.pinned,
           title: newTitle || (sourceGroup.title ? `${sourceGroup.title} (Copy)` : 'Copy'),
           userId: this.userId,
@@ -368,8 +385,14 @@ export class AgentGroupRepository {
       const [newSupervisor] = await trx
         .insert(agents)
         .values({
+          avatar: supervisorAgent?.avatar,
+          backgroundColor: supervisorAgent?.backgroundColor,
+          description: supervisorAgent?.description,
           model: supervisorAgent?.model,
+          params: supervisorAgent?.params,
           provider: supervisorAgent?.provider,
+          systemRole: supervisorAgent?.systemRole,
+          tags: supervisorAgent?.tags,
           title: supervisorAgent?.title || 'Supervisor',
           userId: this.userId,
           virtual: true,

package/packages/model-bank/src/aiModels/qiniu.ts

@@ -78,6 +78,30 @@ const qiniuChatModels: AIChatModelCard[] = [
     },
     type: 'chat',
   },
+  {
+    abilities: {
+      reasoning: true,
+    },
+    contextWindowTokens: 163_840,
+    description: 
+      'DeepSeek Math V2 is a model that has made significant breakthroughs in mathematical reasoning capabilities. Its core innovation lies in the "self-verification" training mechanism, and it has achieved gold medal levels in several top mathematics competitions.',
+    displayName: 'DeepSeek Math V2',
+    enabled: true,
+    id: 'deepseek/deepseek-math-v2',
+    maxOutput: 131_072,
+    pricing: {
+      currency: 'CNY',
+      units: [
+        { name: 'textInput', rate: 4.0, strategy: 'fixed', unit: 'millionTokens' },
+        { name: 'textOutput', rate: 16.0, strategy: 'fixed', unit: 'millionTokens' },
+      ],
+    },
+    releasedAt: '2025-11-27',
+    settings: {
+      searchImpl: 'params',
+    },
+    type: 'chat',
+  },
   {
     abilities: {
       functionCall: true,

package/packages/prompts/src/prompts/fileSystem/formatCommandOutput.test.ts

@@ -0,0 +1,61 @@
+import { describe, expect, it } from 'vitest';
+
+import { formatCommandOutput } from './formatCommandOutput';
+
+describe('formatCommandOutput', () => {
+  it('should format successful output while running', () => {
+    const result = formatCommandOutput({
+      running: true,
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`"Output retrieved. Running: true"`);
+  });
+
+  it('should format successful output when not running', () => {
+    const result = formatCommandOutput({
+      running: false,
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`"Output retrieved. Running: false"`);
+  });
+
+  it('should format successful output with content', () => {
+    const result = formatCommandOutput({
+      output: 'Process output here',
+      running: true,
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`
+      "Output retrieved. Running: true
+
+      Output:
+      Process output here"
+    `);
+  });
+
+  it('should format failed output', () => {
+    const result = formatCommandOutput({
+      error: 'Process not found',
+      running: false,
+      success: false,
+    });
+    expect(result).toMatchInlineSnapshot(`"Failed: Process not found"`);
+  });
+
+  it('should format successful output with error info', () => {
+    const result = formatCommandOutput({
+      error: 'Warning message',
+      output: 'Some output',
+      running: false,
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`
+      "Output retrieved. Running: false
+
+      Output:
+      Some output
+
+      Error: Warning message"
+    `);
+  });
+});

package/packages/prompts/src/prompts/fileSystem/formatCommandOutput.ts

@@ -0,0 +1,21 @@
+export interface FormatCommandOutputParams {
+  error?: string;
+  output?: string;
+  running: boolean;
+  success: boolean;
+}
+
+export const formatCommandOutput = ({
+  success,
+  running,
+  output,
+  error,
+}: FormatCommandOutputParams): string => {
+  const message = success ? `Output retrieved. Running: ${running}` : `Failed: ${error}`;
+
+  const parts: string[] = [message];
+  if (output) parts.push(`Output:\n${output}`);
+  if (error && success) parts.push(`Error: ${error}`);
+
+  return parts.join('\n\n');
+};

package/packages/prompts/src/prompts/fileSystem/formatCommandResult.test.ts

@@ -0,0 +1,87 @@
+import { describe, expect, it } from 'vitest';
+
+import { formatCommandResult } from './formatCommandResult';
+
+describe('formatCommandResult', () => {
+  it('should format successful command without output', () => {
+    const result = formatCommandResult({ success: true });
+    expect(result).toMatchInlineSnapshot(`"Command completed successfully."`);
+  });
+
+  it('should format successful background command', () => {
+    const result = formatCommandResult({
+      shellId: 'shell-123',
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(
+      `"Command started in background with shell_id: shell-123"`,
+    );
+  });
+
+  it('should format successful command with stdout', () => {
+    const result = formatCommandResult({
+      stdout: 'Hello World',
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`
+      "Command completed successfully.
+
+      Output:
+      Hello World"
+    `);
+  });
+
+  it('should format successful command with stderr', () => {
+    const result = formatCommandResult({
+      stderr: 'Warning: deprecated',
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`
+      "Command completed successfully.
+
+      Stderr:
+      Warning: deprecated"
+    `);
+  });
+
+  it('should format successful command with exit code', () => {
+    const result = formatCommandResult({
+      exitCode: 0,
+      success: true,
+    });
+    expect(result).toMatchInlineSnapshot(`
+      "Command completed successfully.
+
+      Exit code: 0"
+    `);
+  });
+
+  it('should format failed command', () => {
+    const result = formatCommandResult({
+      error: 'Permission denied',
+      success: false,
+    });
+    expect(result).toMatchInlineSnapshot(`"Command failed: Permission denied"`);
+  });
+
+  it('should format command with all fields', () => {
+    const result = formatCommandResult({
+      error: 'Command error',
+      exitCode: 1,
+      stderr: 'Error occurred',
+      stdout: 'Some output',
+      success: false,
+    });
+    expect(result).toMatchInlineSnapshot(`
+      "Command failed: Command error
+
+      Output:
+      Some output
+
+      Stderr:
+      Error occurred
+
+      Exit code: 1"
+    `);
+  });
+});