@opensumi/ide-ai-native 3.9.1-next-1747791726.0 → 3.9.1-next-1747836538.0

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

@@ -13,18 +13,25 @@ const color = {
   reset: '\x1b[0m',
 };
 
-export const inputSchema = z.object({
-  command: z.string().describe('The terminal command to execute'),
-  is_background: z.boolean().describe('Whether the command should be run in the background'),
-  explanation: z
-    .string()
-    .describe('One sentence explanation as to why this command needs to be run and how it contributes to the goal.'),
-  require_user_approval: z
-    .boolean()
-    .describe(
-      "Whether the user must approve the command before it is executed. Only set this to false if the command is safe and if it matches the user's requirements for commands that should be executed automatically.",
-    ),
-});
+export const inputSchema = z
+  .object({
+    command: z.string().describe('The terminal command to execute'),
+    is_background: z.boolean().describe('Whether the command should be run in the background'),
+    explanation: z
+      .string()
+      .describe('One sentence explanation as to why this command needs to be run and how it contributes to the goal.'),
+    require_user_approval: z
+      .boolean()
+      .describe(
+        "Whether the user must approve the command before it is executed. Only set this to false if the command is safe and if it matches the user's requirements for commands that should be executed automatically.",
+      ),
+  })
+  .transform((data) => ({
+    command: data.command,
+    isBackground: data.is_background,
+    explanation: data.explanation,
+    requireUserApproval: data.require_user_approval,
+  }));
 
 @Injectable()
 export class RunCommandHandler {
@@ -66,7 +73,7 @@ export class RunCommandHandler {
 
   async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) {
     logger.appendLine(`Executing command: ${args.command}`);
-    if (this.isAlwaysApproval(args.require_user_approval)) {
+    if (this.isAlwaysApproval(args.requireUserApproval)) {
       const def = new Deferred<boolean>();
       this.approvalDeferredMap.set(args.toolCallId, def);
       const approval = await def.promise;
@@ -93,7 +100,7 @@ export class RunCommandHandler {
     const result: { type: string; text: string }[] = [];
     const def = new Deferred<{ isError?: boolean; content: { type: string; text: string }[] }>();
 
-    if (args.is_background) {
+    if (args.isBackground) {
       def.resolve({
         isError: false,
         content: [{ type: 'text', text: `Successful run command ${args.command} in background.` }],

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

@@ -63,18 +63,21 @@ export class ListDirTool implements MCPServerContribution {
 
     // 设置消息的附加数据
     const messages = this.chatInternalService.sessionModel.history.getMessages();
-    this.chatInternalService.sessionModel.history.setMessageAdditional(messages[messages.length - 1].id, {
-      [args.toolCallId]: {
-        files: fileUris,
-        title: `Listed directory "${args.relativeWorkspacePath}"`,
-        details: result.files.map((file) => ({
-          type: file.isDirectory ? 'dir' : 'file',
-          name: file.name,
-          info: file.isDirectory ? `${file.numChildren ?? '?'} items` : `${file.size}KB, ${file.numLines} lines`,
-          lastModified: file.lastModified,
-        })),
-      },
-    });
+    const messageId = messages[messages.length - 1]?.id;
+    if (messageId) {
+      this.chatInternalService.sessionModel.history.setMessageAdditional(messageId, {
+        [args.toolCallId]: {
+          files: fileUris,
+          title: `Listed directory "${args.relativeWorkspacePath}"`,
+          details: result.files.map((file) => ({
+            type: file.isDirectory ? 'dir' : 'file',
+            name: file.name,
+            info: file.isDirectory ? `${file.numChildren ?? '?'} items` : `${file.size}KB, ${file.numLines} lines`,
+            lastModified: file.lastModified,
+          })),
+        },
+      });
+    }
 
     logger.appendLine(`Listed ${fileUris.length} files in directory "${args.relativeWorkspacePath}"`);
 

package/src/node/mcp-server.sse.ts

@@ -1,5 +1,6 @@
 // have to import with extension since the exports map is ./* -> ./dist/cjs/*
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+import { SSEClientTransport } from '@modelcontextprotocol/sdk/client/sse.js';
 import { EventSource } from 'eventsource';
 
 import { ILogger } from '@opensumi/ide-core-common';
@@ -47,8 +48,6 @@ export class SSEMCPServer implements IMCPServer {
     }
     this.logger?.log(`Starting server "${this.name}" with url: ${this.url}`);
 
-    const SSEClientTransport = (await import('@modelcontextprotocol/sdk/client/sse.js')).SSEClientTransport;
-
     const transport = new SSEClientTransport(new URL(this.url), this.transportOptions);
 
     transport.onerror = (error) => {