From 5a89e0cc2b6c19a734db426fde79a17f80d821cc Mon Sep 17 00:00:00 2001
From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com>
Date: Tue, 20 May 2025 09:36:00 -0400
Subject: [PATCH] Fix typing

---
 src/fastmcp/client/transports.py | 2 --
 tests/client/test_progress.py    | 6 ++++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/src/fastmcp/client/transports.py b/src/fastmcp/client/transports.py
index e1597bec8..7faeab613 100644
--- a/src/fastmcp/client/transports.py
+++ b/src/fastmcp/client/transports.py
@@ -22,7 +22,6 @@ from mcp.client.stdio import stdio_client
 from mcp.client.streamable_http import streamablehttp_client
 from mcp.client.websocket import websocket_client
 from mcp.shared.memory import create_connected_server_and_client_session
-from mcp.shared.session import ProgressFnT
 from pydantic import AnyUrl
 from typing_extensions import Unpack
 
@@ -36,7 +35,6 @@ class SessionKwargs(TypedDict, total=False):
     list_roots_callback: ListRootsFnT | None
     logging_callback: LoggingFnT | None
     message_handler: MessageHandlerFnT | None
-    progress_callback: ProgressFnT | None
     read_timeout_seconds: datetime.timedelta | None
 
 
diff --git a/tests/client/test_progress.py b/tests/client/test_progress.py
index 5f61c3487..f67f5c54c 100644
--- a/tests/client/test_progress.py
+++ b/tests/client/test_progress.py
@@ -59,8 +59,10 @@ async def test_progress_handler_can_be_supplied_on_tool_call(fastmcp_server: Fas
 async def test_progress_handler_supplied_on_tool_call_overrides_default(
     fastmcp_server: FastMCP,
 ):
-    async def bad_progress_handler(*args, **kwargs):
-        1 / 0
+    async def bad_progress_handler(
+        progress: float, total: float | None, message: str | None
+    ) -> None:
+        raise Exception("This should not be called")
 
     async with Client(fastmcp_server, progress_handler=bad_progress_handler) as client:
         await client.call_tool("progress_tool", {}, progress_handler=progress_handler)