Skip to content

Commit 463ad81

Browse files
committed
adds max job count for async retriever
1 parent f99970b commit 463ad81

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

airbyte_cdk/sources/declarative/models/declarative_component_schema.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2368,6 +2368,11 @@ class AsyncRetriever(BaseModel):
23682368
description="Component decoding the download response so records can be extracted.",
23692369
title="Download Decoder",
23702370
)
2371+
maximum_job_count: Optional[float] = Field(
2372+
1,
2373+
description="Maximum number of asynchronous jobs to run concurrently.",
2374+
title="Maximum Job Count",
2375+
)
23712376
parameters: Optional[Dict[str, Any]] = Field(None, alias="$parameters")
23722377

23732378

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2888,8 +2888,7 @@ def _get_download_retriever() -> SimpleRetrieverTestReadDecorator | SimpleRetrie
28882888
job_orchestrator_factory=lambda stream_slices: AsyncJobOrchestrator(
28892889
job_repository,
28902890
stream_slices,
2891-
JobTracker(1),
2892-
# FIXME eventually make the number of concurrent jobs in the API configurable. Until then, we limit to 1
2891+
JobTracker(model.maximum_job_count),
28932892
self._message_repository,
28942893
has_bulk_parent=False,
28952894
# FIXME work would need to be done here in order to detect if a stream as a parent stream that is bulk

0 commit comments

Comments
 (0)