@opensumi/ide-ai-native 3.9.1-next-1747733557.0 → 3.9.1-next-1747829468.0

package/src/browser/chat/apply.service.ts

@@ -1,7 +1,6 @@
 import { Autowired, Injectable } from '@opensumi/di';
 import { AIBackSerivcePath, IAIBackService, IChatProgress, URI, path } from '@opensumi/ide-core-browser';
 import { IEditorDocumentModelService } from '@opensumi/ide-editor/lib/browser';
-import { ICodeEditor } from '@opensumi/ide-monaco';
 import { SumiReadableStream } from '@opensumi/ide-utils/lib/stream';
 import { Range } from '@opensumi/monaco-editor-core';
 
@@ -13,10 +12,6 @@ import { ChatProxyService } from './chat-proxy.service';
 
 @Injectable()
 export class ApplyService extends BaseApplyService {
-  protected postApplyHandler(editor: ICodeEditor): Promise<void> {
-    return Promise.resolve();
-  }
-
   @Autowired(IEditorDocumentModelService)
   private readonly modelService: IEditorDocumentModelService;
 

package/src/browser/mcp/base-apply.service.ts

@@ -63,8 +63,6 @@ export abstract class BaseApplyService extends WithEventBus {
 
   private currentSessionId?: string;
 
-  protected abstract postApplyHandler(editor: ICodeEditor): Promise<void>;
-
   constructor() {
     super();
     this.addDispose(
@@ -319,13 +317,6 @@ export abstract class BaseApplyService extends WithEventBus {
       codeBlock.applyResult = res.result;
       this.updateCodeBlock(codeBlock);
 
-      // apply 结果流式输出完成后，自动尝试修复代码中的 lint 错误
-      if (this.postApplyHandler && typeof this.postApplyHandler === 'function') {
-        try {
-          await this.postApplyHandler(result.group.codeEditor.monacoEditor);
-        } catch (error) {}
-      }
-
       return codeBlock;
     } catch (err) {
       codeBlock.status = 'failed';

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

@@ -44,7 +44,12 @@ export class MCPServerRegistry implements IMCPServerRegistry {
   }
 
   registerMCPTool(tool: MCPToolDefinition): void {
-    this.tools.push(tool);
+    const existingIndex = this.tools.findIndex((t) => t.name === tool.name);
+    if (existingIndex !== -1) {
+      this.tools[existingIndex] = tool;
+    } else {
+      this.tools.push(tool);
+    }
   }
 
   registerToolComponent(

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

@@ -80,11 +80,14 @@ export class FileSearchTool implements MCPServerContribution {
     });
 
     const messages = this.chatInternalService.sessionModel.history.getMessages();
-    this.chatInternalService.sessionModel.history.setMessageAdditional(messages[messages.length - 1].id, {
-      [args.toolCallId]: {
-        files,
-      },
-    });
+    const messageId = messages[messages.length - 1]?.id;
+    if (messageId) {
+      this.chatInternalService.sessionModel.history.setMessageAdditional(messageId, {
+        [args.toolCallId]: {
+          files,
+        },
+      });
+    }
 
     logger.appendLine(`Found ${files.length} files matching "${args.query}"`);
 

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

@@ -111,11 +111,14 @@ export class GrepSearchTool implements MCPServerContribution {
       }
       deferred.resolve(results.join('\n\n'));
       const messages = this.chatInternalService.sessionModel.history.getMessages();
-      this.chatInternalService.sessionModel.history.setMessageAdditional(messages[messages.length - 1].id, {
-        [args.toolCallId]: {
-          files,
-        },
-      });
+      const messageId = messages[messages.length - 1]?.id;
+      if (messageId) {
+        this.chatInternalService.sessionModel.history.setMessageAdditional(messageId, {
+          [args.toolCallId]: {
+            files,
+          },
+        });
+      }
     });
     const text = await deferred.promise;
     return {

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) => {