fix partition key

Author: leandrodamascena

aws_lambda_powertools/utilities/batch/base.py

@@ -374,12 +374,12 @@ def _collect_dynamodb_failures(self):
     def _collect_kafka_failures(self):
         failures = []
         for msg in self.fail_messages:
-            # Kafka uses a composite identifier with topic-partition and offset
+            # Kafka uses a composite identifier with partition and offset
             # Both data class and Pydantic model use the same field names
             failures.append(
                 {
                     "itemIdentifier": {
-                        "topic-partition": f"{msg.topic}-{msg.partition}",
+                        "partition": f"{msg.topic}-{msg.partition}",
                         "offset": msg.offset,
                     },
                 },

aws_lambda_powertools/utilities/batch/types.py

@@ -29,9 +29,9 @@
 
 
 class KafkaItemIdentifier(TypedDict):
-    """Kafka uses a composite identifier with topic-partition and offset."""
+    """Kafka uses a composite identifier with partition and offset."""
 
-    topic_partition: str  # Maps to "topic-partition" in the actual response
+    partition: str
     offset: int
 
 
@@ -40,7 +40,7 @@ class PartialItemFailures(TypedDict):
     Represents a partial item failure response.
 
     For SQS, Kinesis, and DynamoDB: itemIdentifier is a string (message_id or sequence_number)
-    For Kafka: itemIdentifier is a KafkaItemIdentifier dict with topic-partition and offset
+    For Kafka: itemIdentifier is a KafkaItemIdentifier dict with partition and offset
     """
 
     itemIdentifier: str | KafkaItemIdentifier

tests/functional/batch/required_dependencies/test_utilities_batch_kafka.py

@@ -150,7 +150,7 @@ def test_kafka_batch_processor_partial_failure(self, kafka_event_factory, kafka_
         # THEN - Kafka uses composite identifier
         assert len(result["batchItemFailures"]) == 1
         assert result["batchItemFailures"][0]["itemIdentifier"] == {
-            "topic-partition": "mytopic-0",
+            "partition": "mytopic-0",
             "offset": 1,
         }
 
@@ -203,7 +203,7 @@ def test_kafka_batch_processor_multi_topic_partition(self, kafka_event_factory,
 
         # THEN
         assert len(result["batchItemFailures"]) == 2
-        topic_partitions = [f["itemIdentifier"]["topic-partition"] for f in result["batchItemFailures"]]
+        topic_partitions = [f["itemIdentifier"]["partition"] for f in result["batchItemFailures"]]
         assert "topic1-0" in topic_partitions
         assert "topic2-1" in topic_partitions
 
@@ -332,7 +332,7 @@ def test_async_kafka_batch_processor_partial_failure(self, kafka_event_factory,
         # THEN
         assert len(result["batchItemFailures"]) == 1
         assert result["batchItemFailures"][0]["itemIdentifier"] == {
-            "topic-partition": "mytopic-0",
+            "partition": "mytopic-0",
             "offset": 1,
         }