Add retry on 429 error code

trinodb · Mar 1, 2024 · d18dae0 · d18dae0
1 parent 00435d6
commit d18dae0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/trino/client.py b/trino/client.py
@@ -507,9 +507,9 @@ def max_attempts(self, value) -> None:
             self._handle_retry,
             handled_exceptions=self._exceptions,
             conditions=(
-                # need retry when there is no exception but the status code is 502, 503, or 504
+                # need retry when there is no exception but the status code is 429, 502, 503, or 504
                 lambda response: getattr(response, "status_code", None)
-                in (502, 503, 504),
+                in (429, 502, 503, 504),
             ),
             max_attempts=self._max_attempts,
         )