diff --git a/README.md b/README.md index 26f43cfd9..4c8bb90dc 100644 --- a/README.md +++ b/README.md @@ -160,7 +160,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.fastmcp import FastMCP # Create a named server mcp = FastMCP("My App") @@ -192,9 +192,10 @@ 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() -> str: """Tool that uses initialized resources""" - db = ctx.request_context.lifespan_context.db + ctx = mcp.get_context() + db = ctx.request_context.lifespan_context["db"] return db.query() ``` @@ -631,7 +632,7 @@ server = Server("example-server", lifespan=server_lifespan) # Access lifespan context in handlers @server.call_tool() async def query_db(name: str, arguments: dict) -> list: - ctx = server.get_context() + ctx = server.request_context db = ctx.lifespan_context["db"] return await db.query(arguments["query"]) ```