FastMCP框架中Context对象注入机制解析

背景介绍

FastMCP是一个基于Python的异步Web框架，它提供了资源管理和工具调用的功能。在2.2.5版本之前，框架中存在一个关于Context对象注入的重要限制：Context对象可以自动注入到被@mcp.tool装饰的函数中，但无法注入到被@mcp.resource装饰的资源函数中。

问题本质

在FastMCP框架中，Context对象包含了请求上下文和生命周期上下文等重要信息。开发者通常需要通过这个对象来：

1. 访问请求特定的信息
2. 获取应用生命周期中维护的状态
3. 实现请求级别的依赖注入

在2.2.5版本之前，资源函数无法像工具函数那样通过类型提示自动获取Context对象，这导致开发者无法在资源处理逻辑中访问这些重要信息。

解决方案实现

FastMCP 2.2.5版本通过引入新的注入机制解决了这个问题。现在，资源函数可以通过以下方式声明Context参数：

@mcp.resource("data://test/{item_id}")
def resource_handler(item_id: str, ctx: Context) -> dict:
    # 现在可以正常访问ctx对象
    lifespan_ctx = ctx.request_context.lifespan_context
    return {"status": "success"}

框架会自动识别ctx: Context类型提示，并在调用时注入正确的Context实例。

技术实现细节

这个改进涉及框架核心的几个关键修改：

1. 参数解析器增强：改进了资源函数的参数解析逻辑，使其能够识别Context类型提示
2. 依赖注入系统：统一了工具函数和资源函数的依赖注入机制
3. 类型检查兼容：确保修改不会破坏现有的类型检查和行为

最佳实践

在使用这个新特性时，建议开发者：

1. 始终明确声明Context参数的类型提示
2. 不要在资源函数中修改Context对象的状态
3. 对于资源模板，确保路径参数和Context参数的顺序正确
4. 考虑使用Pydantic模型来验证从Context中获取的数据

性能考量

虽然自动注入带来了便利性，但开发者应该注意：

1. Context对象的创建和注入会有轻微性能开销
2. 对于高频调用的资源端点，考虑缓存从Context中获取的数据
3. 避免在资源函数中进行复杂的上下文处理

总结

FastMCP 2.2.5版本对Context注入机制的改进，使得资源函数和工具函数在依赖注入方面达到了行为一致性，大大提高了框架的可用性和开发体验。这一改进使得开发者能够以统一的方式访问请求和生命周期上下文，同时保持了框架的简洁性和性能。