Add http_transport_retries parameter to Groundlight class

Author: CoreyEWood

src/groundlight/client.py

@@ -37,6 +37,7 @@
     PaginatedImageQueryList,
 )
 from urllib3.exceptions import InsecureRequestWarning
+from urllib3.util.retry import Retry
 
 from groundlight.binary_labels import Label, convert_internal_label_to_display
 from groundlight.config import API_TOKEN_MISSING_HELP_MESSAGE, API_TOKEN_VARIABLE_NAME, DISABLE_TLS_VARIABLE_NAME
@@ -133,26 +134,31 @@ def __init__(
         endpoint: Optional[str] = None,
         api_token: Optional[str] = None,
         disable_tls_verification: Optional[bool] = None,
+        http_transport_retries: Optional[Union[int, Retry]] = None,
     ):
         """
         Initialize a new Groundlight client instance.
 
         :param endpoint: Optional custom API endpoint URL. If not specified, uses the default Groundlight endpoint.
         :param api_token: Authentication token for API access.
-                        If not provided, will attempt to read from the "GROUNDLIGHT_API_TOKEN" environment variable.
+            If not provided, will attempt to read from the "GROUNDLIGHT_API_TOKEN" environment variable.
         :param disable_tls_verification: If True, disables SSL/TLS certificate verification for API calls.
-                                       When not specified, checks the "DISABLE_TLS_VERIFY" environment variable
-                                       (1=disable, 0=enable). Certificate verification is enabled by default.
+            When not specified, checks the "DISABLE_TLS_VERIFY" environment variable (1=disable, 0=enable).
+            Certificate verification is enabled by default.
 
-                                       Warning: Only disable verification when connecting to a Groundlight Edge
-                                       Endpoint using self-signed certificates. For security, always keep
-                                       verification enabled when using the Groundlight cloud service.
+            Warning: Only disable verification when connecting to a Groundlight Edge Endpoint using self-signed
+            certificates. For security, always keep verification enabled when using the Groundlight cloud service.
+        :param http_transport_retries: Overrides urllib3 `PoolManager` retry policy for HTTP/HTTPS (forwarded to
+            `Configuration.retries`). Not the same as SDK 5xx retries handled by `RequestsRetryDecorator`.
 
         :return: Groundlight client
         """
         # Specify the endpoint
         self.endpoint = sanitize_endpoint_url(endpoint)
         self.configuration = Configuration(host=self.endpoint)
+        if http_transport_retries is not None:
+            # Once we upgrade openapitools to ^7.7.0, retries can be passed into the constructor of Configuration above.
+            self.configuration.retries = http_transport_retries
 
         if not api_token:
             try: