@@ -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