Add NUCLEUS_SKIP_SSL_VERIFY for different environments (#431)

---------

Co-authored-by: Sam Videlock <sam.videlock@scalegov.com>

Author: gatli

CHANGELOG.md

@@ -14,6 +14,10 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Fixes
 - Fix test `test_models.test_remove_invalid_tag_from_model`
 
+## [0.17.1] - 2024-02-22
+
+### Added
+- Environment variable `NUCLEUS_SKIP_SSL_VERIFY` to skip SSL verification on requests
 
 ## [0.16.18](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.16.18) - 2024-02-06
 

nucleus/connection.py

@@ -1,3 +1,4 @@
+import os
 import time
 from typing import Optional
 
@@ -68,6 +69,7 @@ def make_request(
                 headers={"Content-Type": "application/json"},
                 auth=(self.api_key, ""),
                 timeout=DEFAULT_NETWORK_TIMEOUT_SEC,
+                verify=os.environ.get("NUCLEUS_SKIP_SSL_VERIFY", None) is None,
             )
             logger.info(
                 "API request has response code %s", response.status_code

nucleus/utils.py

@@ -361,7 +361,11 @@ def serialize_and_write(
 
 def upload_to_presigned_url(presigned_url: str, file_pointer: IO):
     # TODO optimize this further to deal with truly huge files and flaky internet connection.
-    upload_response = requests.put(presigned_url, file_pointer)
+    upload_response = requests.put(
+        presigned_url,
+        file_pointer,
+        verify=os.environ.get("NUCLEUS_SKIP_SSL_VERIFY", None) is None,
+    )
     if not upload_response.ok:
         raise HTTPError(  # type: ignore
             f"Tried to put a file to url, but failed with status {upload_response.status_code}. The detailed error was: {upload_response.text}"

pyproject.toml

@@ -25,7 +25,7 @@ ignore = ["E501", "E741", "E731", "F401"]  # Easy ignore for getting it running
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.17.0"
+version = "0.17.1"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]