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

package/src/browser/components/ChatReply.tsx

@@ -38,6 +38,7 @@ import {
   IChatComponent,
   IChatContent,
   IChatResponseProgressFileTreeData,
+  IChatToolContent,
   URI,
 } from '@opensumi/ide-core-common';
 import { IIconService } from '@opensumi/ide-theme';
@@ -148,6 +149,33 @@ const TreeRenderer = (props: { treeData: IChatResponseProgressFileTreeData }) =>
   );
 };
 
+const ToolCallRender = (props: { toolCall: IChatToolContent['content'] }) => {
+  const { toolCall } = props;
+  const chatAgentViewService = useInjectable<IChatAgentViewService>(ChatAgentViewServiceToken);
+  const [node, setNode] = useState<React.JSX.Element | null>(null);
+
+  useEffect(() => {
+    const config = chatAgentViewService.getChatComponent('toolCall');
+    if (config) {
+      const { component: Component, initialProps } = config;
+      setNode(<Component {...initialProps} value={toolCall} />);
+      return;
+    }
+    setNode(
+      <div>
+        <Loading />
+        <span style={{ marginLeft: 4 }}>正在加载组件</span>
+      </div>,
+    );
+    const deferred = chatAgentViewService.getChatComponentDeferred('toolCall')!;
+    deferred.promise.then(({ component: Component, initialProps }) => {
+      setNode(<Component {...initialProps} value={toolCall} />);
+    });
+  }, [toolCall.state]);
+
+  return node;
+};
+
 const ComponentRender = (props: { component: string; value?: unknown }) => {
   const chatAgentViewService = useInjectable<IChatAgentViewService>(ChatAgentViewServiceToken);
   const [node, setNode] = useState<React.JSX.Element | null>(null);
@@ -274,6 +302,8 @@ export const ChatReply = (props: IChatReplyProps) => {
     <ComponentRender component={componentId} value={value} />
   );
 
+  const renderToolCall = (toolCall: IChatToolContent['content']) => <ToolCallRender toolCall={toolCall} />;
+
   const contentNode = React.useMemo(
     () =>
       request.response.responseContents.map((item, index) => {
@@ -284,6 +314,8 @@ export const ChatReply = (props: IChatReplyProps) => {
           node = renderTreeData(item.treeData);
         } else if (item.kind === 'component') {
           node = renderComponent(item.component, item.value);
+        } else if (item.kind === 'toolCall') {
+          node = renderToolCall(item.content);
         } else {
           node = renderMarkdown(item.content);
         }

package/src/browser/components/ChatToolRender.module.less

@@ -0,0 +1,86 @@
+.chat-tool-render {
+  margin: 8px 0;
+  border: 1px solid var(--design-borderColor);
+  border-radius: 6px;
+  overflow: hidden;
+
+  .tool-header {
+    display: flex;
+    align-items: center;
+    justify-content: space-between;
+    padding: 8px 12px;
+    background-color: var(--design-block-background);
+    cursor: pointer;
+    user-select: none;
+
+    &:hover {
+      background-color: var(--design-block-hoverBackground);
+    }
+  }
+
+  .tool-name {
+    display: flex;
+    align-items: center;
+    font-weight: 500;
+    color: var(--design-text-foreground);
+  }
+
+  .expand-icon {
+    display: inline-block;
+    margin-right: 8px;
+    transition: transform 0.2s;
+    color: var(--design-text-placeholderForeground);
+
+    &.expanded {
+      transform: rotate(90deg);
+    }
+  }
+
+  .tool-state {
+    display: flex;
+    align-items: center;
+    font-size: 12px;
+    color: var(--design-text-placeholderForeground);
+  }
+
+  .state-icon {
+    display: flex;
+    align-items: center;
+    margin-right: 6px;
+  }
+
+  .loading-icon {
+    width: 12px;
+    height: 12px;
+  }
+
+  .state-label {
+    margin-left: 4px;
+  }
+
+  .tool-content {
+    max-height: 0;
+    overflow: hidden;
+    transition: max-height 0.3s ease-out;
+    background-color: var(--design-container-background);
+
+    &.expanded {
+      max-height: 1000px;
+    }
+  }
+
+  .tool-arguments,
+  .tool-result {
+    padding: 12px;
+  }
+
+  .section-label {
+    font-size: 12px;
+    color: var(--design-text-placeholderForeground);
+    margin-bottom: 8px;
+  }
+
+  .tool-result {
+    border-top: 1px solid var(--design-borderColor);
+  }
+}

package/src/browser/components/ChatToolRender.tsx

@@ -0,0 +1,77 @@
+import cls from 'classnames';
+import React, { useState } from 'react';
+
+import { Icon } from '@opensumi/ide-core-browser/lib/components';
+import { Loading } from '@opensumi/ide-core-browser/lib/components/ai-native';
+import { IChatToolContent, uuid } from '@opensumi/ide-core-common';
+
+import { CodeEditorWithHighlight } from './ChatEditor';
+import styles from './ChatToolRender.module.less';
+
+export const ChatToolRender = (props: { value: IChatToolContent['content'] }) => {
+  const { value } = props;
+  const [isExpanded, setIsExpanded] = useState(false);
+
+  if (!value || !value.function || !value.id) {
+    return null;
+  }
+
+  const getStateInfo = (state?: string): { label: string; icon: React.ReactNode } => {
+    switch (state) {
+      case 'streaming-start':
+      case 'streaming':
+        return { label: 'Generating', icon: <Loading /> };
+      case 'complete':
+        return { label: 'Complete', icon: <Icon iconClass="codicon codicon-check" /> };
+      case 'result':
+        return { label: 'Result Ready', icon: <Icon iconClass="codicon codicon-check-all" /> };
+      default:
+        return { label: state || 'Unknown', icon: <Icon iconClass="codicon codicon-question" /> };
+    }
+  };
+
+  const toggleExpand = () => {
+    setIsExpanded(!isExpanded);
+  };
+
+  const stateInfo = getStateInfo(value.state);
+
+  return (
+    <div className={styles['chat-tool-render']}>
+      <div className={styles['tool-header']} onClick={toggleExpand}>
+        <div className={styles['tool-name']}>
+          <span className={cls(styles['expand-icon'], { [styles.expanded]: isExpanded })}>▶</span>
+          {value?.function?.name}
+        </div>
+        {value.state && (
+          <div className={styles['tool-state']}>
+            <span className={styles['state-icon']}>{stateInfo.icon}</span>
+            <span className={styles['state-label']}>{stateInfo.label}</span>
+          </div>
+        )}
+      </div>
+      <div className={cls(styles['tool-content'], { [styles.expanded]: isExpanded })}>
+        {value?.function?.arguments && (
+          <div className={styles['tool-arguments']}>
+            <div className={styles['section-label']}>Arguments</div>
+            <CodeEditorWithHighlight
+              input={value?.function?.arguments}
+              language={'json'}
+              relationId={uuid(4)}
+            />
+          </div>
+        )}
+        {value?.result && (
+          <div className={styles['tool-result']}>
+            <div className={styles['section-label']}>Result</div>
+            <CodeEditorWithHighlight
+              input={value.result}
+              language={'json'}
+              relationId={uuid(4)}
+            />
+          </div>
+        )}
+      </div>
+    </div>
+  );
+};

package/src/browser/components/components.module.less

@@ -244,44 +244,45 @@
   }
 }
 
-.code_block {
+.monaco_wrapper {
   position: relative;
-  min-width: 100px;
-  margin-top: 4px;
-  .monaco_wrapper {
-    position: relative;
-    min-width: 130px;
-    > pre {
-      margin-bottom: 10px;
-    }
-    .editor {
-      border-radius: 8px;
-      font-size: 12px;
-      padding: 32px 8px 8px 8px;
-      line-height: 18px;
-      &::-webkit-scrollbar {
-        width: auto;
-        height: 4px;
-      }
+  min-width: 130px;
+  > pre {
+    margin-bottom: 10px;
+  }
+  .editor {
+    border-radius: 8px;
+    font-size: 12px;
+    padding: 32px 8px 8px 8px;
+    line-height: 18px;
+    &::-webkit-scrollbar {
+      width: auto;
+      height: 4px;
     }
+  }
 
-    .action_toolbar {
-      display: flex;
-      position: absolute;
-      right: 8px;
-      top: 6px;
-      z-index: 100;
-      height: 20px;
-      align-items: center;
-      overflow: hidden;
+  .action_toolbar {
+    display: flex;
+    position: absolute;
+    right: 8px;
+    top: 6px;
+    z-index: 100;
+    height: 20px;
+    align-items: center;
+    overflow: hidden;
 
-      :global {
-        .kt-popover {
-          height: inherit;
-        }
+    :global {
+      .kt-popover {
+        height: inherit;
       }
     }
   }
+}
+
+.code_block {
+  position: relative;
+  min-width: 100px;
+  margin-top: 4px;
 
   :global {
     .hljs {

package/src/browser/context/llm-context.contribution.ts

@@ -0,0 +1,14 @@
+import { Autowired } from '@opensumi/di';
+import { ClientAppContribution, Domain } from '@opensumi/ide-core-browser';
+
+import { LLMContextService, LLMContextServiceToken } from '../../common/llm-context';
+
+@Domain(ClientAppContribution)
+export class LlmContextContribution implements ClientAppContribution {
+  @Autowired(LLMContextServiceToken)
+  protected readonly llmContextService: LLMContextService;
+
+  initialize() {
+    this.llmContextService.startAutoCollection();
+  }
+}

package/src/browser/context/llm-context.service.ts

@@ -0,0 +1,156 @@
+import { Autowired, Injectable } from '@opensumi/di';
+import { AppConfig } from '@opensumi/ide-core-browser/lib/react-providers/config-provider';
+import { WithEventBus } from '@opensumi/ide-core-common/lib/event-bus/event-decorator';
+import { MarkerSeverity } from '@opensumi/ide-core-common/lib/types/markers/markers';
+import { Emitter, URI } from '@opensumi/ide-core-common/lib/utils';
+import {
+  EditorDocumentModelCreationEvent,
+  EditorDocumentModelRemovalEvent,
+  EditorDocumentModelSavedEvent,
+  IEditorDocumentModelService,
+} from '@opensumi/ide-editor/lib/browser/doc-model/types';
+import { EditorSelectionChangeEvent } from '@opensumi/ide-editor/lib/browser/types';
+import { IMarkerService } from '@opensumi/ide-markers/lib/common/types';
+
+import { FileContext, LLMContextService, SerializedContext } from '../../common/llm-context';
+
+@Injectable()
+export class LLMContextServiceImpl extends WithEventBus implements LLMContextService {
+  @Autowired(AppConfig)
+  protected readonly appConfig: AppConfig;
+
+  @Autowired(IEditorDocumentModelService)
+  protected readonly docModelManager: IEditorDocumentModelService;
+
+  @Autowired(IMarkerService)
+  protected readonly markerService: IMarkerService;
+
+  private isAutoCollecting = false;
+
+  private contextFiles: Map<string, FileContext> = new Map();
+
+  private onDidContextFilesChangeEmitter = new Emitter<FileContext[]>();
+  onDidContextFilesChangeEvent = this.onDidContextFilesChangeEmitter.event;
+
+  addFileToContext(uri: URI, selection?: [number, number], isManual = true): void {
+    this.contextFiles.set(uri.toString(), {
+      uri,
+      selection,
+      isManual,
+    });
+    this.onDidContextFilesChangeEmitter.fire(this.getAllContextFiles());
+  }
+
+  cleanFileContext() {
+    this.contextFiles.clear();
+    this.onDidContextFilesChangeEmitter.fire(this.getAllContextFiles());
+  }
+
+  private getAllContextFiles() {
+    return Array.from(this.contextFiles.values());
+  }
+
+  removeFileFromContext(uri: URI): void {
+    this.contextFiles.delete(uri.toString());
+    this.onDidContextFilesChangeEmitter.fire(this.getAllContextFiles());
+  }
+
+  startAutoCollection(): void {
+    if (this.isAutoCollecting) {
+      return;
+    }
+    this.isAutoCollecting = true;
+
+    this.startAutoCollectionInternal();
+  }
+
+  private startAutoCollectionInternal(): void {
+    // 文件打开
+    this.disposables.push(
+      this.eventBus.on(EditorDocumentModelCreationEvent, (event) => {
+        if (event.payload.uri.scheme !== 'file') {
+          return;
+        }
+        // TODO: 是否自动添加文件到上下文？
+        // this.addFileToContext(event.payload.uri);
+      }),
+    );
+
+    // 删除
+    this.disposables.push(
+      this.eventBus.on(EditorDocumentModelRemovalEvent, (event) => {
+        if (event.payload.scheme !== 'file') {
+          return;
+        }
+      }),
+    );
+
+    // 保存
+    this.disposables.push(
+      this.eventBus.on(EditorDocumentModelSavedEvent, (event) => {
+        if (event.payload.scheme !== 'file') {
+          return;
+        }
+      }),
+    );
+
+    // 光标选中
+    this.disposables.push(
+      this.eventBus.on(EditorSelectionChangeEvent, (event) => {
+        if (event.payload.selections.length > 0) {
+          const selection = [
+            event.payload.selections[0].selectionStartLineNumber,
+            event.payload.selections[0].positionLineNumber,
+          ].sort() as [number, number];
+          if (selection[0] === selection[1]) {
+            // TODO: 是否自动添加文件到上下文？
+            // this.addFileToContext(event.payload.editorUri, undefined);
+          } else {
+            this.addFileToContext(
+              event.payload.editorUri,
+              selection.sort((a, b) => a - b),
+            );
+          }
+        }
+      }),
+    );
+  }
+
+  stopAutoCollection(): void {
+    this.dispose();
+  }
+
+  serialize(): SerializedContext {
+    const files = this.getAllContextFiles();
+    const recentlyViewFiles = files
+      .filter((v) => !v.selection)
+      .map((file) => URI.file(this.appConfig.workspaceDir).relative(file.uri)!.toString())
+      .filter(Boolean);
+    const attachedFiles = files
+      .filter((v) => v.selection)
+      .map((file) => {
+        const ref = this.docModelManager.getModelReference(file.uri);
+        const content = ref!.instance.getText();
+        const lineErrors = this.markerService
+          .getManager()
+          .getMarkers({
+            resource: file.uri.toString(),
+            severities: MarkerSeverity.Error,
+          })
+          .map((marker) => marker.message);
+
+        return {
+          content,
+          lineErrors,
+          path: URI.file(this.appConfig.workspaceDir).relative(file.uri)!.toString(),
+          language: ref?.instance.languageId!,
+        };
+      })
+      .filter(Boolean);
+
+    return {
+      recentlyViewFiles,
+      attachedFiles,
+    };
+  }
+}

package/src/browser/index.ts

@@ -19,8 +19,17 @@ import {
   TerminalRegistryToken,
 } from '@opensumi/ide-core-common';
 
-import { ChatProxyServiceToken, IChatAgentService, IChatInternalService, IChatManagerService } from '../common';
-import { IAIInlineCompletionsProvider } from '../common';
+import {
+  ChatProxyServiceToken,
+  IAIInlineCompletionsProvider,
+  IChatAgentService,
+  IChatInternalService,
+  IChatManagerService,
+  SumiMCPServerProxyServicePath,
+  TokenMCPServerProxyService,
+} from '../common';
+import { LLMContextServiceToken } from '../common/llm-context';
+import { MCPServerManager, MCPServerManagerPath } from '../common/mcp-server-manager';
 
 import { AINativeBrowserContribution } from './ai-core.contribution';
 import { ChatAgentService } from './chat/chat-agent.service';
@@ -31,6 +40,8 @@ import { ChatService } from './chat/chat.api.service';
 import { ChatFeatureRegistry } from './chat/chat.feature.registry';
 import { ChatInternalService } from './chat/chat.internal.service';
 import { ChatRenderRegistry } from './chat/chat.render.registry';
+import { LlmContextContribution } from './context/llm-context.contribution';
+import { LLMContextServiceImpl } from './context/llm-context.service';
 import { AICodeActionContribution } from './contrib/code-action/code-action.contribution';
 import { AIInlineCompletionsProvider } from './contrib/inline-completions/completeProvider';
 import { IntelligentCompletionsContribution } from './contrib/intelligent-completions/intelligent-completions.contribution';
@@ -43,8 +54,22 @@ import { RenameCandidatesProviderRegistry } from './contrib/rename/rename.featur
 import { TerminalAIContribution } from './contrib/terminal/terminal-ai.contributon';
 import { TerminalFeatureRegistry } from './contrib/terminal/terminal.feature.registry';
 import { LanguageParserService } from './languages/service';
+import { MCPServerProxyService } from './mcp/mcp-server-proxy.service';
+import { MCPServerRegistry } from './mcp/mcp-server.feature.registry';
+import { CreateNewFileWithTextTool } from './mcp/tools/createNewFileWithText';
+import { FindFilesByNameSubstringTool } from './mcp/tools/findFilesByNameSubstring';
+import { GetCurrentFilePathTool } from './mcp/tools/getCurrentFilePath';
+import { GetDiagnosticsByPathTool } from './mcp/tools/getDiagnosticsByPath';
+import { GetFileTextByPathTool } from './mcp/tools/getFileTextByPath';
+import { GetOpenEditorFileDiagnosticsTool } from './mcp/tools/getOpenEditorFileDiagnostics';
+import { GetOpenEditorFileTextTool } from './mcp/tools/getOpenEditorFileText';
+import { GetSelectedTextTool } from './mcp/tools/getSelectedText';
+import { ListDirTool } from './mcp/tools/listDir';
+import { ReadFileTool } from './mcp/tools/readFile';
+import { ReplaceOpenEditorFileByDiffPreviewerTool } from './mcp/tools/replaceOpenEditorFileByDiffPreviewer';
+import { RunTerminalCommandTool } from './mcp/tools/runTerminalCmd';
 import { AINativePreferencesContribution } from './preferences';
-import { AINativeCoreContribution } from './types';
+import { AINativeCoreContribution, MCPServerContribution, TokenMCPServerRegistry } from './types';
 import { InlineChatFeatureRegistry } from './widget/inline-chat/inline-chat.feature.registry';
 import { InlineChatService } from './widget/inline-chat/inline-chat.service';
 import { InlineDiffService } from './widget/inline-diff';
@@ -59,7 +84,7 @@ export class AINativeModule extends BrowserModule {
     this.aiNativeConfig.setAINativeModuleLoaded(true);
   }
 
-  contributionProvider = AINativeCoreContribution;
+  contributionProvider = [AINativeCoreContribution, MCPServerContribution];
   providers: Provider[] = [
     AINativeBrowserContribution,
     InterfaceNavigationContribution,
@@ -68,6 +93,37 @@ export class AINativeModule extends BrowserModule {
     AICodeActionContribution,
     AINativePreferencesContribution,
     IntelligentCompletionsContribution,
+
+    // MCP Server Contributions START
+    ListDirTool,
+    ReadFileTool,
+    CreateNewFileWithTextTool,
+    GetSelectedTextTool,
+    GetOpenEditorFileDiagnosticsTool,
+    GetOpenEditorFileTextTool,
+    GetFileTextByPathTool,
+    GetCurrentFilePathTool,
+    FindFilesByNameSubstringTool,
+    GetDiagnosticsByPathTool,
+    RunTerminalCommandTool,
+    ReplaceOpenEditorFileByDiffPreviewerTool,
+    // MCP Server Contributions END
+
+    // Context Service
+    LlmContextContribution,
+    {
+      token: LLMContextServiceToken,
+      useClass: LLMContextServiceImpl,
+    },
+
+    {
+      token: TokenMCPServerRegistry,
+      useClass: MCPServerRegistry,
+    },
+    {
+      token: TokenMCPServerProxyService,
+      useClass: MCPServerProxyService,
+    },
     {
       token: InlineChatFeatureRegistryToken,
       useClass: InlineChatFeatureRegistry,
@@ -148,5 +204,13 @@ export class AINativeModule extends BrowserModule {
       token: AIBackSerivceToken,
       clientToken: ChatProxyServiceToken,
     },
+    {
+      servicePath: MCPServerManagerPath,
+      token: MCPServerManager,
+    },
+    {
+      clientToken: TokenMCPServerProxyService,
+      servicePath: SumiMCPServerProxyServicePath,
+    },
   ];
 }

package/src/browser/mcp/mcp-server-proxy.service.ts

@@ -0,0 +1,53 @@
+import { Autowired, Injectable } from '@opensumi/di';
+import { ILogger } from '@opensumi/ide-core-browser';
+import { Emitter, Event } from '@opensumi/ide-core-common';
+
+import { ISumiMCPServerBackend, SumiMCPServerProxyServicePath } from '../../common';
+import { IMCPServerProxyService } from '../../common/types';
+import { IMCPServerRegistry, TokenMCPServerRegistry } from '../types';
+
+@Injectable()
+export class MCPServerProxyService implements IMCPServerProxyService {
+  @Autowired(TokenMCPServerRegistry)
+  private readonly mcpServerRegistry: IMCPServerRegistry;
+
+  @Autowired(ILogger)
+  private readonly logger: ILogger;
+
+  @Autowired(SumiMCPServerProxyServicePath)
+  private readonly sumiMCPServerProxyService: ISumiMCPServerBackend;
+
+  private readonly _onChangeMCPServers = new Emitter<any>();
+  public readonly onChangeMCPServers: Event<any> = this._onChangeMCPServers.event;
+
+  // 调用 OpenSumi 内部注册的 MCP 工具
+  $callMCPTool(name: string, args: any) {
+    return this.mcpServerRegistry.callMCPTool(name, args);
+  }
+
+  // 获取 OpenSumi 内部注册的 MCP tools
+  async $getMCPTools() {
+    const tools = await this.mcpServerRegistry.getMCPTools().map((tool) =>
+      // 不要传递 handler
+       ({
+        name: tool.name,
+        description: tool.description,
+        inputSchema: tool.inputSchema,
+        providerName: 'sumi-builtin',
+      }),
+    );
+
+    this.logger.log('SUMI MCP tools', tools);
+
+    return tools;
+  }
+
+  // 通知前端 MCP 服务注册表发生了变化
+  async $updateMCPServers() {
+    this._onChangeMCPServers.fire('update');
+  }
+
+  async getAllMCPTools() {
+    return this.sumiMCPServerProxyService.getAllMCPTools();
+  }
+}

package/src/browser/mcp/mcp-server.feature.registry.ts

@@ -0,0 +1,54 @@
+// OpenSumi as MCP Server 前端的代理服务
+import { Autowired, Injectable } from '@opensumi/di';
+import { IAIBackService, ILogger } from '@opensumi/ide-core-common';
+
+import { IMCPServerRegistry, MCPLogger, MCPToolDefinition } from '../types';
+
+class LoggerAdapter implements MCPLogger {
+  constructor(private readonly logger: ILogger) { }
+
+  appendLine(message: string): void {
+    this.logger.log(message);
+  }
+}
+
+@Injectable()
+export class MCPServerRegistry implements IMCPServerRegistry {
+  private tools: MCPToolDefinition[] = [];
+
+  @Autowired(ILogger)
+  private readonly baseLogger: ILogger;
+
+  private get logger(): MCPLogger {
+    return new LoggerAdapter(this.baseLogger);
+  }
+
+  registerMCPTool(tool: MCPToolDefinition): void {
+    this.tools.push(tool);
+  }
+
+  getMCPTools(): MCPToolDefinition[] {
+    return this.tools;
+  }
+
+  async callMCPTool(
+    name: string,
+    args: any,
+  ): Promise<{
+    content: { type: string; text: string }[];
+    isError?: boolean;
+  }> {
+    try {
+      const tool = this.tools.find((tool) => tool.name === name);
+      if (!tool) {
+        throw new Error(`MCP tool ${name} not found`);
+      }
+      return await tool.handler(args, this.logger);
+    } catch (error) {
+      return {
+        content: [{ type: 'text', text: `The tool ${name} failed to execute. Error: ${error}` }],
+        isError: true,
+      };
+    }
+  }
+}