fix: improve types according to new typecheck

Author: AkMo3

examples/doc-examples/example_net_stream.py

@@ -193,7 +193,7 @@ async def run_enhanced_demo(
             print("🖥️  ENHANCED ECHO SERVER MODE")
             print("=" * 60)
 
-            # Set the enhanced stream handler
+            # type: ignore: Stream is type of NetStream
             host.set_stream_handler(PROTOCOL_ID, enhanced_echo_handler)
 
             print(
@@ -217,7 +217,8 @@ async def run_enhanced_demo(
 
             # Create stream and run enhanced demo
             stream = await host.new_stream(info.peer_id, [PROTOCOL_ID])
-            await enhanced_client_demo(stream)
+            if isinstance(stream, NetStream):
+                await enhanced_client_demo(stream)
 
             print("\n" + "=" * 60)
             print("CLIENT DEMO COMPLETE")

libp2p/network/stream/net_stream.py

@@ -1,9 +1,6 @@
 from enum import (
     Enum,
 )
-from typing import (
-    Optional,
-)
 
 import trio
 
@@ -102,11 +99,11 @@ class NetStream(INetStream):
     """
 
     muxed_stream: IMuxedStream
-    protocol_id: Optional[TProtocol]
+    protocol_id: TProtocol | None
     __stream_state: StreamState
 
     def __init__(
-        self, muxed_stream: IMuxedStream, nursery: Optional[trio.Nursery] = None
+        self, muxed_stream: IMuxedStream, nursery: trio.Nursery | None = None
     ) -> None:
         super().__init__()
 
@@ -142,7 +139,7 @@ async def state(self) -> StreamState:
         async with self._state_lock:
             return self.__stream_state
 
-    async def read(self, n: Optional[int] = None) -> bytes:
+    async def read(self, n: int | None = None) -> bytes:
         """
         Read from stream.
 
@@ -279,7 +276,7 @@ async def _notify_closed(self) -> None:
                 if hasattr(swarm, "refs") and hasattr(swarm.refs, "done"):
                     swarm.refs.done()
 
-    def get_remote_address(self) -> Optional[tuple[str, int]]:
+    def get_remote_address(self) -> tuple[str, int] | None:
         """Delegate to the underlying muxed stream."""
         return self.muxed_stream.get_remote_address()