Skip to content

Commit 8e1d947

Browse files
committed
refactor: replace lowlevel Server decorators with on_* constructor kwargs
Replace the decorator-based handler registration on the lowlevel Server with direct on_* keyword arguments on the constructor. Handlers are raw callables with a uniform (ctx, params) -> result signature. - Server constructor takes on_list_tools, on_call_tool, etc. - String-keyed dispatch instead of type-keyed - Remove RequestT generic from Server (transport-specific, not bound at construction) - Delete handler.py and func_inspection.py (no longer needed) - Update ExperimentalHandlers to use callback-based registration - Update MCPServer to pass on_* kwargs via _create_handler_kwargs() - Update migration docs and docstrings
1 parent b1f7eec commit 8e1d947

File tree

13 files changed

+581
-757
lines changed

13 files changed

+581
-757
lines changed

docs/experimental/index.md

Lines changed: 3 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -27,10 +27,9 @@ Tasks are useful for:
2727
Experimental features are accessed via the `.experimental` property:
2828

2929
```python
30-
# Server-side
31-
@server.experimental.get_task()
32-
async def handle_get_task(request: GetTaskRequest) -> GetTaskResult:
33-
...
30+
# Server-side: enable task support (auto-registers default handlers)
31+
server = Server(name="my-server")
32+
server.experimental.enable_tasks()
3433

3534
# Client-side
3635
result = await session.experimental.call_tool_as_task("tool_name", {"arg": "value"})

docs/migration.md

Lines changed: 167 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -351,7 +351,6 @@ The nested `RequestParams.Meta` Pydantic model class has been replaced with a to
351351
- `RequestParams.Meta` (Pydantic model) → `RequestParamsMeta` (TypedDict)
352352
- Attribute access (`meta.progress_token`) → Dictionary access (`meta.get("progress_token")`)
353353
- `progress_token` field changed from `ProgressToken | None = None` to `NotRequired[ProgressToken]`
354-
`
355354

356355
**In request context handlers:**
357356

@@ -364,11 +363,12 @@ async def handle_tool(name: str, arguments: dict) -> list[TextContent]:
364363
await ctx.session.send_progress_notification(ctx.meta.progress_token, 0.5, 100)
365364

366365
# After (v2)
367-
@server.call_tool()
368-
async def handle_tool(name: str, arguments: dict) -> list[TextContent]:
369-
ctx = server.request_context
366+
async def handle_call_tool(
367+
ctx: RequestContext, params: CallToolRequestParams
368+
) -> CallToolResult:
370369
if ctx.meta and "progress_token" in ctx.meta:
371370
await ctx.session.send_progress_notification(ctx.meta["progress_token"], 0.5, 100)
371+
...
372372
```
373373

374374
### `RequestContext` and `ProgressContext` type parameters simplified
@@ -470,6 +470,158 @@ await client.read_resource("test://resource")
470470
await client.read_resource(str(my_any_url))
471471
```
472472

473+
### Lowlevel `Server`: decorator-based handlers replaced with constructor `on_*` params
474+
475+
The lowlevel `Server` class no longer uses decorator methods for handler registration. Instead, handlers are passed as `on_*` keyword arguments to the constructor.
476+
477+
**Before (v1):**
478+
479+
```python
480+
from mcp.server.lowlevel.server import Server
481+
482+
server = Server("my-server")
483+
484+
@server.list_tools()
485+
async def handle_list_tools():
486+
return [types.Tool(name="my_tool", description="A tool", inputSchema={})]
487+
488+
@server.call_tool()
489+
async def handle_call_tool(name: str, arguments: dict):
490+
return [types.TextContent(type="text", text=f"Called {name}")]
491+
```
492+
493+
**After (v2):**
494+
495+
```python
496+
from mcp.server.lowlevel import Server
497+
from mcp.shared.context import RequestContext
498+
from mcp.types import (
499+
CallToolRequestParams,
500+
CallToolResult,
501+
ListToolsResult,
502+
PaginatedRequestParams,
503+
TextContent,
504+
Tool,
505+
)
506+
507+
async def handle_list_tools(
508+
ctx: RequestContext, params: PaginatedRequestParams | None
509+
) -> ListToolsResult:
510+
return ListToolsResult(tools=[
511+
Tool(name="my_tool", description="A tool", inputSchema={})
512+
])
513+
514+
async def handle_call_tool(
515+
ctx: RequestContext, params: CallToolRequestParams
516+
) -> CallToolResult:
517+
return CallToolResult(
518+
content=[TextContent(type="text", text=f"Called {params.name}")],
519+
is_error=False,
520+
)
521+
522+
server = Server(
523+
"my-server",
524+
on_list_tools=handle_list_tools,
525+
on_call_tool=handle_call_tool,
526+
)
527+
```
528+
529+
**Key differences:**
530+
531+
- Handlers receive `(ctx, params)` instead of the full request object or unpacked arguments. `ctx` is a `RequestContext` with `session`, `lifespan_context`, and `experimental` fields (plus `request_id`, `meta`, etc. for request handlers). `params` is the typed request params object.
532+
- Handlers return the full result type (e.g. `ListToolsResult`) rather than unwrapped values (e.g. `list[Tool]`).
533+
- The automatic `jsonschema` input/output validation that the old `call_tool()` decorator performed has been removed. There is no built-in replacement — if you relied on schema validation in the lowlevel server, you will need to validate inputs yourself in your handler.
534+
535+
**Notification handlers:**
536+
537+
```python
538+
from mcp.server.lowlevel import Server
539+
from mcp.shared.context import RequestContext
540+
from mcp.types import ProgressNotificationParams
541+
542+
async def handle_progress(
543+
ctx: RequestContext, params: ProgressNotificationParams
544+
) -> None:
545+
print(f"Progress: {params.progress}/{params.total}")
546+
547+
server = Server(
548+
"my-server",
549+
on_progress=handle_progress,
550+
)
551+
```
552+
553+
### Lowlevel `Server`: `request_context` property removed
554+
555+
The `server.request_context` property has been removed. Request context is now passed directly to handlers as the first argument (`ctx`). The `request_ctx` module-level contextvar still exists but should not be needed — use `ctx` directly instead.
556+
557+
**Before (v1):**
558+
559+
```python
560+
from mcp.server.lowlevel.server import request_ctx
561+
562+
@server.call_tool()
563+
async def handle_call_tool(name: str, arguments: dict):
564+
ctx = server.request_context # or request_ctx.get()
565+
await ctx.session.send_log_message(level="info", data="Processing...")
566+
return [types.TextContent(type="text", text="Done")]
567+
```
568+
569+
**After (v2):**
570+
571+
```python
572+
from mcp.shared.context import RequestContext
573+
from mcp.types import CallToolRequestParams, CallToolResult, TextContent
574+
575+
async def handle_call_tool(
576+
ctx: RequestContext, params: CallToolRequestParams
577+
) -> CallToolResult:
578+
await ctx.session.send_log_message(level="info", data="Processing...")
579+
return CallToolResult(
580+
content=[TextContent(type="text", text="Done")],
581+
is_error=False,
582+
)
583+
```
584+
585+
### `RequestContext`: request-specific fields are now optional
586+
587+
The `RequestContext` class now uses optional fields for request-specific data (`request_id`, `meta`, etc.) so it can be used for both request and notification handlers. In notification handlers, these fields are `None`.
588+
589+
```python
590+
from mcp.shared.context import RequestContext
591+
592+
# request_id, meta, etc. are available in request handlers
593+
# but None in notification handlers
594+
```
595+
596+
### Experimental: task handler decorators removed
597+
598+
The experimental decorator methods on `ExperimentalHandlers` (`@server.experimental.list_tasks()`, `@server.experimental.get_task()`, etc.) have been removed.
599+
600+
Default task handlers are still registered automatically via `server.experimental.enable_tasks()`.
601+
602+
**Before (v1):**
603+
604+
```python
605+
server = Server("my-server")
606+
server.experimental.enable_tasks(task_store)
607+
608+
@server.experimental.get_task()
609+
async def custom_get_task(request: GetTaskRequest) -> GetTaskResult:
610+
...
611+
```
612+
613+
**After (v2):**
614+
615+
```python
616+
from mcp.server.lowlevel import Server
617+
from mcp.types import GetTaskRequestParams, GetTaskResult
618+
619+
server = Server("my-server")
620+
server.experimental.enable_tasks(task_store)
621+
# Default handlers are registered automatically.
622+
# Custom task handlers are not yet supported via the constructor.
623+
```
624+
473625
## Deprecations
474626

475627
<!-- Add deprecations below -->
@@ -505,16 +657,20 @@ params = CallToolRequestParams(
505657
The `streamable_http_app()` method is now available directly on the lowlevel `Server` class, not just `MCPServer`. This allows using the streamable HTTP transport without the MCPServer wrapper.
506658

507659
```python
508-
from mcp.server.lowlevel.server import Server
660+
from mcp.server.lowlevel import Server
661+
from mcp.shared.context import RequestContext
662+
from mcp.types import ListToolsResult, PaginatedRequestParams
509663

510-
server = Server("my-server")
664+
async def handle_list_tools(
665+
ctx: RequestContext, params: PaginatedRequestParams | None
666+
) -> ListToolsResult:
667+
return ListToolsResult(tools=[...])
511668

512-
# Register handlers...
513-
@server.list_tools()
514-
async def list_tools():
515-
return [...]
669+
server = Server(
670+
"my-server",
671+
on_list_tools=handle_list_tools,
672+
)
516673

517-
# Create a Starlette app for streamable HTTP
518674
app = server.streamable_http_app(
519675
streamable_http_path="/mcp",
520676
json_response=False,

src/mcp/server/experimental/request_context.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -160,10 +160,7 @@ async def run_task(
160160
RuntimeError: If task support is not enabled or task_metadata is missing
161161
162162
Example:
163-
@server.call_tool()
164-
async def handle_tool(name: str, args: dict):
165-
ctx = server.request_context
166-
163+
async def handle_tool(ctx: RequestContext, params: CallToolRequestParams) -> CallToolResult:
167164
async def work(task: ServerTaskContext) -> CallToolResult:
168165
result = await task.elicit(
169166
message="Are you sure?",

src/mcp/server/experimental/task_result_handler.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -47,14 +47,12 @@ class TaskResultHandler:
4747
# Create handler with store and queue
4848
handler = TaskResultHandler(task_store, message_queue)
4949
50-
# Register it with the server
51-
@server.experimental.get_task_result()
52-
async def handle_task_result(req: GetTaskPayloadRequest) -> GetTaskPayloadResult:
53-
ctx = server.request_context
54-
return await handler.handle(req, ctx.session, ctx.request_id)
55-
56-
# Or use the convenience method
57-
handler.register(server)
50+
# Register as a handler with the lowlevel server
51+
async def handle_task_result(ctx, params):
52+
return await handler.handle(
53+
GetTaskPayloadRequest(params=params), ctx.session, ctx.request_id
54+
)
55+
server = Server(on_call_tool=..., on_list_tools=...)
5856
"""
5957

6058
def __init__(
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,3 @@
11
from .server import NotificationOptions, Server
22

3-
__all__ = ["Server", "NotificationOptions"]
3+
__all__ = ["NotificationOptions", "Server"]

0 commit comments

Comments
 (0)