feat: support MCP sampling requests

src/main/events.ts

@@ -112,7 +112,10 @@ export const MCP_EVENTS = {
   TOOL_CALL_RESULT: 'mcp:tool-call-result',
   SERVER_STATUS_CHANGED: 'mcp:server-status-changed',
   CLIENT_LIST_UPDATED: 'mcp:client-list-updated',
-  INITIALIZED: 'mcp:initialized' // 新增：MCP初始化完成事件
+  INITIALIZED: 'mcp:initialized', // 新增：MCP初始化完成事件
+  SAMPLING_REQUEST: 'mcp:sampling-request',
+  SAMPLING_DECISION: 'mcp:sampling-decision',
+  SAMPLING_CANCELLED: 'mcp:sampling-cancelled'
 }
 
 // 同步相关事件

src/main/presenter/mcpPresenter/index.ts

@@ -8,7 +8,9 @@ import {
   Prompt,
   ResourceListEntry,
   Resource,
-  PromptListEntry
+  PromptListEntry,
+  McpSamplingRequestPayload,
+  McpSamplingDecision
 } from '@shared/presenter'
 import { ServerManager } from './serverManager'
 import { ToolManager } from './toolManager'
@@ -85,6 +87,10 @@ export class McpPresenter implements IMCPPresenter {
   private isInitialized: boolean = false
   // McpRouter
   private mcprouter?: McpRouterManager
+  private pendingSamplingRequests = new Map<
+    string,
+    { resolve: (decision: McpSamplingDecision) => void; reject: (error: Error) => void }
+  >()
 
   constructor(configPresenter?: IConfigPresenter) {
     console.log('Initializing MCP Presenter')
@@ -580,6 +586,60 @@ export class McpPresenter implements IMCPPresenter {
     return { content: formattedContent, rawData: toolCallResult }
   }
 
+  async handleSamplingRequest(request: McpSamplingRequestPayload): Promise<McpSamplingDecision> {
+    if (!request || !request.requestId) {
+      throw new Error('Invalid sampling request: missing requestId')
+    }
+
+    return new Promise<McpSamplingDecision>((resolve, reject) => {
+      try {
+        this.pendingSamplingRequests.set(request.requestId, { resolve, reject })
+        eventBus.sendToRenderer(MCP_EVENTS.SAMPLING_REQUEST, SendTarget.DEFAULT_TAB, request)
+      } catch (error) {
+        this.pendingSamplingRequests.delete(request.requestId)
+        reject(error instanceof Error ? error : new Error(String(error)))
+      }
+    })
+  }
+
+  async submitSamplingDecision(decision: McpSamplingDecision): Promise<void> {
+    if (!decision || !decision.requestId) {
+      throw new Error('Invalid sampling decision: missing requestId')
+    }
+
+    const pending = this.pendingSamplingRequests.get(decision.requestId)
+    if (!pending) {
+      console.warn(
+        `[MCP] Sampling request ${decision.requestId} not found when submitting decision`
+      )
+      return
+    }
+
+    this.pendingSamplingRequests.delete(decision.requestId)
+    pending.resolve(decision)
+
+    eventBus.sendToRenderer(MCP_EVENTS.SAMPLING_DECISION, SendTarget.ALL_WINDOWS, decision)
+  }
+
+  async cancelSamplingRequest(requestId: string, reason?: string): Promise<void> {
+    if (!requestId) {
+      return
+    }
+
+    const pending = this.pendingSamplingRequests.get(requestId)
+    if (!pending) {
+      return
+    }
+
+    this.pendingSamplingRequests.delete(requestId)
+    pending.reject(new Error(reason ?? 'Sampling request cancelled'))
+
+    eventBus.sendToRenderer(MCP_EVENTS.SAMPLING_CANCELLED, SendTarget.ALL_WINDOWS, {
+      requestId,
+      reason: reason ?? 'cancelled'
+    })
+  }
+
   // Convert MCPToolDefinition to MCPTool
   private mcpToolDefinitionToMcpTool(
     toolDefinition: MCPToolDefinition,