@opensumi/ide-ai-native 3.7.2-next-1739848467.0 → 3.7.2-next-1739945875.0

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

@@ -0,0 +1,91 @@
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+
+import { Autowired, Injectable } from '@opensumi/di';
+import { Domain } from '@opensumi/ide-core-common';
+import { WorkbenchEditorService } from '@opensumi/ide-editor';
+import { Selection, SelectionDirection } from '@opensumi/monaco-editor-core/esm/vs/editor/common/core/selection';
+
+import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';
+import { LiveInlineDiffPreviewer } from '../../widget/inline-diff/inline-diff-previewer';
+import { InlineDiffController } from '../../widget/inline-diff/inline-diff.controller';
+
+const inputSchema = z.object({
+  text: z.string().describe('The new content to replace the entire file with'),
+});
+
+@Domain(MCPServerContribution)
+export class ReplaceOpenEditorFileByDiffPreviewerTool implements MCPServerContribution {
+  @Autowired(WorkbenchEditorService)
+  private readonly editorService: WorkbenchEditorService;
+
+  registerMCPServer(registry: IMCPServerRegistry): void {
+    registry.registerMCPTool(this.getToolDefinition());
+  }
+
+  getToolDefinition(): MCPToolDefinition {
+    return {
+      name: 'replace_open_in_editor_file_text',
+      description:
+        'Replaces the entire content of the currently active file in the IDE editor with specified new text using diff previewer. ' +
+        "Use this tool when you need to completely overwrite the current file's content with diff preview. " +
+        'Requires a text parameter containing the new content. ' +
+        'Returns one of three possible responses: ' +
+        '"ok" if the file content was successfully replaced, ' +
+        '"no file open" if no editor is active, ' +
+        '"unknown error" if the operation fails.',
+      inputSchema: zodToJsonSchema(inputSchema),
+      handler: this.handler.bind(this),
+    };
+  }
+
+  private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
+    try {
+      const editor = this.editorService.currentEditor;
+      if (!editor || !editor.monacoEditor) {
+        logger.appendLine('Error: No active text editor found');
+        return {
+          content: [{ type: 'text', text: 'no file open' }],
+          isError: true,
+        };
+      }
+
+      // Get the model and its full range
+      const model = editor.monacoEditor.getModel();
+      if (!model) {
+        logger.appendLine('Error: No model found for current editor');
+        return {
+          content: [{ type: 'text', text: 'unknown error' }],
+          isError: true,
+        };
+      }
+
+      const fullRange = model.getFullModelRange();
+      const inlineDiffHandler = InlineDiffController.get(editor.monacoEditor)!;
+
+      // Create diff previewer
+      const previewer = inlineDiffHandler.createDiffPreviewer(
+        editor.monacoEditor,
+        Selection.fromRange(fullRange, SelectionDirection.LTR),
+        {
+          disposeWhenEditorClosed: false,
+          renderRemovedWidgetImmediately: true,
+        },
+      ) as LiveInlineDiffPreviewer;
+
+      // Set the new content
+      previewer.setValue(args.text);
+
+      logger.appendLine('Successfully created diff preview with new content');
+      return {
+        content: [{ type: 'text', text: 'ok' }],
+      };
+    } catch (error) {
+      logger.appendLine(`Error during file content replacement: ${error}`);
+      return {
+        content: [{ type: 'text', text: 'unknown error' }],
+        isError: true,
+      };
+    }
+  }
+}

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

@@ -0,0 +1,107 @@
+import { z } from 'zod';
+import { zodToJsonSchema } from 'zod-to-json-schema';
+
+import { Autowired } from '@opensumi/di';
+import { AppConfig } from '@opensumi/ide-core-browser';
+import { Deferred, Domain } from '@opensumi/ide-core-common';
+import { ITerminalController, ITerminalGroupViewService } from '@opensumi/ide-terminal-next/lib/common/controller';
+
+import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';
+
+const color = {
+  italic: '\x1b[3m',
+  reset: '\x1b[0m',
+};
+
+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.",
+    ),
+});
+
+@Domain(MCPServerContribution)
+export class RunTerminalCommandTool implements MCPServerContribution {
+  @Autowired(ITerminalController)
+  protected readonly terminalController: ITerminalController;
+
+  @Autowired(AppConfig)
+  protected readonly appConfig: AppConfig;
+
+  @Autowired(ITerminalGroupViewService)
+  protected readonly terminalView: ITerminalGroupViewService;
+
+  private terminalId = 0;
+
+  registerMCPServer(registry: IMCPServerRegistry): void {
+    registry.registerMCPTool(this.getToolDefinition());
+  }
+
+  getToolDefinition(): MCPToolDefinition {
+    return {
+      name: 'run_terminal_cmd',
+      description:
+        "PROPOSE a command to run on behalf of the user.\nIf you have this tool, note that you DO have the ability to run commands directly on the USER's system.\n\nAdhere to these rules:\n1. Based on the contents of the conversation, you will be told if you are in the same shell as a previous step or a new shell.\n2. If in a new shell, you should `cd` to the right directory and do necessary setup in addition to running the command.\n3. If in the same shell, the state will persist, no need to do things like `cd` to the same directory.\n4. For ANY commands that would use a pager, you should append ` | cat` to the command (or whatever is appropriate). You MUST do this for: git, less, head, tail, more, etc.\n5. For commands that are long running/expected to run indefinitely until interruption, please run them in the background. To run jobs in the background, set `is_background` to true rather than changing the details of the command.\n6. Dont include any newlines in the command.",
+      inputSchema: zodToJsonSchema(inputSchema),
+      handler: this.handler.bind(this),
+    };
+  }
+
+  getShellLaunchConfig(command: string) {
+    return {
+      name: `MCP:Terminal_${this.terminalId++}`,
+      cwd: this.appConfig.workspaceDir,
+      args: ['-c', command],
+    };
+  }
+
+  private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
+    if (args.require_user_approval) {
+      // FIXME: support approval
+    }
+
+    const terminalClient = await this.terminalController.createTerminalWithWidget({
+      config: this.getShellLaunchConfig(args.command),
+      closeWhenExited: false,
+    });
+
+    this.terminalController.showTerminalPanel();
+
+    const result: { type: string; text: string }[] = [];
+    const def = new Deferred<{ isError?: boolean; content: { type: string; text: string }[] }>();
+
+    terminalClient.onOutput((e) => {
+      result.push({
+        type: 'output',
+        text: e.data.toString(),
+      });
+    });
+
+    terminalClient.onExit((e) => {
+      const isError = e.code !== 0;
+      def.resolve({
+        isError,
+        content: result,
+      });
+
+      terminalClient.term.writeln(
+        `\n${color.italic}> Command ${args.command} executed successfully. Terminal will close in ${
+          3000 / 1000
+        } seconds.${color.reset}\n`,
+      );
+
+      setTimeout(() => {
+        terminalClient.dispose();
+        this.terminalView.removeWidget(terminalClient.id);
+      }, 3000);
+    });
+
+    return def.promise;
+  }
+}

package/src/browser/preferences/schema.ts

@@ -58,6 +58,66 @@ export const aiNativePreferenceSchema: PreferenceSchema = {
       type: 'boolean',
       default: false,
     },
+    [AINativeSettingSectionsId.LLMModelSelection]: {
+      type: 'string',
+      default: 'deepseek',
+      enum: ['deepseek', 'anthropic', 'openai'],
+      description: localize('preference.ai.native.llm.model.selection.description'),
+    },
+    [AINativeSettingSectionsId.DeepseekApiKey]: {
+      type: 'string',
+      default: '',
+      description: localize('preference.ai.native.deepseek.apiKey.description'),
+    },
+    [AINativeSettingSectionsId.AnthropicApiKey]: {
+      type: 'string',
+      default: '',
+      description: localize('preference.ai.native.anthropic.apiKey.description'),
+    },
+    [AINativeSettingSectionsId.OpenaiApiKey]: {
+      type: 'string',
+      default: '',
+      description: localize('preference.ai.native.openai.apiKey.description'),
+    },
+    [AINativeSettingSectionsId.OpenaiBaseURL]: {
+      type: 'string',
+      default: '',
+      description: localize('preference.ai.native.openai.baseURL.description'),
+    },
+    [AINativeSettingSectionsId.MCPServers]: {
+      type: 'array',
+      default: [],
+      description: localize('preference.ai.native.mcp.servers.description'),
+      items: {
+        type: 'object',
+        required: ['name', 'command', 'args'],
+        properties: {
+          name: {
+            type: 'string',
+            description: localize('preference.ai.native.mcp.servers.name.description'),
+          },
+          command: {
+            type: 'string',
+            description: localize('preference.ai.native.mcp.servers.command.description'),
+          },
+          args: {
+            type: 'array',
+            items: {
+              type: 'string',
+            },
+            description: localize('preference.ai.native.mcp.servers.args.description'),
+          },
+          env: {
+            type: 'object',
+            additionalProperties: {
+              type: 'string',
+            },
+            description: localize('preference.ai.native.mcp.servers.env.description'),
+            default: {},
+          },
+        },
+      },
+    },
     [AINativeSettingSectionsId.CodeEditsTyping]: {
       type: 'boolean',
       default: false,

package/src/browser/types.ts

@@ -26,6 +26,7 @@ import { SumiReadableStream } from '@opensumi/ide-utils/lib/stream';
 import { IMarker } from '@opensumi/monaco-editor-core/esm/vs/platform/markers/common/markers';
 
 import { IChatWelcomeMessageContent, ISampleQuestions, ITerminalCommandSuggestionDesc } from '../common';
+import { SerializedContext } from '../common/llm-context';
 
 import {
   ICodeEditsContextBean,
@@ -325,6 +326,51 @@ export interface AINativeCoreContribution {
    * proposed api
    */
   registerIntelligentCompletionFeature?(registry: IIntelligentCompletionsRegistry): void;
+
+  /**
+   * 注册 Agent 模式下的 chat prompt provider
+   * @param provider
+   */
+  registerChatAgentPromptProvider?(): void;
+}
+
+// MCP Server 的 贡献点
+export const MCPServerContribution = Symbol('MCPServerContribution');
+
+export const TokenMCPServerRegistry = Symbol('TokenMCPServerRegistry');
+
+export interface MCPServerContribution {
+  registerMCPServer(registry: IMCPServerRegistry): void;
+}
+
+export interface MCPLogger {
+  appendLine(message: string): void;
+}
+
+export interface MCPToolDefinition {
+  name: string;
+  description: string;
+  inputSchema: any; // JSON Schema
+  handler: (
+    args: any,
+    logger: MCPLogger,
+  ) => Promise<{
+    content: { type: string; text: string }[];
+    isError?: boolean;
+  }>;
+}
+
+export interface IMCPServerRegistry {
+  registerMCPTool(tool: MCPToolDefinition): void;
+  getMCPTools(): MCPToolDefinition[];
+  callMCPTool(
+    name: string,
+    args: any,
+  ): Promise<{
+    content: { type: string; text: string }[];
+    isError?: boolean;
+  }>;
+  // 后续支持其他 MCP 功能
 }
 
 export interface IChatComponentConfig {
@@ -359,3 +405,13 @@ export interface IAIMiddleware {
     provideInlineCompletions?: IProvideInlineCompletionsSignature;
   };
 }
+
+export const ChatAgentPromptProvider = Symbol('ChatAgentPromptProvider');
+
+export interface ChatAgentPromptProvider {
+  /**
+   * 提供上下文提示
+   * @param context 上下文
+   */
+  provideContextPrompt(context: SerializedContext, userMessage: string): MaybePromise<string>;
+}

package/src/common/index.ts

@@ -16,6 +16,9 @@ import { IChatMessage } from '@opensumi/ide-core-common/lib/types/ai-native';
 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 { MCPTool } from './types';
+
 export const IAINativeService = Symbol('IAINativeService');
 
 /**
@@ -116,6 +119,17 @@ export const IChatAgentService = Symbol('IChatAgentService');
 
 export const ChatProxyServiceToken = Symbol('ChatProxyServiceToken');
 
+// 暴露给 Node.js 层，使其可以感知 Opensumi 注册的 MCP 能力
+export const TokenMCPServerProxyService = Symbol('TokenMCPServerProxyService');
+
+export interface ISumiMCPServerBackend {
+  initBuiltinMCPServer(): void;
+  initExternalMCPServers(servers: MCPServerDescription[]): void;
+  getAllMCPTools(): Promise<MCPTool[]>;
+}
+
+export const SumiMCPServerProxyServicePath = 'SumiMCPServerProxyServicePath';
+
 export interface IChatAgentService {
   readonly onDidChangeAgents: Event<void>;
   readonly onDidSendMessage: Event<IChatProgress>;

package/src/common/llm-context.ts

@@ -0,0 +1,41 @@
+import { Event, URI } from '@opensumi/ide-core-common/lib/utils';
+
+export interface LLMContextService {
+  startAutoCollection(): void;
+
+  stopAutoCollection(): void;
+
+  /**
+   * 添加文件到 context 中
+   */
+  addFileToContext(uri: URI, selection?: [number, number], isManual?: boolean): void;
+
+  /**
+   * 清除上下文
+   */
+  cleanFileContext(): void;
+
+  onDidContextFilesChangeEvent: Event<FileContext[]>;
+
+  /**
+   * 从 context 中移除文件
+   * @param uri URI
+   */
+  removeFileFromContext(uri: URI): void;
+
+  /** 导出为可序列化格式 */
+  serialize(): SerializedContext;
+}
+
+export interface FileContext {
+  uri: URI;
+  selection?: [number, number];
+  isManual: boolean;
+}
+
+export const LLMContextServiceToken = Symbol('LLMContextService');
+
+export interface SerializedContext {
+  recentlyViewFiles: string[];
+  attachedFiles: Array<{ content: string; lineErrors: string[]; path: string; language: string }>;
+}

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

@@ -0,0 +1,46 @@
+import { Client } from '@modelcontextprotocol/sdk/client/index.js';
+
+export interface MCPServerManager {
+    callTool(serverName: string, toolName: string, arg_string: string): ReturnType<Client['callTool']>;
+    removeServer(name: string): void;
+    addOrUpdateServer(description: MCPServerDescription): void;
+    // invoke in node.js only
+    addOrUpdateServerDirectly(server: any): void;
+    initBuiltinServer(builtinMCPServer: any): void;
+    getTools(serverName: string): ReturnType<Client['listTools']>;
+    getServerNames(): Promise<string[]>;
+    startServer(serverName: string): Promise<void>;
+    stopServer(serverName: string): Promise<void>;
+    getStartedServers(): Promise<string[]>;
+    registerTools(serverName: string): Promise<void>;
+    addExternalMCPServers(servers: MCPServerDescription[]): void;
+}
+
+export type MCPTool = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number];
+
+export type MCPToolParameter = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number]['inputSchema'];
+
+export interface MCPServerDescription {
+  /**
+   * The unique name of the MCP server.
+   */
+  name: string;
+
+  /**
+   * The command to execute the MCP server.
+   */
+  command: string;
+
+  /**
+   * An array of arguments to pass to the command.
+   */
+  args?: string[];
+
+  /**
+   * Optional environment variables to set when starting the server.
+   */
+  env?: { [key: string]: string };
+}
+
+export const MCPServerManager = Symbol('MCPServerManager');
+export const MCPServerManagerPath = 'ServicesMCPServerManager';

package/src/common/tool-invocation-registry.ts

@@ -0,0 +1,170 @@
+import { z } from 'zod';
+
+import { Injectable } from '@opensumi/di';
+
+import { MCPToolParameter } from './mcp-server-manager';
+
+export const ToolParameterSchema = z.object({
+    type: z.enum(['string', 'number', 'boolean', 'object', 'array']),
+    description: z.string().optional(),
+    enum: z.array(z.any()).optional(),
+    items: z.lazy(() => ToolParameterSchema).optional(),
+    properties: z.record(z.lazy(() => ToolParameterSchema)).optional(),
+    required: z.array(z.string()).optional(),
+});
+
+export type ToolParameter = z.infer<typeof ToolParameterSchema>;
+
+export interface ToolRequest {
+    id: string;
+    name: string;
+    parameters?: any;
+    description?: string;
+    handler: (arg_string: string) => Promise<any>;
+    providerName?: string;
+}
+
+export namespace ToolRequest {
+    export function isToolParameter(obj: unknown): obj is ToolParameter {
+        return ToolParameterSchema.safeParse(obj).success;
+    }
+}
+
+export const ToolInvocationRegistry = Symbol('ToolInvocationRegistry');
+
+/**
+ * 为 Agent 提供的所有可用函数调用的注册表
+ */
+export interface ToolInvocationRegistry {
+    /**
+     * 在注册表中注册一个工具
+     *
+     * @param tool - 要注册的 `ToolRequest` 对象
+     */
+    registerTool(tool: ToolRequest): void;
+
+    /**
+     * 从注册表中获取特定的 `ToolRequest`
+     *
+     * @param toolId - 要获取的工具的唯一标识符
+     * @returns 对应提供的工具 ID 的 `ToolRequest` 对象，
+     *          如果在注册表中找不到该工具，则返回 `undefined`
+     */
+    getFunction(toolId: string): ToolRequest | undefined;
+
+    /**
+     * 从注册表中获取多个 `ToolRequest`
+     *
+     * @param toolIds - 要获取的工具 ID 列表
+     * @returns 指定工具 ID 的 `ToolRequest` 对象数组
+     *          如果找不到某个工具 ID，将在返回的数组中跳过该工具
+     */
+    getFunctions(...toolIds: string[]): ToolRequest[];
+
+    /**
+     * 获取当前注册表中的所有 `ToolRequest`
+     *
+     * @returns 注册表中所有 `ToolRequest` 对象的数组
+     */
+    getAllFunctions(): ToolRequest[];
+
+    /**
+     * 注销特定工具提供者的所有工具
+     *
+     * @param providerName - 要移除其工具的工具提供者名称（在 `ToolRequest` 中指定）
+     */
+    unregisterAllTools(providerName: string): void;
+}
+
+export const ToolProvider = Symbol('ToolProvider');
+export interface ToolProvider {
+    getTool(): ToolRequest;
+}
+
+export class ToolInvocationRegistryImpl implements ToolInvocationRegistry {
+    private tools: Map<string, ToolRequest> = new Map<string, ToolRequest>();
+
+    unregisterAllTools(providerName: string): void {
+        const toolsToRemove: string[] = [];
+        for (const [id, tool] of this.tools.entries()) {
+            if (tool.providerName === providerName) {
+                toolsToRemove.push(id);
+            }
+        }
+        toolsToRemove.forEach((id) => this.tools.delete(id));
+    }
+
+    getAllFunctions(): ToolRequest[] {
+        return Array.from(this.tools.values());
+    }
+
+    registerTool(tool: ToolRequest): void {
+        if (this.tools.has(tool.id)) {
+            // TODO: 使用适当的日志机制
+            this.tools.set(tool.id, tool);
+        } else {
+            this.tools.set(tool.id, tool);
+        }
+    }
+
+    getFunction(toolId: string): ToolRequest | undefined {
+        return this.tools.get(toolId);
+    }
+
+    getFunctions(...toolIds: string[]): ToolRequest[] {
+        const tools: ToolRequest[] = toolIds.map((toolId) => {
+            const tool = this.tools.get(toolId);
+            if (tool) {
+                return tool;
+            } else {
+                throw new Error(`找不到 ID 为 ${toolId} 的函数`);
+            }
+        });
+        return tools;
+    }
+}
+
+/**
+ * 管理多个 ToolInvocationRegistry 实例的管理器，每个实例与一个 clientId 关联
+ */
+export interface IToolInvocationRegistryManager {
+    /**
+     * 获取或创建特定 clientId 的 ToolInvocationRegistry
+     */
+    getRegistry(clientId: string): ToolInvocationRegistry;
+
+    /**
+     * 移除特定 clientId 的 ToolInvocationRegistry
+     */
+    removeRegistry(clientId: string): void;
+
+    /**
+     * 检查特定 clientId 是否存在对应的注册表
+     */
+    hasRegistry(clientId: string): boolean;
+}
+
+export const ToolInvocationRegistryManager = Symbol('ToolInvocationRegistryManager');
+
+@Injectable()
+export class ToolInvocationRegistryManagerImpl implements IToolInvocationRegistryManager {
+    private registries: Map<string, ToolInvocationRegistry> = new Map();
+
+    getRegistry(clientId: string): ToolInvocationRegistry {
+        let registry = this.registries.get(clientId);
+        if (!registry) {
+            registry = new ToolInvocationRegistryImpl();
+            this.registries.set(clientId, registry);
+        }
+        return registry;
+    }
+
+    removeRegistry(clientId: string): void {
+        this.registries.delete(clientId);
+    }
+
+    hasRegistry(clientId: string): boolean {
+        return this.registries.has(clientId);
+    }
+}
+

package/src/common/types.ts

@@ -18,3 +18,25 @@ export interface INearestCodeBlock {
   offset: number;
   type?: NearestCodeBlockType;
 }
+
+// SUMI MCP Server 网页部分暴露给 Node.js 部分的能力
+export interface IMCPServerProxyService {
+  $callMCPTool(
+    name: string,
+    args: any,
+  ): Promise<{
+    content: { type: string; text: string }[];
+    isError?: boolean;
+  }>;
+  // 获取 browser 层注册的 MCP 工具列表 (Browser tab 维度)
+  $getMCPTools(): Promise<MCPTool[]>;
+  // 通知前端 MCP 服务注册表发生了变化
+  $updateMCPServers(): Promise<void>;
+}
+
+export interface MCPTool {
+  name: string;
+  description: string;
+  inputSchema: any;
+  providerName: string;
+}

package/src/node/anthropic/anthropic-language-model.ts

@@ -0,0 +1,25 @@
+import { AnthropicProvider, createAnthropic } from '@ai-sdk/anthropic';
+
+import { Injectable } from '@opensumi/di';
+import { IAIBackServiceOption } from '@opensumi/ide-core-common';
+import { AINativeSettingSectionsId } from '@opensumi/ide-core-common/lib/settings/ai-native';
+
+import { BaseLanguageModel } from '../base-language-model';
+
+export const AnthropicModelIdentifier = Symbol('AnthropicModelIdentifier');
+
+@Injectable()
+export class AnthropicModel extends BaseLanguageModel {
+  protected initializeProvider(options: IAIBackServiceOption): AnthropicProvider {
+    const apiKey = options.apiKey;
+    if (!apiKey) {
+      throw new Error(`Please provide Anthropic API Key in preferences (${AINativeSettingSectionsId.AnthropicApiKey})`);
+    }
+
+    return createAnthropic({ apiKey });
+  }
+
+  protected getModelIdentifier(provider: AnthropicProvider) {
+    return provider('claude-3-5-sonnet-20241022');
+  }
+}