Merge branch 'lazebnyi/add-state-delegating-retriever' of github.com:…

…airbytehq/airbyte-python-cdk into lazebnyi/add-state-delegating-retriever
airbytehq · Mar 7, 2025 · 5bf46a7 · 5bf46a7
2 parents 63b156e + 571ffa9
commit 5bf46a7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py b/airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py
@@ -2766,7 +2766,9 @@ def create_state_delegating_retriever(
         client_side_incremental_sync: Optional[Dict[str, Any]] = None,
         transformations: List[RecordTransformation],
     ) -> Optional[StateDelegatingRetriever]:
-        if not isinstance(stream_slicer, DatetimeBasedCursor) and not isinstance(stream_slicer, PerPartitionCursor):
+        if not isinstance(stream_slicer, DatetimeBasedCursor) and not isinstance(
+            stream_slicer, PerPartitionCursor
+        ):
             raise ValueError("StateDelegatingRetriever requires a DatetimeBasedCursor")
 
         full_refresh_retriever = self._create_component_from_model(
@@ -2797,7 +2799,7 @@ def create_state_delegating_retriever(
             full_data_retriever=full_refresh_retriever,
             incremental_data_retriever=incremental_retriever,
             cursor=stream_slicer,
-            started_with_state=bool(self._connector_state_manager.get_stream_state(name, None))
+            started_with_state=bool(self._connector_state_manager.get_stream_state(name, None)),
         )
 
     def _create_async_job_status_mapping(

diff --git a/airbyte_cdk/sources/declarative/retrievers/state_delegating_retriever.py b/airbyte_cdk/sources/declarative/retrievers/state_delegating_retriever.py
@@ -56,9 +56,7 @@ def __setattr__(self, name: str, value: Any) -> None:
     @property
     def retriever(self) -> Retriever:
         return (
-            self.incremental_data_retriever
-            if self.started_with_state
-            else self.full_data_retriever
+            self.incremental_data_retriever if self.started_with_state else self.full_data_retriever
         )
 
     @property