feat(client) : make ssl option for all calls to OH api

openhexa/sdk/datasets/dataset.py

@@ -11,7 +11,7 @@
 
 import requests
 
-from openhexa.sdk.utils import Iterator, Page, graphql, read_content
+from openhexa.sdk.utils import Iterator, Page, Settings, graphql, read_content
 
 
 class DatasetFile:
@@ -38,7 +38,7 @@ def __init__(
 
     def read(self):
         """Download the file content and return it."""
-        response = requests.get(self.download_url, stream=True)
+        response = requests.get(self.download_url, stream=True, verify=Settings.verify_ssl())
         response.raise_for_status()
         return response.content
 
@@ -261,7 +261,9 @@ def add_file(
 
         upload_url = upload_url_result["generateDatasetUploadUrl"]["uploadUrl"]
         with read_content(source) as content:
-            response = requests.put(upload_url, data=content, headers={"Content-Type": mime_type})
+            response = requests.put(
+                upload_url, data=content, headers={"Content-Type": mime_type}, verify=Settings.verify_ssl()
+            )
         response.raise_for_status()
 
         data = graphql(

openhexa/sdk/pipelines/pipeline.py

@@ -17,7 +17,7 @@
 import requests
 from multiprocess import get_context  # NOQA
 
-from openhexa.sdk.utils import Environment, get_environment
+from openhexa.sdk.utils import Environment, Settings, get_environment
 
 from .parameter import FunctionWithParameter, Parameter, ParameterValueError
 from .task import PipelineWithTask, Task
@@ -186,6 +186,7 @@ def _update_progress(self, progress: int):
                     "query": query,
                     "variables": {"input": {"percent": progress}},
                 },
+                verify=Settings.verify_ssl(),
             )
             r.raise_for_status()
         else:

openhexa/sdk/pipelines/runtime.py

@@ -21,6 +21,7 @@
     Parameter,
     validate_parameters,
 )
+from openhexa.sdk.utils import Settings
 
 from .pipeline import Pipeline
 
@@ -91,11 +92,14 @@ def download_pipeline(url: str, token: str, run_id: str, target_dir: str) -> Non
     """
 
     try:
+        import os
+
         response = requests.post(
             f"{url}/graphql/",
             headers={"Authorization": f"Bearer {token}"},
             json={"query": query, "variables": {"id": run_id}},
             timeout=30,
+            verify=Settings.verify_ssl(),
         )
         response.raise_for_status()
 

openhexa/sdk/utils.py

@@ -10,6 +10,20 @@
 from openhexa.utils import create_requests_session
 
 
+class Settings:
+    """Centralized settings for the OpenHexa SDK."""
+
+    @staticmethod
+    def verify_ssl() -> bool:
+        """Return the SSL verification flag from environment variables."""
+        return os.getenv("HEXA_VERIFY_SSL", "True").lower() not in ("0", "false")
+
+    @staticmethod
+    def debug() -> bool:
+        """Return the debug flag from environment variables."""
+        return bool(os.getenv("DEBUG") or os.getenv("HEXA_DEBUG"))
+
+
 class Environment(enum.Enum):
     """Enumeration of supported runtime environments."""
 
@@ -35,7 +49,7 @@ def graphql(operation: str, variables: dict[str | typing.Any] | None = None) ->
         "HEXA_TOKEN"
     ]  # Works for notebooks with the membership token and pipelines with the run token
     headers = {"Authorization": f"Bearer {auth_token}"}
-    session = create_requests_session()
+    session = create_requests_session(verify=Settings.verify_ssl())
 
     req = session.post(
         f"{os.environ['HEXA_SERVER_URL'].rstrip('/')}/graphql/",
@@ -66,9 +80,7 @@ def __init__(self, token: str | None = None, server_url: str | None = None):
         url = server_url or f"{os.environ['HEXA_SERVER_URL'].rstrip('/')}/graphql/"
         token = token or os.getenv("HEXA_TOKEN")
 
-        verify_ssl = os.getenv("HEXA_VERIFY_SSL", "True").lower() not in ("0", "false")
-
-        super().__init__(url=url, token=token, verify=verify_ssl)
+        super().__init__(url=url, token=token, verify=Settings.verify_ssl())
 
 
 class Iterator(metaclass=abc.ABCMeta):