Skip to content

Commit 040ff9e

Browse files
committed
Format
1 parent b6bcdd7 commit 040ff9e

File tree

1 file changed

+5
-15
lines changed

1 file changed

+5
-15
lines changed

airbyte_cdk/sources/declarative/parsers/model_to_component_factory.py

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -2969,9 +2969,7 @@ def create_api_budget(self, model: APIBudgetModel, config: Config, **kwargs: Any
29692969

29702970
return APIBudget(
29712971
policies=policies,
2972-
maximum_attempts_to_acquire=model.maximum_attempts_to_acquire
2973-
if model.maximum_attempts_to_acquire
2974-
else 100000,
2972+
maximum_attempts_to_acquire=model.maximum_attempts_to_acquire or 100000,
29752973
)
29762974

29772975
def create_http_api_budget(
@@ -2984,18 +2982,10 @@ def create_http_api_budget(
29842982

29852983
return HttpAPIBudget(
29862984
policies=policies,
2987-
maximum_attempts_to_acquire=model.maximum_attempts_to_acquire
2988-
if model.maximum_attempts_to_acquire
2989-
else 100000,
2990-
ratelimit_reset_header=model.ratelimit_reset_header
2991-
if model.ratelimit_reset_header
2992-
else "ratelimit-reset",
2993-
ratelimit_remaining_header=model.ratelimit_remaining_header
2994-
if model.ratelimit_remaining_header
2995-
else "ratelimit-remaining",
2996-
status_codes_for_ratelimit_hit=model.status_codes_for_ratelimit_hit
2997-
if model.status_codes_for_ratelimit_hit
2998-
else (429,),
2985+
maximum_attempts_to_acquire=model.maximum_attempts_to_acquire or 100000,
2986+
ratelimit_reset_header=model.ratelimit_reset_header or "ratelimit-reset",
2987+
ratelimit_remaining_header=model.ratelimit_remaining_header or "ratelimit-remaining",
2988+
status_codes_for_ratelimit_hit=model.status_codes_for_ratelimit_hit or (429,),
29992989
)
30002990

30012991
def create_fixed_window_call_rate_policy(

0 commit comments

Comments
 (0)