@opensumi/ide-ai-native 3.8.1-next-1741250387.0 → 3.8.1-next-1741251026.0

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

@@ -3,8 +3,11 @@ import { z } from 'zod';
 import { Autowired } from '@opensumi/di';
 import { Domain } from '@opensumi/ide-core-common';
 
+import { IChatInternalService } from '../../../common';
+import { ChatInternalService } from '../../chat/chat.internal.service';
 import { IMCPServerRegistry, MCPLogger, MCPServerContribution, MCPToolDefinition } from '../../types';
 
+import { ListDirToolComponent } from './components/ExpandableFileList';
 import { ListDirHandler } from './handlers/ListDir';
 
 const inputSchema = z
@@ -26,8 +29,12 @@ export class ListDirTool implements MCPServerContribution {
   @Autowired(ListDirHandler)
   private readonly listDirHandler: ListDirHandler;
 
+  @Autowired(IChatInternalService)
+  private readonly chatInternalService: ChatInternalService;
+
   registerMCPServer(registry: IMCPServerRegistry): void {
     registry.registerMCPTool(this.getToolDefinition());
+    registry.registerToolComponent('list_dir', ListDirToolComponent);
   }
 
   getToolDefinition(): MCPToolDefinition {
@@ -41,16 +48,40 @@ export class ListDirTool implements MCPServerContribution {
     };
   }
 
-  private async handler(args: z.infer<typeof inputSchema>, logger: MCPLogger) {
+  private async handler(args: z.infer<typeof inputSchema> & { toolCallId: string }, logger: MCPLogger) {
     const result = await this.listDirHandler.handler(args);
+
+    // 构建文件 URI 列表，用于前端渲染
+    const fileUris = result.files.map((file) => {
+      const filePath = `${this.listDirHandler.getWorkspaceDir()}/${result.directoryRelativeWorkspacePath}/${file.name}`;
+      return {
+        uri: filePath,
+        isDirectory: file.isDirectory,
+      };
+    });
+
+    // 设置消息的附加数据
+    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,
+        })),
+      },
+    });
+
+    logger.appendLine(`Listed ${fileUris.length} files in directory "${args.relativeWorkspacePath}"`);
+
     return {
       content: [
         {
           type: 'text',
-          text: `Contents of directory:
-
-
-          ${result.files
+          text: `Contents of directory "${args.relativeWorkspacePath}":\n${result.files
             .map(
               (file) =>
                 `[${file.isDirectory ? 'dir' : 'file'}] ${file.name} ${

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

@@ -57,11 +57,6 @@ export class MsgHistoryManager extends Disposable {
 
   private startIndex = 0;
 
-  private get totalTokens(): number {
-    const list = this.messageList.slice(this.startIndex);
-    return list.reduce((acc, msg) => acc + (msg.content.length || 0), 0) / 3;
-  }
-
   public get slicedMessageCount(): number {
     return this.startIndex;
   }
@@ -71,12 +66,7 @@ export class MsgHistoryManager extends Disposable {
     return list[list.length - 1]?.id;
   }
 
-  public getMessages(maxTokens?: number): IHistoryChatMessage[] {
-    if (maxTokens && this.totalTokens > maxTokens) {
-      while (this.totalTokens > maxTokens) {
-        this.startIndex++;
-      }
-    }
+  public getMessages(): IHistoryChatMessage[] {
     return this.messageList.slice(this.startIndex);
   }
 
@@ -128,6 +118,10 @@ export class MsgHistoryManager extends Disposable {
     return this.messageAdditionalMap.get(id) || {};
   }
 
+  public get sessionAdditionals() {
+    return this.messageAdditionalMap;
+  }
+
   toJSON() {
     return {
       messages: this.getMessages(),

package/src/browser/preferences/schema.ts

@@ -69,9 +69,14 @@ export const aiNativePreferenceSchema: PreferenceSchema = {
     [AINativeSettingSectionsId.LLMModelSelection]: {
       type: 'string',
       default: 'deepseek',
-      enum: ['deepseek', 'anthropic', 'openai'],
+      enum: ['deepseek', 'anthropic', 'openai', 'openai-compatible'],
       description: localize('preference.ai.native.llm.model.selection.description'),
     },
+    [AINativeSettingSectionsId.ModelID]: {
+      type: 'string',
+      default: 'deepseek-chat',
+      description: localize('preference.ai.native.llm.model.id'),
+    },
     [AINativeSettingSectionsId.DeepseekApiKey]: {
       type: 'string',
       default: '',
@@ -108,11 +113,27 @@ 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: {
@@ -132,7 +153,6 @@ export const aiNativePreferenceSchema: PreferenceSchema = {
     },
     [AINativeSettingSectionsId.SystemPrompt]: {
       type: 'string',
-      default: '',
       description: localize('preference.ai.native.chat.system.prompt.description'),
     },
   },

package/src/browser/types.ts

@@ -372,6 +372,7 @@ export interface IMCPServerToolComponentProps {
 }
 
 export interface IMCPServerRegistry {
+  activeMessageInfo: { messageId: string; sessionId: string };
   registerMCPTool(tool: MCPToolDefinition): void;
   getMCPTools(): MCPToolDefinition[];
   getMCPTool(name: string): MCPToolDefinition | undefined;

package/src/browser/widget/inline-chat/inline-chat-controller.ts

@@ -83,7 +83,11 @@ export class InlineChatController {
         this._onData.fire(reply);
       },
       onEnd: () => {
-        this._onEnd.fire();
+        if (!wholeContent) {
+          this._onError.fire(new ErrorResponse(new Error('No content')));
+        } else {
+          this._onEnd.fire();
+        }
       },
       onError: (error) => {
         if (AbortError.is(error)) {

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

@@ -1,38 +1,160 @@
-import React, { useEffect, useState } from 'react';
+import React, { useCallback, useEffect, useMemo, useState } from 'react';
 
-import { Button } from '@opensumi/ide-components';
-import { localize, useInjectable } from '@opensumi/ide-core-browser';
-import { IResource } from '@opensumi/ide-editor';
+import { Icon, Popover } from '@opensumi/ide-components';
+import { AppConfig, IDisposable, URI, localize, path, useInjectable } from '@opensumi/ide-core-browser';
+import { IResource, WorkbenchEditorService } from '@opensumi/ide-editor';
+import { Path } from '@opensumi/ide-utils/lib/path';
 
 import { BaseApplyService } from '../../mcp/base-apply.service';
 
 import styles from './inline-diff-widget.module.less';
 
+const IconWithPopover = (props: {
+  icon: string;
+  content: string;
+  id: string;
+  onClick?: () => void;
+  disabled?: boolean;
+}) => (
+  <Popover content={props.content} id={props.id} onClick={props.onClick}>
+    <Icon iconClass={props.icon} className={props.disabled ? styles.disabled : ''} />
+  </Popover>
+);
+
 export const InlineDiffManager: React.FC<{ resource: IResource }> = (props) => {
+  const { resource } = props;
   const applyService = useInjectable<BaseApplyService>(BaseApplyService);
+  const editorService = useInjectable<WorkbenchEditorService>(WorkbenchEditorService);
+  const appConfig = useInjectable<AppConfig>(AppConfig);
   const [show, setShow] = useState(true);
+  const [changesCount, setChangesCount] = useState(0);
+  const [currentChangeIndex, setCurrentChangeIndex] = useState(0);
+  const [filePaths, setFilePaths] = useState<string[]>(applyService.getPendingPaths());
+
+  const currentFilePath = useMemo(
+    () => path.relative(appConfig.workspaceDir, resource.uri.path.toString()),
+    [resource],
+  );
+
   useEffect(() => {
-    applyService.onCodeBlockUpdate((codeBlock) => {
-      setShow(codeBlock.status === 'pending');
+    const toDispose = applyService.onCodeBlockUpdate((codeBlock) => {
+      if (path.relative(appConfig.workspaceDir, props.resource.uri.path.toString()) === codeBlock.relativePath) {
+        setShow(codeBlock.status === 'pending');
+      }
+      const pendingPaths = applyService.getPendingPaths();
+      setFilePaths(pendingPaths);
     });
+    return () => {
+      toDispose.dispose();
+    };
   }, []);
+
+  // 不同编辑器是不同实例，所以不需要监听
+  const decorationModelService = applyService.currentPreviewer?.getNode()?.livePreviewDiffDecorationModel;
+
+  useEffect(() => {
+    let toDispose: IDisposable | undefined;
+    if (decorationModelService) {
+      setChangesCount(decorationModelService.partialEditWidgetCount);
+      toDispose = decorationModelService.onPartialEditWidgetListChange((e) => {
+        setChangesCount(e.filter((item) => item.status === 'pending').length);
+      });
+    }
+    return () => {
+      toDispose?.dispose();
+    };
+  }, [decorationModelService]);
+
+  const handleSiblingChange = useCallback(
+    (direction: 'up' | 'down') => {
+      const index = decorationModelService?.revealSiblingChange(direction);
+      if (index !== undefined) {
+        setCurrentChangeIndex(index);
+      }
+    },
+    [decorationModelService],
+  );
+
+  const handleSiblingFile = useCallback(
+    (direction: 'up' | 'down') => {
+      const index = filePaths.indexOf(currentFilePath!);
+      if (index === -1) {
+        return;
+      }
+      const uri = URI.file(path.join(appConfig.workspaceDir, filePaths[index + (direction === 'up' ? -1 : 1)]));
+      editorService.open(uri);
+    },
+    [currentFilePath, filePaths],
+  );
+
+  const changeTip = useMemo(() => {
+    if (changesCount === 0) {
+      return '';
+    }
+    return ` ${currentChangeIndex + 1} of ${changesCount}`;
+  }, [changesCount, currentChangeIndex]);
+
+  const fileTip = useMemo(() => {
+    if (!currentFilePath) {
+      return '';
+    }
+    return ` ${filePaths.indexOf(currentFilePath) + 1} of ${filePaths.length}`;
+  }, [currentFilePath, filePaths]);
+
   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 className={styles.left}>
+        <IconWithPopover
+          icon='codicon codicon-issues'
+          content={localize('aiNative.inlineDiff.reveal')}
+          id='inline-diff-manager-reveal'
+          onClick={handleReveal}
+        />
+      </div> */}
+      <div className={styles.mid}>
+        <IconWithPopover
+          icon='codicon codicon-check'
+          onClick={() => applyService.processAll(props.resource.uri, 'accept')}
+          content={localize('aiNative.inlineDiff.acceptAll')}
+          id='inline-diff-manager-accept-all'
+        />
+        <IconWithPopover
+          icon='codicon codicon-close'
+          onClick={() => applyService.processAll(props.resource.uri, 'reject')}
+          content={localize('aiNative.inlineDiff.rejectAll')}
+          id='inline-diff-manager-reject-all'
+        />
+        <IconWithPopover
+          icon='codicon codicon-arrow-up'
+          content={localize('aiNative.inlineDiff.up') + changeTip}
+          id='inline-diff-manager-up'
+          disabled={currentChangeIndex === 0}
+          onClick={() => handleSiblingChange('up')}
+        />
+        <IconWithPopover
+          icon='codicon codicon-arrow-down'
+          content={localize('aiNative.inlineDiff.down') + changeTip}
+          id='inline-diff-manager-down'
+          disabled={currentChangeIndex === changesCount - 1}
+          onClick={() => handleSiblingChange('down')}
+        />
+      </div>
+      <div className={styles.right}>
+        <IconWithPopover
+          icon='codicon codicon-arrow-left'
+          onClick={() => handleSiblingFile('up')}
+          disabled={filePaths.length === 0 || filePaths[0] === currentFilePath}
+          content={localize('aiNative.inlineDiff.left') + fileTip}
+          id='inline-diff-manager-left'
+        />
+        <IconWithPopover
+          icon='codicon codicon-arrow-right'
+          onClick={() => handleSiblingFile('down')}
+          disabled={filePaths.length === 0 || filePaths[filePaths.length - 1] === currentFilePath}
+          content={localize('aiNative.inlineDiff.right') + fileTip}
+          id='inline-diff-manager-right'
+        />
+      </div>
     </div>
   );
 };

package/src/browser/widget/inline-diff/inline-diff-previewer.ts

@@ -1,5 +1,6 @@
 import { Autowired, INJECTOR_TOKEN, Injectable, Injector } from '@opensumi/di';
 import { Disposable, ErrorResponse, IDisposable, ReplyResponse } from '@opensumi/ide-core-common';
+import { WorkbenchEditorService } from '@opensumi/ide-editor';
 import { EOL, ICodeEditor, IPosition, ITextModel, Position, Selection } from '@opensumi/ide-monaco';
 import { ContentWidgetPositionPreference } from '@opensumi/ide-monaco/lib/browser/monaco-exports/editor';
 import { empty, getLeadingWhitespace } from '@opensumi/ide-utils/lib/strings';
@@ -14,7 +15,12 @@ import {
 
 import { EResultKind } from '../inline-chat/inline-chat.service';
 import { AIInlineContentWidget } from '../inline-chat/inline-content-widget';
-import { EComputerMode, InlineStreamDiffHandler } from '../inline-stream-diff/inline-stream-diff.handler';
+import {
+  BaseInlineStreamDiffHandler,
+  EComputerMode,
+  InlineStreamDiffHandler,
+  ReverseInlineStreamDiffHandler,
+} from '../inline-stream-diff/inline-stream-diff.handler';
 
 import { InlineDiffWidget } from './inline-diff-widget';
 
@@ -29,6 +35,11 @@ export interface IDiffPreviewerOptions {
    * 默认情况下，removed widget 会在 `runWhenIdle` 内被添加，如果需要立即添加，可以设置为 true
    */
   renderRemovedWidgetImmediately?: boolean;
+
+  /**
+   * 是否为回退模式，即 setValue 内容作为 original，编辑器 getValue 内容作为 modified
+   */
+  reverse?: boolean;
 }
 
 export interface IInlineDiffPreviewerNode extends IDisposable {
@@ -41,6 +52,8 @@ export abstract class BaseInlineDiffPreviewer<N extends IInlineDiffPreviewerNode
   @Autowired(INJECTOR_TOKEN)
   protected readonly injector: Injector;
 
+  @Autowired(WorkbenchEditorService)
+  protected readonly editorService: WorkbenchEditorService;
   protected inlineContentWidget: AIInlineContentWidget | null = null;
   protected selection: Selection;
   protected model: ITextModel;
@@ -124,7 +137,7 @@ export abstract class BaseInlineDiffPreviewer<N extends IInlineDiffPreviewerNode
     return Disposable.NULL;
   }
 
-  protected abstract createNode(): N;
+  protected abstract createNode(reverse?: boolean): N;
   abstract onData(data: ReplyResponse): void;
   abstract handleAction(action: EResultKind): void;
   abstract getPosition(): IPosition;
@@ -136,7 +149,7 @@ export abstract class BaseInlineDiffPreviewer<N extends IInlineDiffPreviewerNode
     },
   ): void {
     this.selection = selection;
-    this.node = this.createNode();
+    this.node = this.createNode(options.reverse);
     this.node.setPreviewerOptions(options);
   }
 
@@ -311,8 +324,8 @@ export class SideBySideInlineDiffWidget extends BaseInlineDiffPreviewer<InlineDi
 }
 
 @Injectable({ multiple: true })
-export class LiveInlineDiffPreviewer extends BaseInlineDiffPreviewer<InlineStreamDiffHandler> {
-  private listenNode(node: InlineStreamDiffHandler): void {
+export class LiveInlineDiffPreviewer extends BaseInlineDiffPreviewer<BaseInlineStreamDiffHandler> {
+  private listenNode(node: BaseInlineStreamDiffHandler): void {
     node.addDispose(node.onDidEditChange(() => this.layout()));
     node.addDispose(
       node.onPartialEditWidgetListChange((widgets) => {
@@ -331,8 +344,13 @@ export class LiveInlineDiffPreviewer extends BaseInlineDiffPreviewer<InlineStrea
     this.addDispose(node);
   }
 
-  createNode(): InlineStreamDiffHandler {
-    const node = this.injector.get(InlineStreamDiffHandler, [this.monacoEditor]);
+  createNode(reverse?: boolean): BaseInlineStreamDiffHandler {
+    let node: BaseInlineStreamDiffHandler;
+    if (reverse) {
+      node = this.injector.get(ReverseInlineStreamDiffHandler, [this.monacoEditor]);
+    } else {
+      node = this.injector.get(InlineStreamDiffHandler, [this.monacoEditor]);
+    }
     node.initialize(this.selection);
     this.listenNode(node);
     return node;

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

@@ -22,12 +22,33 @@
 
 .inlineDiffManager {
   display: flex;
-  padding: 12px 16px;
-  justify-content: center;
+  border-radius: 14px;
+  align-items: center;
   position: absolute;
-  bottom: 0;
+  bottom: 18px;
   left: 50%;
   transform: translateX(-50%);
-  gap: 12px;
+  overflow: hidden;
   z-index: 999;
+  > div {
+    display: flex;
+    align-items: center;
+  }
+  .mid {
+    border-right: 1px solid var(--design-borderColor);
+  }
+  :global(.codicon) {
+    font-size: 16px;
+    padding: 6px;
+    color: var(--kt-primaryButton-foreground);
+    background-color: var(--kt-primaryButton-hoverBackground);
+    &:hover {
+      background-color: var(--kt-primaryButton-background);
+    }
+  }
+}
+.disabled {
+  color: var(--kt-button-disableForeground) !important;
+  background-color: var(--kt-button-disableBackground) !important;
+  cursor: not-allowed;
 }

package/src/browser/widget/inline-diff/inline-diff.controller.ts

@@ -3,6 +3,7 @@ import {
   AINativeSettingSectionsId,
   ChatResponse,
   Disposable,
+  DisposableCollection,
   Emitter,
   Event,
   IDisposable,
@@ -23,7 +24,7 @@ import { BaseAIMonacoEditorController } from '../../contrib/base';
 import { EInlineDiffPreviewMode } from '../../preferences/schema';
 import { InlineChatController } from '../inline-chat/inline-chat-controller';
 import { EResultKind } from '../inline-chat/inline-chat.service';
-import { InlineStreamDiffHandler } from '../inline-stream-diff/inline-stream-diff.handler';
+import { BaseInlineStreamDiffHandler } from '../inline-stream-diff/inline-stream-diff.handler';
 
 import {
   BaseInlineDiffPreviewer,
@@ -33,7 +34,7 @@ import {
 } from './inline-diff-previewer';
 import { InlineDiffWidget } from './inline-diff-widget';
 
-type IInlineDiffPreviewer = BaseInlineDiffPreviewer<InlineDiffWidget | InlineStreamDiffHandler>;
+type IInlineDiffPreviewer = BaseInlineDiffPreviewer<InlineDiffWidget | BaseInlineStreamDiffHandler>;
 
 export class InlineDiffController extends BaseAIMonacoEditorController {
   public static readonly ID = 'editor.contrib.ai.inline.diff';
@@ -118,28 +119,26 @@ export class InlineDiffController extends BaseAIMonacoEditorController {
       chatResponse?: ChatResponse | InlineChatController;
       previewerOptions?: IDiffPreviewerOptions;
     },
-  ): BaseInlineDiffPreviewer<InlineDiffWidget | InlineStreamDiffHandler> {
+  ): BaseInlineDiffPreviewer<InlineDiffWidget | BaseInlineStreamDiffHandler> {
     const { crossSelection, chatResponse } = options;
 
     const disposable = new Disposable();
 
     const previewer = this.createDiffPreviewer(monacoEditor, crossSelection, options.previewerOptions);
-    transaction((tx) => {
-      this.currentPreviewer.set(previewer, tx);
-      this.previewerStore.set(previewer.modelId, previewer);
-    });
 
     const onFinish = () => {
       previewer.layout();
       disposable.dispose();
     };
 
+    const previewerDisposable = new DisposableCollection();
+
     disposable.addDispose(
       previewer.onReady(() => {
         if (InlineChatController.is(chatResponse)) {
           const controller = chatResponse as InlineChatController;
 
-          disposable.addDispose([
+          previewerDisposable.pushAll([
             controller.onData((data) => {
               if (ReplyResponse.is(data)) {
                 this.renderDiff(previewer, data);
@@ -167,6 +166,10 @@ export class InlineDiffController extends BaseAIMonacoEditorController {
       }),
     );
 
+    previewer.onDispose(() => {
+      previewerDisposable.dispose();
+    });
+    disposable.addDispose(previewerDisposable);
     previewer.layout();
     return previewer;
   }
@@ -196,6 +199,11 @@ export class InlineDiffController extends BaseAIMonacoEditorController {
       }),
     );
 
+    transaction((tx) => {
+      this.currentPreviewer.set(previewer, tx);
+      this.previewerStore.set(previewer.modelId, previewer);
+    });
+
     return previewer;
   }