Jeremiah Lowin commited on
Commit
f0e5fd6
·
1 Parent(s): c27f039

Fix typing

Browse files
src/fastmcp/client/transports.py CHANGED
@@ -22,7 +22,6 @@ from mcp.client.stdio import stdio_client
22
  from mcp.client.streamable_http import streamablehttp_client
23
  from mcp.client.websocket import websocket_client
24
  from mcp.shared.memory import create_connected_server_and_client_session
25
- from mcp.shared.session import ProgressFnT
26
  from pydantic import AnyUrl
27
  from typing_extensions import Unpack
28
 
@@ -36,7 +35,6 @@ class SessionKwargs(TypedDict, total=False):
36
  list_roots_callback: ListRootsFnT | None
37
  logging_callback: LoggingFnT | None
38
  message_handler: MessageHandlerFnT | None
39
- progress_callback: ProgressFnT | None
40
  read_timeout_seconds: datetime.timedelta | None
41
 
42
 
 
22
  from mcp.client.streamable_http import streamablehttp_client
23
  from mcp.client.websocket import websocket_client
24
  from mcp.shared.memory import create_connected_server_and_client_session
 
25
  from pydantic import AnyUrl
26
  from typing_extensions import Unpack
27
 
 
35
  list_roots_callback: ListRootsFnT | None
36
  logging_callback: LoggingFnT | None
37
  message_handler: MessageHandlerFnT | None
 
38
  read_timeout_seconds: datetime.timedelta | None
39
 
40
 
tests/client/test_progress.py CHANGED
@@ -59,8 +59,10 @@ async def test_progress_handler_can_be_supplied_on_tool_call(fastmcp_server: Fas
59
  async def test_progress_handler_supplied_on_tool_call_overrides_default(
60
  fastmcp_server: FastMCP,
61
  ):
62
- async def bad_progress_handler(*args, **kwargs):
63
- 1 / 0
 
 
64
 
65
  async with Client(fastmcp_server, progress_handler=bad_progress_handler) as client:
66
  await client.call_tool("progress_tool", {}, progress_handler=progress_handler)
 
59
  async def test_progress_handler_supplied_on_tool_call_overrides_default(
60
  fastmcp_server: FastMCP,
61
  ):
62
+ async def bad_progress_handler(
63
+ progress: float, total: float | None, message: str | None
64
+ ) -> None:
65
+ raise Exception("This should not be called")
66
 
67
  async with Client(fastmcp_server, progress_handler=bad_progress_handler) as client:
68
  await client.call_tool("progress_tool", {}, progress_handler=progress_handler)