Skip to content

Commit 93852df

Browse files
fix(specs): endpoint level timeout for ingestion (generated)
algolia/api-clients-automation#4251 Co-authored-by: algolia-bot <[email protected]> Co-authored-by: Clément Vannicatte <[email protected]>
1 parent 05ac319 commit 93852df

File tree

2 files changed

+44
-3
lines changed

2 files changed

+44
-3
lines changed

algoliasearch/http/request_options.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ def merge(
5858
query_parameters: Optional[Dict[str, Any]] = None,
5959
headers: Optional[Dict[str, str]] = None,
6060
data: Optional[str] = None,
61+
timeouts: Dict[str, int] = {},
6162
user_request_options: Optional[Union[Self, Dict[str, Any]]] = None,
6263
) -> Self:
6364
"""
@@ -74,9 +75,9 @@ def merge(
7475
"headers": headers,
7576
"query_parameters": query_parameters,
7677
"timeouts": {
77-
"read": self._config.read_timeout,
78-
"write": self._config.write_timeout,
79-
"connect": self._config.connect_timeout,
78+
"read": timeouts.get("read", self._config.read_timeout),
79+
"write": timeouts.get("write", self._config.write_timeout),
80+
"connect": timeouts.get("connect", self._config.connect_timeout),
8081
},
8182
"data": data,
8283
}

algoliasearch/ingestion/client.py

Lines changed: 40 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3512,6 +3512,11 @@ async def push_task_with_http_info(
35123512
request_options=self._request_options.merge(
35133513
query_parameters=_query_parameters,
35143514
data=dumps(body_serializer(_data)),
3515+
timeouts={
3516+
"read": 180000,
3517+
"write": 180000,
3518+
"connect": 180000,
3519+
},
35153520
user_request_options=request_options,
35163521
),
35173522
use_read_transporter=False,
@@ -4144,6 +4149,11 @@ async def trigger_docker_source_discover_with_http_info(
41444149
"{sourceID}", quote(str(source_id), safe="")
41454150
),
41464151
request_options=self._request_options.merge(
4152+
timeouts={
4153+
"read": 180000,
4154+
"write": 180000,
4155+
"connect": 180000,
4156+
},
41474157
user_request_options=request_options,
41484158
),
41494159
use_read_transporter=False,
@@ -4783,6 +4793,11 @@ async def validate_source_with_http_info(
47834793
path="/1/sources/validate",
47844794
request_options=self._request_options.merge(
47854795
data=dumps(body_serializer(_data)),
4796+
timeouts={
4797+
"read": 180000,
4798+
"write": 180000,
4799+
"connect": 180000,
4800+
},
47864801
user_request_options=request_options,
47874802
),
47884803
use_read_transporter=False,
@@ -4854,6 +4869,11 @@ async def validate_source_before_update_with_http_info(
48544869
),
48554870
request_options=self._request_options.merge(
48564871
data=dumps(body_serializer(_data)),
4872+
timeouts={
4873+
"read": 180000,
4874+
"write": 180000,
4875+
"connect": 180000,
4876+
},
48574877
user_request_options=request_options,
48584878
),
48594879
use_read_transporter=False,
@@ -8267,6 +8287,11 @@ def push_task_with_http_info(
82678287
request_options=self._request_options.merge(
82688288
query_parameters=_query_parameters,
82698289
data=dumps(body_serializer(_data)),
8290+
timeouts={
8291+
"read": 180000,
8292+
"write": 180000,
8293+
"connect": 180000,
8294+
},
82708295
user_request_options=request_options,
82718296
),
82728297
use_read_transporter=False,
@@ -8899,6 +8924,11 @@ def trigger_docker_source_discover_with_http_info(
88998924
"{sourceID}", quote(str(source_id), safe="")
89008925
),
89018926
request_options=self._request_options.merge(
8927+
timeouts={
8928+
"read": 180000,
8929+
"write": 180000,
8930+
"connect": 180000,
8931+
},
89028932
user_request_options=request_options,
89038933
),
89048934
use_read_transporter=False,
@@ -9534,6 +9564,11 @@ def validate_source_with_http_info(
95349564
path="/1/sources/validate",
95359565
request_options=self._request_options.merge(
95369566
data=dumps(body_serializer(_data)),
9567+
timeouts={
9568+
"read": 180000,
9569+
"write": 180000,
9570+
"connect": 180000,
9571+
},
95379572
user_request_options=request_options,
95389573
),
95399574
use_read_transporter=False,
@@ -9605,6 +9640,11 @@ def validate_source_before_update_with_http_info(
96059640
),
96069641
request_options=self._request_options.merge(
96079642
data=dumps(body_serializer(_data)),
9643+
timeouts={
9644+
"read": 180000,
9645+
"write": 180000,
9646+
"connect": 180000,
9647+
},
96089648
user_request_options=request_options,
96099649
),
96109650
use_read_transporter=False,

0 commit comments

Comments
 (0)