diff --git a/README.md b/README.md index 0ca039ae..ed7ae7ab 100644 --- a/README.md +++ b/README.md @@ -161,6 +161,7 @@ from dataclasses import dataclass from fake_database import Database # Replace with your actual DB type from mcp.server.fastmcp import Context, FastMCP +from mcp.server.session import ServerSessionT # Create a named server mcp = FastMCP("My App") @@ -192,7 +193,7 @@ mcp = FastMCP("My App", lifespan=app_lifespan) # Access type-safe lifespan context in tools @mcp.tool() -def query_db(ctx: Context) -> str: +def query_db(ctx: Context[ServerSessionT, AppContext]) -> str: """Tool that uses initialized resources""" db = ctx.request_context.lifespan_context.db return db.query()