Skip to content

Commit 9e6368f

Browse files
author
maxime.c
committed
fix case where request option provider is stream slicer
1 parent e5ecf41 commit 9e6368f

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2061,7 +2061,7 @@ def create_default_stream(
20612061
name=model.name,
20622062
primary_key=primary_key,
20632063
request_options_provider=request_options_provider,
2064-
stream_slicer=stream_slicer,
2064+
partition_router=partition_router,
20652065
stop_condition_cursor=concurrent_cursor
20662066
if self._is_stop_condition_on_cursor(model)
20672067
else None,
@@ -3139,6 +3139,7 @@ def create_simple_retriever(
31393139
] = None,
31403140
use_cache: Optional[bool] = None,
31413141
log_formatter: Optional[Callable[[Response], Any]] = None,
3142+
partition_router: Optional[PartitionRouter] = None,
31423143
**kwargs: Any,
31433144
) -> SimpleRetriever:
31443145
def _get_url(req: Requester) -> str:
@@ -3236,6 +3237,8 @@ def _get_url(req: Requester) -> str:
32363237

32373238
if not request_options_provider:
32383239
request_options_provider = DefaultRequestOptionsProvider(parameters={})
3240+
if isinstance(request_options_provider, DefaultRequestOptionsProvider) and isinstance(partition_router, PartitionRouter):
3241+
request_options_provider = partition_router
32393242

32403243
paginator = (
32413244
self._create_component_from_model(

0 commit comments

Comments
 (0)