revert(backoff)

deepinfra/clients/deepinfra.py

@@ -1,7 +1,7 @@
-import json
 import time
 import requests
 
+from deepinfra.exceptions import MaxRetriesExceededError
 from deepinfra.constants import (
     MAX_RETRIES,
     USER_AGENT,
@@ -19,6 +19,10 @@ def __init__(self, url, auth_token):
         self.url = url
         self.auth_token = auth_token
 
+    def backoff_delay(self, attempt):
+        delay = self.initial_backoff if attempt == 1 else self.subsequent_backoff
+        time.sleep(delay)
+
     def post(self, data, config=None):
         """
         Performs a POST request.
@@ -36,9 +40,18 @@ def post(self, data, config=None):
             "User-Agent": USER_AGENT,
             "Authorization": f"Bearer {self.auth_token}",
         }
-        try:
-            response = requests.post(self.url, data=data, headers=headers)
-            response.raise_for_status()
-            return response
-        except requests.RequestException as error:
-            raise error
+        for attempt in range(self.max_retries + 1):
+            try:
+                response = requests.post(self.url, data=data, headers=headers)
+                response.raise_for_status()
+                return response
+            except requests.RequestException as error:
+                if attempt < self.max_retries:
+                    print(
+                        f"Request failed, retrying... Attempt {attempt + 1}/{self.max_retries}"
+                    )
+                    self.backoff_delay(attempt + 1)
+                else:
+                    raise error
+
+        raise MaxRetriesExceededError()

deepinfra/constants/client.py

@@ -2,7 +2,7 @@
 This module contains the constants used by the DeepInfra API client.
 """
 
-MAX_RETRIES = 0
+MAX_RETRIES = 5
 INITIAL_BACKOFF = 5000
 SUBSEQUENT_BACKOFF = 2000
 USER_AGENT = "DeepInfra Python API Client"

deepinfra/exceptions/max_retries_exceeded.py

@@ -0,0 +1,12 @@
+"""
+This error is raised when the maximum number of retries is exceeded by DeepInfraClient.
+"""
+
+
+class MaxRetriesExceededError(Exception):
+    """
+    This error is raised when the maximum number of retries is exceeded by DeepInfraClient.
+    """
+
+    def __init__(self, message="Maximum retries exceeded"):
+        super().__init__(message)