MCP-Agent项目中的WebSocket支持与授权头传递实现

背景介绍

MCP-Agent作为Model Context Protocol协议的实现项目，近期增加了对WebSocket协议的支持。这项改进使得开发者能够通过WebSocket连接与MCP服务器进行实时通信，同时还需要解决在WebSocket连接中传递授权头(Authorization Header)的技术挑战。

WebSocket支持的技术实现

在MCP-Agent项目中，WebSocket支持通过以下几个关键步骤实现：

1. 协议适配层：在现有HTTP协议实现基础上，增加了WebSocket协议的适配层，确保两种协议都能正确处理MCP消息格式。

2. 连接管理：实现了WebSocket特有的连接保持机制，不同于HTTP的请求-响应模式，WebSocket需要维护持久连接状态。

3. 消息序列化：确保通过WebSocket传输的消息能够正确序列化和反序列化，保持与HTTP协议实现的一致性。

授权头传递方案

为了在WebSocket连接中传递授权信息，项目采用了以下设计方案：

1. 配置中心化：授权令牌可以在mcp_agent配置文件中指定，支持通过程序方式或配置文件(mcp_agent.config.yaml/mcp_agent.secrets.yaml)传递。

2. 认证设置扩展：在MCPServerAuthSettings配置类中新增了相关字段，专门用于处理WebSocket连接的认证信息。

3. 通用头支持：除了专门的认证字段外，还实现了通用的headers配置项，允许开发者传递任意HTTP头信息。

实现细节

在实际代码实现中，主要涉及以下几个关键点：

1. 配置解析：系统会优先检查配置中的auth设置，如果存在则自动将其转换为适当的Authorization头。

2. 连接初始化：在建立WebSocket连接时，配置的头部信息会被自动添加到连接请求中。

3. 安全考虑：敏感信息如Bearer令牌会被特殊处理，避免在日志或调试信息中泄露。

使用示例

开发者可以通过以下方式使用新功能：

1. 在配置文件中指定WebSocket服务器地址和认证信息
2. 通过编程方式设置连接参数
3. 使用提供的初始化钩子(init hook)进行更动态的配置

未来展望

虽然当前实现已经满足了基本需求，但随着MCP协议向流式HTTP方向发展，WebSocket支持可能会演变为更通用的流式传输方案。项目维护者将持续关注协议发展，及时调整实现策略。

这项改进为MCP-Agent项目带来了更灵活的连接方式选择，特别是在需要实时双向通信的场景下，WebSocket支持将大大提升系统的实用性。同时，标准化的授权头传递机制也为企业级应用的安全需求提供了可靠保障。