refactor: move inline imports to module level in default_stream.py and concurrent_declarative_source.py

Co-Authored-By: unknown <>

Author: devin-ai-integration[bot]

airbyte_cdk/sources/declarative/concurrent_declarative_source.py

@@ -76,6 +76,9 @@
 from airbyte_cdk.sources.declarative.parsers.model_to_component_factory import (
     ModelToComponentFactory,
 )
+from airbyte_cdk.sources.declarative.partition_routers.substream_partition_router import (
+    SubstreamPartitionRouter,
+)
 from airbyte_cdk.sources.declarative.resolvers import COMPONENTS_RESOLVER_TYPE_MAPPING
 from airbyte_cdk.sources.declarative.spec.spec import Spec
 from airbyte_cdk.sources.declarative.types import Config, ConnectionDefinition
@@ -433,10 +436,6 @@ def _apply_stream_groups(self, streams: List[AbstractStream]) -> None:
         against actual created stream instances by name. Validates that no stream shares a
         group with any of its parent streams, which would cause a deadlock.
         """
-        from airbyte_cdk.sources.declarative.partition_routers.substream_partition_router import (
-            SubstreamPartitionRouter,
-        )
-
         stream_groups = self._source_config.get("stream_groups", {})
         if not stream_groups:
             return

airbyte_cdk/sources/streams/concurrent/default_stream.py

@@ -3,12 +3,16 @@
 #
 
 from logging import Logger
-from typing import TYPE_CHECKING, Any, Callable, Iterable, List, Mapping, Optional, Union
-
-if TYPE_CHECKING:
-    from airbyte_cdk.sources.declarative.partition_routers.partition_router import PartitionRouter
+from typing import Any, Callable, Iterable, List, Mapping, Optional, Union
 
 from airbyte_cdk.models import AirbyteStream, SyncMode
+from airbyte_cdk.sources.declarative.incremental.concurrent_partition_cursor import (
+    ConcurrentPerPartitionCursor,
+)
+from airbyte_cdk.sources.declarative.partition_routers.partition_router import PartitionRouter
+from airbyte_cdk.sources.declarative.stream_slicers.declarative_partition_generator import (
+    StreamSlicerPartitionGenerator,
+)
 from airbyte_cdk.sources.streams.concurrent.abstract_stream import AbstractStream
 from airbyte_cdk.sources.streams.concurrent.availability_strategy import StreamAvailability
 from airbyte_cdk.sources.streams.concurrent.cursor import Cursor, CursorField
@@ -108,15 +112,8 @@ def block_simultaneous_read(self) -> str:
     def block_simultaneous_read(self, value: str) -> None:
         self._block_simultaneous_read = value
 
-    def get_partition_router(self) -> "PartitionRouter | None":
+    def get_partition_router(self) -> PartitionRouter | None:
         """Return the partition router for this stream, or None if not available."""
-        from airbyte_cdk.sources.declarative.incremental.concurrent_partition_cursor import (
-            ConcurrentPerPartitionCursor,
-        )
-        from airbyte_cdk.sources.declarative.stream_slicers.declarative_partition_generator import (
-            StreamSlicerPartitionGenerator,
-        )
-
         if not isinstance(self._stream_partition_generator, StreamSlicerPartitionGenerator):
             return None
         stream_slicer = self._stream_partition_generator._stream_slicer