Merge pull request #10634 from RasaHQ/fix_kafka_headers

Only add rasa_environment header to kafka message if it is set
RasaHQ · Jan 6, 2022 · fb0cb86 · fb0cb86
2 parents f00148b + 33ea59a
commit fb0cb86
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 6 deletions.
diff --git a/changelog/10634.bugfix.md b/changelog/10634.bugfix.md
@@ -0,0 +1 @@
+Add `RASA_ENVIRONMENT` header in Kafka only if the environmental variable is set.
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},"
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		Add `RASA_ENVIRONMENT` header in Kafka only if the environmental variable is set.