@opensumi/ide-ai-native 3.8.3-next-1744969919.0 → 3.8.3-next-1745295615.0

package/src/browser/contrib/inline-completions/prompt/languages.ts

@@ -0,0 +1,65 @@
+export const LANGUAGE_TO_SUFFIX: {
+  [key: string]: string;
+} = {
+  abap: '.abap',
+  bat: '.bat',
+  bibtex: '.bib',
+  c: '.c',
+  clojure: '.clj',
+  coffeescript: '.coffee',
+  cpp: '.cpp',
+  csharp: '.cs',
+  css: '.css',
+  'cuda-cpp': '.cu',
+  dart: '.dart',
+  diff: '.diff',
+  dockerfile: '.dockerfile',
+  fsharp: '.fs',
+  go: '.go',
+  groovy: '.groovy',
+  haml: '.haml',
+  handlebars: '.handlebars',
+  hlsl: '.hlsl',
+  html: '.html',
+  ini: '.ini',
+  jade: '.jade',
+  java: '.java',
+  javascript: '.js',
+  javascriptreact: '.jsx',
+  json: '.json',
+  julia: '.jl',
+  latex: '.tex',
+  less: '.less',
+  lua: '.lua',
+  makefile: '.make',
+  markdown: '.md',
+  'objective-c': '.m',
+  'objective-cpp': '.mm',
+  perl: '.pl',
+  perl6: '.6pl',
+  php: '.php',
+  plaintext: '.txt',
+  powershell: '.ps1',
+  pug: '.pug',
+  python: '.py',
+  r: '.r',
+  razor: '.cshtml',
+  ruby: '.rb',
+  rust: '.rs',
+  sass: '.sass',
+  scss: '.scss',
+  shaderlab: '.shader',
+  shellscript: '.sh',
+  slim: '.slim',
+  sql: '.sql',
+  stylus: '.styl',
+  swift: '.swift',
+  tex: '.tex',
+  typescript: '.ts',
+  typescriptreact: '.tsx',
+  vb: '.vb',
+  vue: '.vue',
+  xml: '.xml',
+  xsl: '.xsl',
+  yaml: '.yaml',
+};

package/src/browser/contrib/merge-conflict/index.ts

@@ -439,9 +439,8 @@ export class MergeConflictContribution
   }
 
   private getCacheResolvedConflicts(currentUri?: string) {
-    currentUri = currentUri || this.getModel()?.uri.toString();
     if (!currentUri) {
-      return new Map();
+      currentUri = this.getModel().uri.toString();
     }
     const cache = this.cacheResolvedConflicts.get(currentUri);
     if (cache) {
@@ -943,7 +942,7 @@ export class MergeConflictContribution
    * 刷新 codelens
    */
   private updateCodeLens() {
-    if (this.getModel()?.uri) {
+    if (this.getModel().uri) {
       // @ts-ignore
       languageFeaturesService.codeLensProvider._onDidChange.fire();
     }

package/src/browser/model/msg-history-manager.ts

@@ -26,10 +26,6 @@ export class MsgHistoryManager extends Disposable {
     super.dispose();
   }
 
-  public get size(): number {
-    return this.messageMap.size;
-  }
-
   public clearMessages() {
     this.messageMap.clear();
     this.messageAdditionalMap.clear();

package/src/common/index.ts

@@ -15,7 +15,7 @@ import {
 import { DESIGN_MENUBAR_CONTAINER_VIEW_ID } from '@opensumi/ide-design/lib/common/constants';
 import { IPosition, ITextModel, InlineCompletionContext } from '@opensumi/ide-monaco/lib/common';
 
-import { MCPServerDescription } from './mcp-server-manager';
+import { IMCPServer, MCPServerDescription } from './mcp-server-manager';
 import { IPartialEditEvent, MCPTool } from './types';
 
 import type { CoreMessage } from 'ai';
@@ -151,6 +151,7 @@ export interface ISumiMCPServerBackend {
   $addOrUpdateServer(description: MCPServerDescription): void;
   $removeServer(name: string): void;
   $syncServer(name: string): Promise<void>;
+  $getMCPServerByName(name: string): IMCPServer | undefined;
 }
 
 export const SumiMCPServerProxyServicePath = 'SumiMCPServerProxyServicePath';

package/src/common/mcp-server-manager.ts

@@ -1,6 +1,6 @@
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 
-import { MCP_SERVER_TYPE } from './types';
+import { MCP_SERVER_TYPE, SSEClientTransportOptions } from './types';
 
 export interface IMCPServer {
   isStarted(): boolean;
@@ -8,6 +8,7 @@ export interface IMCPServer {
   getServerName(): string;
   callTool(toolName: string, toolCallId: string, arg_string: string): ReturnType<Client['callTool']>;
   getTools(): ReturnType<Client['listTools']>;
+  getClient(): Client | null;
   update(command: string, args?: string[], env?: { [key: string]: string }): void;
   stop(): void;
 }
@@ -32,6 +33,7 @@ export interface MCPServerManager {
   registerTools(serverName: string): Promise<void>;
   addExternalMCPServers(servers: MCPServerDescription[]): void;
   getServers(): Map<string, IMCPServer>;
+  getServerByName(name: string): IMCPServer | undefined;
 }
 
 export type MCPTool = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number];
@@ -76,6 +78,7 @@ export interface SSEMCPServerDescription extends BaseMCPServerDescription {
    * The host of the MCP server.
    */
   serverHost: string;
+  transportOptions?: SSEClientTransportOptions;
 }
 
 export interface BuiltinMCPServerDescription {

package/src/common/types.ts

@@ -118,3 +118,10 @@ export interface IPartialEditEvent {
     deletedLinesCount: number;
   };
 }
+
+export interface SSEClientTransportOptions {
+  /**
+   * Customizes recurring POST requests to the server.
+   */
+  requestInit?: RequestInit;
+}

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

@@ -60,6 +60,10 @@ export class SumiMCPServerBackend extends RPCService<IMCPServerProxyService> imp
     return tools;
   }
 
+  $getMCPServerByName(name: string) {
+    return this.mcpServerManager.getServerByName(name);
+  }
+
   async callMCPTool(name: string, args: any) {
     if (!this.client) {
       throw new Error('SUMI MCP RPC Client not initialized');
@@ -232,6 +236,10 @@ export class BuiltinMCPServer implements IMCPServer {
     return BUILTIN_MCP_SERVER_NAME;
   }
 
+  getClient(): Client | null {
+    return null;
+  }
+
   async start(): Promise<void> {
     if (this.started) {
       return;

package/src/node/mcp-server-manager-impl.ts

@@ -92,6 +92,10 @@ export class MCPServerManagerImpl implements MCPServerManager {
     return Array.from(this.servers.keys());
   }
 
+  getServerByName(name: string): IMCPServer | undefined {
+    return this.servers.get(name);
+  }
+
   private convertToToolRequest(tool: MCPTool, serverName: string): ToolRequest {
     const id = getToolName(tool.name, serverName);
 
@@ -153,11 +157,11 @@ export class MCPServerManagerImpl implements MCPServerManager {
         this.servers.set(name, newServer);
       }
     } else if (description.type === MCP_SERVER_TYPE.SSE) {
-      const { name, serverHost } = description;
+      const { name, serverHost, transportOptions } = description;
       if (existingServer) {
         existingServer.update(serverHost);
       } else {
-        const newServer = new SSEMCPServer(name, serverHost, this.logger);
+        const newServer = new SSEMCPServer(name, serverHost, this.logger, transportOptions);
         this.servers.set(name, newServer);
       }
     }

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

@@ -6,17 +6,25 @@ import { ILogger } from '@opensumi/ide-core-common';
 
 import pkg from '../../package.json';
 import { IMCPServer } from '../common/mcp-server-manager';
+import { SSEClientTransportOptions } from '../common/types';
 
 global.EventSource = EventSource as any;
 export class SSEMCPServer implements IMCPServer {
   private name: string;
   public serverHost: string;
+  private transportOptions?: SSEClientTransportOptions;
   private client: Client;
   private started: boolean = false;
 
-  constructor(name: string, serverHost: string, private readonly logger?: ILogger) {
+  constructor(
+    name: string,
+    serverHost: string,
+    private readonly logger?: ILogger,
+    private readonly options?: SSEClientTransportOptions,
+  ) {
     this.name = name;
     this.serverHost = serverHost;
+    this.transportOptions = options;
   }
 
   isStarted(): boolean {
@@ -27,6 +35,10 @@ export class SSEMCPServer implements IMCPServer {
     return this.name;
   }
 
+  getClient(): Client | null {
+    return this.client;
+  }
+
   async start(): Promise<void> {
     if (this.started) {
       return;
@@ -35,7 +47,8 @@ export class SSEMCPServer implements IMCPServer {
 
     const SSEClientTransport = (await import('@modelcontextprotocol/sdk/client/sse.js')).SSEClientTransport;
 
-    const transport = new SSEClientTransport(new URL(this.serverHost));
+    const transport = new SSEClientTransport(new URL(this.serverHost), this.transportOptions);
+
     transport.onerror = (error) => {
       this.logger?.error('Transport Error:', error);
     };

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

@@ -36,6 +36,10 @@ export class StdioMCPServer implements IMCPServer {
     return this.name;
   }
 
+  getClient(): Client | null {
+    return this.client;
+  }
+
   async start(): Promise<void> {
     if (this.started) {
       return;