hiddenlayer-sdk 2.0.10__py3-none-any.whl → 3.0.0__py3-none-any.whl

hiddenlayer/__init__.py

@@ -1,114 +1,109 @@
-import os
-from typing import Callable, Optional
-from urllib.parse import urljoin
-
-import requests
-from requests.auth import HTTPBasicAuth
-
-from hiddenlayer.sdk.rest.api_client import ApiClient
-from hiddenlayer.sdk.rest.configuration import Configuration
-from hiddenlayer.sdk.services.aidr_predictive import AIDRPredictive
-from hiddenlayer.sdk.services.model_scan import ModelScanAPI
-from hiddenlayer.sdk.utils import is_saas
-from hiddenlayer.sdk.version import VERSION
-
-__version__ = VERSION
-
-
-class HiddenlayerServiceAuthError(Exception):
-    pass
-
-
-class HiddenlayerUnsupportedPlatformError(Exception):
-    pass
-
-
-class HiddenlayerServiceClient:
-    """
-    The Hiddenlayer Service Client is a client for the Hiddenlayer services REST API.
-    """
-
-    def __init__(
-        self,
-        *,
-        api_id: Optional[str] = None,
-        api_key: Optional[str] = None,
-        host: str = "https://api.us.hiddenlayer.ai",
-        auth_url: Optional[str] = None,
-    ):
-        self.host = host.strip()
-        if auth_url:
-            self.auth_url = auth_url.strip()
-        else:
-            self.auth_url = None
-        self.is_saas = is_saas(host)
-        refresh_jwt_func: Optional[Callable[[], str]] = None
-
-        if self.is_saas:
-            if not api_id:
-                raise EnvironmentError(
-                    "`api_id` cannot be None when using the SaaS version of HiddenLayer."
-                )
-
-            if not api_key:
-                raise EnvironmentError(
-                    "`api_key` cannot be None when using the SaaS version of HiddenLayer."
-                )
-
-            jwt_token = self._get_jwt(api_id=api_id, api_key=api_key)
-            self._config = Configuration(host=self.host, access_token=jwt_token)
-            refresh_jwt_func = lambda: self._get_jwt(api_id=api_id, api_key=api_key)
-
-        else:
-            self._config = Configuration(host=self.host)
-
-        self._api_client = ApiClient(configuration=self._config)
-        self._aidr_predictive = AIDRPredictive(self._api_client)
-        self._model_scan = ModelScanAPI(
-            self._api_client, refresh_token_func=refresh_jwt_func
-        )
-
-    def _get_jwt(self, *, api_id: str, api_key: str) -> str:
-        "Get the JWT token to auth to the Hiddenlayer API."
-
-        if self.auth_url:
-            auth_url = self.auth_url
-        else:
-            auth_url = os.getenv("HL_AUTH_URL", "https://auth.hiddenlayer.ai")
-
-        token_url = urljoin(auth_url, "/oauth2/token?grant_type=client_credentials")
-
-        resp = requests.post(token_url, auth=HTTPBasicAuth(api_id, api_key))
-
-        if not resp.ok:
-            raise HiddenlayerServiceAuthError(
-                f"Unable to get authentication credentials for the HiddenLayer API: {resp.status_code}: {resp.text}"
-            )
-
-        if "access_token" not in resp.json():
-            raise HiddenlayerServiceAuthError(
-                f"Unable to get authentication credentials for the HiddenLayer API - invalid response: {resp.json()}"
-            )
-
-        return resp.json()["access_token"]
-
-    @property
-    def config(self) -> Configuration:
-        return self._config
-
-    @property
-    def api_client(self) -> ApiClient:
-        return self._api_client
-
-    @property
-    def model_scanner(self) -> ModelScanAPI:
-        return self._model_scan
-
-    @property
-    def aidr_predictive(self) -> AIDRPredictive:
-        # If the platform is not SaaS, it's not supported
-        if not self.is_saas:
-            raise HiddenlayerUnsupportedPlatformError(
-                "AIDR for Predictive Models is currently only supported with the SaaS version of the platform."
-            )
-        return self._aidr_predictive
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+import typing as _t
+
+from . import types
+from .lib import ModelScanner, CommunityScanner, AsyncModelScanner, AsyncCommunityScanner
+from ._types import NOT_GIVEN, Omit, NoneType, NotGiven, Transport, ProxiesTypes, omit, not_given
+from ._utils import file_from_path
+from ._client import (
+    ENVIRONMENTS,
+    Client,
+    Stream,
+    Timeout,
+    Transport,
+    AsyncClient,
+    AsyncStream,
+    HiddenLayer,
+    RequestOptions,
+    AsyncHiddenLayer,
+)
+from ._models import BaseModel
+from ._version import __title__, __version__
+from ._response import APIResponse as APIResponse, AsyncAPIResponse as AsyncAPIResponse
+from ._constants import DEFAULT_TIMEOUT, DEFAULT_MAX_RETRIES, DEFAULT_CONNECTION_LIMITS
+from ._exceptions import (
+    APIError,
+    ConflictError,
+    NotFoundError,
+    APIStatusError,
+    RateLimitError,
+    APITimeoutError,
+    BadRequestError,
+    HiddenLayerError,
+    APIConnectionError,
+    AuthenticationError,
+    InternalServerError,
+    PermissionDeniedError,
+    UnprocessableEntityError,
+    APIResponseValidationError,
+)
+from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
+from ._utils._logs import setup_logging as _setup_logging
+
+__all__ = [
+    "types",
+    "__version__",
+    "__title__",
+    "NoneType",
+    "CommunityScanner",
+    "AsyncCommunityScanner",
+    "ModelScanner",
+    "AsyncModelScanner",
+    "Transport",
+    "ProxiesTypes",
+    "NotGiven",
+    "NOT_GIVEN",
+    "not_given",
+    "Omit",
+    "omit",
+    "HiddenLayerError",
+    "APIError",
+    "APIStatusError",
+    "APITimeoutError",
+    "APIConnectionError",
+    "APIResponseValidationError",
+    "BadRequestError",
+    "AuthenticationError",
+    "PermissionDeniedError",
+    "NotFoundError",
+    "ConflictError",
+    "UnprocessableEntityError",
+    "RateLimitError",
+    "InternalServerError",
+    "Timeout",
+    "RequestOptions",
+    "Client",
+    "AsyncClient",
+    "Stream",
+    "AsyncStream",
+    "HiddenLayer",
+    "AsyncHiddenLayer",
+    "ENVIRONMENTS",
+    "file_from_path",
+    "BaseModel",
+    "DEFAULT_TIMEOUT",
+    "DEFAULT_MAX_RETRIES",
+    "DEFAULT_CONNECTION_LIMITS",
+    "DefaultHttpxClient",
+    "DefaultAsyncHttpxClient",
+    "DefaultAioHttpClient",
+]
+
+if not _t.TYPE_CHECKING:
+    from ._utils._resources_proxy import resources as resources
+
+_setup_logging()
+
+# Update the __module__ attribute for exported symbols so that
+# error messages point to this module instead of the module
+# it was originally defined in, e.g.
+# hiddenlayer._exceptions.NotFoundError -> hiddenlayer.NotFoundError
+__locals = locals()
+for __name in __all__:
+    if not __name.startswith("__"):
+        try:
+            __locals[__name].__module__ = "hiddenlayer"
+        except (TypeError, AttributeError):
+            # Some of our exported symbols are builtins which we can't set attributes for.
+            pass