rename

Author: ihrpr

src/mcp/client/session.py

@@ -8,7 +8,7 @@
 import mcp.types as types
 from mcp.shared.context import RequestContext
 from mcp.shared.message import SessionMessage
-from mcp.shared.session import BaseSession, ProgressCallbackFnT, RequestResponder
+from mcp.shared.session import BaseSession, ProgressFnT, RequestResponder
 from mcp.shared.version import SUPPORTED_PROTOCOL_VERSIONS
 
 DEFAULT_CLIENT_INFO = types.Implementation(name="mcp", version="0.1.0")
@@ -270,7 +270,7 @@ async def call_tool(
         name: str,
         arguments: dict[str, Any] | None = None,
         read_timeout_seconds: timedelta | None = None,
-        progress_callback: ProgressCallbackFnT | None = None,
+        progress_callback: ProgressFnT | None = None,
     ) -> types.CallToolResult:
         """Send a tools/call request with optional progress callback support."""
 

src/mcp/server/session.py

@@ -286,9 +286,6 @@ async def send_progress_notification(
         related_request_id: str | None = None,
     ) -> None:
         """Send a progress notification."""
-        print(
-            f"Sending progress notification: {progress_token}, {progress}, {total}, {message}"
-        )
         await self.send_notification(
             types.ServerNotification(
                 types.ProgressNotification(

src/mcp/shared/session.py

@@ -43,20 +43,12 @@
 RequestId = str | int
 
 
-class ProgressCallbackFnT(Protocol):
+class ProgressFnT(Protocol):
     """Protocol for progress notification callbacks."""
 
     def __call__(
         self, progress: float, total: float | None, message: str | None
-    ) -> None:
-        """Called when progress updates are received.
-
-        Args:
-            progress: Current progress value
-            total: Total progress value (if known), None if indeterminate
-            message: Optional progress message
-        """
-        ...
+    ) -> None: ...
 
 
 class RequestResponder(Generic[ReceiveRequestT, SendResultT]):
@@ -186,7 +178,7 @@ class BaseSession(
     ]
     _request_id: int
     _in_flight: dict[RequestId, RequestResponder[ReceiveRequestT, SendResultT]]
-    _progress_callbacks: dict[RequestId, ProgressCallbackFnT]
+    _progress_callbacks: dict[RequestId, ProgressFnT]
 
     def __init__(
         self,
@@ -233,7 +225,7 @@ async def send_request(
         result_type: type[ReceiveResultT],
         request_read_timeout_seconds: timedelta | None = None,
         metadata: MessageMetadata = None,
-        progress_callback: ProgressCallbackFnT | None = None,
+        progress_callback: ProgressFnT | None = None,
     ) -> ReceiveResultT:
         """
         Sends a request and wait for a response. Raises an McpError if the