add timeout to prevent hanging

Author: sasha-scale

nucleus/__init__.py

@@ -92,6 +92,7 @@
 )
 from .constants import (
     NUCLEUS_ENDPOINT,
+    DEFAULT_TIMEOUT,
     ERRORS_KEY,
     ERROR_ITEMS,
     ERROR_PAYLOAD,
@@ -1021,6 +1022,7 @@ def _make_grequest(
                 session=sess,
                 files=payload,
                 auth=(self.api_key, ""),
+                timeout=DEFAULT_TIMEOUT,
             )
         else:
             post = requests_command(
@@ -1029,6 +1031,7 @@ def _make_grequest(
                 json=payload,
                 headers={"Content-Type": "application/json"},
                 auth=(self.api_key, ""),
+                timeout=DEFAULT_TIMEOUT,
             )
         return post
 
@@ -1052,6 +1055,7 @@ def _make_request_raw(
             json=payload,
             headers={"Content-Type": "application/json"},
             auth=(self.api_key, ""),
+            timeout=DEFAULT_TIMEOUT,
         )
         logger.info("API request has response code %s", response.status_code)
 

nucleus/constants.py

@@ -1,4 +1,5 @@
 NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
+DEFAULT_TIMEOUT = 30
 ITEMS_KEY = "items"
 ITEM_KEY = "item"
 REFERENCE_ID_KEY = "reference_id"