From 4dde24ff15675646e9a8cb5b7fe5509d5844c4eb Mon Sep 17 00:00:00 2001 From: Tomasz Szuba Date: Thu, 2 Jan 2025 11:32:04 +0100 Subject: [PATCH] Pass `extra_fields` in `global_substream_cursor` --- .../declarative/incremental/global_substream_cursor.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/airbyte_cdk/sources/declarative/incremental/global_substream_cursor.py b/airbyte_cdk/sources/declarative/incremental/global_substream_cursor.py index 3b3636236..856774e01 100644 --- a/airbyte_cdk/sources/declarative/incremental/global_substream_cursor.py +++ b/airbyte_cdk/sources/declarative/incremental/global_substream_cursor.py @@ -112,7 +112,7 @@ def stream_slices(self) -> Iterable[StreamSlice]: * Yield the last slice. At that point, once there are as many slices yielded as closes, the global slice will be closed too """ slice_generator = ( - StreamSlice(partition=partition, cursor_slice=cursor_slice) + StreamSlice(partition=partition, cursor_slice=cursor_slice, extra_fields=partition.extra_fields) for partition in self._partition_router.stream_slices() for cursor_slice in self._stream_cursor.stream_slices() ) @@ -128,7 +128,7 @@ def stream_slices(self) -> Iterable[StreamSlice]: def generate_slices_from_partition(self, partition: StreamSlice) -> Iterable[StreamSlice]: slice_generator = ( - StreamSlice(partition=partition, cursor_slice=cursor_slice) + StreamSlice(partition=partition, cursor_slice=cursor_slice, extra_fields=partition.extra_fields) for cursor_slice in self._stream_cursor.stream_slices() )