airbyte-cdk 6.7.0__py3-none-any.whl → 6.7.0.dev4__py3-none-any.whl

airbyte_cdk/sources/file_based/config/excel_format.py

@@ -2,9 +2,8 @@
 # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
 #
 
-from pydantic.v1 import BaseModel, Field
-
 from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
+from pydantic.v1 import BaseModel, Field
 
 
 class ExcelFormat(BaseModel):

airbyte_cdk/sources/file_based/config/file_based_stream_config.py

@@ -5,8 +5,6 @@
 from enum import Enum
 from typing import Any, List, Mapping, Optional, Union
 
-from pydantic.v1 import BaseModel, Field, validator
-
 from airbyte_cdk.sources.file_based.config.avro_format import AvroFormat
 from airbyte_cdk.sources.file_based.config.csv_format import CsvFormat
 from airbyte_cdk.sources.file_based.config.excel_format import ExcelFormat
@@ -15,6 +13,7 @@ from airbyte_cdk.sources.file_based.config.parquet_format import ParquetFormat
 from airbyte_cdk.sources.file_based.config.unstructured_format import UnstructuredFormat
 from airbyte_cdk.sources.file_based.exceptions import ConfigValidationError, FileBasedSourceError
 from airbyte_cdk.sources.file_based.schema_helpers import type_mapping_to_jsonschema
+from pydantic.v1 import BaseModel, Field, validator
 
 PrimaryKeyType = Optional[Union[str, List[str]]]
 

airbyte_cdk/sources/file_based/config/jsonl_format.py

@@ -2,9 +2,8 @@
 # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
 #
 
-from pydantic.v1 import BaseModel, Field
-
 from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
+from pydantic.v1 import BaseModel, Field
 
 
 class JsonlFormat(BaseModel):

airbyte_cdk/sources/file_based/config/parquet_format.py

@@ -3,9 +3,8 @@
 #
 
 
-from pydantic.v1 import BaseModel, Field
-
 from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
+from pydantic.v1 import BaseModel, Field
 
 
 class ParquetFormat(BaseModel):

airbyte_cdk/sources/file_based/config/unstructured_format.py

@@ -4,9 +4,8 @@
 
 from typing import List, Literal, Optional, Union
 
-from pydantic.v1 import BaseModel, Field
-
 from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
+from pydantic.v1 import BaseModel, Field
 
 
 class LocalProcessingConfigModel(BaseModel):

airbyte_cdk/sources/file_based/file_based_source.py

@@ -8,8 +8,6 @@ from abc import ABC
 from collections import Counter
 from typing import Any, Iterator, List, Mapping, Optional, Tuple, Type, Union
 
-from pydantic.v1.error_wrappers import ValidationError
-
 from airbyte_cdk.logger import AirbyteLogFormatter, init_logger
 from airbyte_cdk.models import (
     AirbyteMessage,
@@ -62,6 +60,7 @@ from airbyte_cdk.sources.streams import Stream
 from airbyte_cdk.sources.streams.concurrent.cursor import CursorField
 from airbyte_cdk.utils.analytics_message import create_analytics_message
 from airbyte_cdk.utils.traced_exception import AirbyteTracedException
+from pydantic.v1.error_wrappers import ValidationError
 
 DEFAULT_CONCURRENCY = 100
 MAX_CONCURRENCY = 100

airbyte_cdk/sources/file_based/file_based_stream_reader.py

@@ -10,10 +10,9 @@ from io import IOBase
 from os import makedirs, path
 from typing import Any, Dict, Iterable, List, Optional, Set
 
-from wcmatch.glob import GLOBSTAR, globmatch
-
 from airbyte_cdk.sources.file_based.config.abstract_file_based_spec import AbstractFileBasedSpec
 from airbyte_cdk.sources.file_based.remote_file import RemoteFile
+from wcmatch.glob import GLOBSTAR, globmatch
 
 
 class FileReadMode(Enum):

airbyte_cdk/sources/file_based/file_types/avro_parser.py

@@ -6,7 +6,6 @@ import logging
 from typing import Any, Dict, Iterable, Mapping, Optional, Tuple
 
 import fastavro
-
 from airbyte_cdk.sources.file_based.config.avro_format import AvroFormat
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
 from airbyte_cdk.sources.file_based.exceptions import FileBasedSourceError, RecordParseError

airbyte_cdk/sources/file_based/file_types/csv_parser.py

@@ -12,8 +12,6 @@ from io import IOBase
 from typing import Any, Callable, Dict, Generator, Iterable, List, Mapping, Optional, Set, Tuple
 from uuid import uuid4
 
-from orjson import orjson
-
 from airbyte_cdk.models import FailureType
 from airbyte_cdk.sources.file_based.config.csv_format import (
     CsvFormat,
@@ -31,6 +29,7 @@ from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeP
 from airbyte_cdk.sources.file_based.remote_file import RemoteFile
 from airbyte_cdk.sources.file_based.schema_helpers import TYPE_PYTHON_MAPPING, SchemaType
 from airbyte_cdk.utils.traced_exception import AirbyteTracedException
+from orjson import orjson
 
 DIALECT_NAME = "_config_dialect"
 

airbyte_cdk/sources/file_based/file_types/excel_parser.py

@@ -8,11 +8,6 @@ from pathlib import Path
 from typing import Any, Dict, Iterable, Mapping, Optional, Tuple, Union
 
 import pandas as pd
-from numpy import datetime64, issubdtype
-from numpy import dtype as dtype_
-from orjson import orjson
-from pydantic.v1 import BaseModel
-
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import (
     ExcelFormat,
     FileBasedStreamConfig,
@@ -29,6 +24,11 @@ from airbyte_cdk.sources.file_based.file_based_stream_reader import (
 from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser
 from airbyte_cdk.sources.file_based.remote_file import RemoteFile
 from airbyte_cdk.sources.file_based.schema_helpers import SchemaType
+from numpy import datetime64
+from numpy import dtype as dtype_
+from numpy import issubdtype
+from orjson import orjson
+from pydantic.v1 import BaseModel
 
 
 class ExcelParser(FileTypeParser):

airbyte_cdk/sources/file_based/file_types/jsonl_parser.py

@@ -6,8 +6,6 @@ import json
 import logging
 from typing import Any, Dict, Iterable, Mapping, Optional, Tuple, Union
 
-from orjson import orjson
-
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
 from airbyte_cdk.sources.file_based.exceptions import FileBasedSourceError, RecordParseError
 from airbyte_cdk.sources.file_based.file_based_stream_reader import (
@@ -21,6 +19,7 @@ from airbyte_cdk.sources.file_based.schema_helpers import (
     SchemaType,
     merge_schemas,
 )
+from orjson import orjson
 
 
 class JsonlParser(FileTypeParser):

airbyte_cdk/sources/file_based/file_types/parquet_parser.py

@@ -10,8 +10,6 @@ from urllib.parse import unquote
 
 import pyarrow as pa
 import pyarrow.parquet as pq
-from pyarrow import DictionaryArray, Scalar
-
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import (
     FileBasedStreamConfig,
     ParquetFormat,
@@ -28,6 +26,7 @@ from airbyte_cdk.sources.file_based.file_based_stream_reader import (
 from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser
 from airbyte_cdk.sources.file_based.remote_file import RemoteFile
 from airbyte_cdk.sources.file_based.schema_helpers import SchemaType
+from pyarrow import DictionaryArray, Scalar
 
 
 class ParquetParser(FileTypeParser):

airbyte_cdk/sources/file_based/file_types/unstructured_parser.py

@@ -9,16 +9,7 @@ from typing import Any, Dict, Iterable, List, Mapping, Optional, Tuple, Union
 
 import backoff
 import dpath
-import nltk
 import requests
-from unstructured.file_utils.filetype import (
-    EXT_TO_FILETYPE,
-    FILETYPE_TO_MIMETYPE,
-    STR_TO_FILETYPE,
-    FileType,
-    detect_filetype,
-)
-
 from airbyte_cdk.models import FailureType
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
 from airbyte_cdk.sources.file_based.config.unstructured_format import (
@@ -37,6 +28,14 @@ from airbyte_cdk.sources.file_based.remote_file import RemoteFile
 from airbyte_cdk.sources.file_based.schema_helpers import SchemaType
 from airbyte_cdk.utils import is_cloud_environment
 from airbyte_cdk.utils.traced_exception import AirbyteTracedException
+from unstructured.file_utils.filetype import (
+    EXT_TO_FILETYPE,
+    FILETYPE_TO_MIMETYPE,
+    STR_TO_FILETYPE,
+    FileType,
+    detect_filetype,
+)
+import nltk
 
 unstructured_partition_pdf = None
 unstructured_partition_docx = None

airbyte_cdk/sources/file_based/stream/abstract_file_based_stream.py

@@ -6,8 +6,6 @@ from abc import abstractmethod
 from functools import cache, cached_property, lru_cache
 from typing import Any, Dict, Iterable, List, Mapping, Optional, Type
 
-from deprecated import deprecated
-
 from airbyte_cdk import AirbyteMessage
 from airbyte_cdk.models import SyncMode
 from airbyte_cdk.sources.file_based.availability_strategy import (
@@ -32,6 +30,7 @@ from airbyte_cdk.sources.file_based.stream.cursor import AbstractFileBasedCursor
 from airbyte_cdk.sources.file_based.types import StreamSlice
 from airbyte_cdk.sources.streams import Stream
 from airbyte_cdk.sources.streams.checkpoint import Cursor
+from deprecated import deprecated
 
 
 class AbstractFileBasedStream(Stream):

airbyte_cdk/sources/file_based/stream/concurrent/adapters.py

@@ -7,8 +7,6 @@ import logging
 from functools import cache, lru_cache
 from typing import TYPE_CHECKING, Any, Iterable, List, Mapping, MutableMapping, Optional, Union
 
-from deprecated.classic import deprecated
-
 from airbyte_cdk.models import (
     AirbyteLogMessage,
     AirbyteMessage,
@@ -41,10 +39,11 @@ from airbyte_cdk.sources.streams.concurrent.helpers import (
 )
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
 from airbyte_cdk.sources.streams.concurrent.partitions.partition_generator import PartitionGenerator
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 from airbyte_cdk.sources.streams.core import StreamData
-from airbyte_cdk.sources.types import Record
 from airbyte_cdk.sources.utils.schema_helpers import InternalConfig
 from airbyte_cdk.sources.utils.slice_logger import SliceLogger
+from deprecated.classic import deprecated
 
 if TYPE_CHECKING:
     from airbyte_cdk.sources.file_based.stream.concurrent.cursor import (
@@ -248,7 +247,7 @@ class FileBasedStreamPartition(Partition):
                     self._stream.transformer.transform(
                         data_to_return, self._stream.get_json_schema()
                     )
-                    yield Record(data=data_to_return, stream_name=self.stream_name())
+                    yield Record(data_to_return, self)
                 elif (
                     isinstance(record_data, AirbyteMessage)
                     and record_data.type == Type.RECORD
@@ -266,7 +265,7 @@ class FileBasedStreamPartition(Partition):
                     else:
                         yield Record(
                             data=record_message_data,
-                            stream_name=self.stream_name(),
+                            partition=self,
                             is_file_transfer_message=self._use_file_transfer(),
                         )
                 else:

airbyte_cdk/sources/file_based/stream/concurrent/cursor/abstract_concurrent_file_based_cursor.py

@@ -12,7 +12,7 @@ from airbyte_cdk.sources.file_based.stream.cursor import AbstractFileBasedCursor
 from airbyte_cdk.sources.file_based.types import StreamState
 from airbyte_cdk.sources.streams.concurrent.cursor import Cursor
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
-from airbyte_cdk.sources.types import Record
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 
 if TYPE_CHECKING:
     from airbyte_cdk.sources.file_based.stream.concurrent.adapters import FileBasedStreamPartition

airbyte_cdk/sources/file_based/stream/concurrent/cursor/file_based_concurrent_cursor.py

@@ -19,7 +19,7 @@ from airbyte_cdk.sources.file_based.types import StreamState
 from airbyte_cdk.sources.message.repository import MessageRepository
 from airbyte_cdk.sources.streams.concurrent.cursor import CursorField
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
-from airbyte_cdk.sources.types import Record
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 
 if TYPE_CHECKING:
     from airbyte_cdk.sources.file_based.stream.concurrent.adapters import FileBasedStreamPartition

airbyte_cdk/sources/file_based/stream/concurrent/cursor/file_based_final_state_cursor.py

@@ -16,7 +16,7 @@ from airbyte_cdk.sources.file_based.types import StreamState
 from airbyte_cdk.sources.message import MessageRepository
 from airbyte_cdk.sources.streams import NO_CURSOR_STATE_KEY
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
-from airbyte_cdk.sources.types import Record
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 
 if TYPE_CHECKING:
     from airbyte_cdk.sources.file_based.stream.concurrent.adapters import FileBasedStreamPartition

airbyte_cdk/sources/http_logger.py

@@ -5,7 +5,6 @@
 from typing import Optional, Union
 
 import requests
-
 from airbyte_cdk.sources.message import LogMessage
 
 

airbyte_cdk/sources/streams/call_rate.py

@@ -14,8 +14,9 @@ from urllib import parse
 
 import requests
 import requests_cache
-from pyrate_limiter import InMemoryBucket, Limiter, RateItem, TimeClock
+from pyrate_limiter import InMemoryBucket, Limiter
 from pyrate_limiter import Rate as PyRateRate
+from pyrate_limiter import RateItem, TimeClock
 from pyrate_limiter.exceptions import BucketFullException
 
 # prevents mypy from complaining about missing session attributes in LimiterMixin

airbyte_cdk/sources/streams/concurrent/abstract_stream.py

@@ -5,13 +5,12 @@
 from abc import ABC, abstractmethod
 from typing import Any, Iterable, Mapping, Optional
 
-from deprecated.classic import deprecated
-
 from airbyte_cdk.models import AirbyteStream
 from airbyte_cdk.sources.source import ExperimentalClassWarning
 from airbyte_cdk.sources.streams.concurrent.availability_strategy import StreamAvailability
 from airbyte_cdk.sources.streams.concurrent.cursor import Cursor
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
+from deprecated.classic import deprecated
 
 
 @deprecated("This class is experimental. Use at your own risk.", category=ExperimentalClassWarning)

airbyte_cdk/sources/streams/concurrent/adapters.py

@@ -8,8 +8,6 @@ import logging
 from functools import lru_cache
 from typing import Any, Iterable, List, Mapping, MutableMapping, Optional, Tuple, Union
 
-from deprecated.classic import deprecated
-
 from airbyte_cdk.models import (
     AirbyteLogMessage,
     AirbyteMessage,
@@ -39,10 +37,12 @@ from airbyte_cdk.sources.streams.concurrent.helpers import (
 )
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
 from airbyte_cdk.sources.streams.concurrent.partitions.partition_generator import PartitionGenerator
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 from airbyte_cdk.sources.streams.core import StreamData
-from airbyte_cdk.sources.types import Record
 from airbyte_cdk.sources.utils.schema_helpers import InternalConfig
 from airbyte_cdk.sources.utils.slice_logger import SliceLogger
+from deprecated.classic import deprecated
+
 from airbyte_cdk.utils.slice_hasher import SliceHasher
 
 """
@@ -294,11 +294,7 @@ class StreamPartition(Partition):
                     self._stream.transformer.transform(
                         data_to_return, self._stream.get_json_schema()
                     )
-                    yield Record(
-                        data=data_to_return,
-                        stream_name=self.stream_name(),
-                        associated_slice=self._slice,
-                    )
+                    yield Record(data_to_return, self)
                 else:
                     self._message_repository.emit_message(record_data)
         except Exception as e:

airbyte_cdk/sources/streams/concurrent/availability_strategy.py

@@ -6,9 +6,8 @@ import logging
 from abc import ABC, abstractmethod
 from typing import Optional
 
-from deprecated.classic import deprecated
-
 from airbyte_cdk.sources.source import ExperimentalClassWarning
+from deprecated.classic import deprecated
 
 
 class StreamAvailability(ABC):

airbyte_cdk/sources/streams/concurrent/cursor.py

@@ -3,7 +3,6 @@
 #
 
 import functools
-import logging
 from abc import ABC, abstractmethod
 from typing import Any, Callable, Iterable, List, Mapping, MutableMapping, Optional, Protocol, Tuple
 
@@ -11,13 +10,12 @@ from airbyte_cdk.sources.connector_state_manager import ConnectorStateManager
 from airbyte_cdk.sources.message import MessageRepository
 from airbyte_cdk.sources.streams import NO_CURSOR_STATE_KEY
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 from airbyte_cdk.sources.streams.concurrent.partitions.stream_slicer import StreamSlicer
 from airbyte_cdk.sources.streams.concurrent.state_converters.abstract_stream_state_converter import (
     AbstractStreamStateConverter,
 )
-from airbyte_cdk.sources.types import Record, StreamSlice
-
-LOGGER = logging.getLogger("airbyte")
+from airbyte_cdk.sources.types import StreamSlice
 
 
 def _extract_value(mapping: Mapping[str, Any], path: List[str]) -> Any:
@@ -175,11 +173,9 @@ class ConcurrentCursor(Cursor):
         self.start, self._concurrent_state = self._get_concurrent_state(stream_state)
         self._lookback_window = lookback_window
         self._slice_range = slice_range
-        self._most_recent_cursor_value_per_partition: MutableMapping[StreamSlice, Any] = {}
+        self._most_recent_cursor_value_per_partition: MutableMapping[Partition, Any] = {}
         self._has_closed_at_least_one_slice = False
         self._cursor_granularity = cursor_granularity
-        # Flag to track if the logger has been triggered (per stream)
-        self._should_be_synced_logger_triggered = False
 
     @property
     def state(self) -> MutableMapping[str, Any]:
@@ -214,12 +210,12 @@ class ConcurrentCursor(Cursor):
 
     def observe(self, record: Record) -> None:
         most_recent_cursor_value = self._most_recent_cursor_value_per_partition.get(
-            record.associated_slice
+            record.partition
         )
         cursor_value = self._extract_cursor_value(record)
 
         if most_recent_cursor_value is None or most_recent_cursor_value < cursor_value:
-            self._most_recent_cursor_value_per_partition[record.associated_slice] = cursor_value
+            self._most_recent_cursor_value_per_partition[record.partition] = cursor_value
 
     def _extract_cursor_value(self, record: Record) -> Any:
         return self._connector_state_converter.parse_value(self._cursor_field.extract_value(record))
@@ -235,9 +231,7 @@ class ConcurrentCursor(Cursor):
         self._has_closed_at_least_one_slice = True
 
     def _add_slice_to_state(self, partition: Partition) -> None:
-        most_recent_cursor_value = self._most_recent_cursor_value_per_partition.get(
-            partition.to_slice()
-        )
+        most_recent_cursor_value = self._most_recent_cursor_value_per_partition.get(partition)
 
         if self._slice_boundary_fields:
             if "slices" not in self.state:
@@ -448,21 +442,3 @@ class ConcurrentCursor(Cursor):
             return lower + step
         except OverflowError:
             return self._end_provider()
-
-    def should_be_synced(self, record: Record) -> bool:
-        """
-        Determines if a record should be synced based on its cursor value.
-        :param record: The record to evaluate
-
-        :return: True if the record's cursor value falls within the sync boundaries
-        """
-        try:
-            record_cursor_value: CursorValueType = self._extract_cursor_value(record)  # type: ignore  # cursor_field is converted to an InterpolatedString in __post_init__
-        except ValueError:
-            if not self._should_be_synced_logger_triggered:
-                LOGGER.warning(
-                    f"Could not find cursor field `{self.cursor_field.cursor_field_key}` in record. The incremental sync will assume it needs to be synced"
-                )
-                self._should_be_synced_logger_triggered = True
-            return True
-        return self.start <= record_cursor_value <= self._end_provider()

airbyte_cdk/sources/streams/concurrent/partitions/partition.py

@@ -5,7 +5,7 @@
 from abc import ABC, abstractmethod
 from typing import Any, Iterable, Mapping, Optional
 
-from airbyte_cdk.sources.types import Record
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 
 
 class Partition(ABC):

airbyte_cdk/sources/streams/concurrent/partitions/record.py

@@ -0,0 +1,35 @@
+#
+# Copyright (c) 2023 Airbyte, Inc., all rights reserved.
+#
+
+from typing import TYPE_CHECKING, Any, Mapping
+
+if TYPE_CHECKING:
+    from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
+
+
+class Record:
+    """
+    Represents a record read from a stream.
+    """
+
+    def __init__(
+        self,
+        data: Mapping[str, Any],
+        partition: "Partition",
+        is_file_transfer_message: bool = False,
+    ):
+        self.data = data
+        self.partition = partition
+        self.is_file_transfer_message = is_file_transfer_message
+
+    def __eq__(self, other: Any) -> bool:
+        if not isinstance(other, Record):
+            return False
+        return (
+            self.data == other.data
+            and self.partition.stream_name() == other.partition.stream_name()
+        )
+
+    def __repr__(self) -> str:
+        return f"Record(data={self.data}, stream_name={self.partition.stream_name()})"

airbyte_cdk/sources/streams/concurrent/partitions/types.py

@@ -8,7 +8,7 @@ from airbyte_cdk.sources.concurrent_source.partition_generation_completed_sentin
     PartitionGenerationCompletedSentinel,
 )
 from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
-from airbyte_cdk.sources.types import Record
+from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
 
 
 class PartitionCompleteSentinel:

airbyte_cdk/sources/streams/concurrent/state_converters/datetime_stream_state_converter.py

@@ -7,7 +7,6 @@ from datetime import datetime, timedelta, timezone
 from typing import Any, Callable, List, MutableMapping, Optional, Tuple
 
 import pendulum
-from pendulum.datetime import DateTime
 
 # FIXME We would eventually like the Concurrent package do be agnostic of the declarative package. However, this is a breaking change and
 #  the goal in the short term is only to fix the issue we are seeing for source-declarative-manifest.
@@ -17,6 +16,7 @@ from airbyte_cdk.sources.streams.concurrent.state_converters.abstract_stream_sta
     AbstractStreamStateConverter,
     ConcurrencyCompatibleStateType,
 )
+from pendulum.datetime import DateTime
 
 
 class DateTimeStreamStateConverter(AbstractStreamStateConverter):

airbyte_cdk/sources/streams/core.py

@@ -10,8 +10,6 @@ from dataclasses import dataclass
 from functools import cached_property, lru_cache
 from typing import Any, Dict, Iterable, Iterator, List, Mapping, MutableMapping, Optional, Union
 
-from deprecated import deprecated
-
 import airbyte_cdk.sources.utils.casing as casing
 from airbyte_cdk.models import (
     AirbyteMessage,
@@ -37,6 +35,7 @@ from airbyte_cdk.sources.types import StreamSlice
 from airbyte_cdk.sources.utils.schema_helpers import InternalConfig, ResourceSchemaLoader
 from airbyte_cdk.sources.utils.slice_logger import DebugSliceLogger, SliceLogger
 from airbyte_cdk.sources.utils.transform import TransformConfig, TypeTransformer
+from deprecated import deprecated
 
 # A stream's read method can return one of the following types:
 # Mapping[str, Any]: The content of an AirbyteRecordMessage

airbyte_cdk/sources/streams/http/error_handlers/default_error_mapping.py

@@ -4,13 +4,12 @@
 
 from typing import Mapping, Type, Union
 
-from requests.exceptions import InvalidSchema, InvalidURL, RequestException
-
 from airbyte_cdk.models import FailureType
 from airbyte_cdk.sources.streams.http.error_handlers.response_models import (
     ErrorResolution,
     ResponseAction,
 )
+from requests.exceptions import InvalidSchema, InvalidURL, RequestException
 
 DEFAULT_ERROR_MAPPING: Mapping[Union[int, str, Type[Exception]], ErrorResolution] = {
     InvalidSchema: ErrorResolution(

airbyte_cdk/sources/streams/http/error_handlers/http_status_error_handler.py

@@ -7,7 +7,6 @@ from datetime import timedelta
 from typing import Mapping, Optional, Union
 
 import requests
-
 from airbyte_cdk.models import FailureType
 from airbyte_cdk.sources.streams.http.error_handlers.default_error_mapping import (
     DEFAULT_ERROR_MAPPING,

airbyte_cdk/sources/streams/http/error_handlers/json_error_message_parser.py

@@ -5,7 +5,6 @@
 from typing import Optional
 
 import requests
-
 from airbyte_cdk.sources.streams.http.error_handlers import ErrorMessageParser
 from airbyte_cdk.sources.utils.types import JsonType
 

airbyte_cdk/sources/streams/http/error_handlers/response_models.py

@@ -5,10 +5,9 @@ from enum import Enum
 from typing import Optional, Union
 
 import requests
-from requests import HTTPError
-
 from airbyte_cdk.models import FailureType
 from airbyte_cdk.utils.airbyte_secrets_utils import filter_secrets
+from requests import HTTPError
 
 
 class ResponseAction(Enum):

airbyte_cdk/sources/streams/http/http.py

@@ -9,9 +9,6 @@ from typing import Any, Callable, Iterable, List, Mapping, MutableMapping, Optio
 from urllib.parse import urljoin
 
 import requests
-from deprecated import deprecated
-from requests.auth import AuthBase
-
 from airbyte_cdk.models import AirbyteMessage, FailureType, SyncMode
 from airbyte_cdk.models import Type as MessageType
 from airbyte_cdk.sources.message.repository import InMemoryMessageRepository
@@ -36,6 +33,8 @@ from airbyte_cdk.sources.streams.http.error_handlers.response_models import (
 from airbyte_cdk.sources.streams.http.http_client import HttpClient
 from airbyte_cdk.sources.types import Record, StreamSlice
 from airbyte_cdk.sources.utils.types import JsonType
+from deprecated import deprecated
+from requests.auth import AuthBase
 
 # list of all possible HTTP methods which can be used for sending of request bodies
 BODY_REQUEST_METHODS = ("GET", "POST", "PUT", "PATCH")