|
85 | 85 | from airbyte_cdk.sources.streams.core import StreamData |
86 | 86 | from airbyte_cdk.sources.types import Config, Record, StreamSlice, StreamState |
87 | 87 | from airbyte_cdk.utils.mapping_helpers import combine_mappings |
| 88 | +from airbyte_cdk.sources.streams.concurrent.cursor import Cursor |
| 89 | + |
88 | 90 |
|
89 | 91 | FULL_REFRESH_SYNC_COMPLETE_KEY = "__ab_full_refresh_sync_complete" |
90 | 92 |
|
@@ -187,7 +189,7 @@ def resolve_dependencies( |
187 | 189 | primary_key: Optional[Union[str, List[str], List[List[str]]]] = None, |
188 | 190 | stream_slicer: Optional[StreamSlicer] = None, |
189 | 191 | request_options_provider: Optional[RequestOptionsProvider] = None, |
190 | | - stop_condition_on_cursor: bool = False, |
| 192 | + stop_condition_cursor: Optional[Cursor] = None, |
191 | 193 | client_side_incremental_sync: Optional[Dict[str, Any]] = None, |
192 | 194 | transformations: Optional[List[RecordTransformation]] = None, |
193 | 195 | file_uploader: Optional[DefaultFileUploader] = None, |
@@ -356,15 +358,14 @@ def _get_log_formatter( |
356 | 358 | ), |
357 | 359 | ) |
358 | 360 |
|
359 | | - cursor_used_for_stop_condition = cursor if stop_condition_on_cursor else None |
360 | 361 | paginator = ( |
361 | 362 | dependency_constructor( |
362 | 363 | model=model.paginator, |
363 | 364 | config=config, |
364 | 365 | url_base=_get_url(), |
365 | 366 | extractor_model=model.record_selector.extractor, |
366 | 367 | decoder=decoder, |
367 | | - cursor_used_for_stop_condition=cursor_used_for_stop_condition, |
| 368 | + cursor_used_for_stop_condition=stop_condition_cursor or None, |
368 | 369 | ) |
369 | 370 | if model.paginator |
370 | 371 | else NoPagination(parameters={}) |
|
0 commit comments