Skip to content

Commit 8e0881f

Browse files
author
Oleksandr Bazarnov
committed
updated / formatted
1 parent b3da6d3 commit 8e0881f

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

airbyte_cdk/sources/declarative/retrievers/async_retriever.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,16 +49,22 @@ def exit_on_rate_limit(self) -> bool:
4949
- If the `creation_requester` can place / create the job - it means all other requesters should successfully manage
5050
to complete the results.
5151
"""
52-
return self.stream_slicer._job_orchestrator._job_repository.creation_requester.exit_on_rate_limit # type: ignore[return-value]
52+
job_orchestrator = self.stream_slicer._job_orchestrator
53+
if job_orchestrator is None:
54+
# Default value when orchestrator is not available
55+
return False
56+
return job_orchestrator._job_repository.creation_requester.exit_on_rate_limit # type: ignore
5357

5458
@exit_on_rate_limit.setter
5559
def exit_on_rate_limit(self, value: bool) -> None:
5660
"""
5761
Sets the `exit_on_rate_limit` property of the job repository > creation_requester,
5862
meaning that the Job cannot be placed / created if the rate limit is reached.
59-
Thus no futher work on managing jobs is expected to be done.
63+
Thus no further work on managing jobs is expected to be done.
6064
"""
61-
self.stream_slicer._job_orchestrator._job_repository.creation_requester.exit_on_rate_limit = value # type: ignore[assignment]
65+
job_orchestrator = self.stream_slicer._job_orchestrator
66+
if job_orchestrator is not None:
67+
job_orchestrator._job_repository.creation_requester.exit_on_rate_limit = value # type: ignore[attr-defined, assignment]
6268

6369
@property
6470
def state(self) -> StreamState:

0 commit comments

Comments
 (0)