@opensumi/ide-ai-native 3.8.1-next-1741093151.0 → 3.8.1-next-1741096184.0

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

@@ -1,6 +1,6 @@
 // OpenSumi as MCP Server 前端的代理服务
 import { Autowired, Injectable } from '@opensumi/di';
-import { ILogger } from '@opensumi/ide-core-common';
+import { AIServiceType, IAIReporter, ILogger } from '@opensumi/ide-core-common';
 
 import { BUILTIN_MCP_SERVER_NAME } from '../../common';
 import { getToolName } from '../../common/utils';
@@ -19,13 +19,26 @@ export class MCPServerRegistry implements IMCPServerRegistry {
   private tools: MCPToolDefinition[] = [];
   private toolComponents: Record<string, React.FC<IMCPServerToolComponentProps>> = {};
 
+  private _activeMessageInfo: {
+    messageId: string;
+    sessionId: string;
+  };
+
   @Autowired(ILogger)
   private readonly baseLogger: ILogger;
 
+  @Autowired(IAIReporter)
+  private readonly aiReporter: IAIReporter;
+
   private get logger(): MCPLogger {
     return new LoggerAdapter(this.baseLogger);
   }
 
+  /** 设置当前活跃的消息信息，便于toolCall打点 */
+  set activeMessageInfo(params: { messageId: string; sessionId: string }) {
+    this._activeMessageInfo = params;
+  }
+
   getMCPTool(name: string, serverName = BUILTIN_MCP_SERVER_NAME): MCPToolDefinition | undefined {
     return this.tools.find((tool) => getToolName(tool.name, serverName) === name);
   }
@@ -65,14 +78,33 @@ export class MCPServerRegistry implements IMCPServerRegistry {
       // 统一校验并转换
       const toolCallId = args.toolCallId;
       args = tool.inputSchema.parse(args);
-      return await tool.handler({ ...args, toolCallId }, this.logger);
+      const result = await tool.handler({ ...args, toolCallId }, this.logger);
+      this.reportToolCall(name, args, result, toolCallId);
+      return result;
     } catch (error) {
       // eslint-disable-next-line no-console
       console.error('callMCPTool error:', error);
+      this.reportToolCall(name, args, error, args.toolCallId);
       return {
         content: [{ type: 'text', text: `The tool ${name} failed to execute. Error: ${error}` }],
         isError: true,
       };
     }
   }
+
+  private reportToolCall(name: string, args: any, result: any, toolCallId: string) {
+    const tool = this.tools.find((tool) => tool.name === name);
+    if (!tool) {
+      throw new Error(`MCP tool ${name} not found`);
+    }
+    this.aiReporter.record(
+      {
+        msgType: AIServiceType.ToolCall,
+        message: JSON.stringify({ args, name, result }),
+        messageId: this._activeMessageInfo.messageId,
+        sessionId: this._activeMessageInfo.sessionId,
+      },
+      toolCallId,
+    );
+  }
 }

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;