@opensumi/ide-ai-native 3.8.1-next-1740571693.0 → 3.8.1-next-1740725107.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,6 +154,7 @@
   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,6 +27,7 @@ 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,6 +29,7 @@ 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,6 +49,7 @@ 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);
@@ -89,6 +90,7 @@ 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,6 +32,7 @@ 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/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/llm-context.ts

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

package/src/common/prompts/context-prompt-provider.ts

@@ -1,5 +1,5 @@
-import { Injectable } from '@opensumi/di';
-import { MaybePromise } from '@opensumi/ide-core-common/lib/utils';
+import { Autowired, Injectable } from '@opensumi/di';
+import { WorkbenchEditorService } from '@opensumi/ide-editor/lib/browser/types';
 
 import { SerializedContext } from '../llm-context';
 
@@ -10,37 +10,47 @@ export interface ChatAgentPromptProvider {
    * 提供上下文提示
    * @param context 上下文
    */
-  provideContextPrompt(context: SerializedContext, userMessage: string): MaybePromise<string>;
+  provideContextPrompt(context: SerializedContext, userMessage: string): string;
 }
 
 @Injectable()
 export class DefaultChatAgentPromptProvider implements ChatAgentPromptProvider {
-  provideContextPrompt(context: SerializedContext, userMessage: string): MaybePromise<string> {
+  @Autowired(WorkbenchEditorService)
+  protected readonly workbenchEditorService: WorkbenchEditorService;
+
+  provideContextPrompt(context: SerializedContext, userMessage: string): string {
+    const editor = this.workbenchEditorService.currentEditor;
+    const currentModel = editor?.currentDocumentModel;
+    currentModel
     return `
-          <additional_data>
-          Below are some potentially helpful/relevant pieces of information for figuring out to respond
-          <recently_viewed_files>
-          ${context.recentlyViewFiles.map((file, idx) => `${idx + 1} : ${file}`)}
-          </recently_viewed_files>
-          <attached_files>
-          ${context.attachedFiles.map(
-            (file) =>
-              `
-          <file_contents>
-          \`\`\`${file.language} ${file.path}
-          ${file.content}
-          \`\`\`
-          </file_contents>
-          <linter_errors>
-          ${file.lineErrors.join('\n')}
-          </linter_errors>
-          `,
-          )}
-          
-          </attached_files>
-          </additional_data>
-          <user_query>
-          ${userMessage}
-          </user_query>`;
+<additional_data>
+  Below are some potentially helpful/relevant pieces of information for figuring out to respond
+  <recently_viewed_files>
+${context.recentlyViewFiles.map((file, idx) => `    ${idx + 1}: ${file}`).join('\n')}
+  </recently_viewed_files>
+  <attached_files>
+    ${context.attachedFiles.map(
+      (file) =>
+        `
+    <file_contents>
+    \`\`\`${file.language} ${file.path}
+    ${file.content}
+    \`\`\`
+    </file_contents>
+    <linter_errors>
+    ${file.lineErrors.join('\n')}
+    </linter_errors>
+              `,
+    )}
+  </attached_files>
+${currentModel ? `<current_opened_file>
+  \`\`\`${currentModel.languageId} ${currentModel.uri.toString()}
+${currentModel.getText()}
+  \`\`\`
+  </current_opened_file>` : ''}
+</additional_data>
+<user_query>
+${userMessage}
+</user_query>`;
   }
 }

package/src/common/types.ts

@@ -1,3 +1,5 @@
+import { IMarker } from '@opensumi/ide-core-browser';
+
 export enum NearestCodeBlockType {
   Block = 'block',
   Line = 'line',
@@ -46,3 +48,21 @@ export interface MCPTool {
   inputSchema: any;
   providerName: string;
 }
+
+export interface CodeBlockData {
+  toolCallId: string;
+  codeEdit: string;
+  updatedCode?: string;
+  relativePath: string;
+  status: CodeBlockStatus;
+  iterationCount: number;
+  createdAt: number;
+  version: number;
+  instructions?: string;
+  applyResult?: {
+    diff: string;
+    diagnosticInfos: IMarker[];
+  };
+}
+
+export type CodeBlockStatus = 'generating' | 'pending' | 'success' | 'rejected' | 'failed' | 'cancelled';