Skip to content

Commit 8e0a3d4

Browse files
committed
style: apply ruff formatting to pass pre-commit checks
Apply automatic formatting from ruff to ensure code meets project standards: - Remove trailing whitespace - Adjust line breaks for consistency - Format function arguments according to line length limits
1 parent 305b2eb commit 8e0a3d4

File tree

2 files changed

+19
-28
lines changed

2 files changed

+19
-28
lines changed

src/mcp/server/lowlevel/server.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -424,9 +424,7 @@ async def handler(_: Any):
424424
return decorator
425425

426426
def list_tools_paginated(self):
427-
def decorator(
428-
func: Callable[[types.Cursor | None], Awaitable[types.ListToolsResult]]
429-
):
427+
def decorator(func: Callable[[types.Cursor | None], Awaitable[types.ListToolsResult]]):
430428
logger.debug("Registering paginated handler for ListToolsRequest")
431429

432430
async def handler(request: types.ListToolsRequest):

tests/server/lowlevel/test_server_pagination.py

Lines changed: 18 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -18,29 +18,26 @@
1818
async def test_list_prompts_pagination() -> None:
1919
server = Server("test")
2020
test_cursor = "test-cursor-123"
21-
21+
2222
# Track what cursor was received
2323
received_cursor: Cursor | None = None
24-
24+
2525
@server.list_prompts_paginated()
2626
async def handle_list_prompts(cursor: Cursor | None) -> ListPromptsResult:
2727
nonlocal received_cursor
2828
received_cursor = cursor
2929
return ListPromptsResult(prompts=[], nextCursor="next")
30-
30+
3131
handler = server.request_handlers[ListPromptsRequest]
32-
32+
3333
# Test: No cursor provided -> handler receives None
3434
request = ListPromptsRequest(method="prompts/list", params=None)
3535
result = await handler(request)
3636
assert received_cursor is None
3737
assert isinstance(result, ServerResult)
38-
38+
3939
# Test: Cursor provided -> handler receives exact cursor value
40-
request_with_cursor = ListPromptsRequest(
41-
method="prompts/list",
42-
params=PaginatedRequestParams(cursor=test_cursor)
43-
)
40+
request_with_cursor = ListPromptsRequest(method="prompts/list", params=PaginatedRequestParams(cursor=test_cursor))
4441
result2 = await handler(request_with_cursor)
4542
assert received_cursor == test_cursor
4643
assert isinstance(result2, ServerResult)
@@ -50,28 +47,27 @@ async def handle_list_prompts(cursor: Cursor | None) -> ListPromptsResult:
5047
async def test_list_resources_pagination() -> None:
5148
server = Server("test")
5249
test_cursor = "resource-cursor-456"
53-
50+
5451
# Track what cursor was received
5552
received_cursor: Cursor | None = None
56-
53+
5754
@server.list_resources_paginated()
5855
async def handle_list_resources(cursor: Cursor | None) -> ListResourcesResult:
5956
nonlocal received_cursor
6057
received_cursor = cursor
6158
return ListResourcesResult(resources=[], nextCursor="next")
62-
59+
6360
handler = server.request_handlers[ListResourcesRequest]
64-
61+
6562
# Test: No cursor provided -> handler receives None
6663
request = ListResourcesRequest(method="resources/list", params=None)
6764
result = await handler(request)
6865
assert received_cursor is None
6966
assert isinstance(result, ServerResult)
70-
67+
7168
# Test: Cursor provided -> handler receives exact cursor value
7269
request_with_cursor = ListResourcesRequest(
73-
method="resources/list",
74-
params=PaginatedRequestParams(cursor=test_cursor)
70+
method="resources/list", params=PaginatedRequestParams(cursor=test_cursor)
7571
)
7672
result2 = await handler(request_with_cursor)
7773
assert received_cursor == test_cursor
@@ -82,29 +78,26 @@ async def handle_list_resources(cursor: Cursor | None) -> ListResourcesResult:
8278
async def test_list_tools_pagination() -> None:
8379
server = Server("test")
8480
test_cursor = "tools-cursor-789"
85-
81+
8682
# Track what cursor was received
8783
received_cursor: Cursor | None = None
88-
84+
8985
@server.list_tools_paginated()
9086
async def handle_list_tools(cursor: Cursor | None) -> ListToolsResult:
9187
nonlocal received_cursor
9288
received_cursor = cursor
9389
return ListToolsResult(tools=[], nextCursor="next")
94-
90+
9591
handler = server.request_handlers[ListToolsRequest]
96-
92+
9793
# Test: No cursor provided -> handler receives None
9894
request = ListToolsRequest(method="tools/list", params=None)
9995
result = await handler(request)
10096
assert received_cursor is None
10197
assert isinstance(result, ServerResult)
102-
98+
10399
# Test: Cursor provided -> handler receives exact cursor value
104-
request_with_cursor = ListToolsRequest(
105-
method="tools/list",
106-
params=PaginatedRequestParams(cursor=test_cursor)
107-
)
100+
request_with_cursor = ListToolsRequest(method="tools/list", params=PaginatedRequestParams(cursor=test_cursor))
108101
result2 = await handler(request_with_cursor)
109102
assert received_cursor == test_cursor
110103
assert isinstance(result2, ServerResult)

0 commit comments

Comments
 (0)