feat: Add retry handling to DataFrameClient (#116)

Co-authored-by: Richard Bell <[email protected]>

Author: santhoshramaraj

nisystemlink/clients/dataframe/_data_frame_client.py

@@ -13,11 +13,17 @@
 )
 from nisystemlink.clients.core.helpers import IteratorFileLike
 from requests.models import Response
-from uplink import Body, Field, Path, Query
+from uplink import Body, Field, Path, Query, retry
 
 from . import models
 
 
+# retry for common http status codes and any Connection error
+@retry(
+    when=retry.when.status([429, 502, 503, 504]),
+    stop=retry.stop.after_attempt(5),
+    on_exception=retry.CONNECTION_ERROR,
+)
 class DataFrameClient(BaseClient):
     def __init__(self, configuration: Optional[core.HttpConfiguration] = None):
         """Initialize an instance.