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
160
160
161
161
from fake_database import Database # Replace with your actual DB type
162
162
163
- from mcp.server.fastmcp import Context, FastMCP
163
+ from mcp.server.fastmcp import FastMCP
164
164
165
165
# Create a named server
166
166
mcp = FastMCP(" My App" )
@@ -192,9 +192,10 @@ mcp = FastMCP("My App", lifespan=app_lifespan)
192
192
193
193
# Access type-safe lifespan context in tools
194
194
@mcp.tool ()
195
- def query_db (ctx : Context ) -> str :
195
+ def query_db () -> str :
196
196
""" 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" ]
198
199
return db.query()
199
200
```
200
201
@@ -631,7 +632,7 @@ server = Server("example-server", lifespan=server_lifespan)
631
632
# Access lifespan context in handlers
632
633
@server.call_tool ()
633
634
async def query_db (name : str , arguments : dict ) -> list :
634
- ctx = server.get_context()
635
+ ctx = server.request_context
635
636
db = ctx.lifespan_context[" db" ]
636
637
return await db.query(arguments[" query" ])
637
638
```
You can’t perform that action at this time.
0 commit comments