Narrow message_handler's parameter to notifications and exceptions (#3168)

examples/stories/stickynotes/client.py

@@ -4,7 +4,7 @@ import anyio
 import mcp_types as types
 from mcp_types.version import HANDSHAKE_PROTOCOL_VERSIONS
 
-from mcp.client import Client, ClientRequestContext
+from mcp.client import Client, ClientRequestContext, IncomingMessage
 from stories._harness import Target, run_client
 
 
@@ -18,7 +18,7 @@ async def main(target: Target, *, mode: str = "auto") -> None:
             return types.ElicitResult(action="cancel")
         return types.ElicitResult(action="accept", content={"confirm": answer == "confirm"})
 
-    async def on_message(message: object) -> None:
+    async def on_message(message: IncomingMessage) -> None:
         if isinstance(message, types.ResourceListChangedNotification):
             list_changed.set()