From 1cbe6ddc4db3f88815b7d949119375e90768e7ee Mon Sep 17 00:00:00 2001
From: Nick Macholl <nick@databento.com>
Date: Mon, 22 Jul 2024 09:50:33 -0700
Subject: [PATCH] MOD: Export BBOMsg and CBBOMsg

---
 CHANGELOG.md          |  1 +
 databento/__init__.py | 18 ++++++++++++------
 2 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 62fedb7..1e938b3 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -10,6 +10,7 @@ This release adds a new feature to the `Live` client for automatically reconnect
 - Added `Live.add_reconnect_callback` method for specifying a callback to handle client reconnections
 - Added platform information to the user agent reported by the `Historical` and `Live` clients
 - Upgraded `databento-dbn` to 0.19.1
+- Added `BBOMsg`, `CBBOMsg`, and `StatusMsg` exports to the root `databento` package
 
 #### Breaking changes
 - Calling `Live.stop` will now clear all user streams and callbacks
diff --git a/databento/__init__.py b/databento/__init__.py
index e8c2542..2b45028 100644
--- a/databento/__init__.py
+++ b/databento/__init__.py
@@ -1,6 +1,8 @@
 import logging
 import warnings
 
+from databento_dbn import BBOMsg
+from databento_dbn import CBBOMsg
 from databento_dbn import Compression
 from databento_dbn import Encoding
 from databento_dbn import ErrorMsg
@@ -14,6 +16,7 @@
 from databento_dbn import Schema
 from databento_dbn import StatMsg
 from databento_dbn import StatType
+from databento_dbn import StatusMsg
 from databento_dbn import SType
 from databento_dbn import SymbolMappingMsg
 from databento_dbn import SystemMsg
@@ -75,22 +78,25 @@
     "RollRule",
     "Schema",
     "SplitDuration",
+    "StatType",
     "SType",
     "SymbologyResolution",
     # DBN Record Types
     "Metadata",
+    "BBOMsg",
+    "CBBOMsg",
+    "ErrorMsg",
+    "ImbalanceMsg",
+    "InstrumentDefMsg",
     "MBOMsg",
     "MBP1Msg",
     "MBP10Msg",
-    "TradeMsg",
     "OHLCVMsg",
     "StatMsg",
-    "InstrumentDefMsg",
-    "ImbalanceMsg",
-    "ErrorMsg",
-    "StatType",
-    "SystemMsg",
+    "StatusMsg",
     "SymbolMappingMsg",
+    "SystemMsg",
+    "TradeMsg",
 ]
 
 # Setup logging