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

…airbytehq/airbyte-python-cdk into lazebnyi/add-state-delegating-retriever
airbytehq · Feb 6, 2025 · e46a88a · e46a88a
2 parents ff57a28 + f85a68e
commit e46a88a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 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
@@ -2588,7 +2588,11 @@ def create_state_delegating_retriever(
             transformations=transformations,
         )
 
-        return StateDelegatingRetriever(full_data_retriever=full_data_retriever, incremental_data_retriever=incremental_data_retriever, cursor=stream_slicer)
+        return StateDelegatingRetriever(
+            full_data_retriever=full_data_retriever,
+            incremental_data_retriever=incremental_data_retriever,
+            cursor=stream_slicer,
+        )
 
     def _create_async_job_status_mapping(
         self, model: AsyncJobStatusMapModel, config: Config, **kwargs: Any

diff --git a/airbyte_cdk/sources/declarative/retrievers/state_delegating_retriever.py b/airbyte_cdk/sources/declarative/retrievers/state_delegating_retriever.py
@@ -2,11 +2,11 @@
 # Copyright (c) 2025 Airbyte, Inc., all rights reserved.
 #
 
+from dataclasses import dataclass
 from typing import (
     Any,
     MutableMapping,
 )
-from dataclasses import dataclass
 
 from typing_extensions import deprecated
 
@@ -48,7 +48,11 @@ def __setattr__(self, name: str, value: Any) -> None:
 
     @property
     def retriever(self) -> Retriever:
-        return self.incremental_data_retriever if self.cursor.get_stream_state() else self.full_data_retriever
+        return (
+            self.incremental_data_retriever
+            if self.cursor.get_stream_state()
+            else self.full_data_retriever
+        )
 
     @property
     def state(self) -> MutableMapping[str, Any]: