MINOR: Kinesis missing nexttoken

This commit is contained in:
Sasha Malahov 2025-04-16 09:27:31 -04:00 committed by GitHub
parent 654529ab7a
commit 105ba064a9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -185,6 +185,9 @@ class KinesisSource(MessagingServiceSource):
try:
while has_more_partitions:
partitions = self.kinesis.list_shards(**args.dict())
# Handle the case when NextToken is not present
if "NextToken" not in partitions:
partitions["NextToken"] = None
kinesis_partitions_model = KinesisPartitions(**partitions)
all_partitions.extend(
[