fix imports

memphis/consumer.py

@@ -5,6 +5,7 @@
 
 from memphis.exceptions import MemphisError
 from memphis.utils import default_error_handler, get_internal_name
+from memphis.message import Message
 
 
 class Consumer:
@@ -157,7 +158,7 @@ async def __ping_consumer(self, callback):
                 )
 
             except Exception as e:
-                callback(e)
+                callback(MemphisError(str(e)))
 
     async def destroy(self):
         """Destroy the consumer."""

memphis/memphis.py

@@ -33,10 +33,9 @@
 from memphis.exceptions import MemphisConnectError, MemphisError, MemphisHeaderError
 from memphis.headers import Headers
 from memphis.producer import Producer
+from memphis.station import Station
 from memphis.types import Retention, Storage
-from memphis.utils import get_internal_name
-
-schemaVFailAlertType = "schema_validation_fail_alert"
+from memphis.utils import get_internal_name, random_bytes
 
 
 class Memphis:
@@ -198,9 +197,9 @@ async def station(
 
             createStationReq = {
                 "name": name,
-                "retention_type": retention_type,
+                "retention_type": retention_type.value,
                 "retention_value": retention_value,
-                "storage_type": storage_type,
+                "storage_type": storage_type.value,
                 "replicas": replicas,
                 "idempotency_window_in_ms": idempotency_window_ms,
                 "schema_name": schema_name,

memphis/producer.py

@@ -13,6 +13,8 @@
 from memphis.headers import Headers
 from memphis.utils import get_internal_name
 
+schemaVFailAlertType = "schema_validation_fail_alert"
+
 
 class Producer:
     def __init__(