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]>
algolia · Dec 18, 2024 · 93852df · 93852df
1 parent 05ac319
commit 93852df
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 3 deletions.
diff --git a/algoliasearch/http/request_options.py b/algoliasearch/http/request_options.py
@@ -58,6 +58,7 @@ def merge(
         query_parameters: Optional[Dict[str, Any]] = None,
         headers: Optional[Dict[str, str]] = None,
         data: Optional[str] = None,
+        timeouts: Dict[str, int] = {},
         user_request_options: Optional[Union[Self, Dict[str, Any]]] = None,
     ) -> Self:
         """
@@ -74,9 +75,9 @@ def merge(
             "headers": headers,
             "query_parameters": query_parameters,
             "timeouts": {
-                "read": self._config.read_timeout,
-                "write": self._config.write_timeout,
-                "connect": self._config.connect_timeout,
+                "read": timeouts.get("read", self._config.read_timeout),
+                "write": timeouts.get("write", self._config.write_timeout),
+                "connect": timeouts.get("connect", self._config.connect_timeout),
             },
             "data": data,
         }

diff --git a/algoliasearch/ingestion/client.py b/algoliasearch/ingestion/client.py
@@ -3512,6 +3512,11 @@ async def push_task_with_http_info(
             request_options=self._request_options.merge(
                 query_parameters=_query_parameters,
                 data=dumps(body_serializer(_data)),
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -4144,6 +4149,11 @@ async def trigger_docker_source_discover_with_http_info(
                 "{sourceID}", quote(str(source_id), safe="")
             ),
             request_options=self._request_options.merge(
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -4783,6 +4793,11 @@ async def validate_source_with_http_info(
             path="/1/sources/validate",
             request_options=self._request_options.merge(
                 data=dumps(body_serializer(_data)),
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -4854,6 +4869,11 @@ async def validate_source_before_update_with_http_info(
             ),
             request_options=self._request_options.merge(
                 data=dumps(body_serializer(_data)),
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -8267,6 +8287,11 @@ def push_task_with_http_info(
             request_options=self._request_options.merge(
                 query_parameters=_query_parameters,
                 data=dumps(body_serializer(_data)),
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -8899,6 +8924,11 @@ def trigger_docker_source_discover_with_http_info(
                 "{sourceID}", quote(str(source_id), safe="")
             ),
             request_options=self._request_options.merge(
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -9534,6 +9564,11 @@ def validate_source_with_http_info(
             path="/1/sources/validate",
             request_options=self._request_options.merge(
                 data=dumps(body_serializer(_data)),
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,
@@ -9605,6 +9640,11 @@ def validate_source_before_update_with_http_info(
             ),
             request_options=self._request_options.merge(
                 data=dumps(body_serializer(_data)),
+                timeouts={
+                    "read": 180000,
+                    "write": 180000,
+                    "connect": 180000,
+                },
                 user_request_options=request_options,
             ),
             use_read_transporter=False,