Added uvloop for better async for UNIX systems

Author: sudoRicheek

benchmarks/bench_latency.py

@@ -20,6 +20,7 @@
 
 sys.path.insert(0, str(Path(__file__).parent.parent / "src"))
 
+import cortex
 from cortex.core.publisher import Publisher
 from cortex.core.subscriber import Subscriber
 from cortex.discovery.daemon import DiscoveryDaemon
@@ -103,9 +104,16 @@ async def subscriber_main():
             message_type=LatencyMessage,
             node_name="latency_subscriber",
             wait_for_topic=True,
-            topic_timeout=10.0,
+            topic_timeout=30.0,
         )
 
+        # Wait for topic to be available before signaling ready
+        if not sub.is_connected:
+            connected = await sub._async_connect()
+            if not connected:
+                results_queue.put({"received": 0, "latencies": [], "error": "timeout"})
+                return
+
         # Signal ready
         ready_event.set()
 
@@ -138,7 +146,7 @@ async def subscriber_main():
             }
         )
 
-    asyncio.run(subscriber_main())
+    cortex.run(subscriber_main())
 
 
 def run_latency_benchmark(

benchmarks/bench_throughput.py

@@ -16,6 +16,7 @@
 
 import numpy as np
 
+import cortex
 from cortex import Publisher, Subscriber
 from cortex.discovery import DiscoveryDaemon
 from cortex.messages import ArrayMessage
@@ -121,7 +122,7 @@ async def run_sub():
                     except Exception:
                         break
 
-            asyncio.run(run_sub())
+            cortex.run(run_sub())
 
         sub_thread = threading.Thread(target=subscriber_loop, daemon=True)
         sub_thread.start()

examples/multi_node_system.py

@@ -23,6 +23,7 @@
 
 import numpy as np
 
+import cortex
 from cortex import Message, Node
 from cortex.messages.base import MessageHeader
 
@@ -236,4 +237,4 @@ async def main() -> None:
 
 if __name__ == "__main__":
     with contextlib.suppress(KeyboardInterrupt):
-        asyncio.run(main())
+        cortex.run(main())

examples/publisher_dict.py

@@ -22,6 +22,7 @@
 
 import numpy as np
 
+import cortex
 from cortex import DictMessage, Node
 
 
@@ -108,4 +109,4 @@ async def main() -> None:
 
 if __name__ == "__main__":
     with contextlib.suppress(KeyboardInterrupt):
-        asyncio.run(main())
+        cortex.run(main())

examples/publisher_numpy.py

@@ -16,10 +16,9 @@
     python examples/subscriber_numpy.py
 """
 
-import asyncio
-
 import numpy as np
 
+import cortex
 from cortex import ArrayMessage, Node
 
 
@@ -84,4 +83,4 @@ async def main():
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    cortex.run(main())

examples/publisher_tensor.py

@@ -16,14 +16,13 @@
     python examples/subscriber_tensor.py
 """
 
-import asyncio
-
 try:
     import torch
 except ImportError:
     print("This example requires PyTorch. Install with: pip install torch")
     exit(1)
 
+import cortex
 from cortex import Node, TensorMessage
 
 
@@ -84,4 +83,4 @@ async def main():
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    cortex.run(main())

examples/subscriber_dict.py

@@ -19,6 +19,7 @@
 import asyncio
 import contextlib
 
+import cortex
 from cortex import DictMessage, Node
 from cortex.messages.base import MessageHeader
 
@@ -28,19 +29,17 @@ class DictSubscriberNode(Node):
 
     def __init__(self) -> None:
         super().__init__(name="dict_subscriber")
+        # Create a subscriber - connection happens asynchronously in run()
         print("Waiting for publisher on /robot/state...")
-        sub = self.create_subscriber(
+        self.create_subscriber(
             topic_name="/robot/state",
             message_type=DictMessage,
             callback=self._on_state_received,
             wait_for_topic=True,
             topic_timeout=30.0,
         )
 
-        if not sub.is_connected:
-            raise RuntimeError("Failed to connect to topic!")
-
-        print("Connected! Receiving messages...")
+        print("Subscriber created, will connect when run() is called...")
         print("Press Ctrl+C to stop")
         print()
 
@@ -78,11 +77,7 @@ async def main() -> None:
     """Run the dictionary subscriber example."""
     print("Starting dictionary subscriber...")
 
-    try:
-        node = DictSubscriberNode()
-    except RuntimeError as e:
-        print(e)
-        return
+    node = DictSubscriberNode()
 
     try:
         await node.run()
@@ -95,4 +90,4 @@ async def main() -> None:
 
 if __name__ == "__main__":
     with contextlib.suppress(KeyboardInterrupt):
-        asyncio.run(main())
+        cortex.run(main())

examples/subscriber_numpy.py

@@ -16,8 +16,7 @@
     python examples/subscriber_numpy.py
 """
 
-import asyncio
-
+import cortex
 from cortex import ArrayMessage, Node
 from cortex.messages.base import MessageHeader
 
@@ -41,7 +40,7 @@ class ArraySubscriberNode(Node):
     def __init__(self):
         super().__init__(name="array_subscriber")
 
-        # Create a subscriber
+        # Create a subscriber - connection happens asynchronously in run()
         print("Waiting for publisher on /sensor/array_data...")
         self.sub = self.create_subscriber(
             topic_name="/sensor/array_data",
@@ -51,10 +50,7 @@ def __init__(self):
             topic_timeout=30.0,
         )
 
-        if not self.sub.is_connected:
-            raise RuntimeError("Failed to connect to topic!")
-
-        print("Connected! Receiving messages...")
+        print("Subscriber created, will connect when run() is called...")
         print("Press Ctrl+C to stop")
         print()
 
@@ -63,17 +59,15 @@ async def main():
     """Run the subscriber example."""
     print("Starting NumPy array subscriber...")
 
+    node = ArraySubscriberNode()
+
     try:
-        node = ArraySubscriberNode()
         await node.run()
-    except RuntimeError as e:
-        print(f"Error: {e}")
     except KeyboardInterrupt:
         print("\nShutting down...")
     finally:
-        if "node" in locals():
-            await node.close()
+        await node.close()
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    cortex.run(main())

examples/subscriber_tensor.py

@@ -16,14 +16,13 @@
     python examples/subscriber_tensor.py
 """
 
-import asyncio
-
 try:
     import torch
 except ImportError:
     print("This example requires PyTorch. Install with: pip install torch")
     exit(1)
 
+import cortex
 from cortex import Node, TensorMessage
 from cortex.messages.base import MessageHeader
 
@@ -50,7 +49,7 @@ class TensorSubscriberNode(Node):
     def __init__(self):
         super().__init__(name="tensor_subscriber")
 
-        # Create a subscriber
+        # Create a subscriber - connection happens asynchronously in run()
         print("Waiting for publisher on /model/features...")
         self.sub = self.create_subscriber(
             topic_name="/model/features",
@@ -60,10 +59,7 @@ def __init__(self):
             topic_timeout=30.0,
         )
 
-        if not self.sub.is_connected:
-            raise RuntimeError("Failed to connect to topic!")
-
-        print("Connected! Receiving messages...")
+        print("Subscriber created, will connect when run() is called...")
         print("Press Ctrl+C to stop")
         print()
 
@@ -73,17 +69,15 @@ async def main():
     print("Starting PyTorch tensor subscriber...")
     print(f"PyTorch version: {torch.__version__}")
 
+    node = TensorSubscriberNode()
+
     try:
-        node = TensorSubscriberNode()
         await node.run()
-    except RuntimeError as e:
-        print(f"Error: {e}")
     except KeyboardInterrupt:
         print("\nShutting down...")
     finally:
-        if "node" in locals():
-            await node.close()
+        await node.close()
 
 
 if __name__ == "__main__":
-    asyncio.run(main())
+    cortex.run(main())

pyproject.toml

@@ -7,10 +7,10 @@ name = "cortex"
 version = "0.1.0"
 description = "A lightweight framework using ZeroMQ for inter-process communication"
 readme = "README.md"
-license = {text = "MIT"}
+license = {text = "Apache-2.0"}
 requires-python = ">=3.10"
 authors = [
-    {name = "Cortex Authors"}
+    {name = "Richeek Das", email = "richeek@seas.upenn.edu"}
 ]
 classifiers = [
     "Development Status :: 3 - Alpha",
@@ -20,11 +20,14 @@ classifiers = [
     "Programming Language :: Python :: 3.10",
     "Programming Language :: Python :: 3.11",
     "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
 ]
 dependencies = [
     "pyzmq>=27.0.0",
     "numpy>=1.26.4",
     "msgpack>=1.0.0",
+    "uvloop>=0.19.0; sys_platform != 'win32'",
 ]
 
 [project.optional-dependencies]