From 010309df5d3b860af4a31c5e9dc434e8cdfefdf4 Mon Sep 17 00:00:00 2001
From: Alexandre Girard <alexandre@airbyte.io>
Date: Sun, 9 Oct 2022 01:25:38 -0700
Subject: [PATCH] Add missing types to the registry (#17763)

* Add missing types to the registry

* bump
---
 airbyte-cdk/python/CHANGELOG.md                           | 4 ++++
 .../sources/declarative/parsers/class_types_registry.py   | 8 ++++++++
 airbyte-cdk/python/setup.py                               | 5 +++--
 3 files changed, 15 insertions(+), 2 deletions(-)

diff --git a/airbyte-cdk/python/CHANGELOG.md b/airbyte-cdk/python/CHANGELOG.md
index 42336f27519c..e839b8a16578 100644
--- 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
index 3c6dbcb58814..2419443089aa 100644
--- 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
index 91fc927ab34d..cb0d13dd4f5a 100644
--- 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",