🐞 fix: if message's type not str

mic1on · Nov 1, 2023 · 7af9003 · 7af9003
1 parent bcf9dac
commit 7af9003
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/src/onestep/broker/base.py b/src/onestep/broker/base.py
@@ -157,10 +157,13 @@ def __str__(self):
 class BaseLocalConsumer(BaseConsumer):
 
     def _to_message(self, data: Any):
-        try:
-            message = json.loads(data)
-        except json.JSONDecodeError:
-            message = {"body": data}
+        if isinstance(data, (str, bytes, bytearray)):
+            try:
+                message = json.loads(data)
+            except json.JSONDecodeError:
+                message = {"body": data}
+        else:
+            message = data
         if not isinstance(message, dict):
             message = {"body": message}
         if "body" not in message: