ModelContextProtocol Python SDK 中SSE初始化问题的分析与解决

问题背景

在使用ModelContextProtocol(MCP) Python SDK进行开发时，许多开发者遇到了一个共同的问题：当使用FastMCP服务器配合SSE(Server-Sent Events)传输协议时，客户端在调用session.initialize()方法时会无限期挂起，无法完成初始化握手过程。这个问题严重影响了开发者的使用体验，特别是在构建基于SSE的实时通信应用时。

问题现象

根据开发者报告，当按照MCP规范的生命周期流程进行操作时：

1. 客户端发送initialize请求
2. 服务器应返回initialize结果(包含服务器能力信息)
3. 客户端发送initialized通知

但在实际使用中，虽然服务器能够正确接受初始的SSE连接GET请求和包含客户端initialize消息的POST请求，却从未通过SSE流发送InitializeResult回客户端。这导致客户端的await session.initialize()调用无限期挂起，最终因超时而失败。

环境配置

出现问题的典型环境包括：

• mcp-client 1.6.0版本
• Python 3.10及以上版本
• httpx作为mcp-client的依赖
• uvicorn作为FastMCP SSE模式的依赖
• anyio作为FastMCP的依赖

问题根源分析

经过深入调查和开发者社区的讨论，发现问题主要源于对ClientSession对象的使用方式不当。许多开发者尝试直接创建ClientSession实例并调用initialize()方法，而没有正确使用Python的异步上下文管理器(async with)模式。

ClientSession类设计为必须通过上下文管理器使用，因为它内部管理着重要的资源(如网络连接和消息流)。当不通过上下文管理器使用时，会话无法正确建立双向通信通道，导致初始化握手过程无法完成。

解决方案

正确的使用方式有以下两种：

方案一：使用嵌套的上下文管理器

from mcp.client.sse import sse_client

async with sse_client(server_url) as (read_stream, write_stream):
    async with ClientSession(read_stream, write_stream) as session:
        print("开始初始化会话")
        await session.initialize()
        print("会话初始化成功")

方案二：手动管理上下文

# 创建SSE上下文管理器但不立即进入
sse_context = sse_client(server_url)
# 手动进入上下文管理器
streams = await sse_context.__aenter__()
read_stream, write_stream = streams

# 创建ClientSession上下文管理器
session_context = ClientSession(read_stream, write_stream)
# 手动进入上下文管理器
session = await session_context.__aenter__()

# 初始化会话
init_result = await session.initialize()
print(f"会话初始化结果: {init_result}")

最佳实践建议

1. 始终使用上下文管理器：无论是sse_client还是ClientSession，都应该通过async with语句使用，确保资源正确释放。

2. 错误处理：在初始化过程中添加适当的错误处理，捕获可能发生的连接超时或协议错误。

3. 日志记录：在关键步骤添加日志记录，便于调试和问题追踪。

4. 超时设置：为初始化操作设置合理的超时时间，避免无限期等待。

5. 资源清理：如果采用手动管理上下文的方式，务必在最后调用相应的__aexit__方法清理资源。

总结

MCP Python SDK中的SSE初始化问题主要源于对ClientSession使用方式的理解不足。通过正确使用上下文管理器模式，可以确保会话初始化的完整生命周期得到正确执行。开发者社区的经验表明，遵循这一模式后，初始化挂起的问题能够得到彻底解决。

对于SDK维护者而言，考虑在未来的版本中增加运行时检查，当检测到ClientSession被错误使用时抛出明确的异常，可以帮助开发者更快地识别和解决问题。