@opensumi/ide-ai-native 3.8.1-next-1740475512.0 → 3.8.1-next-1740556231.0

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

@@ -1,5 +1,5 @@
 import cls from 'classnames';
-import React, { useEffect, useMemo } from 'react';
+import React, { useEffect, useMemo, useState } from 'react';
 
 import { Icon, Popover } from '@opensumi/ide-components';
 import {
@@ -10,7 +10,6 @@ import {
   Uri,
   detectModeId,
   path,
-  useAutorun,
   useInjectable,
 } from '@opensumi/ide-core-browser';
 import { Loading } from '@opensumi/ide-core-browser/lib/components/ai-native';
@@ -18,61 +17,24 @@ import { ILanguageService } from '@opensumi/monaco-editor-core/esm/vs/editor/com
 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 { CodeBlockData } from '../../../../common/types';
 import { ChatMarkdown } from '../../../components/ChatMarkdown';
 import { IMCPServerToolComponentProps } from '../../../types';
-import { BaseApplyService, CodeBlockData } from '../../base-apply.service';
+import { BaseApplyService } 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 [mode, setMode] = useState<'code' | 'diff'>('code');
   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 [codeBlockData, setCodeBlockData] = useState<CodeBlockData | undefined>(
+    applyService.getCodeBlock(toolCallId, messageId),
+  );
 
   const icon = useMemo(() => {
     if (!target_file) {
@@ -91,40 +53,66 @@ export const EditFileToolComponent = (props: IMCPServerToolComponentProps) => {
     return detectedModeId;
   }, [target_file, absolutePath]);
 
+  useEffect(() => {
+    const disposable = applyService.onCodeBlockUpdate((codeBlockData) => {
+      setCodeBlockData({ ...codeBlockData });
+    });
+    return () => {
+      disposable.dispose();
+    };
+  }, []);
+
   // 多次迭代时，仅在首处tool组件中展示
-  if (!args || !codeBlockData || (toolCallId && toolCallId !== codeBlockData.initToolCallId)) {
+  // FIXME: 这个优化有必要吗？每次都展示也挺好？
+  if (!args || !codeBlockData) {
     return null;
   }
 
   return [
     instructions && <p>{instructions}</p>,
-    <div className={styles['edit-file-tool']} key={`edit-file-tool-${codeBlockData.id}`}>
+    <div className={styles['edit-file-tool']} key={'edit-file-tool'}>
       <div
         className={cls(styles['edit-file-tool-header'], {
           clickable: codeBlockData.status === 'pending' || codeBlockData.status === 'success',
         })}
         onClick={() => {
           if (codeBlockData.status === 'pending') {
-            applyService.reRenderPendingApply();
+            applyService.renderApplyResult(codeBlockData, codeBlockData.updatedCode!);
           } else if (codeBlockData.status === 'success') {
-            applyService.revealApplyPosition(codeBlockData.id);
+            applyService.revealApplyPosition(codeBlockData);
           }
         }}
       >
-        {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 className={styles.left}>
+          {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>
+        <div className={styles.right}>
+          <Popover title={'Show Code'} id={'edit-file-tool-show-code'}>
+            <Icon iconClass='codicon codicon-file-code' onClick={() => setMode('code')} />
+          </Popover>
+          {codeBlockData.applyResult?.diff && (
+            <Popover title={'Show Diff'} id={'edit-file-tool-show-diff'}>
+              <Icon iconClass='codicon codicon-diff-multiple' onClick={() => setMode('diff')} />
+            </Popover>
+          )}
+        </div>
       </div>
-      <ChatMarkdown markdown={`\`\`\`${languageId || ''}\n${code_edit}\n\`\`\``} hideInsert={true} />
+      <ChatMarkdown
+        markdown={
+          mode === 'code'
+            ? `\`\`\`${languageId || ''}\n${code_edit}\n\`\`\``
+            : `\`\`\`diff\n${codeBlockData.applyResult?.diff}\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['edit-file-tool-diagnostic-errors']} key={'edit-file-tool-diagnostic-errors'}>
         <div className={styles['title']}>Found Lints:</div>
         {codeBlockData.applyResult?.diagnosticInfos.map((info) => (
           <div
@@ -142,3 +130,37 @@ export const EditFileToolComponent = (props: IMCPServerToolComponentProps) => {
     ),
   ];
 };
+
+const renderStatus = (codeBlockData: CodeBlockData) => {
+  const status = codeBlockData.status;
+  switch (status) {
+    case 'generating':
+      return <Loading />;
+    case 'pending':
+      return (
+        <Popover title='Pending' id={'edit-file-tool-status-pending'}>
+          <Icon iconClass='codicon codicon-circle-large' />
+        </Popover>
+      );
+    case 'success':
+      return (
+        <Popover title='Success' id={'edit-file-tool-status-success'}>
+          <Icon iconClass='codicon codicon-check-all' />
+        </Popover>
+      );
+    case 'failed':
+      return (
+        <Popover title='Failed' id={'edit-file-tool-status-failed'}>
+          <Icon iconClass='codicon codicon-error' style={{ color: 'var(--debugConsole-errorForeground)' }} />
+        </Popover>
+      );
+    case 'cancelled':
+      return (
+        <Popover title='Cancelled' id={'edit-file-tool-status-cancelled'}>
+          <Icon iconClass='codicon codicon-close' style={{ color: 'var(--input-placeholderForeground)' }} />
+        </Popover>
+      );
+    default:
+      return null;
+  }
+};

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

@@ -1,15 +1,14 @@
 .edit-file-tool-header {
   display: flex;
   align-items: center;
+  justify-content: space-between;
   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;
-  }
+  white-space: nowrap;
   :global(span.codicon) {
     font-size: 12px;
   }
@@ -18,6 +17,26 @@
     align-items: center;
     justify-content: center;
   }
+  .left,
+  .right {
+    display: flex;
+    align-items: center;
+  }
+  &::after {
+    display: none;
+  }
+  .left {
+    > span {
+      margin-right: 4px;
+    }
+  }
+  .right > div {
+    margin-left: 4px;
+  }
+}
+.edit-file-tool-file-name {
+  text-overflow: ellipsis;
+  overflow: hidden;
 }
 .edit-file-tool {
   border: 1px solid var(--vscode-commandCenter-inactiveBorder);
@@ -135,7 +154,6 @@
   background-color: var(--design-chatInput-background);
   padding: 10px;
   border-radius: 4px;
-  margin: 10px 0px;
 
   .command_title {
     display: flex;

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

@@ -27,7 +27,6 @@ export class CreateNewFileWithTextTool implements MCPServerContribution {
   getToolDefinition(): MCPToolDefinition {
     return {
       name: 'create_new_file_with_text',
-      label: 'Create File',
       description:
         'Creates a new file at the specified path within the project directory and populates it with the provided text. ' +
         'Use this tool to generate new files in your project structure. ' +

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

@@ -69,8 +69,8 @@ You should specify the following arguments before the others: [target_file]`,
     };
   }
 
-  private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
-    const result = await this.editFileHandler.handler(args.targetFile, args.codeEdit, args.instructions);
+  private async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) {
+    const result = await this.editFileHandler.handler(args, args.toolCallId);
     return {
       content: [
         {

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

@@ -29,7 +29,6 @@ export class GetDiagnosticsByPathTool implements MCPServerContribution {
   getToolDefinition(): MCPToolDefinition {
     return {
       name: 'get_diagnostics_by_path',
-      label: 'Get Diagnostics',
       description:
         'Retrieves diagnostic information (errors, warnings, etc.) from a specific file in the project. ' +
         'Use this tool to get information about problems in any project file. ' +

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

@@ -12,10 +12,10 @@ 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);
+  async handler(params: { targetFile: string; codeEdit: string; instructions?: string }, toolCallId: string) {
+    const { targetFile, codeEdit } = params;
+    const block = this.applyService.registerCodeBlock(targetFile, codeEdit, toolCallId);
+    const blockData = await this.applyService.apply(block);
     return blockData;
   }
 }

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

@@ -49,7 +49,6 @@ export class RunCommandHandler {
   }
 
   async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) {
-    logger.appendLine(`Executing command: ${args.command}`);
     if (args.require_user_approval) {
       const def = new Deferred<boolean>();
       this.approvalDeferredMap.set(args.toolCallId, def);
@@ -90,7 +89,6 @@ export class RunCommandHandler {
         content: result,
       });
 
-      logger.appendLine(`Command ${args.command} finished with exit code: ${e.code}`);
       terminalClient.term.writeln(
         `\n${color.italic}> Command ${args.command} executed successfully. Terminal will close in ${
           3000 / 1000

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

@@ -32,7 +32,6 @@ export class RunTerminalCommandTool implements MCPServerContribution {
   getToolDefinition(): MCPToolDefinition {
     return {
       name: 'run_terminal_cmd',
-      label: 'Run Command',
       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,

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

@@ -66,6 +66,11 @@ export class MsgHistoryManager extends Disposable {
     return this.startIndex;
   }
 
+  public get lastMessageId(): string | undefined {
+    const list = this.messageList;
+    return list[list.length - 1]?.id;
+  }
+
   public getMessages(maxTokens?: number): IHistoryChatMessage[] {
     if (maxTokens && this.totalTokens > maxTokens) {
       while (this.totalTokens > maxTokens) {
@@ -109,9 +114,14 @@ export class MsgHistoryManager extends Disposable {
       return;
     }
 
-    this.messageAdditionalMap.set(id, additional);
+    const oldAdditional = this.messageAdditionalMap.get(id) || {};
+    const newAdditional = {
+      ...oldAdditional,
+      ...additional,
+    };
 
-    this._onMessageAdditionalChange.fire(additional);
+    this.messageAdditionalMap.set(id, newAdditional);
+    this._onMessageAdditionalChange.fire(newAdditional);
   }
 
   public getMessageAdditional(id: string): Record<string, any> {

package/src/browser/preferences/schema.ts

@@ -108,27 +108,11 @@ export const aiNativePreferenceSchema: PreferenceSchema = {
             type: 'string',
             description: localize('preference.ai.native.mcp.servers.command.description'),
           },
-          type: {
-            type: 'string',
-            enum: ['stdio', 'sse'],
-            enumDescriptions: [
-              localize('preference.ai.native.mcp.servers.type.stdio'),
-              localize('preference.ai.native.mcp.servers.type.sse'),
-            ],
-            description: localize('preference.ai.native.mcp.servers.type.description'),
-            default: 'stdio',
-          },
-          enabled: {
-            type: 'boolean',
-            description: localize('preference.ai.native.mcp.servers.enabled.description'),
-            default: true,
-          },
           args: {
             type: 'array',
             items: {
               type: 'string',
             },
-            default: [],
             description: localize('preference.ai.native.mcp.servers.args.description'),
           },
           env: {

package/src/browser/types.ts

@@ -368,7 +368,7 @@ export interface IMCPServerToolComponentProps {
   result?: any;
   index?: number;
   messageId?: string;
-  toolCallId?: string;
+  toolCallId: string;
 }
 
 export interface IMCPServerRegistry {

package/src/browser/widget/inline-diff/inline-diff-manager.tsx

@@ -0,0 +1,38 @@
+import React, { useEffect, useState } from 'react';
+
+import { Button } from '@opensumi/ide-components';
+import { localize, useInjectable } from '@opensumi/ide-core-browser';
+import { IResource } from '@opensumi/ide-editor';
+
+import { BaseApplyService } from '../../mcp/base-apply.service';
+
+import styles from './inline-diff-widget.module.less';
+
+export const InlineDiffManager: React.FC<{ resource: IResource }> = (props) => {
+  const applyService = useInjectable<BaseApplyService>(BaseApplyService);
+  const [show, setShow] = useState(true);
+  useEffect(() => {
+    applyService.onCodeBlockUpdate((codeBlock) => {
+      setShow(codeBlock.status === 'pending');
+    });
+  }, []);
+  return (
+    <div className={styles.inlineDiffManager} style={{ display: show ? 'flex' : 'none' }}>
+      <Button
+        onClick={() => {
+          applyService.processAll(props.resource.uri, 'accept');
+        }}
+      >
+        {localize('aiNative.inlineDiff.acceptAll')}
+      </Button>
+      <Button
+        type='ghost'
+        onClick={() => {
+          applyService.processAll(props.resource.uri, 'reject');
+        }}
+      >
+        {localize('aiNative.inlineDiff.rejectAll')}
+      </Button>
+    </div>
+  );
+};

package/src/browser/widget/inline-diff/inline-diff-widget.module.less

@@ -19,3 +19,15 @@
   display: flex;
   position: relative;
 }
+
+.inlineDiffManager {
+  display: flex;
+  padding: 12px 16px;
+  justify-content: center;
+  position: absolute;
+  bottom: 0;
+  left: 50%;
+  transform: translateX(-50%);
+  gap: 12px;
+  z-index: 999;
+}

package/src/browser/widget/inline-stream-diff/inline-stream-diff.handler.tsx

@@ -52,6 +52,9 @@ export class InlineStreamDiffHandler extends Disposable implements IInlineDiffPr
   protected readonly _onDidEditChange = this.registerDispose(new Emitter<void>());
   public readonly onDidEditChange: Event<void> = this._onDidEditChange.event;
 
+  protected readonly onDiffFinishedEmitter = this.registerDispose(new Emitter<IComputeDiffData>());
+  public readonly onDiffFinished: Event<IComputeDiffData> = this.onDiffFinishedEmitter.event;
+
   public previewerOptions: IDiffPreviewerOptions;
 
   private originalModel: ITextModel;
@@ -445,6 +448,8 @@ export class InlineStreamDiffHandler extends Disposable implements IInlineDiffPr
           this.diffModel.set(currentDiffModel, tx);
         });
 
+        this.onDiffFinishedEmitter.fire(currentDiffModel);
+
         if (this.originalModel.id === this.monacoEditor.getModel()?.id) {
           this.renderDiffEdits(currentDiffModel);
         }
@@ -471,10 +476,13 @@ export class InlineStreamDiffHandler extends Disposable implements IInlineDiffPr
   }
 
   public pushRateFinallyDiffStack(diffModel: IComputeDiffData): void {
-    // 可能存在 rate editor controller 处理完之后接口层流式才结束
-    if (this.isEditing === false) {
-      this.finallyRender(diffModel);
-    }
+    transaction((tx) => {
+      this.finallyDiffModel.set(diffModel, tx);
+      // 可能存在 rate editor controller 处理完之后接口层流式才结束
+      if (this.isEditing === false) {
+        this.finallyRender(diffModel);
+      }
+    });
   }
 
   public finallyRender(diffModel: IComputeDiffData): void {
@@ -487,6 +495,7 @@ export class InlineStreamDiffHandler extends Disposable implements IInlineDiffPr
       return;
     }
 
+    this.onDiffFinishedEmitter.fire(diffModel);
     this.renderPartialEditWidgets(diffModel);
     this.renderDiffEdits(diffModel);
     this.pushStackElement();

package/src/common/index.ts

@@ -31,9 +31,6 @@ export const AI_CHAT_CONTAINER_ID = 'AI-Chat-Container';
 export const AI_CHAT_LOGO_AVATAR_ID = 'AI-Chat-Logo-Avatar';
 export const AI_MENU_BAR_DEBUG_TOOLBAR = 'AI_MENU_BAR_DEBUG_TOOLBAR';
 
-// 内置 MCP 服务器名称
-export const BUILTIN_MCP_SERVER_NAME = 'sumi-builtin';
-
 /**
  * @deprecated Use {@link DESIGN_MENUBAR_CONTAINER_VIEW_ID} instead
  */
@@ -126,12 +123,9 @@ export const ChatProxyServiceToken = Symbol('ChatProxyServiceToken');
 export const TokenMCPServerProxyService = Symbol('TokenMCPServerProxyService');
 
 export interface ISumiMCPServerBackend {
-  initBuiltinMCPServer(enabled: boolean): void;
+  initBuiltinMCPServer(): void;
   initExternalMCPServers(servers: MCPServerDescription[]): void;
   getAllMCPTools(): Promise<MCPTool[]>;
-  getServers(): Promise<Array<{ name: string; isStarted: boolean }>>;
-  startServer(serverName: string): Promise<void>;
-  stopServer(serverName: string): Promise<void>;
 }
 
 export const SumiMCPServerProxyServicePath = 'SumiMCPServerProxyServicePath';

package/src/common/llm-context.ts

@@ -15,13 +15,13 @@ export interface LLMContextService {
    */
   cleanFileContext(): void;
 
-  onDidContextFilesChangeEvent: Event<{ viewed: FileContext[]; attached: FileContext[] }>;
+  onDidContextFilesChangeEvent: Event<FileContext[]>;
 
   /**
    * 从 context 中移除文件
    * @param uri URI
    */
-  removeFileFromContext(uri: URI, isManual?: boolean): void;
+  removeFileFromContext(uri: URI): void;
 
   /** 导出为可序列化格式 */
   serialize(): SerializedContext;
@@ -30,18 +30,12 @@ export interface LLMContextService {
 export interface FileContext {
   uri: URI;
   selection?: [number, number];
+  isManual: boolean;
 }
 
 export const LLMContextServiceToken = Symbol('LLMContextService');
 
-export interface AttachFileContext {
-  content: string;
-  lineErrors: string[];
-  path: string;
-  language: string;
-}
-
 export interface SerializedContext {
   recentlyViewFiles: string[];
-  attachedFiles: Array<AttachFileContext>;
+  attachedFiles: Array<{ content: string; lineErrors: string[]; path: string; language: string }>;
 }

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

@@ -1,15 +1,5 @@
 import { Client } from '@modelcontextprotocol/sdk/client/index.js';
 
-export interface IMCPServer {
-  isStarted(): boolean;
-  start(): Promise<void>;
-  getServerName(): string;
-  callTool(toolName: string, toolCallId: string, arg_string: string): ReturnType<Client['callTool']>;
-  getTools(): ReturnType<Client['listTools']>;
-  update(command: string, args?: string[], env?: { [key: string]: string }): void;
-  stop(): void;
-}
-
 export interface MCPServerManager {
   callTool(
     serverName: string,
@@ -21,7 +11,7 @@ export interface MCPServerManager {
   addOrUpdateServer(description: MCPServerDescription): void;
   // invoke in node.js only
   addOrUpdateServerDirectly(server: any): void;
-  initBuiltinServer(builtinMCPServer: any, enabled: boolean): void;
+  initBuiltinServer(builtinMCPServer: any): void;
   getTools(serverName: string): ReturnType<Client['listTools']>;
   getServerNames(): Promise<string[]>;
   startServer(serverName: string): Promise<void>;
@@ -29,7 +19,6 @@ export interface MCPServerManager {
   getStartedServers(): Promise<string[]>;
   registerTools(serverName: string): Promise<void>;
   addExternalMCPServers(servers: MCPServerDescription[]): void;
-  getServers(): Map<string, IMCPServer>;
 }
 
 export type MCPTool = Awaited<ReturnType<MCPServerManager['getTools']>>['tools'][number];
@@ -56,11 +45,6 @@ export interface MCPServerDescription {
    * Optional environment variables to set when starting the server.
    */
   env?: { [key: string]: string };
-
-  /**
-   * Whether to enable the MCP server.
-   */
-  enabled?: boolean;
 }
 
 export const MCPServerManager = Symbol('MCPServerManager');