fix: correct type signature of BasicHost.run method

Author: acul71

libp2p/abc.py

@@ -8,6 +8,7 @@
     KeysView,
     Sequence,
 )
+from contextlib import AbstractAsyncContextManager
 from types import (
     TracebackType,
 )
@@ -1178,7 +1179,9 @@ def get_live_peers(self) -> list[ID]:
         """
 
     @abstractmethod
-    def run(self, listen_addrs: Sequence[Multiaddr]) -> AsyncContextManager[None]:
+    def run(
+        self, listen_addrs: Sequence[Multiaddr]
+    ) -> AbstractAsyncContextManager[None]:
         """
         Run the host and start listening on the specified multiaddresses.
 

libp2p/host/basic_host.py

@@ -3,6 +3,7 @@
     Sequence,
 )
 from contextlib import (
+    AbstractAsyncContextManager,
     asynccontextmanager,
 )
 import logging
@@ -147,19 +148,23 @@ def get_connected_peers(self) -> list[ID]:
         """
         return list(self._network.connections.keys())
 
-    @asynccontextmanager
-    async def run(
+    def run(
         self, listen_addrs: Sequence[multiaddr.Multiaddr]
-    ) -> AsyncIterator[None]:
+    ) -> AbstractAsyncContextManager[None]:
         """
         Run the host instance and listen to ``listen_addrs``.
 
         :param listen_addrs: a sequence of multiaddrs that we want to listen to
         """
-        network = self.get_network()
-        async with background_trio_service(network):
-            await network.listen(*listen_addrs)
-            yield
+
+        @asynccontextmanager
+        async def _run() -> AsyncIterator[None]:
+            network = self.get_network()
+            async with background_trio_service(network):
+                await network.listen(*listen_addrs)
+                yield
+
+        return _run()
 
     def set_stream_handler(
         self, protocol_id: TProtocol, stream_handler: StreamHandlerFn