use defined types in decorator

Author: davemssavage

src/mcp/server/lowlevel/server.py

@@ -85,7 +85,7 @@ async def main():
 from mcp.shared.context import RequestContext
 from mcp.shared.exceptions import McpError
 from mcp.shared.message import SessionMessage
-from mcp.shared.session import RequestResponder
+from mcp.shared.session import RequestId, RequestResponder
 
 logger = logging.getLogger(__name__)
 
@@ -427,7 +427,7 @@ async def handler(req: types.CallToolRequest):
 
     def progress_notification(self):
         def decorator(
-            func: Callable[[str | int, float, float | None], Awaitable[None]],
+            func: Callable[[types.ProgressToken, float, float | None], Awaitable[None]],
         ):
             logger.debug("Registering handler for ProgressNotification")
 
@@ -443,7 +443,7 @@ async def handler(req: types.ProgressNotification):
 
     def cancel_notification(self):
         def decorator(
-            func: Callable[[str | int, str | None], Awaitable[None]],
+            func: Callable[[RequestId, str | None], Awaitable[None]],
         ):
             logger.debug("Registering handler for CancelledNotification")