From 540861a2a054f3518ef1c26667d54f60926747b4 Mon Sep 17 00:00:00 2001
From: shay23b <shay@memphis.dev>
Date: Thu, 2 Mar 2023 12:55:17 +0200
Subject: [PATCH] fix imports

---
 memphis/consumer.py | 3 ++-
 memphis/memphis.py  | 9 ++++-----
 memphis/producer.py | 2 ++
 3 files changed, 8 insertions(+), 6 deletions(-)

diff --git a/memphis/consumer.py b/memphis/consumer.py
index aad4625..598155c 100644
--- a/memphis/consumer.py
+++ b/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."""
diff --git a/memphis/memphis.py b/memphis/memphis.py
index 0d15ccd..cc0d91b 100644
--- a/memphis/memphis.py
+++ b/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,
diff --git a/memphis/producer.py b/memphis/producer.py
index 1c8eded..204c6d2 100644
--- a/memphis/producer.py
+++ b/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__(