Skip to content

Commit a85068e

Browse files
committed
some pyright fixes
1 parent 4e6800e commit a85068e

File tree

7 files changed

+21
-21
lines changed

7 files changed

+21
-21
lines changed

examples/clients/simple-task-interactive-client/mcp_simple_task_interactive_client/main.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010
from typing import Any
1111

1212
import click
13-
from mcp import ClientSession
13+
from mcp import ClientSession, ClientTransportSession
1414
from mcp.client.streamable_http import streamable_http_client
1515
from mcp.shared.context import RequestContext
1616
from mcp.types import (
@@ -24,7 +24,7 @@
2424

2525

2626
async def elicitation_callback(
27-
context: RequestContext[ClientSession, Any],
27+
context: RequestContext[ClientTransportSession, Any],
2828
params: ElicitRequestParams,
2929
) -> ElicitResult:
3030
"""Handle elicitation requests from the server."""
@@ -39,7 +39,7 @@ async def elicitation_callback(
3939

4040

4141
async def sampling_callback(
42-
context: RequestContext[ClientSession, Any],
42+
context: RequestContext[ClientTransportSession, Any],
4343
params: CreateMessageRequestParams,
4444
) -> CreateMessageResult:
4545
"""Handle sampling requests from the server."""

examples/servers/simple-streamablehttp/mcp_simple_streamablehttp/server.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,8 @@
1313
from starlette.routing import Mount
1414
from starlette.types import Receive, Scope, Send
1515

16+
from pydantic import AnyUrl
17+
1618
from .event_store import InMemoryEventStore
1719

1820
# Configure logging
@@ -73,7 +75,7 @@ async def call_tool(name: str, arguments: dict[str, Any]) -> list[types.ContentB
7375

7476
# This will send a resource notificaiton though standalone SSE
7577
# established by GET request
76-
await ctx.session.send_resource_updated(uri="http:///test_resource")
78+
await ctx.session.send_resource_updated(uri=AnyUrl("http:///test_resource"))
7779
return [
7880
types.TextContent(
7981
type="text",

examples/snippets/clients/stdio_client.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import os
77

88
from mcp import ClientSession, StdioServerParameters, types
9-
from pydantic import AnyUrl
109

1110
from mcp import ClientSession, ClientTransportSession, StdioServerParameters, types
1211
from mcp.client.stdio import stdio_client

examples/snippets/clients/url_elicitation_client.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,15 +30,15 @@
3030
from typing import Any
3131
from urllib.parse import urlparse
3232

33-
from mcp import ClientSession, types
33+
from mcp import ClientSession, ClientTransportSession, types
3434
from mcp.client.sse import sse_client
3535
from mcp.shared.context import RequestContext
3636
from mcp.shared.exceptions import McpError, UrlElicitationRequiredError
3737
from mcp.types import URL_ELICITATION_REQUIRED
3838

3939

4040
async def handle_elicitation(
41-
context: RequestContext[ClientSession, Any],
41+
context: RequestContext[ClientTransportSession, Any],
4242
params: types.ElicitRequestParams,
4343
) -> types.ElicitResult | types.ErrorData:
4444
"""Handle elicitation requests from the server.

src/mcp/client/transport_session.py

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -41,19 +41,19 @@ async def set_logging_level(
4141
raise NotImplementedError
4242

4343
@abstractmethod
44-
async def list_resources(
45-
self,
46-
cursor: str | None = None,
47-
) -> types.ListResourcesResult:
44+
async def list_resources(self, *, params: types.PaginatedRequestParams | None = None) -> types.ListResourcesResult:
4845
"""Send a resources/list request."""
4946
raise NotImplementedError
5047

5148
@abstractmethod
5249
async def list_resource_templates(
53-
self,
54-
cursor: str | None = None,
50+
self, *, params: types.PaginatedRequestParams | None = None
5551
) -> types.ListResourceTemplatesResult:
56-
"""Send a resources/templates/list request."""
52+
"""Send a resources/templates/list request.
53+
54+
Args:
55+
params: Full pagination parameters including cursor and any future fields
56+
"""
5757
raise NotImplementedError
5858

5959
@abstractmethod
@@ -62,22 +62,24 @@ async def read_resource(self, uri: AnyUrl) -> types.ReadResourceResult:
6262
raise NotImplementedError
6363

6464
@abstractmethod
65-
async def subscribe_resource(self, uri: AnyUrl) -> types.EmptyResult:
65+
async def subscribe_resource(self, uri: str, *, meta: RequestParamsMeta | None = None) -> types.EmptyResult:
6666
"""Send a resources/subscribe request."""
6767
raise NotImplementedError
6868

6969
@abstractmethod
70-
async def unsubscribe_resource(self, uri: AnyUrl) -> types.EmptyResult:
70+
async def unsubscribe_resource(self, uri: str, *, meta: RequestParamsMeta | None = None) -> types.EmptyResult:
7171
"""Send a resources/unsubscribe request."""
7272
raise NotImplementedError
7373

7474
@abstractmethod
7575
async def call_tool(
7676
self,
7777
name: str,
78-
arguments: Any | None = None,
79-
read_timeout_seconds: timedelta | None = None,
78+
arguments: dict[str, Any] | None = None,
79+
read_timeout_seconds: float | None = None,
8080
progress_callback: ProgressFnT | None = None,
81+
*,
82+
meta: RequestParamsMeta | None = None,
8183
) -> types.CallToolResult:
8284
"""Send a tools/call request with optional progress callback support."""
8385
raise NotImplementedError

tests/server/test_cancel_handling.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,6 @@
66
import pytest
77

88
import mcp.types as types
9-
from mcp.client.session import ClientSession
109
from mcp import Client
1110
from mcp.server.lowlevel.server import Server
1211
from mcp.shared.exceptions import McpError

tests/shared/test_memory.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,6 @@
11
import pytest
22

33
from mcp import Client
4-
from mcp.client.session import ClientSession
5-
from mcp.client.transport_session import ClientTransportSession
64
from mcp.server import Server
75
from mcp.types import EmptyResult, Resource
86

0 commit comments

Comments
 (0)