only add env header if it exists

taytzehao · Jan 6, 2022 · 527de37 · 527de37
1 parent f00148b
commit 527de37
Showing 1 changed file with 8 additions and 6 deletions.
diff --git a/rasa/core/brokers/kafka.py b/rasa/core/brokers/kafka.py
@@ -192,12 +192,14 @@ def _publish(self, event: Dict[Text, Any]) -> None:
         else:
             partition_key = None
 
-        headers = [
-            (
-                "RASA_ENVIRONMENT",
-                bytes(self.rasa_environment, encoding=DEFAULT_ENCODING),
-            )
-        ]
+        headers = []
+        if self.rasa_environment:
+            headers = [
+                (
+                    "RASA_ENVIRONMENT",
+                    bytes(self.rasa_environment, encoding=DEFAULT_ENCODING),
+                )
+            ]
 
         logger.debug(
             f"Calling kafka send({self.topic}, value={event},"