@@ -2062,6 +2062,7 @@ def create_default_stream(
2062
2062
primary_key = primary_key ,
2063
2063
request_options_provider = request_options_provider ,
2064
2064
stream_slicer = stream_slicer ,
2065
+ partition_router = partition_router ,
2065
2066
stop_condition_cursor = concurrent_cursor
2066
2067
if self ._is_stop_condition_on_cursor (model )
2067
2068
else None ,
@@ -2524,7 +2525,7 @@ def create_dynamic_schema_loader(
2524
2525
config = config ,
2525
2526
name = name ,
2526
2527
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 ),
2528
2529
transformations = [],
2529
2530
use_cache = True ,
2530
2531
log_formatter = (
@@ -3139,6 +3140,7 @@ def create_simple_retriever(
3139
3140
] = None ,
3140
3141
use_cache : Optional [bool ] = None ,
3141
3142
log_formatter : Optional [Callable [[Response ], Any ]] = None ,
3143
+ partition_router : Optional [PartitionRouter ] = None ,
3142
3144
** kwargs : Any ,
3143
3145
) -> SimpleRetriever :
3144
3146
def _get_url (req : Requester ) -> str :
@@ -3236,6 +3238,8 @@ def _get_url(req: Requester) -> str:
3236
3238
3237
3239
if not request_options_provider :
3238
3240
request_options_provider = DefaultRequestOptionsProvider (parameters = {})
3241
+ if isinstance (request_options_provider , DefaultRequestOptionsProvider ) and isinstance (partition_router , PartitionRouter ):
3242
+ request_options_provider = partition_router
3239
3243
3240
3244
paginator = (
3241
3245
self ._create_component_from_model (
0 commit comments