[airbyte-cdk]: add condition to check type retriever (#39309)

airbytehq · Jun 10, 2024 · 4e91046 · 4e91046
1 parent 86549e8
commit 4e91046
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py b/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py
@@ -671,7 +671,11 @@ def create_declarative_stream(self, model: DeclarativeStreamModel, config: Confi
 
     def _merge_stream_slicers(self, model: DeclarativeStreamModel, config: Config) -> Optional[StreamSlicer]:
         stream_slicer = None
-        if hasattr(model.retriever, "partition_router") and model.retriever.partition_router:
+        if (
+            hasattr(model.retriever, "partition_router")
+            and isinstance(model.retriever, SimpleRetrieverModel)
+            and model.retriever.partition_router
+        ):
             stream_slicer_model = model.retriever.partition_router
             if isinstance(stream_slicer_model, list):
                 stream_slicer = CartesianProductStreamSlicer(