From a1488a1773350746b27603094a1fa17dbe82eec5 Mon Sep 17 00:00:00 2001
From: Tyler Goodlet <jgbt@protonmail.com>
Date: Sun, 13 Jun 2021 18:02:27 -0400
Subject: [PATCH] Expose streaming components at top level

---
 tractor/__init__.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/tractor/__init__.py b/tractor/__init__.py
index 9ff62cc..a7cadb9 100644
--- a/tractor/__init__.py
+++ b/tractor/__init__.py
@@ -15,7 +15,11 @@ from ._streaming import (
 from ._discovery import get_arbiter, find_actor, wait_for_actor
 from ._trionics import open_nursery
 from ._state import current_actor, is_root_process
-from ._exceptions import RemoteActorError, ModuleNotExposed
+from ._exceptions import (
+    RemoteActorError,
+    ModuleNotExposed,
+    ContextCancelled,
+)
 from ._debug import breakpoint, post_mortem
 from . import msg
 from ._root import run, run_daemon, open_root_actor
@@ -27,6 +31,7 @@ __all__ = [
     'ModuleNotExposed',
     'MultiError',
     'RemoteActorError',
+    'ContextCancelled',
     'breakpoint',
     'current_actor',
     'find_actor',
@@ -40,6 +45,8 @@ __all__ = [
     'run_daemon',
     'stream',
     'context',
+    'ReceiveMsgStream',
+    'MsgStream',
     'to_asyncio',
     'wait_for_actor',
 ]