Skip to content

Commit 294c69c

Browse files
committed
fix: _retry_params
1 parent 1d03009 commit 294c69c

File tree

2 files changed

+24
-4
lines changed

2 files changed

+24
-4
lines changed

openfga_sdk/api/open_fga_api.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,11 @@ async def execute_api_request(
211211
query_params_list = builder.build_query_params_list()
212212
final_headers = builder.build_headers()
213213

214-
retry_params = options.get("retry_params") if options else None
214+
retry_params = (
215+
options.get("retry_params") or options.get("_retry_params")
216+
if options
217+
else None
218+
)
215219
request_timeout = options.get("_request_timeout") if options else None
216220
async_req = options.get("async_req") if options else None
217221

@@ -341,7 +345,12 @@ async def execute_streamed_api_request(
341345
query_params_list = builder.build_query_params_list()
342346
final_headers = builder.build_headers()
343347

344-
retry_params = options.get("retry_params") if options else None
348+
retry_params = (
349+
options.get("retry_params") or options.get("_retry_params")
350+
if options
351+
else None
352+
)
353+
request_timeout = options.get("_request_timeout") if options else None
345354

346355
if telemetry_attributes is None:
347356
telemetry_attributes = {
@@ -370,6 +379,7 @@ async def execute_streamed_api_request(
370379
auth_settings=[],
371380
_return_http_data_only=True,
372381
_preload_content=True,
382+
_request_timeout=request_timeout,
373383
_retry_params=retry_params,
374384
_oauth2_client=self._oauth2_client,
375385
_telemetry_attributes=telemetry_attributes,

openfga_sdk/sync/open_fga_api.py

Lines changed: 12 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -211,7 +211,11 @@ def execute_api_request(
211211
query_params_list = builder.build_query_params_list()
212212
final_headers = builder.build_headers()
213213

214-
retry_params = options.get("retry_params") if options else None
214+
retry_params = (
215+
options.get("retry_params") or options.get("_retry_params")
216+
if options
217+
else None
218+
)
215219
request_timeout = options.get("_request_timeout") if options else None
216220
async_req = options.get("async_req") if options else None
217221

@@ -341,7 +345,12 @@ def execute_streamed_api_request(
341345
query_params_list = builder.build_query_params_list()
342346
final_headers = builder.build_headers()
343347

344-
retry_params = options.get("retry_params") if options else None
348+
retry_params = (
349+
options.get("retry_params") or options.get("_retry_params")
350+
if options
351+
else None
352+
)
353+
request_timeout = options.get("_request_timeout") if options else None
345354

346355
if telemetry_attributes is None:
347356
telemetry_attributes = {
@@ -370,6 +379,7 @@ def execute_streamed_api_request(
370379
auth_settings=[],
371380
_return_http_data_only=True,
372381
_preload_content=True,
382+
_request_timeout=request_timeout,
373383
_retry_params=retry_params,
374384
_oauth2_client=self._oauth2_client,
375385
_telemetry_attributes=telemetry_attributes,

0 commit comments

Comments
 (0)