File tree Expand file tree Collapse file tree 1 file changed +4
-1
lines changed
airbyte_cdk/sources/declarative/parsers Expand file tree Collapse file tree 1 file changed +4
-1
lines changed Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments