diff --git a/homeassistant/components/apache_kafka/__init__.py b/homeassistant/components/apache_kafka/__init__.py
index 454b748dcc2..8be70458a8f 100644
--- a/homeassistant/components/apache_kafka/__init__.py
+++ b/homeassistant/components/apache_kafka/__init__.py
@@ -141,7 +141,8 @@ class KafkaManager:
 
     async def write(self, event: Event[EventStateChangedData]) -> None:
         """Write a binary payload to Kafka."""
+        key = event.data["entity_id"].encode("utf-8")
         payload = self._encode_event(event)
 
         if payload:
-            await self._producer.send_and_wait(self._topic, payload)
+            await self._producer.send_and_wait(self._topic, payload, key)