Adds config_change_callback to get_source and get_destination

airbytehq · Nov 4, 2024 · 5650728 · 5650728
1 parent abe154e
commit 5650728
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 2 deletions.
diff --git a/airbyte/destinations/base.py b/airbyte/destinations/base.py
@@ -48,7 +48,7 @@ def __init__(
         executor: Executor,
         name: str,
         config: dict[str, Any] | None = None,
-        config_change_callback: Callable[[dict[str, Any]], None] | None = None,
+        config_change_callback: Callable[[dict[str, Any], int], None] | None = None,
         *,
         validate: bool = False,
     ) -> None:

diff --git a/airbyte/destinations/util.py b/airbyte/destinations/util.py
@@ -19,6 +19,7 @@
 def get_destination(
     name: str,
     config: dict[str, Any] | None = None,
+    config_change_callback: Callable[[dict[str, Any], int], None] | None = None,
     *,
     version: str | None = None,
     pip_url: str | None = None,
@@ -58,6 +59,7 @@ def get_destination(
     return Destination(
         name=name,
         config=config,
+        config_change_callback=config_change_callback,
         executor=get_connector_executor(
             name=name,
             version=version,

diff --git a/airbyte/sources/base.py b/airbyte/sources/base.py
@@ -58,7 +58,7 @@ def __init__(
         executor: Executor,
         name: str,
         config: dict[str, Any] | None = None,
-        config_change_callback: Callable[[dict[str, Any]], None] | None = None,
+        config_change_callback: Callable[[dict[str, Any], int], None] | None = None,
         streams: str | list[str] | None = None,
         *,
         validate: bool = False,

diff --git a/airbyte/sources/util.py b/airbyte/sources/util.py
@@ -45,6 +45,7 @@ def get_connector(
 def get_source(  # noqa: PLR0913 # Too many arguments
     name: str,
     config: dict[str, Any] | None = None,
+    config_change_callback: Callable[[dict[str, Any], int], None] | None = None,
     *,
     streams: str | list[str] | None = None,
     version: str | None = None,
@@ -103,6 +104,7 @@ def get_source(  # noqa: PLR0913 # Too many arguments
     return Source(
         name=name,
         config=config,
+        config_change_callback=config_change_callback,
         streams=streams,
         executor=get_connector_executor(
             name=name,