@opensumi/ide-ai-native 3.7.2-next-1740066619.0 → 3.7.2-next-1740107209.0

package/src/browser/mcp/mcp-server-proxy.service.ts

@@ -1,3 +1,5 @@
+import { zodToJsonSchema } from 'zod-to-json-schema';
+
 import { Autowired, Injectable } from '@opensumi/di';
 import { ILogger } from '@opensumi/ide-core-browser';
 import { Emitter, Event } from '@opensumi/ide-core-common';
@@ -29,10 +31,10 @@ export class MCPServerProxyService implements IMCPServerProxyService {
   async $getMCPTools() {
     const tools = await this.mcpServerRegistry.getMCPTools().map((tool) =>
       // 不要传递 handler
-       ({
+      ({
         name: tool.name,
         description: tool.description,
-        inputSchema: tool.inputSchema,
+        inputSchema: zodToJsonSchema(tool.inputSchema),
         providerName: 'sumi-builtin',
       }),
     );

package/src/browser/mcp/mcp-server.feature.registry.ts

@@ -1,11 +1,12 @@
 // OpenSumi as MCP Server 前端的代理服务
 import { Autowired, Injectable } from '@opensumi/di';
-import { IAIBackService, ILogger } from '@opensumi/ide-core-common';
+import { ILogger } from '@opensumi/ide-core-common';
 
-import { IMCPServerRegistry, MCPLogger, MCPToolDefinition } from '../types';
+import { getToolName } from '../../common/utils';
+import { IMCPServerRegistry, IMCPServerToolComponentProps, MCPLogger, MCPToolDefinition } from '../types';
 
 class LoggerAdapter implements MCPLogger {
-  constructor(private readonly logger: ILogger) { }
+  constructor(private readonly logger: ILogger) {}
 
   appendLine(message: string): void {
     this.logger.log(message);
@@ -15,6 +16,7 @@ class LoggerAdapter implements MCPLogger {
 @Injectable()
 export class MCPServerRegistry implements IMCPServerRegistry {
   private tools: MCPToolDefinition[] = [];
+  private toolComponents: Record<string, React.FC<IMCPServerToolComponentProps>> = {};
 
   @Autowired(ILogger)
   private readonly baseLogger: ILogger;
@@ -23,10 +25,26 @@ export class MCPServerRegistry implements IMCPServerRegistry {
     return new LoggerAdapter(this.baseLogger);
   }
 
+  getMCPTool(name: string, serverName = 'sumi-builtin'): MCPToolDefinition | undefined {
+    return this.tools.find((tool) => getToolName(tool.name, serverName) === name);
+  }
+
   registerMCPTool(tool: MCPToolDefinition): void {
     this.tools.push(tool);
   }
 
+  registerToolComponent(
+    name: string,
+    component: React.FC<IMCPServerToolComponentProps>,
+    serverName = 'sumi-builtin',
+  ): void {
+    this.toolComponents[getToolName(name, serverName)] = component;
+  }
+
+  getToolComponent(name: string): React.FC<IMCPServerToolComponentProps> | undefined {
+    return this.toolComponents[name];
+  }
+
   getMCPTools(): MCPToolDefinition[] {
     return this.tools;
   }
@@ -43,8 +61,12 @@ export class MCPServerRegistry implements IMCPServerRegistry {
       if (!tool) {
         throw new Error(`MCP tool ${name} not found`);
       }
+      // 统一校验并转换
+      args = tool.inputSchema.parse(args);
       return await tool.handler(args, this.logger);
     } catch (error) {
+      // eslint-disable-next-line no-console
+      console.error('callMCPTool error:', error);
       return {
         content: [{ type: 'text', text: `The tool ${name} failed to execute. Error: ${error}` }],
         isError: true,

package/src/browser/mcp/tools/components/EditFile.tsx

@@ -0,0 +1,144 @@
+import cls from 'classnames';
+import React, { useEffect, useMemo } from 'react';
+
+import { Icon, Popover } from '@opensumi/ide-components';
+import {
+  AppConfig,
+  LabelService,
+  MarkerSeverity,
+  URI,
+  Uri,
+  detectModeId,
+  path,
+  useAutorun,
+  useInjectable,
+} from '@opensumi/ide-core-browser';
+import { Loading } from '@opensumi/ide-core-browser/lib/components/ai-native';
+import { ILanguageService } from '@opensumi/monaco-editor-core/esm/vs/editor/common/languages/language';
+import { IModelService } from '@opensumi/monaco-editor-core/esm/vs/editor/common/services/model';
+import { StandaloneServices } from '@opensumi/monaco-editor-core/esm/vs/editor/standalone/browser/standaloneServices';
+
+import { ChatMarkdown } from '../../../components/ChatMarkdown';
+import { IMCPServerToolComponentProps } from '../../../types';
+import { BaseApplyService, CodeBlockData } from '../../base-apply.service';
+
+import styles from './index.module.less';
+
+const renderStatus = (codeBlockData: CodeBlockData) => {
+  const status = codeBlockData.status;
+  switch (status) {
+    case 'generating':
+      return <Loading />;
+    case 'pending':
+      return (
+        <Popover title={status} id={'edit-file-tool-status-pending'}>
+          <Icon iconClass='codicon codicon-circle-large' />
+        </Popover>
+      );
+    case 'success':
+      return (
+        <Popover title={status} id={'edit-file-tool-status-success'}>
+          <Icon iconClass='codicon codicon-check-all' />
+        </Popover>
+      );
+    case 'failed':
+      return (
+        <Popover title={status} id={'edit-file-tool-status-failed'}>
+          <Icon iconClass='codicon codicon-error' color='var(--vscode-input-errorForeground)' />
+        </Popover>
+      );
+    case 'cancelled':
+      return (
+        <Popover title={status} id={'edit-file-tool-status-cancelled'}>
+          <Icon iconClass='codicon codicon-close' color='var(--vscode-input-placeholderForeground)' />
+        </Popover>
+      );
+    default:
+      return null;
+  }
+};
+
+export const EditFileToolComponent = (props: IMCPServerToolComponentProps) => {
+  const { args, messageId, toolCallId } = props;
+  const labelService = useInjectable(LabelService);
+  const appConfig = useInjectable<AppConfig>(AppConfig);
+  const applyService = useInjectable<BaseApplyService>(BaseApplyService);
+  const { target_file = '', code_edit, instructions } = args || {};
+  const absolutePath = path.join(appConfig.workspaceDir, target_file);
+
+  const codeBlockData = applyService.getCodeBlock(absolutePath, messageId);
+
+  useAutorun(applyService.codeBlockMapObservable);
+
+  if (toolCallId && codeBlockData) {
+    applyService.initToolCallId(codeBlockData.id, toolCallId);
+  }
+
+  const icon = useMemo(() => {
+    if (!target_file) {
+      return;
+    }
+    const icon = `file-icon ${labelService.getIcon(URI.file(absolutePath))}`;
+    return icon;
+  }, [target_file, absolutePath]);
+  const languageId = useMemo(() => {
+    if (!target_file) {
+      return;
+    }
+    const modelService = StandaloneServices.get(IModelService);
+    const languageService = StandaloneServices.get(ILanguageService);
+    const detectedModeId = detectModeId(modelService, languageService, Uri.file(absolutePath));
+    return detectedModeId;
+  }, [target_file, absolutePath]);
+
+  // 多次迭代时，仅在首处tool组件中展示
+  if (!args || !codeBlockData || (toolCallId && toolCallId !== codeBlockData.initToolCallId)) {
+    return null;
+  }
+
+  return [
+    instructions && <p>{instructions}</p>,
+    <div className={styles['edit-file-tool']} key={`edit-file-tool-${codeBlockData.id}`}>
+      <div
+        className={cls(styles['edit-file-tool-header'], {
+          clickable: codeBlockData.status === 'pending' || codeBlockData.status === 'success',
+        })}
+        onClick={() => {
+          if (codeBlockData.status === 'pending') {
+            applyService.reRenderPendingApply();
+          } else if (codeBlockData.status === 'success') {
+            applyService.revealApplyPosition(codeBlockData.id);
+          }
+        }}
+      >
+        {icon && <span className={icon}></span>}
+        <span className={styles['edit-file-tool-file-name']}>{target_file}</span>
+        {codeBlockData.iterationCount > 1 && (
+          <span className={styles['edit-file-tool-iteration-count']}>{codeBlockData.iterationCount}/3</span>
+        )}
+        {renderStatus(codeBlockData)}
+      </div>
+      <ChatMarkdown markdown={`\`\`\`${languageId || ''}\n${code_edit}\n\`\`\``} hideInsert={true} />
+    </div>,
+    codeBlockData.applyResult && codeBlockData.applyResult.diagnosticInfos.length > 0 && (
+      <div
+        className={styles['edit-file-tool-diagnostic-errors']}
+        key={`edit-file-tool-diagnostic-errors-${codeBlockData.id}`}
+      >
+        <div className={styles['title']}>Found Lints:</div>
+        {codeBlockData.applyResult?.diagnosticInfos.map((info) => (
+          <div
+            key={info.message}
+            className={cls({
+              [styles['error']]: info.severity === MarkerSeverity.Error,
+              [styles['warning']]: info.severity === MarkerSeverity.Warning,
+            })}
+          >
+            <Icon className={`codicon codicon-${info.severity === MarkerSeverity.Error ? 'error' : 'warning'}`} />
+            {info.message.split('\n')[0]}
+          </div>
+        ))}
+      </div>
+    ),
+  ];
+};

package/src/browser/mcp/tools/components/index.module.less

@@ -0,0 +1,67 @@
+.edit-file-tool-header {
+  display: flex;
+  align-items: center;
+  padding: 2px 8px;
+  border-bottom: 1px solid var(--vscode-commandCenter-inactiveBorder);
+  background-color: var(--design-block-background);
+  font-size: 10px;
+  margin-bottom: -4px;
+  border-radius: 8px 8px 0 0;
+  > span {
+    margin-right: 4px;
+  }
+  :global(span.codicon) {
+    font-size: 12px;
+  }
+  :global(.kt-popover-trigger) {
+    display: flex;
+    align-items: center;
+    justify-content: center;
+  }
+}
+.edit-file-tool {
+  border: 1px solid var(--vscode-commandCenter-inactiveBorder);
+  border-radius: 4px;
+  margin: 8px 0;
+  :global(.language-badge) {
+    border-top-left-radius: 0;
+  }
+  pre > code {
+    border-radius: 0 0 8px 8px !important;
+  }
+}
+.edit-file-tool-iteration-count {
+  color: var(--vscode-input-placeholderForeground);
+  margin-left: 4px;
+}
+:global(.clickable) {
+  cursor: pointer;
+}
+.edit-file-tool-diagnostic-errors {
+  padding: 8px;
+  border: 1px solid var(--vscode-commandCenter-inactiveBorder);
+  background-color: var(--design-block-background);
+  border-radius: 8px;
+  font-size: 12px;
+  margin: 4px 0;
+  > div {
+    display: flex;
+    align-items: center;
+    padding: 2px 0;
+  }
+  :global(.codicon) {
+    margin-right: 4px;
+  }
+  .title {
+    margin-bottom: 3px;
+    display: inline-block;
+  }
+}
+.error,
+.error > span {
+  color: var(--debugConsole-errorForeground);
+}
+.warning,
+.warning > span {
+  color: var(--debugConsole-warningForeground);
+}

package/src/browser/mcp/tools/createNewFileWithText.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired } from '@opensumi/di';
 import { Domain, URI, path } from '@opensumi/ide-core-common';
@@ -38,7 +37,7 @@ export class CreateNewFileWithTextTool implements MCPServerContribution {
         '"ok" if the file was successfully created and populated, ' +
         '"can\'t find project dir" if the project directory cannot be determined. ' +
         'Note: Creates any necessary parent directories automatically.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/editFile.ts

@@ -0,0 +1,100 @@
+import { z } from 'zod';
+
+import { Autowired } from '@opensumi/di';
+import { Domain } from '@opensumi/ide-core-common';
+
+import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';
+
+import { EditFileToolComponent } from './components/EditFile';
+import { EditFileHandler } from './handlers/EditFile';
+const inputSchema = z
+  .object({
+    target_file: z
+      .string()
+      .describe(
+        'The target file to modify. Always specify the target file as the first argument and use the relative path in the workspace of the file to edit',
+      ),
+    instructions: z
+      .string()
+      .optional()
+      .describe(
+        'A single sentence instruction describing what you are going to do for the sketched edit. This is used to assist the less intelligent model in applying the edit. Please use the first person to describe what you are going to do. Dont repeat what you have said previously in normal messages. And use it to disambiguate uncertainty in the edit.',
+      ),
+    code_edit: z
+      .string()
+      .describe(
+        "Specify ONLY the precise lines of code that you wish to edit. **NEVER specify or write out unchanged code**. Instead, represent all unchanged code using the comment of the language you're editing in - example: `// ... existing code ...`",
+      ),
+  })
+  .transform((data) => ({
+    targetFile: data.target_file,
+    instructions: data.instructions,
+    codeEdit: data.code_edit,
+  }));
+
+@Domain(MCPServerContribution)
+export class EditFileTool implements MCPServerContribution {
+  @Autowired(EditFileHandler)
+  private readonly editFileHandler: EditFileHandler;
+
+  registerMCPServer(registry: IMCPServerRegistry): void {
+    registry.registerMCPTool(this.getToolDefinition());
+    registry.registerToolComponent('edit_file', EditFileToolComponent);
+  }
+
+  getToolDefinition(): MCPToolDefinition {
+    return {
+      name: 'edit_file',
+      label: 'Edit File',
+      description: `Use this tool to propose an edit to an existing file.
+This will be read by a less intelligent model, which will quickly apply the edit. You should make it clear what the edit is, while also minimizing the unchanged code you write.
+When writing the edit, you should specify each edit in sequence, with the special comment \`// ... existing code ...\` to represent unchanged code in between edited lines.
+For example:
+\`\`\`
+// ... existing code ...
+FIRST_EDIT
+// ... existing code ...
+SECOND_EDIT
+// ... existing code ...
+THIRD_EDIT
+// ... existing code ...
+\`\`\`
+You should bias towards repeating as few lines of the original file as possible to convey the change.
+But, each edit should contain sufficient context of unchanged lines around the code you're editing to resolve ambiguity.
+DO NOT omit spans of pre-existing code without using the \`// ... existing code ...\` comment to indicate its absence.
+Make sure it is clear what the edit should be.
+You should specify the following arguments before the others: [target_file]`,
+      inputSchema,
+      handler: this.handler.bind(this),
+    };
+  }
+
+  private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
+    const result = await this.editFileHandler.handler(args.targetFile, args.codeEdit, args.instructions);
+    return {
+      content: [
+        {
+          type: 'text',
+          // TODO: lint error
+          text: result.applyResult
+            ? `The apply model made the following changes to the file:
+
+\`\`\`
+${result.applyResult.diff}
+\`\`\`
+${
+  result.applyResult.diagnosticInfos.length > 0
+    ? `The edit introduced the following new linter errors:
+${result.applyResult.diagnosticInfos
+  .map((error) => `Line ${error.startLineNumber}: ${error.message.split('\n')[0]}`)
+  .join('\n')}
+
+Please fix the linter errors if it is clear how to (or you can easily figure out how to). Do not make uneducated guesses. And do not loop more than 3 times on fixing linter errors on the same file.`
+    : ''
+}`
+            : 'User cancelled the edit.',
+        },
+      ],
+    };
+  }
+}

package/src/browser/mcp/tools/findFilesByNameSubstring.ts

@@ -1,7 +1,6 @@
 import * as path from 'path';
 
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain, URI } from '@opensumi/ide-core-common';
@@ -38,7 +37,7 @@ export class FindFilesByNameSubstringTool implements MCPServerContribution {
         '- name: File name ' +
         'Returns an empty array ([]) if no matching files are found. ' +
         'Note: Only searches through files within the project directory, excluding libraries and external dependencies.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/getCurrentFilePath.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain } from '@opensumi/ide-core-common';
@@ -25,7 +24,7 @@ export class GetCurrentFilePathTool implements MCPServerContribution {
         'Retrieves the absolute path of the currently active file in the VS Code editor. ' +
         'Use this tool to get the file location for tasks requiring file path information. ' +
         'Returns an empty string if no file is currently open.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/getDiagnosticsByPath.ts

@@ -1,7 +1,6 @@
 import * as path from 'path';
 
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain, URI } from '@opensumi/ide-core-common';
@@ -47,7 +46,7 @@ export class GetDiagnosticsByPathTool implements MCPServerContribution {
         '- "error": Must be fixed immediately as they indicate critical issues that will prevent code from working correctly. ' +
         '- "warning": For user code, preserve unless the warning indicates a clear improvement opportunity. For generated code, optimize to remove warnings. ' +
         '- "information"/"hint": For user code, preserve as they might reflect intentional patterns. For generated code, optimize if it improves code quality without changing functionality.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/getFileTextByPath.ts

@@ -1,7 +1,6 @@
 import * as path from 'path';
 
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain, URI } from '@opensumi/ide-core-common';
@@ -38,7 +37,7 @@ export class GetFileTextByPathTool implements MCPServerContribution {
         '- error "project dir not found" if project directory cannot be determined ' +
         '- error "file not found" if the file doesn\'t exist or is outside project scope ' +
         'Note: Automatically refreshes the file system before reading',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/getOpenEditorFileDiagnostics.ts

@@ -1,7 +1,6 @@
 import * as path from 'path';
 
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain, URI } from '@opensumi/ide-core-common';
@@ -47,7 +46,7 @@ export class GetOpenEditorFileDiagnosticsTool implements MCPServerContribution {
         '- "error": Must be fixed immediately as they indicate critical issues that will prevent code from working correctly. ' +
         '- "warning": For user code, preserve unless the warning indicates a clear improvement opportunity. For generated code, optimize to remove warnings. ' +
         '- "information"/"hint": For user code, preserve as they might reflect intentional patterns. For generated code, optimize if it improves code quality without changing functionality.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/getOpenEditorFileText.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain } from '@opensumi/ide-core-common';
@@ -25,7 +24,7 @@ export class GetOpenEditorFileTextTool implements MCPServerContribution {
         'Retrieves the complete text content of the currently active file in the IDE editor. ' +
         "Use this tool to access and analyze the file's contents for tasks such as code review, content inspection, or text processing. " +
         'Returns empty string if no file is currently open.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/getSelectedText.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired, Injectable } from '@opensumi/di';
 import { Domain } from '@opensumi/ide-core-common';
@@ -25,7 +24,7 @@ export class GetSelectedTextTool implements MCPServerContribution {
         'Retrieves the currently selected text from the active editor in VS Code. ' +
         'Use this tool when you need to access and analyze text that has been highlighted/selected by the user. ' +
         'Returns an empty string if no text is selected or no editor is open.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }

package/src/browser/mcp/tools/handlers/EditFile.ts

@@ -0,0 +1,21 @@
+import { Autowired, Injectable } from '@opensumi/di';
+
+import { BaseApplyService } from '../../base-apply.service';
+
+/**
+ * TODO: 代码块改动版本号，次数，流式工具调用？
+ * 基础文件编辑处理类
+ * 用于处理代码改动的应用、保存等操作
+ */
+@Injectable()
+export class EditFileHandler {
+  @Autowired(BaseApplyService)
+  private applyService: BaseApplyService;
+
+  async handler(relativePath: string, updateContent: string, instructions?: string) {
+    // TODO: ignore file
+    this.applyService.registerCodeBlock(relativePath, updateContent);
+    const blockData = await this.applyService.apply(relativePath, updateContent, instructions);
+    return blockData;
+  }
+}

package/src/browser/mcp/tools/handlers/ReadFile.ts

@@ -12,6 +12,11 @@ export class FileHandler {
   private static readonly MAX_CHARS = 1e5;
   private static readonly NEWLINE = '\n';
 
+  private fileResultMap: Map<
+    string,
+    { content: string; startLineOneIndexed: number; endLineOneIndexedInclusive: number }
+  > = new Map();
+
   @Autowired(IEditorDocumentModelService)
   protected modelService: IEditorDocumentModelService;
 
@@ -141,7 +146,14 @@ export class FileHandler {
           didShortenCharRange = true;
           selectedContent = this.trimContent(selectedContent, FileHandler.MAX_CHARS);
         }
-
+        // 文件的浏览窗口需要记录，应用的时候需要用
+        if (didShortenLineRange) {
+          this.fileResultMap.set(fileParams.relativeWorkspacePath, {
+            content: selectedContent,
+            startLineOneIndexed: adjustedStart,
+            endLineOneIndexedInclusive: adjustedEnd,
+          });
+        }
         return {
           contents: selectedContent,
           didDowngradeToLineRange: shouldForceLimitLines,
@@ -171,4 +183,10 @@ export class FileHandler {
       modelReference?.dispose();
     }
   }
+
+  getFileReadResult(
+    relativeWorkspacePath: string,
+  ): { content: string; startLineOneIndexed: number; endLineOneIndexedInclusive: number } | undefined {
+    return this.fileResultMap.get(relativeWorkspacePath);
+  }
 }

package/src/browser/mcp/tools/handlers/utils.ts

@@ -0,0 +1,3 @@
+export function generateCodeBlockId(composerId: string, messageId: string): string {
+  return `${composerId}:${messageId}`;
+}

package/src/browser/mcp/tools/listDir.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired } from '@opensumi/di';
 import { Domain } from '@opensumi/ide-core-common';
@@ -33,16 +32,15 @@ export class ListDirTool implements MCPServerContribution {
   getToolDefinition(): MCPToolDefinition {
     return {
       name: 'list_dir',
+      label: 'List Directory',
       description:
         'List the contents of a directory. The quick tool to use for discovery, before using more targeted tools like semantic search or file reading. Useful to try to understand the file structure before diving deeper into specific files. Can be used to explore the codebase.',
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }
 
   private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
-    // TODO: 应该添加统一的 validate 逻辑
-    args = inputSchema.parse(args);
     const result = await this.listDirHandler.handler(args);
     return {
       content: [

package/src/browser/mcp/tools/readFile.ts

@@ -1,5 +1,4 @@
 import { z } from 'zod';
-import { zodToJsonSchema } from 'zod-to-json-schema';
 
 import { Autowired } from '@opensumi/di';
 import { Domain } from '@opensumi/ide-core-common';
@@ -37,6 +36,7 @@ export class ReadFileTool implements MCPServerContribution {
   getToolDefinition(): MCPToolDefinition {
     return {
       name: 'read_file',
+      label: 'Read File',
       description: `Read the contents of a file (and the outline).
 
 When using this tool to gather information, it's your responsibility to ensure you have the COMPLETE context. Each time you call this command you should:
@@ -48,14 +48,12 @@ When using this tool to gather information, it's your responsibility to ensure y
 If reading a range of lines is not enough, you may choose to read the entire file.
 Reading entire files is often wasteful and slow, especially for large files (i.e. more than a few hundred lines). So you should use this option sparingly.
 Reading the entire file is not allowed in most cases. You are only allowed to read the entire file if it has been edited or manually attached to the conversation by the user.`,
-      inputSchema: zodToJsonSchema(inputSchema),
+      inputSchema,
       handler: this.handler.bind(this),
     };
   }
 
   private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
-    // TODO: 应该添加统一的 validate 逻辑
-    args = inputSchema.parse(args);
     const result = await this.fileHandler.readFile(args);
     return {
       content: [