Skip to content

Commit f0443aa

Browse files
authored
feat: share API budget with parent streams (#828)
1 parent ab9f644 commit f0443aa

File tree

3 files changed

+31
-555
lines changed

3 files changed

+31
-555
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -685,6 +685,7 @@ def __init__(
685685
connector_state_manager: Optional[ConnectorStateManager] = None,
686686
max_concurrent_async_job_count: Optional[int] = None,
687687
configured_catalog: Optional[ConfiguredAirbyteCatalog] = None,
688+
api_budget: Optional[APIBudget] = None,
688689
):
689690
self._init_mappings()
690691
self._limit_pages_fetched_per_slice = limit_pages_fetched_per_slice
@@ -699,7 +700,7 @@ def __init__(
699700
configured_catalog
700701
)
701702
self._connector_state_manager = connector_state_manager or ConnectorStateManager()
702-
self._api_budget: Optional[Union[APIBudget, HttpAPIBudget]] = None
703+
self._api_budget: Optional[Union[APIBudget]] = api_budget
703704
self._job_tracker: JobTracker = JobTracker(max_concurrent_async_job_count or 1)
704705
# placeholder for deprecation warnings
705706
self._collected_deprecation_logs: List[ConnectorBuilderLogMessage] = []
@@ -3945,6 +3946,7 @@ def create_parent_stream_config_with_substream_wrapper(
39453946
self._evaluate_log_level(self._emit_connector_builder_messages),
39463947
),
39473948
),
3949+
api_budget=self._api_budget,
39483950
)
39493951

39503952
return substream_factory.create_parent_stream_config(

0 commit comments

Comments
 (0)