ModelContextProtocol C SDK实现按应用ID动态过滤工具集

在基于ModelContextProtocol C# SDK开发服务端应用时，我们经常需要根据不同客户端需求动态提供不同的工具能力。本文将深入探讨如何通过配置会话选项实现按应用ID过滤工具集的高级功能。

核心实现原理

SDK提供了HttpServerTransportOptions.ConfigureSessionOptions回调接口，允许开发者在建立SSE连接时动态配置会话参数。通过该机制，我们可以实现以下功能架构：

1. 建立应用ID与工具集的映射关系
2. 在连接建立时解析查询参数
3. 根据应用ID筛选对应的工具集
4. 动态构建会话能力描述

具体实现方案

首先需要创建应用ID与工具集的字典映射，这是实现动态过滤的基础数据结构。建议在服务启动前完成初始化以避免并发问题。

var appDictionary = new Dictionary<string, McpServerTool[]>();

在配置HTTP传输时，通过ConfigureSessionOptions回调实现动态过滤逻辑。该回调会在每个SSE连接建立时触发，开发者可以在此处访问HTTP上下文并修改会话选项。

关键实现要点包括：

• 从查询参数解析appId
• 从预置字典获取对应工具集
• 动态构建Capabilities配置
• 安全处理各种边界情况

完整示例代码

以下是一个完整的实现示例，展示了如何为不同应用ID注册不同工具集：

var builder = WebApplication.CreateBuilder(args);
builder.Services.AddMcpServer()
    .WithHttpTransport(httpOptions =>
    {
        httpOptions.ConfigureSessionOptions = (httpContext, mcpServerOptions, cancellationToken) =>
        {
            if (httpContext.Request.Query.TryGetValue("appId", out var appId) &&
                appDictionary.TryGetValue(appId.ToString(), out var tools))
            {
                mcpServerOptions.Capabilities ??= new();
                mcpServerOptions.Capabilities.Tools ??= new();
                var toolCollection = mcpServerOptions.Capabilities.Tools.ToolCollection ??= [];
                foreach (var tool in tools)
                {
                    toolCollection.TryAdd(tool);
                }
            }
            return Task.CompletedTask;
        };
    });

实际应用建议

1. 性能优化：对于工具集较大的场景，建议使用缓存机制
2. 安全考虑：应对appId参数进行验证和过滤
3. 错误处理：提供默认工具集应对未知appId的情况
4. 动态加载：可结合配置中心实现工具集的热更新

总结

通过本文介绍的方法，开发者可以灵活地为不同客户端应用提供差异化的工具能力。这种动态过滤机制特别适合多租户场景或需要细粒度控制工具访问权限的系统。ModelContextProtocol C# SDK提供的会话配置接口为这类需求提供了优雅的实现方案。