✨ feat: add message_group_id

mic1on · Nov 20, 2023 · c4b3595 · c4b3595
1 parent 6ad9e2f
commit c4b3595
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/src/onestep/broker/sqs.py b/src/onestep/broker/sqs.py
@@ -12,10 +12,9 @@
 
 class SQSBroker(BaseBroker):
 
-    def __init__(self, queue_name, params: Optional[Dict] = None, prefetch: Optional[int] = 1,
-                 auto_create=True, queue_params: Optional[Dict] = None, message_group_id=None,
-                 *args,
-                 **kwargs):
+    def __init__(self, queue_name, message_group_id, params: Optional[Dict] = None, prefetch: Optional[int] = 1,
+                 auto_create=True, queue_params: Optional[Dict] = None,
+                 *args, **kwargs):
         super().__init__(*args, **kwargs)
         self.queue_name = queue_name
         self.queue = Queue()