Skip to content

Commit f14aed6

Browse files
committed
Update partition_router validation for asynx retriever
1 parent b79177b commit f14aed6

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -1900,6 +1900,8 @@ def _merge_stream_slicers(
19001900
) -> Optional[StreamSlicer]:
19011901
retriever_model = model.retriever
19021902

1903+
stream_slicer = self._build_stream_slicer_from_partition_router(retriever_model, config)
1904+
19031905
if retriever_model.type == "AsyncRetriever":
19041906
is_not_datetime_cursor = (
19051907
model.incremental_sync.type != "DatetimeBasedCursor"
@@ -1919,13 +1921,11 @@ def _merge_stream_slicers(
19191921
"AsyncRetriever with cursor other than DatetimeBasedCursor is not supported yet."
19201922
)
19211923

1922-
if is_partition_router:
1924+
if is_partition_router and not stream_slicer:
19231925
# Note that this development is also done in parallel to the per partition development which once merged
19241926
# we could support here by calling create_concurrent_cursor_from_perpartition_cursor
19251927
raise ValueError("Per partition state is not supported yet for AsyncRetriever.")
19261928

1927-
stream_slicer = self._build_stream_slicer_from_partition_router(retriever_model, config)
1928-
19291929
if model.incremental_sync:
19301930
return self._build_incremental_cursor(model, stream_slicer, config)
19311931

0 commit comments

Comments
 (0)