Skip to content

Commit f2f1363

Browse files
committed
fix case where request option provider is stream slicer
1 parent e5ecf41 commit f2f1363

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2062,6 +2062,7 @@ def create_default_stream(
20622062
primary_key=primary_key,
20632063
request_options_provider=request_options_provider,
20642064
stream_slicer=stream_slicer,
2065+
partition_router=partition_router,
20652066
stop_condition_cursor=concurrent_cursor
20662067
if self._is_stop_condition_on_cursor(model)
20672068
else None,
@@ -2524,7 +2525,7 @@ def create_dynamic_schema_loader(
25242525
config=config,
25252526
name=name,
25262527
primary_key=None,
2527-
stream_slicer=self._build_stream_slicer_from_partition_router(model.retriever, config),
2528+
partition_router=self._build_stream_slicer_from_partition_router(model.retriever, config),
25282529
transformations=[],
25292530
use_cache=True,
25302531
log_formatter=(
@@ -3139,6 +3140,7 @@ def create_simple_retriever(
31393140
] = None,
31403141
use_cache: Optional[bool] = None,
31413142
log_formatter: Optional[Callable[[Response], Any]] = None,
3143+
partition_router: Optional[PartitionRouter] = None,
31423144
**kwargs: Any,
31433145
) -> SimpleRetriever:
31443146
def _get_url(req: Requester) -> str:
@@ -3236,6 +3238,8 @@ def _get_url(req: Requester) -> str:
32363238

32373239
if not request_options_provider:
32383240
request_options_provider = DefaultRequestOptionsProvider(parameters={})
3241+
if isinstance(request_options_provider, DefaultRequestOptionsProvider) and isinstance(partition_router, PartitionRouter):
3242+
request_options_provider = partition_router
32393243

32403244
paginator = (
32413245
self._create_component_from_model(

0 commit comments

Comments
 (0)