airbyte-cdk 6.6.8__py3-none-any.whl → 6.6.8rc1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (106) hide show
  1. airbyte_cdk/cli/source_declarative_manifest/_run.py +1 -2
  2. airbyte_cdk/config_observation.py +1 -2
  3. airbyte_cdk/connector.py +0 -1
  4. airbyte_cdk/connector_builder/connector_builder_handler.py +1 -1
  5. airbyte_cdk/connector_builder/main.py +1 -2
  6. airbyte_cdk/destinations/destination.py +1 -2
  7. airbyte_cdk/destinations/vector_db_based/config.py +1 -2
  8. airbyte_cdk/destinations/vector_db_based/document_processor.py +3 -4
  9. airbyte_cdk/destinations/vector_db_based/embedder.py +4 -5
  10. airbyte_cdk/entrypoint.py +2 -3
  11. airbyte_cdk/logger.py +1 -2
  12. airbyte_cdk/models/airbyte_protocol.py +1 -2
  13. airbyte_cdk/sources/config.py +1 -2
  14. airbyte_cdk/sources/declarative/auth/jwt.py +0 -1
  15. airbyte_cdk/sources/declarative/auth/oauth.py +0 -1
  16. airbyte_cdk/sources/declarative/auth/selective_authenticator.py +0 -1
  17. airbyte_cdk/sources/declarative/auth/token.py +1 -2
  18. airbyte_cdk/sources/declarative/auth/token_provider.py +2 -3
  19. airbyte_cdk/sources/declarative/concurrent_declarative_source.py +60 -8
  20. airbyte_cdk/sources/declarative/decoders/json_decoder.py +2 -3
  21. airbyte_cdk/sources/declarative/decoders/noop_decoder.py +0 -1
  22. airbyte_cdk/sources/declarative/decoders/pagination_decoder_decorator.py +0 -1
  23. airbyte_cdk/sources/declarative/decoders/xml_decoder.py +0 -1
  24. airbyte_cdk/sources/declarative/extractors/dpath_extractor.py +0 -1
  25. airbyte_cdk/sources/declarative/extractors/http_selector.py +0 -1
  26. airbyte_cdk/sources/declarative/extractors/record_selector.py +0 -1
  27. airbyte_cdk/sources/declarative/extractors/response_to_file_extractor.py +1 -2
  28. airbyte_cdk/sources/declarative/incremental/datetime_based_cursor.py +1 -2
  29. airbyte_cdk/sources/declarative/interpolation/jinja.py +4 -5
  30. airbyte_cdk/sources/declarative/manifest_declarative_source.py +3 -4
  31. airbyte_cdk/sources/declarative/migrations/legacy_to_per_partition_state_migration.py +1 -1
  32. airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py +6 -3
  33. airbyte_cdk/sources/declarative/partition_routers/substream_partition_router.py +0 -1
  34. airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/constant_backoff_strategy.py +0 -1
  35. airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/exponential_backoff_strategy.py +0 -1
  36. airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/wait_time_from_header_backoff_strategy.py +0 -1
  37. airbyte_cdk/sources/declarative/requesters/error_handlers/backoff_strategies/wait_until_time_from_header_backoff_strategy.py +0 -1
  38. airbyte_cdk/sources/declarative/requesters/error_handlers/composite_error_handler.py +0 -1
  39. airbyte_cdk/sources/declarative/requesters/error_handlers/default_error_handler.py +0 -1
  40. airbyte_cdk/sources/declarative/requesters/error_handlers/default_http_response_filter.py +0 -1
  41. airbyte_cdk/sources/declarative/requesters/error_handlers/http_response_filter.py +0 -1
  42. airbyte_cdk/sources/declarative/requesters/http_job_repository.py +1 -2
  43. airbyte_cdk/sources/declarative/requesters/http_requester.py +0 -1
  44. airbyte_cdk/sources/declarative/requesters/paginators/default_paginator.py +0 -1
  45. airbyte_cdk/sources/declarative/requesters/paginators/no_pagination.py +0 -1
  46. airbyte_cdk/sources/declarative/requesters/paginators/paginator.py +0 -1
  47. airbyte_cdk/sources/declarative/requesters/paginators/strategies/cursor_pagination_strategy.py +0 -1
  48. airbyte_cdk/sources/declarative/requesters/paginators/strategies/offset_increment.py +0 -1
  49. airbyte_cdk/sources/declarative/requesters/paginators/strategies/page_increment.py +0 -1
  50. airbyte_cdk/sources/declarative/requesters/paginators/strategies/pagination_strategy.py +0 -1
  51. airbyte_cdk/sources/declarative/requesters/paginators/strategies/stop_condition.py +0 -1
  52. airbyte_cdk/sources/declarative/requesters/request_options/interpolated_request_options_provider.py +1 -2
  53. airbyte_cdk/sources/declarative/requesters/requester.py +0 -1
  54. airbyte_cdk/sources/declarative/retrievers/async_retriever.py +1 -2
  55. airbyte_cdk/sources/declarative/retrievers/simple_retriever.py +0 -1
  56. airbyte_cdk/sources/declarative/stream_slicers/declarative_partition_generator.py +1 -1
  57. airbyte_cdk/sources/declarative/transformations/add_fields.py +0 -1
  58. airbyte_cdk/sources/declarative/transformations/remove_fields.py +0 -1
  59. airbyte_cdk/sources/declarative/yaml_declarative_source.py +0 -1
  60. airbyte_cdk/sources/embedded/tools.py +0 -1
  61. airbyte_cdk/sources/file_based/config/abstract_file_based_spec.py +1 -2
  62. airbyte_cdk/sources/file_based/config/avro_format.py +1 -2
  63. airbyte_cdk/sources/file_based/config/csv_format.py +1 -2
  64. airbyte_cdk/sources/file_based/config/excel_format.py +1 -2
  65. airbyte_cdk/sources/file_based/config/file_based_stream_config.py +1 -2
  66. airbyte_cdk/sources/file_based/config/jsonl_format.py +1 -2
  67. airbyte_cdk/sources/file_based/config/parquet_format.py +1 -2
  68. airbyte_cdk/sources/file_based/config/unstructured_format.py +1 -2
  69. airbyte_cdk/sources/file_based/file_based_source.py +1 -2
  70. airbyte_cdk/sources/file_based/file_based_stream_reader.py +1 -2
  71. airbyte_cdk/sources/file_based/file_types/avro_parser.py +0 -1
  72. airbyte_cdk/sources/file_based/file_types/csv_parser.py +1 -2
  73. airbyte_cdk/sources/file_based/file_types/excel_parser.py +5 -5
  74. airbyte_cdk/sources/file_based/file_types/jsonl_parser.py +1 -2
  75. airbyte_cdk/sources/file_based/file_types/parquet_parser.py +1 -2
  76. airbyte_cdk/sources/file_based/file_types/unstructured_parser.py +8 -9
  77. airbyte_cdk/sources/file_based/stream/abstract_file_based_stream.py +1 -2
  78. airbyte_cdk/sources/file_based/stream/concurrent/adapters.py +1 -2
  79. airbyte_cdk/sources/http_logger.py +0 -1
  80. airbyte_cdk/sources/streams/call_rate.py +2 -1
  81. airbyte_cdk/sources/streams/concurrent/abstract_stream.py +1 -2
  82. airbyte_cdk/sources/streams/concurrent/adapters.py +2 -2
  83. airbyte_cdk/sources/streams/concurrent/availability_strategy.py +1 -2
  84. airbyte_cdk/sources/streams/concurrent/state_converters/datetime_stream_state_converter.py +1 -1
  85. airbyte_cdk/sources/streams/core.py +1 -2
  86. airbyte_cdk/sources/streams/http/error_handlers/default_error_mapping.py +1 -2
  87. airbyte_cdk/sources/streams/http/error_handlers/http_status_error_handler.py +0 -1
  88. airbyte_cdk/sources/streams/http/error_handlers/json_error_message_parser.py +0 -1
  89. airbyte_cdk/sources/streams/http/error_handlers/response_models.py +1 -2
  90. airbyte_cdk/sources/streams/http/http.py +2 -3
  91. airbyte_cdk/sources/streams/http/http_client.py +2 -34
  92. airbyte_cdk/sources/streams/http/requests_native_auth/abstract_oauth.py +1 -2
  93. airbyte_cdk/sources/streams/http/requests_native_auth/oauth.py +0 -1
  94. airbyte_cdk/sources/utils/schema_helpers.py +2 -3
  95. airbyte_cdk/sql/secrets.py +1 -2
  96. airbyte_cdk/sql/shared/sql_processor.py +6 -8
  97. airbyte_cdk/test/entrypoint_wrapper.py +3 -4
  98. airbyte_cdk/test/mock_http/mocker.py +0 -1
  99. airbyte_cdk/utils/schema_inferrer.py +1 -2
  100. airbyte_cdk/utils/slice_hasher.py +1 -1
  101. airbyte_cdk/utils/traced_exception.py +1 -2
  102. {airbyte_cdk-6.6.8.dist-info → airbyte_cdk-6.6.8rc1.dist-info}/METADATA +1 -8
  103. {airbyte_cdk-6.6.8.dist-info → airbyte_cdk-6.6.8rc1.dist-info}/RECORD +106 -106
  104. {airbyte_cdk-6.6.8.dist-info → airbyte_cdk-6.6.8rc1.dist-info}/LICENSE.txt +0 -0
  105. {airbyte_cdk-6.6.8.dist-info → airbyte_cdk-6.6.8rc1.dist-info}/WHEEL +0 -0
  106. {airbyte_cdk-6.6.8.dist-info → airbyte_cdk-6.6.8rc1.dist-info}/entry_points.txt +0 -0
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass, field
6
6
  from typing import Any, List, Mapping, MutableMapping, Optional, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.requesters.error_handlers.default_http_response_filter import (
11
10
  DefaultHttpResponseFilter,
12
11
  )
@@ -5,7 +5,6 @@
5
5
  from typing import Optional, Union
6
6
 
7
7
  import requests
8
-
9
8
  from airbyte_cdk.sources.declarative.requesters.error_handlers.http_response_filter import (
10
9
  HttpResponseFilter,
11
10
  )
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass
6
6
  from typing import Any, Mapping, Optional, Set, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.models import FailureType
11
10
  from airbyte_cdk.sources.declarative.interpolation import InterpolatedString
12
11
  from airbyte_cdk.sources.declarative.interpolation.interpolated_boolean import InterpolatedBoolean
@@ -6,8 +6,6 @@ from datetime import timedelta
6
6
  from typing import Any, Dict, Iterable, Mapping, Optional
7
7
 
8
8
  import requests
9
- from requests import Response
10
-
11
9
  from airbyte_cdk import AirbyteMessage
12
10
  from airbyte_cdk.logger import lazy_log
13
11
  from airbyte_cdk.models import FailureType, Type
@@ -25,6 +23,7 @@ from airbyte_cdk.sources.declarative.requesters.requester import Requester
25
23
  from airbyte_cdk.sources.declarative.retrievers.simple_retriever import SimpleRetriever
26
24
  from airbyte_cdk.sources.types import Record, StreamSlice
27
25
  from airbyte_cdk.utils import AirbyteTracedException
26
+ from requests import Response
28
27
 
29
28
  LOGGER = logging.getLogger("airbyte")
30
29
 
@@ -9,7 +9,6 @@ from typing import Any, Callable, Mapping, MutableMapping, Optional, Union
9
9
  from urllib.parse import urljoin
10
10
 
11
11
  import requests
12
-
13
12
  from airbyte_cdk.sources.declarative.auth.declarative_authenticator import (
14
13
  DeclarativeAuthenticator,
15
14
  NoAuth,
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass, field
6
6
  from typing import Any, Mapping, MutableMapping, Optional, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.decoders import (
11
10
  Decoder,
12
11
  JsonDecoder,
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass
6
6
  from typing import Any, Mapping, MutableMapping, Optional, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.requesters.paginators.paginator import Paginator
11
10
  from airbyte_cdk.sources.types import Record, StreamSlice, StreamState
12
11
 
@@ -7,7 +7,6 @@ from dataclasses import dataclass
7
7
  from typing import Any, Mapping, Optional
8
8
 
9
9
  import requests
10
-
11
10
  from airbyte_cdk.sources.declarative.requesters.request_options.request_options_provider import (
12
11
  RequestOptionsProvider,
13
12
  )
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass, field
6
6
  from typing import Any, Dict, Mapping, Optional, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.decoders import (
11
10
  Decoder,
12
11
  JsonDecoder,
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass, field
6
6
  from typing import Any, Mapping, Optional, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.decoders import (
11
10
  Decoder,
12
11
  JsonDecoder,
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass
6
6
  from typing import Any, Mapping, Optional, Union
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.interpolation import InterpolatedString
11
10
  from airbyte_cdk.sources.declarative.requesters.paginators.strategies.pagination_strategy import (
12
11
  PaginationStrategy,
@@ -7,7 +7,6 @@ from dataclasses import dataclass
7
7
  from typing import Any, Optional
8
8
 
9
9
  import requests
10
-
11
10
  from airbyte_cdk.sources.types import Record
12
11
 
13
12
 
@@ -6,7 +6,6 @@ from abc import ABC, abstractmethod
6
6
  from typing import Any, Optional
7
7
 
8
8
  import requests
9
-
10
9
  from airbyte_cdk.sources.declarative.incremental.declarative_cursor import DeclarativeCursor
11
10
  from airbyte_cdk.sources.declarative.requesters.paginators.strategies.pagination_strategy import (
12
11
  PaginationStrategy,
@@ -5,8 +5,6 @@
5
5
  from dataclasses import InitVar, dataclass, field
6
6
  from typing import Any, Mapping, MutableMapping, Optional, Union
7
7
 
8
- from deprecated import deprecated
9
-
10
8
  from airbyte_cdk.sources.declarative.interpolation.interpolated_nested_mapping import NestedMapping
11
9
  from airbyte_cdk.sources.declarative.requesters.request_options.interpolated_nested_request_input_provider import (
12
10
  InterpolatedNestedRequestInputProvider,
@@ -19,6 +17,7 @@ from airbyte_cdk.sources.declarative.requesters.request_options.request_options_
19
17
  )
20
18
  from airbyte_cdk.sources.source import ExperimentalClassWarning
21
19
  from airbyte_cdk.sources.types import Config, StreamSlice, StreamState
20
+ from deprecated import deprecated
22
21
 
23
22
  RequestInput = Union[str, Mapping[str, str]]
24
23
  ValidRequestTypes = (str, list)
@@ -7,7 +7,6 @@ from enum import Enum
7
7
  from typing import Any, Callable, Mapping, MutableMapping, Optional, Union
8
8
 
9
9
  import requests
10
-
11
10
  from airbyte_cdk.sources.declarative.auth.declarative_authenticator import DeclarativeAuthenticator
12
11
  from airbyte_cdk.sources.declarative.requesters.request_options.request_options_provider import (
13
12
  RequestOptionsProvider,
@@ -4,8 +4,6 @@
4
4
  from dataclasses import InitVar, dataclass, field
5
5
  from typing import Any, Callable, Iterable, Mapping, Optional
6
6
 
7
- from deprecated.classic import deprecated
8
-
9
7
  from airbyte_cdk.models import FailureType
10
8
  from airbyte_cdk.sources.declarative.async_job.job_orchestrator import (
11
9
  AsyncJobOrchestrator,
@@ -19,6 +17,7 @@ from airbyte_cdk.sources.source import ExperimentalClassWarning
19
17
  from airbyte_cdk.sources.streams.core import StreamData
20
18
  from airbyte_cdk.sources.types import Config, StreamSlice, StreamState
21
19
  from airbyte_cdk.utils.traced_exception import AirbyteTracedException
20
+ from deprecated.classic import deprecated
22
21
 
23
22
 
24
23
  @deprecated("This class is experimental. Use at your own risk.", category=ExperimentalClassWarning)
@@ -20,7 +20,6 @@ from typing import (
20
20
  )
21
21
 
22
22
  import requests
23
-
24
23
  from airbyte_cdk.models import AirbyteMessage
25
24
  from airbyte_cdk.sources.declarative.extractors.http_selector import HttpSelector
26
25
  from airbyte_cdk.sources.declarative.incremental import ResumableFullRefreshCursor
@@ -1,6 +1,6 @@
1
1
  # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
2
2
 
3
- from typing import Any, Callable, Iterable, Mapping, Optional
3
+ from typing import Iterable, Optional, Mapping, Any, Callable
4
4
 
5
5
  from airbyte_cdk.sources.declarative.retrievers import Retriever
6
6
  from airbyte_cdk.sources.message import MessageRepository
@@ -6,7 +6,6 @@ from dataclasses import InitVar, dataclass, field
6
6
  from typing import Any, Dict, List, Mapping, Optional, Type, Union
7
7
 
8
8
  import dpath
9
-
10
9
  from airbyte_cdk.sources.declarative.interpolation.interpolated_string import InterpolatedString
11
10
  from airbyte_cdk.sources.declarative.transformations import RecordTransformation
12
11
  from airbyte_cdk.sources.types import Config, FieldPointer, StreamSlice, StreamState
@@ -7,7 +7,6 @@ from typing import Any, Dict, List, Mapping, Optional
7
7
 
8
8
  import dpath
9
9
  import dpath.exceptions
10
-
11
10
  from airbyte_cdk.sources.declarative.interpolation.interpolated_boolean import InterpolatedBoolean
12
11
  from airbyte_cdk.sources.declarative.transformations import RecordTransformation
13
12
  from airbyte_cdk.sources.types import Config, FieldPointer, StreamSlice, StreamState
@@ -6,7 +6,6 @@ import pkgutil
6
6
  from typing import Any, List, Mapping, Optional
7
7
 
8
8
  import yaml
9
-
10
9
  from airbyte_cdk.models import AirbyteStateMessage, ConfiguredAirbyteCatalog
11
10
  from airbyte_cdk.sources.declarative.concurrent_declarative_source import (
12
11
  ConcurrentDeclarativeSource,
@@ -5,7 +5,6 @@
5
5
  from typing import Any, Callable, Dict, Iterable, Optional
6
6
 
7
7
  import dpath
8
-
9
8
  from airbyte_cdk.models import AirbyteStream
10
9
 
11
10
 
@@ -7,11 +7,10 @@ from abc import abstractmethod
7
7
  from typing import Any, Dict, List, Literal, Optional, Union
8
8
 
9
9
  import dpath
10
- from pydantic.v1 import AnyUrl, BaseModel, Field
11
-
12
10
  from airbyte_cdk import OneOfOptionConfig
13
11
  from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
14
12
  from airbyte_cdk.sources.utils import schema_helpers
13
+ from pydantic.v1 import AnyUrl, BaseModel, Field
15
14
 
16
15
 
17
16
  class DeliverRecords(BaseModel):
@@ -3,9 +3,8 @@
3
3
  #
4
4
 
5
5
 
6
- from pydantic.v1 import BaseModel, Field
7
-
8
6
  from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
7
+ from pydantic.v1 import BaseModel, Field
9
8
 
10
9
 
11
10
  class AvroFormat(BaseModel):
@@ -6,11 +6,10 @@ import codecs
6
6
  from enum import Enum
7
7
  from typing import Any, Dict, List, Optional, Set, Union
8
8
 
9
+ from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
9
10
  from pydantic.v1 import BaseModel, Field, root_validator, validator
10
11
  from pydantic.v1.error_wrappers import ValidationError
11
12
 
12
- from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
13
-
14
13
 
15
14
  class InferenceType(Enum):
16
15
  NONE = "None"
@@ -2,9 +2,8 @@
2
2
  # Copyright (c) 2024 Airbyte, Inc., all rights reserved.
3
3
  #
4
4
 
5
- from pydantic.v1 import BaseModel, Field
6
-
7
5
  from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
6
+ from pydantic.v1 import BaseModel, Field
8
7
 
9
8
 
10
9
  class ExcelFormat(BaseModel):
@@ -5,8 +5,6 @@
5
5
  from enum import Enum
6
6
  from typing import Any, List, Mapping, Optional, Union
7
7
 
8
- from pydantic.v1 import BaseModel, Field, validator
9
-
10
8
  from airbyte_cdk.sources.file_based.config.avro_format import AvroFormat
11
9
  from airbyte_cdk.sources.file_based.config.csv_format import CsvFormat
12
10
  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
15
13
  from airbyte_cdk.sources.file_based.config.unstructured_format import UnstructuredFormat
16
14
  from airbyte_cdk.sources.file_based.exceptions import ConfigValidationError, FileBasedSourceError
17
15
  from airbyte_cdk.sources.file_based.schema_helpers import type_mapping_to_jsonschema
16
+ from pydantic.v1 import BaseModel, Field, validator
18
17
 
19
18
  PrimaryKeyType = Optional[Union[str, List[str]]]
20
19
 
@@ -2,9 +2,8 @@
2
2
  # Copyright (c) 2023 Airbyte, Inc., all rights reserved.
3
3
  #
4
4
 
5
- from pydantic.v1 import BaseModel, Field
6
-
7
5
  from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
6
+ from pydantic.v1 import BaseModel, Field
8
7
 
9
8
 
10
9
  class JsonlFormat(BaseModel):
@@ -3,9 +3,8 @@
3
3
  #
4
4
 
5
5
 
6
- from pydantic.v1 import BaseModel, Field
7
-
8
6
  from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
7
+ from pydantic.v1 import BaseModel, Field
9
8
 
10
9
 
11
10
  class ParquetFormat(BaseModel):
@@ -4,9 +4,8 @@
4
4
 
5
5
  from typing import List, Literal, Optional, Union
6
6
 
7
- from pydantic.v1 import BaseModel, Field
8
-
9
7
  from airbyte_cdk.utils.oneof_option_config import OneOfOptionConfig
8
+ from pydantic.v1 import BaseModel, Field
10
9
 
11
10
 
12
11
  class LocalProcessingConfigModel(BaseModel):
@@ -8,8 +8,6 @@ from abc import ABC
8
8
  from collections import Counter
9
9
  from typing import Any, Iterator, List, Mapping, Optional, Tuple, Type, Union
10
10
 
11
- from pydantic.v1.error_wrappers import ValidationError
12
-
13
11
  from airbyte_cdk.logger import AirbyteLogFormatter, init_logger
14
12
  from airbyte_cdk.models import (
15
13
  AirbyteMessage,
@@ -62,6 +60,7 @@ from airbyte_cdk.sources.streams import Stream
62
60
  from airbyte_cdk.sources.streams.concurrent.cursor import CursorField
63
61
  from airbyte_cdk.utils.analytics_message import create_analytics_message
64
62
  from airbyte_cdk.utils.traced_exception import AirbyteTracedException
63
+ from pydantic.v1.error_wrappers import ValidationError
65
64
 
66
65
  DEFAULT_CONCURRENCY = 100
67
66
  MAX_CONCURRENCY = 100
@@ -10,10 +10,9 @@ from io import IOBase
10
10
  from os import makedirs, path
11
11
  from typing import Any, Dict, Iterable, List, Optional, Set
12
12
 
13
- from wcmatch.glob import GLOBSTAR, globmatch
14
-
15
13
  from airbyte_cdk.sources.file_based.config.abstract_file_based_spec import AbstractFileBasedSpec
16
14
  from airbyte_cdk.sources.file_based.remote_file import RemoteFile
15
+ from wcmatch.glob import GLOBSTAR, globmatch
17
16
 
18
17
 
19
18
  class FileReadMode(Enum):
@@ -6,7 +6,6 @@ import logging
6
6
  from typing import Any, Dict, Iterable, Mapping, Optional, Tuple
7
7
 
8
8
  import fastavro
9
-
10
9
  from airbyte_cdk.sources.file_based.config.avro_format import AvroFormat
11
10
  from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
12
11
  from airbyte_cdk.sources.file_based.exceptions import FileBasedSourceError, RecordParseError
@@ -12,8 +12,6 @@ from io import IOBase
12
12
  from typing import Any, Callable, Dict, Generator, Iterable, List, Mapping, Optional, Set, Tuple
13
13
  from uuid import uuid4
14
14
 
15
- from orjson import orjson
16
-
17
15
  from airbyte_cdk.models import FailureType
18
16
  from airbyte_cdk.sources.file_based.config.csv_format import (
19
17
  CsvFormat,
@@ -31,6 +29,7 @@ from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeP
31
29
  from airbyte_cdk.sources.file_based.remote_file import RemoteFile
32
30
  from airbyte_cdk.sources.file_based.schema_helpers import TYPE_PYTHON_MAPPING, SchemaType
33
31
  from airbyte_cdk.utils.traced_exception import AirbyteTracedException
32
+ from orjson import orjson
34
33
 
35
34
  DIALECT_NAME = "_config_dialect"
36
35
 
@@ -8,11 +8,6 @@ from pathlib import Path
8
8
  from typing import Any, Dict, Iterable, Mapping, Optional, Tuple, Union
9
9
 
10
10
  import pandas as pd
11
- from numpy import datetime64, issubdtype
12
- from numpy import dtype as dtype_
13
- from orjson import orjson
14
- from pydantic.v1 import BaseModel
15
-
16
11
  from airbyte_cdk.sources.file_based.config.file_based_stream_config import (
17
12
  ExcelFormat,
18
13
  FileBasedStreamConfig,
@@ -29,6 +24,11 @@ from airbyte_cdk.sources.file_based.file_based_stream_reader import (
29
24
  from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser
30
25
  from airbyte_cdk.sources.file_based.remote_file import RemoteFile
31
26
  from airbyte_cdk.sources.file_based.schema_helpers import SchemaType
27
+ from numpy import datetime64
28
+ from numpy import dtype as dtype_
29
+ from numpy import issubdtype
30
+ from orjson import orjson
31
+ from pydantic.v1 import BaseModel
32
32
 
33
33
 
34
34
  class ExcelParser(FileTypeParser):
@@ -6,8 +6,6 @@ import json
6
6
  import logging
7
7
  from typing import Any, Dict, Iterable, Mapping, Optional, Tuple, Union
8
8
 
9
- from orjson import orjson
10
-
11
9
  from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
12
10
  from airbyte_cdk.sources.file_based.exceptions import FileBasedSourceError, RecordParseError
13
11
  from airbyte_cdk.sources.file_based.file_based_stream_reader import (
@@ -21,6 +19,7 @@ from airbyte_cdk.sources.file_based.schema_helpers import (
21
19
  SchemaType,
22
20
  merge_schemas,
23
21
  )
22
+ from orjson import orjson
24
23
 
25
24
 
26
25
  class JsonlParser(FileTypeParser):
@@ -10,8 +10,6 @@ from urllib.parse import unquote
10
10
 
11
11
  import pyarrow as pa
12
12
  import pyarrow.parquet as pq
13
- from pyarrow import DictionaryArray, Scalar
14
-
15
13
  from airbyte_cdk.sources.file_based.config.file_based_stream_config import (
16
14
  FileBasedStreamConfig,
17
15
  ParquetFormat,
@@ -28,6 +26,7 @@ from airbyte_cdk.sources.file_based.file_based_stream_reader import (
28
26
  from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser
29
27
  from airbyte_cdk.sources.file_based.remote_file import RemoteFile
30
28
  from airbyte_cdk.sources.file_based.schema_helpers import SchemaType
29
+ from pyarrow import DictionaryArray, Scalar
31
30
 
32
31
 
33
32
  class ParquetParser(FileTypeParser):
@@ -9,16 +9,7 @@ from typing import Any, Dict, Iterable, List, Mapping, Optional, Tuple, Union
9
9
 
10
10
  import backoff
11
11
  import dpath
12
- import nltk
13
12
  import requests
14
- from unstructured.file_utils.filetype import (
15
- EXT_TO_FILETYPE,
16
- FILETYPE_TO_MIMETYPE,
17
- STR_TO_FILETYPE,
18
- FileType,
19
- detect_filetype,
20
- )
21
-
22
13
  from airbyte_cdk.models import FailureType
23
14
  from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
24
15
  from airbyte_cdk.sources.file_based.config.unstructured_format import (
@@ -37,6 +28,14 @@ from airbyte_cdk.sources.file_based.remote_file import RemoteFile
37
28
  from airbyte_cdk.sources.file_based.schema_helpers import SchemaType
38
29
  from airbyte_cdk.utils import is_cloud_environment
39
30
  from airbyte_cdk.utils.traced_exception import AirbyteTracedException
31
+ from unstructured.file_utils.filetype import (
32
+ EXT_TO_FILETYPE,
33
+ FILETYPE_TO_MIMETYPE,
34
+ STR_TO_FILETYPE,
35
+ FileType,
36
+ detect_filetype,
37
+ )
38
+ import nltk
40
39
 
41
40
  unstructured_partition_pdf = None
42
41
  unstructured_partition_docx = None
@@ -6,8 +6,6 @@ from abc import abstractmethod
6
6
  from functools import cache, cached_property, lru_cache
7
7
  from typing import Any, Dict, Iterable, List, Mapping, Optional, Type
8
8
 
9
- from deprecated import deprecated
10
-
11
9
  from airbyte_cdk import AirbyteMessage
12
10
  from airbyte_cdk.models import SyncMode
13
11
  from airbyte_cdk.sources.file_based.availability_strategy import (
@@ -32,6 +30,7 @@ from airbyte_cdk.sources.file_based.stream.cursor import AbstractFileBasedCursor
32
30
  from airbyte_cdk.sources.file_based.types import StreamSlice
33
31
  from airbyte_cdk.sources.streams import Stream
34
32
  from airbyte_cdk.sources.streams.checkpoint import Cursor
33
+ from deprecated import deprecated
35
34
 
36
35
 
37
36
  class AbstractFileBasedStream(Stream):
@@ -7,8 +7,6 @@ import logging
7
7
  from functools import cache, lru_cache
8
8
  from typing import TYPE_CHECKING, Any, Iterable, List, Mapping, MutableMapping, Optional, Union
9
9
 
10
- from deprecated.classic import deprecated
11
-
12
10
  from airbyte_cdk.models import (
13
11
  AirbyteLogMessage,
14
12
  AirbyteMessage,
@@ -45,6 +43,7 @@ from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
45
43
  from airbyte_cdk.sources.streams.core import StreamData
46
44
  from airbyte_cdk.sources.utils.schema_helpers import InternalConfig
47
45
  from airbyte_cdk.sources.utils.slice_logger import SliceLogger
46
+ from deprecated.classic import deprecated
48
47
 
49
48
  if TYPE_CHECKING:
50
49
  from airbyte_cdk.sources.file_based.stream.concurrent.cursor import (
@@ -5,7 +5,6 @@
5
5
  from typing import Optional, Union
6
6
 
7
7
  import requests
8
-
9
8
  from airbyte_cdk.sources.message import LogMessage
10
9
 
11
10
 
@@ -14,8 +14,9 @@ from urllib import parse
14
14
 
15
15
  import requests
16
16
  import requests_cache
17
- from pyrate_limiter import InMemoryBucket, Limiter, RateItem, TimeClock
17
+ from pyrate_limiter import InMemoryBucket, Limiter
18
18
  from pyrate_limiter import Rate as PyRateRate
19
+ from pyrate_limiter import RateItem, TimeClock
19
20
  from pyrate_limiter.exceptions import BucketFullException
20
21
 
21
22
  # prevents mypy from complaining about missing session attributes in LimiterMixin
@@ -5,13 +5,12 @@
5
5
  from abc import ABC, abstractmethod
6
6
  from typing import Any, Iterable, Mapping, Optional
7
7
 
8
- from deprecated.classic import deprecated
9
-
10
8
  from airbyte_cdk.models import AirbyteStream
11
9
  from airbyte_cdk.sources.source import ExperimentalClassWarning
12
10
  from airbyte_cdk.sources.streams.concurrent.availability_strategy import StreamAvailability
13
11
  from airbyte_cdk.sources.streams.concurrent.cursor import Cursor
14
12
  from airbyte_cdk.sources.streams.concurrent.partitions.partition import Partition
13
+ from deprecated.classic import deprecated
15
14
 
16
15
 
17
16
  @deprecated("This class is experimental. Use at your own risk.", category=ExperimentalClassWarning)
@@ -8,8 +8,6 @@ import logging
8
8
  from functools import lru_cache
9
9
  from typing import Any, Iterable, List, Mapping, MutableMapping, Optional, Tuple, Union
10
10
 
11
- from deprecated.classic import deprecated
12
-
13
11
  from airbyte_cdk.models import (
14
12
  AirbyteLogMessage,
15
13
  AirbyteMessage,
@@ -43,6 +41,8 @@ from airbyte_cdk.sources.streams.concurrent.partitions.record import Record
43
41
  from airbyte_cdk.sources.streams.core import StreamData
44
42
  from airbyte_cdk.sources.utils.schema_helpers import InternalConfig
45
43
  from airbyte_cdk.sources.utils.slice_logger import SliceLogger
44
+ from deprecated.classic import deprecated
45
+
46
46
  from airbyte_cdk.utils.slice_hasher import SliceHasher
47
47
 
48
48
  """
@@ -6,9 +6,8 @@ import logging
6
6
  from abc import ABC, abstractmethod
7
7
  from typing import Optional
8
8
 
9
- from deprecated.classic import deprecated
10
-
11
9
  from airbyte_cdk.sources.source import ExperimentalClassWarning
10
+ from deprecated.classic import deprecated
12
11
 
13
12
 
14
13
  class StreamAvailability(ABC):
@@ -7,7 +7,6 @@ from datetime import datetime, timedelta, timezone
7
7
  from typing import Any, Callable, List, MutableMapping, Optional, Tuple
8
8
 
9
9
  import pendulum
10
- from pendulum.datetime import DateTime
11
10
 
12
11
  # FIXME We would eventually like the Concurrent package do be agnostic of the declarative package. However, this is a breaking change and
13
12
  # 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
17
16
  AbstractStreamStateConverter,
18
17
  ConcurrencyCompatibleStateType,
19
18
  )
19
+ from pendulum.datetime import DateTime
20
20
 
21
21
 
22
22
  class DateTimeStreamStateConverter(AbstractStreamStateConverter):
@@ -10,8 +10,6 @@ from dataclasses import dataclass
10
10
  from functools import cached_property, lru_cache
11
11
  from typing import Any, Dict, Iterable, Iterator, List, Mapping, MutableMapping, Optional, Union
12
12
 
13
- from deprecated import deprecated
14
-
15
13
  import airbyte_cdk.sources.utils.casing as casing
16
14
  from airbyte_cdk.models import (
17
15
  AirbyteMessage,
@@ -37,6 +35,7 @@ from airbyte_cdk.sources.types import StreamSlice
37
35
  from airbyte_cdk.sources.utils.schema_helpers import InternalConfig, ResourceSchemaLoader
38
36
  from airbyte_cdk.sources.utils.slice_logger import DebugSliceLogger, SliceLogger
39
37
  from airbyte_cdk.sources.utils.transform import TransformConfig, TypeTransformer
38
+ from deprecated import deprecated
40
39
 
41
40
  # A stream's read method can return one of the following types:
42
41
  # Mapping[str, Any]: The content of an AirbyteRecordMessage
@@ -4,13 +4,12 @@
4
4
 
5
5
  from typing import Mapping, Type, Union
6
6
 
7
- from requests.exceptions import InvalidSchema, InvalidURL, RequestException
8
-
9
7
  from airbyte_cdk.models import FailureType
10
8
  from airbyte_cdk.sources.streams.http.error_handlers.response_models import (
11
9
  ErrorResolution,
12
10
  ResponseAction,
13
11
  )
12
+ from requests.exceptions import InvalidSchema, InvalidURL, RequestException
14
13
 
15
14
  DEFAULT_ERROR_MAPPING: Mapping[Union[int, str, Type[Exception]], ErrorResolution] = {
16
15
  InvalidSchema: ErrorResolution(
@@ -7,7 +7,6 @@ from datetime import timedelta
7
7
  from typing import Mapping, Optional, Union
8
8
 
9
9
  import requests
10
-
11
10
  from airbyte_cdk.models import FailureType
12
11
  from airbyte_cdk.sources.streams.http.error_handlers.default_error_mapping import (
13
12
  DEFAULT_ERROR_MAPPING,