hiddenlayer-sdk 0.1.0__tar.gz → 0.1.2__tar.gz

{hiddenlayer_sdk-0.1.0 → hiddenlayer_sdk-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hiddenlayer-sdk
-Version: 0.1.0
+Version: 0.1.2
 Summary: Official HiddenLayer Python SDK
 Author-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>
 Maintainer-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>

{hiddenlayer_sdk-0.1.0 → hiddenlayer_sdk-0.1.2}/hiddenlayer/__init__.py

@@ -1,3 +1,4 @@
+import os
 from typing import Optional
 
 import requests
@@ -34,6 +35,7 @@ class HiddenlayerServiceClient:
         api_key: Optional[str] = None,
         host: str = "https://api.us.hiddenlayer.ai",
     ):
+        self.host = host.strip()
         self.is_saas = is_saas(host)
 
         if self.is_saas:
@@ -48,10 +50,10 @@ class HiddenlayerServiceClient:
                 )
 
             jwt_token = self._get_jwt(api_id=api_id, api_key=api_key)
-            self._config = Configuration(host=host, access_token=jwt_token)
+            self._config = Configuration(host=self.host, access_token=jwt_token)
 
         else:
-            self._config = Configuration(host=host)
+            self._config = Configuration(host=self.host)
 
         self._api_client = ApiClient(configuration=self._config)
         self._aidr_predictive = AIDRPredictive(self._api_client)
@@ -61,9 +63,9 @@ class HiddenlayerServiceClient:
     def _get_jwt(self, *, api_id: str, api_key: str) -> str:
         "Get the JWT token to auth to the Hiddenlayer API."
 
-        token_url = (
-            "https://auth.hiddenlayer.ai/oauth2/token?grant_type=client_credentials"
-        )
+        auth_url = os.getenv("HL_AUTH_URL", "https://auth.hiddenlayer.ai")
+
+        token_url = f"{auth_url}/oauth2/token?grant_type=client_credentials"
 
         resp = requests.post(token_url, auth=HTTPBasicAuth(api_id, api_key))
 

hiddenlayer_sdk-0.1.2/hiddenlayer/sdk/version.py

@@ -0,0 +1 @@
+VERSION = "0.1.2"

{hiddenlayer_sdk-0.1.0 → hiddenlayer_sdk-0.1.2}/hiddenlayer_sdk.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: hiddenlayer-sdk
-Version: 0.1.0
+Version: 0.1.2
 Summary: Official HiddenLayer Python SDK
 Author-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>
 Maintainer-email: HiddenLayer Integrations Team <integrations@hiddenlayer.com>

{hiddenlayer_sdk-0.1.0 → hiddenlayer_sdk-0.1.2}/pyproject.toml

@@ -61,6 +61,8 @@ venv = ".venv"
 
 [tool.ruff]
 exclude = [".venv", "hiddenlayer/sdk/rest", "tests/rest", "docs/", "hiddenlayer/sdk/enterprise"]
+
+[tool.ruff.lint]
 select = [
   "I" # isort formatting
 ]

hiddenlayer_sdk-0.1.0/hiddenlayer/sdk/version.py

@@ -1 +0,0 @@
-VERSION = "0.1.0"