Make context reference safe

NeonGeckoCom · Aug 11, 2023 · 3594f47 · 3594f47
1 parent 75d6384
commit 3594f47
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neon_mq_connector/connector.py b/neon_mq_connector/connector.py
@@ -369,7 +369,7 @@ def emit_mq_message(cls,
             raise ValueError(f'No request data provided')
 
         # Ensure `message_id` in data will match context in messagebus connector
-        request_data.setdefault('message_id', request_data["context"]
+        request_data.setdefault('message_id', request_data.get("context", {})
                                 .get("mq", {}).get("message_id") or
                                 cls.create_unique_id())