File tree Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Expand file tree Collapse file tree 1 file changed +5
-4
lines changed Original file line number Diff line number Diff line change @@ -160,7 +160,7 @@ from dataclasses import dataclass
160160
161161from fake_database import Database # Replace with your actual DB type
162162
163- from mcp.server.fastmcp import Context, FastMCP
163+ from mcp.server.fastmcp import FastMCP
164164
165165# Create a named server
166166mcp = FastMCP(" My App" )
@@ -192,9 +192,10 @@ mcp = FastMCP("My App", lifespan=app_lifespan)
192192
193193# Access type-safe lifespan context in tools
194194@mcp.tool ()
195- def query_db (ctx : Context ) -> str :
195+ def query_db () -> str :
196196 """ Tool that uses initialized resources"""
197- db = ctx.request_context.lifespan_context.db
197+ ctx = mcp.get_context()
198+ db = ctx.request_context.lifespan_context[" db" ]
198199 return db.query()
199200```
200201
@@ -631,7 +632,7 @@ server = Server("example-server", lifespan=server_lifespan)
631632# Access lifespan context in handlers
632633@server.call_tool ()
633634async def query_db (name : str , arguments : dict ) -> list :
634- ctx = server.get_context()
635+ ctx = server.request_context
635636 db = ctx.lifespan_context[" db" ]
636637 return await db.query(arguments[" query" ])
637638```
You can’t perform that action at this time.
0 commit comments