Add missing types to the registry (airbytehq#17763)

* Add missing types to the registry * bump
jhammarstedt · Oct 31, 2022 · 010309d · 010309d
1 parent 8ef2b2a
commit 010309d
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/airbyte-cdk/python/CHANGELOG.md b/airbyte-cdk/python/CHANGELOG.md
@@ -1,5 +1,9 @@
 # Changelog
 
+## 0.1.98
+
+- Low-code: Expose WaitUntilTimeFromHeader strategy and WaitTimeFromHeader as component type
+
 ## 0.1.97
 
 - Revert 0.1.96

diff --git a/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/class_types_registry.py b/airbyte-cdk/python/airbyte_cdk/sources/declarative/parsers/class_types_registry.py
@@ -17,6 +17,12 @@
 from airbyte_cdk.sources.declarative.requesters.error_handlers.backoff_strategies.exponential_backoff_strategy import (
     ExponentialBackoffStrategy,
 )
+from airbyte_cdk.sources.declarative.requesters.error_handlers.backoff_strategies.wait_time_from_header_backoff_strategy import (
+    WaitTimeFromHeaderBackoffStrategy,
+)
+from airbyte_cdk.sources.declarative.requesters.error_handlers.backoff_strategies.wait_until_time_from_header_backoff_strategy import (
+    WaitUntilTimeFromHeaderBackoffStrategy,
+)
 from airbyte_cdk.sources.declarative.requesters.error_handlers.composite_error_handler import CompositeErrorHandler
 from airbyte_cdk.sources.declarative.requesters.error_handlers.default_error_handler import DefaultErrorHandler
 from airbyte_cdk.sources.declarative.requesters.http_requester import HttpRequester
@@ -69,4 +75,6 @@
     "SimpleRetriever": SimpleRetriever,
     "SingleSlice": SingleSlice,
     "SubstreamSlicer": SubstreamSlicer,
+    "WaitUntilTimeFromHeader": WaitUntilTimeFromHeaderBackoffStrategy,
+    "WaitTimeFromHeader": WaitTimeFromHeaderBackoffStrategy,
 }
diff --git a/airbyte-cdk/python/setup.py b/airbyte-cdk/python/setup.py
@@ -15,7 +15,7 @@
 
 setup(
     name="airbyte-cdk",
-    version="0.1.97",
+    version="0.1.98",
     description="A framework for writing Airbyte Connectors.",
     long_description=README,
     long_description_content_type="text/markdown",
@@ -44,7 +44,8 @@
     packages=find_packages(exclude=("unit_tests",)),
     install_requires=[
         "backoff",
-        "dataclasses-jsonschema==2.15.1",  # pinned to the last working version for us temporarily while we fix
+        # pinned to the last working version for us temporarily while we fix
+        "dataclasses-jsonschema==2.15.1",
         "dpath~=2.0.1",
         "jsonschema~=3.2.0",
         "jsonref~=0.2",